Fix tests

ITensor · May 8, 2024 · 1babfec · 1babfec
1 parent 8f16905
commit 1babfec
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/ITensorGPU.jl b/src/ITensorGPU.jl
@@ -29,9 +29,9 @@ end
 function cuMPO(args...; kwargs...)
   return adapt(CuArray, MPO(args...; kwargs...))
 end
+cuMPO(tn::MPO) = adapt(CuArray, tn)
 function randomCuMPO(args...; kwargs...)
   return adapt(CuArray, randomMPO(args...; kwargs...))
 end
-cuMPO(tn::MPO) = cu(tn)
 export cuMPO, cuMPS, productCuMPS, randomCuMPO, randomCuMPS
 end