diff --git a/fbgemm_gpu/CMakeLists.txt b/fbgemm_gpu/CMakeLists.txt index 9da0960890..45bf2ef786 100644 --- a/fbgemm_gpu/CMakeLists.txt +++ b/fbgemm_gpu/CMakeLists.txt @@ -162,7 +162,6 @@ file(GLOB_RECURSE asmjit_sources set(COMMON_OPTIMIZERS adagrad - lars_sgd partial_rowwise_adam partial_rowwise_lamb rowwise_adagrad @@ -176,6 +175,7 @@ set(CPU_ONLY_OPTIMIZERS "") set(GPU_ONLY_OPTIMIZERS adam lamb + lars_sgd rowwise_adagrad_with_weight_decay approx_rowwise_adagrad_with_weight_decay) diff --git a/fbgemm_gpu/codegen/embedding_backward_code_generator.py b/fbgemm_gpu/codegen/embedding_backward_code_generator.py index 0188947f1b..d84cde1bb8 100644 --- a/fbgemm_gpu/codegen/embedding_backward_code_generator.py +++ b/fbgemm_gpu/codegen/embedding_backward_code_generator.py @@ -1399,7 +1399,7 @@ def lars_sgd() -> None: split_weight_update=split_weight_update, split_post_update="", split_weight_update_cpu=split_weight_update_cpu, - has_cpu_support=True, + has_cpu_support=False, has_gpu_support=True, has_vbe_support=False, )