From a9f9d2ad26e34d12042800e209420362939e892f Mon Sep 17 00:00:00 2001 From: alexmul1114 <96952477+alexmul1114@users.noreply.github.com> Date: Thu, 29 Feb 2024 15:34:56 -0500 Subject: [PATCH] Rename benchmark file (new) --- src/gcp-opt.jl | 14 ++------------ 1 file changed, 2 insertions(+), 12 deletions(-) diff --git a/src/gcp-opt.jl b/src/gcp-opt.jl index 393a8bf..a8026cf 100644 --- a/src/gcp-opt.jl +++ b/src/gcp-opt.jl @@ -131,20 +131,10 @@ function gcp_grad_U!( I in CartesianIndices(X) ] # Use faster MTTKRPs algorithm - #faster_mttkrps!(GU, M, Y) - #return ntuple(Val(N)) do k - # return rmul!(GU[k], Diagonal(M.λ)) - #end - # MTTKRPs (inefficient but simple) + faster_mttkrps!(GU, M, Y) 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 end function _gcp(