Add table_names as a instance variable of SplitTableBatchedEmbeddingBagsCodegen

Joey Yang · meta-codesync[bot] · commit cb62d163698d · 2025-11-14T10:50:20.000-08:00
Differential Revision: D87089649
diff --git a/fbgemm_gpu/fbgemm_gpu/split_table_batched_embeddings_ops_training.py b/fbgemm_gpu/fbgemm_gpu/split_table_batched_embeddings_ops_training.py
@@ -710,6 +710,7 @@ def __init__(  # noqa C901
             f"Feature Gates: {[(feature.name, feature.is_enabled()) for feature in FeatureGateName]}"
         )
 
+        self.table_names: Optional[list[str]] = table_names
         self.logging_table_name: str = self.get_table_name_for_logging(table_names)
         self.enable_raw_embedding_streaming: bool = enable_raw_embedding_streaming
         self.pooling_mode = pooling_mode

Original file line number	Diff line number	Diff line change
`@@ -710,6 +710,7 @@ def __init__( # noqa C901`
`710`	`710`	`f"Feature Gates: {[(feature.name, feature.is_enabled()) for feature in FeatureGateName]}"`
`711`	`711`	`)`
`712`	`712`
	`713`	`+ self.table_names: Optional[list[str]] = table_names`
`713`	`714`	`self.logging_table_name: str = self.get_table_name_for_logging(table_names)`
`714`	`715`	`self.enable_raw_embedding_streaming: bool = enable_raw_embedding_streaming`
`715`	`716`	`self.pooling_mode = pooling_mode`