Move ComputeDevice to split_table_batched_embeddings_ops_common (#4034)

Summary:
X-link: facebookresearch/FBGEMM#1118


    - Moved `ComputeDevice` to `split_table_batched_embeddings_ops_common`

Reviewed By: spcyppt, q10

Differential Revision: D73608722

Author: gchalump

fbgemm_gpu/fbgemm_gpu/split_table_batched_embeddings_ops_common.py

@@ -106,6 +106,12 @@ class BoundsCheckMode(enum.IntEnum):
     V2_FATAL = 6
 
 
+class ComputeDevice(enum.IntEnum):
+    CPU = 0
+    CUDA = 1
+    MTIA = 2
+
+
 class EmbeddingSpecInfo(enum.IntEnum):
     feature_names = 0
     rows = 1

fbgemm_gpu/fbgemm_gpu/split_table_batched_embeddings_ops_training.py

@@ -37,6 +37,7 @@
     BoundsCheckMode,
     CacheAlgorithm,
     CacheState,
+    ComputeDevice,
     construct_cache_state,
     EmbeddingLocation,
     MAX_PREFETCH_DEPTH,
@@ -80,12 +81,6 @@ class DoesNotHavePrefix(Exception):
     pass
 
 
-class ComputeDevice(enum.IntEnum):
-    CPU = 0
-    CUDA = 1
-    MTIA = 2
-
-
 class WeightDecayMode(enum.IntEnum):
     NONE = 0
     L2 = 1