Remove CPU support for partial_rowwise_lamb (pytorch#1775)

Summary:
Pull Request resolved: pytorch#1775

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

Changes include:
- Setting `has_cpu_support` of `partial_rowwise_lamb` 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: D45845467

fbshipit-source-id: d4f6f2dd89212a7a9eb4c7edc3ef65e2fc1b5762

fbgemm_gpu/CMakeLists.txt

@@ -168,7 +168,6 @@ file(GLOB_RECURSE asmjit_sources
 
 set(COMMON_OPTIMIZERS
     adagrad
-    partial_rowwise_lamb
     rowwise_adagrad
     rowwise_adagrad_with_counter
     rowwise_weighted_adagrad
@@ -181,6 +180,7 @@ set(GPU_ONLY_OPTIMIZERS
   adam
   lamb
   partial_rowwise_adam
+  partial_rowwise_lamb
   lars_sgd
   rowwise_adagrad_with_weight_decay
   approx_rowwise_adagrad_with_weight_decay)

fbgemm_gpu/codegen/embedding_backward_code_generator.py

@@ -1219,7 +1219,7 @@ def partial_rowwise_lamb() -> 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,
     )