From 97c31a36dc8d0e177490c496767ef831835777a8 Mon Sep 17 00:00:00 2001 From: mtfishman Date: Sun, 23 Jun 2024 18:56:40 -0400 Subject: [PATCH] Try fixing copy in cuTENSOR contract --- NDTensors/ext/NDTensorscuTENSORExt/contract.jl | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/NDTensors/ext/NDTensorscuTENSORExt/contract.jl b/NDTensors/ext/NDTensorscuTENSORExt/contract.jl index 0edbcce646..a1d866bb79 100644 --- a/NDTensors/ext/NDTensorscuTENSORExt/contract.jl +++ b/NDTensors/ext/NDTensorscuTENSORExt/contract.jl @@ -9,7 +9,7 @@ function to_zero_offset_cuarray(a::CuArray) return iszero(a.offset) ? a : copy(a) end function to_zero_offset_cuarray(a::ReshapedArray) - return copy(a) + return copy(expose(a)) end function NDTensors.contract!(