format

ITensor · Oct 9, 2023 · 2d46f5a · 2d46f5a
1 parent 78ecb1b
commit 2d46f5a
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/NDTensors/src/abstractarray/similar.jl b/NDTensors/src/abstractarray/similar.jl
@@ -58,7 +58,9 @@ end
 
 # For when there are CUArray specific issues inline
 iscu(A::AbstractArray) = iscu(typeof(A))
-iscu(A::Type{<:AbstractArray}) = (leaf_parenttype(A) == A ? false : iscu(leaf_parenttype(A)))
+function iscu(A::Type{<:AbstractArray})
+  return (leaf_parenttype(A) == A ? false : iscu(leaf_parenttype(A)))
+end
 # This function actually allocates the data.
 # Catches conversions of dimensions specified by ranges
 # dimensions specified by integers with `Base.to_shape`.

diff --git a/src/mps/dmrg.jl b/src/mps/dmrg.jl
@@ -281,7 +281,11 @@ function dmrg(PH, psi0::MPS, sweeps::Sweeps; kwargs...)
         ## 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
         ## 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 = if NDTensors.iscu(tensor(phi))
+          itensor(adapt(typeof(data(phi)), tensor(vecs[1])))
+        else
+          vecs[1]
+        end
         #phi::ITensor = vecs[1]
 
         ortho = ha == 1 ? "left" : "right"