Replace norm function

dahong67 · Jan 18, 2024 · f4696c0 · f4696c0
1 parent ab40583
commit f4696c0
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/ext/CUDAExt.jl b/ext/CUDAExt.jl
@@ -36,7 +36,7 @@ function _gcp(
         for n in 1:N
             V = reduce(.*, U[i]'U[i] for i in setdiff(1:N, n))
             U[n] = GCPDecompositions.mttkrp(X, U, n) / V
-            λ = CuArray(CUDA.norm.(eachcol(U[n])))
+            λ = vec(sqrt.(sum(abs2, U[n]; dims=1)))
             U[n] = U[n] ./ permutedims(λ)
         end
     end

diff --git a/src/gcp-opt.jl b/src/gcp-opt.jl
@@ -166,7 +166,7 @@ function _gcp(
         for n in 1:N
             V = reduce(.*, U[i]'U[i] for i in setdiff(1:N, n))
             U[n] = mttkrp(X, U, n) / V
-            λ = norm.(eachcol(U[n]))
+            λ = vec(sqrt.(sum(abs2, U[n]; dims=1)))
             U[n] = U[n] ./ permutedims(λ)
         end
     end