Skip to content

Commit 396ad52

Browse files
emlinmeta-codesync[bot]
authored andcommitted
enable feature score auto collection in EBC (#5031)
Summary: Pull Request resolved: #5031 X-link: meta-pytorch/torchrec#3475 X-link: https://github.com/facebookresearch/FBGEMM/pull/2044 Enable feature score auto collection for EBC in the similar way of EC. The configuration has no difference in embedding table config: virtual_table_eviction_policy=FeatureScoreBasedEvictionPolicy( training_id_eviction_trigger_count=260_000_000, # 260M training_id_keep_count=160_000_000, # 160M enable_auto_feature_score_collection=True, feature_score_mapping={ "sparse_public_original_content_creator": 1.0, }, feature_score_default_value=0.5, ), Reviewed By: EddyLXJ Differential Revision: D85017179 fbshipit-source-id: 3d62f8adbe201d6e30c445aaed88710bbbcd6557
1 parent 6c3a720 commit 396ad52

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

fbgemm_gpu/fbgemm_gpu/split_table_batched_embeddings_ops_common.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -241,6 +241,7 @@ class KVZCHParams(NamedTuple):
241241
backend_return_whole_row: bool = False
242242
eviction_policy: EvictionPolicy = EvictionPolicy()
243243
embedding_cache_mode: bool = False
244+
feature_score_collection_enabled: bool = False
244245

245246
def validate(self) -> None:
246247
assert len(self.bucket_offsets) == len(self.bucket_sizes), (

0 commit comments

Comments
 (0)