From 1babfec2f264d634864ee601749c881271ef828b Mon Sep 17 00:00:00 2001
From: mtfishman <mfishman@flatironinstitute.org>
Date: Wed, 8 May 2024 19:26:35 -0400
Subject: [PATCH] Fix tests

---
 src/ITensorGPU.jl | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/ITensorGPU.jl b/src/ITensorGPU.jl
index 4a29e72..18f39d2 100644
--- 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