added changes from julia 1.11 branch

euro-hpc-pl · Nov 8, 2024 · 6f5e005 · 6f5e005
1 parent 9a32b6a
commit 6f5e005
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/zipper.jl b/src/zipper.jl
@@ -163,6 +163,7 @@ function _left_sweep_var_site!(env::EnvironmentMixed, site; kwargs...)   # site:
     _, Q = rq_fact(B; toGPU = env.onGPU, kwargs...)
     # @cast C[l, r, t] := Q[l, (r, t)] (t ∈ 1:size(A, 3))
     C = reshape(Q, size(Q, 1), size(Q, 2) ÷ size(A, 3), size(A, 3))
+    !env.onGPU && (C = collect(C))
     if site == :central
         env.C = C
     else
@@ -181,6 +182,7 @@ function _right_sweep_var_site!(env::EnvironmentMixed, site; kwargs...)
     # @cast C[l, t, r] := Q[(l, t), r] (t ∈ 1:size(A, 3))
     C = reshape(Q, size(Q, 1) ÷ size(A, 3), size(A, 3), size(Q, 2))
     C = permutedims(C, (1, 3, 2))  # [l, r, t]
+    !env.onGPU && (C = collect(C))
     if site == :central
         env.C = C
     else