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 2d5acba444..f9c452c297 100644 --- a/fbgemm_gpu/codegen/embedding_backward_code_generator.py +++ b/fbgemm_gpu/codegen/embedding_backward_code_generator.py @@ -1344,7 +1344,7 @@ def lars_sgd() -> None: split_precomputation=split_precomputation, split_weight_update=split_weight_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, )