Fix weight decay integration problem.

ROCm · Jul 22, 2024 · 7024318 · 7024318
1 parent 6a4f65b
commit 7024318
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/fbgemm_gpu/include/fbgemm_gpu/hip_kernel_inc/split_tbe_common.h b/fbgemm_gpu/include/fbgemm_gpu/hip_kernel_inc/split_tbe_common.h
@@ -369,7 +369,7 @@ struct rowwise_adagrad_kernel_arg_t
     float eps;
     float learning_rate;
     float weight_decay;
-    // int64_t weight_decay_mode;
+    int64_t weight_decay_mode;
 };
 
 typedef struct