From 59c0e7365a4564dd2f9c568daa367d0d271ae675 Mon Sep 17 00:00:00 2001 From: Xinya Zhang Date: Mon, 22 Jul 2024 19:08:10 +0000 Subject: [PATCH] Fix weight decay integration problem. --- fbgemm_gpu/include/fbgemm_gpu/hip_kernel_inc/split_tbe_common.h | 2 +- 1 file changed, 1 insertion(+), 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 index 9d690d43e3..afc6c097dc 100644 --- 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