From 6e1e2771a931c064ffc40ed97a93aad3f1c2dccf Mon Sep 17 00:00:00 2001 From: Kathy Xu Date: Fri, 10 Oct 2025 09:51:22 -0700 Subject: [PATCH] Typo fix Summary: as title Reviewed By: emlin Differential Revision: D83699190 --- torchrec/distributed/batched_embedding_kernel.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/torchrec/distributed/batched_embedding_kernel.py b/torchrec/distributed/batched_embedding_kernel.py index cbcd0d78b..882726d0b 100644 --- a/torchrec/distributed/batched_embedding_kernel.py +++ b/torchrec/distributed/batched_embedding_kernel.py @@ -337,10 +337,10 @@ def _populate_zero_collision_tbe_params( table_names = [table.name for table in config.embedding_tables] l2_cache_size = tbe_params["l2_cache_size"] if "kvzch_eviction_trigger_mode" in tbe_params: - eviction_tirgger_mode = tbe_params["kvzch_eviction_trigger_mode"] + eviction_trigger_mode = tbe_params["kvzch_eviction_trigger_mode"] tbe_params.pop("kvzch_eviction_trigger_mode") else: - eviction_tirgger_mode = 2 # 2 means mem_util based eviction + eviction_trigger_mode = 2 # 2 means mem_util based eviction for i, table in enumerate(config.embedding_tables): policy_t = table.virtual_table_eviction_policy if policy_t is not None: @@ -409,7 +409,7 @@ def _populate_zero_collision_tbe_params( f"Unsupported eviction policy {policy_t} for table {table.name}" ) eviction_policy = EvictionPolicy( - eviction_trigger_mode=eviction_tirgger_mode, + eviction_trigger_mode=eviction_trigger_mode, eviction_mem_threshold_gb=l2_cache_size, eviction_strategy=eviction_strategy, counter_thresholds=counter_thresholds,