Rename benchmark file (baseline)

dahong67 · Feb 29, 2024 · 9b25f93 · 9b25f93
1 parent 4bb9716
commit 9b25f93
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 1 deletion.
diff --git a/benchmark/suites/gcp_large.jl → benchmark/suites/gcp-large.jl b/benchmark/suites/gcp_large.jl → benchmark/suites/gcp-large.jl
diff --git a/src/gcp-opt.jl b/src/gcp-opt.jl
@@ -131,8 +131,18 @@ function gcp_grad_U!(
         I in CartesianIndices(X)
     ]
     # Use faster MTTKRPs algorithm
-    faster_mttkrps!(GU, M, Y)
+    #faster_mttkrps!(GU, M, Y)
+    #return ntuple(Val(N)) do k
+    #    return rmul!(GU[k], Diagonal(M.λ))
+    #end
+    # MTTKRPs (inefficient but simple)
     return ntuple(Val(N)) do k
+        Yk = reshape(PermutedDimsArray(Y, [k; setdiff(1:N, k)]), size(X, k), :)
+        Zk = similar(Yk, prod(size(X)[setdiff(1:N, k)]), ncomponents(M))
+        for j in Base.OneTo(ncomponents(M))
+            Zk[:, j] = reduce(kron, [view(M.U[i], :, j) for i in reverse(setdiff(1:N, k))])
+        end
+        mul!(GU[k], Yk, Zk)
         return rmul!(GU[k], Diagonal(M.λ))
     end
 end