array necessary here because .= fails for CUDA gpu

ITensor · Jun 14, 2024 · d8cb8bd · d8cb8bd
1 parent dcf778e
commit d8cb8bd
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/NDTensors/src/tensor/tensor.jl b/NDTensors/src/tensor/tensor.jl
@@ -369,7 +369,7 @@ function diag(tensor::Tensor)
   tensordiag = NDTensors.similar(
     dense(typeof(tensor)), eltype(tensor), (diaglength(tensor),)
   )
-  tensordiag .= diagview(tensor)
+  array(tensordiag) .= diagview(tensor)
   return tensordiag
 end