From 7c1a4de77ad0ab9b0a11cd2a78708e5772bf038d Mon Sep 17 00:00:00 2001 From: Joey Yang Date: Mon, 17 Nov 2025 15:39:24 -0800 Subject: [PATCH] Add table_names as a instance variable of SplitTableBatchedEmbeddingBagsCodegen Differential Revision: D87089649 --- .../fbgemm_gpu/split_table_batched_embeddings_ops_training.py | 1 + 1 file changed, 1 insertion(+) 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 index 0a2a7ab0a1..8b7c5538f3 100644 --- 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