Use iscu to just adapt CUDA computations. skip on CPU

ITensor · Oct 9, 2023 · 78ecb1b · 78ecb1b
1 parent 0788770
commit 78ecb1b
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/mps/dmrg.jl b/src/mps/dmrg.jl
@@ -280,7 +280,8 @@ function dmrg(PH, psi0::MPS, sweeps::Sweeps; kwargs...)
         energy = vals[1]
         ## Right now there is a conversion problem in CUDA.jl where `UnifiedMemory` Arrays are being converted 
         ## into `DeviceMemory`. This conversion line is here temporarily to fix that problem when it arises
-        phi::ITensor = itensor(convert(typeof(tensor(phi)), tensor(vecs[1])))
+        ## Adapt is only called when using CUDA backend. CPU will work as implemented previously.
+        phi::ITensor = NDTensors.iscu(tensor(phi)) ? itensor(adapt(typeof(data(phi)), tensor(vecs[1]))) : vecs[1]
         #phi::ITensor = vecs[1]
 
         ortho = ha == 1 ? "left" : "right"