Fix weight decay integration problem.

ROCm · Aug 7, 2024 · 59c0e73 · 59c0e73
1 parent 4052548
commit 59c0e73
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