Skip to content

Commit

Permalink
Remove CPU support for lars_sgd (#1774)
Browse files Browse the repository at this point in the history
Summary:
Pull Request resolved: #1774

This diff is a part of FBGEMM TBE optimizer deprecation plan (see
#1727)

Changes include:
- Setting `has_cpu_support` of `lars_sgd` to
  `False` in the code gen script
- Updating `codegen/TARGETS` and `CMakeLists.txt`

See D45835048 for how the changes affect the code generation

Reviewed By: csmiler

Differential Revision: D45845181

fbshipit-source-id: 2cf5f40f5aceea997a59eb1eef97b27da0fb4a18
  • Loading branch information
q10 authored and facebook-github-bot committed Jun 1, 2023
1 parent 66e629c commit 29bc6eb
Show file tree
Hide file tree
Showing 2 changed files with 2 additions and 2 deletions.
2 changes: 1 addition & 1 deletion fbgemm_gpu/CMakeLists.txt
Original file line number Diff line number Diff line change
Expand Up @@ -165,7 +165,6 @@ file(GLOB_RECURSE asmjit_sources

set(COMMON_OPTIMIZERS
adagrad
lars_sgd
partial_rowwise_adam
partial_rowwise_lamb
rowwise_adagrad
Expand All @@ -179,6 +178,7 @@ set(CPU_ONLY_OPTIMIZERS "")
set(GPU_ONLY_OPTIMIZERS
adam
lamb
lars_sgd
rowwise_adagrad_with_weight_decay
approx_rowwise_adagrad_with_weight_decay)

Expand Down
2 changes: 1 addition & 1 deletion fbgemm_gpu/codegen/embedding_backward_code_generator.py
Original file line number Diff line number Diff line change
Expand Up @@ -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,
)
Expand Down

0 comments on commit 29bc6eb

Please sign in to comment.