From 0739a38b3e2961f2982e193b2887bf9c60e45242 Mon Sep 17 00:00:00 2001
From: alexmul1114 <96952477+alexmul1114@users.noreply.github.com>
Date: Tue, 26 Nov 2024 13:39:30 -0500
Subject: [PATCH] add multiplication by constant for gradU sym case

---
 src/gcp-losses.jl | 1 +
 1 file changed, 1 insertion(+)

diff --git a/src/gcp-losses.jl b/src/gcp-losses.jl
index 1f2cf54..dd9674c 100644
--- a/src/gcp-losses.jl
+++ b/src/gcp-losses.jl
@@ -118,6 +118,7 @@ function grad_U!(
     for j in 1:K
         if sym_data
             mttkrp!(GU[j], Y, tuple([M.U[k] for k in M.S]...), findall(M.S .== j)[1])
+            rmul!(GU[j], count(M.S .== j))
         else
             for (index, mode) in enumerate(findall(M.S .== j))
                 if index == 1