Skip to content

Commit 1371bfc

Browse files
Kathy Xumeta-codesync[bot]
authored andcommitted
Typo fix (#3449)
Summary: Pull Request resolved: #3449 as title Reviewed By: emlin Differential Revision: D83699190 fbshipit-source-id: 9ea36d1d198552fa56a28b79217dad1c71a15300
1 parent c89d22b commit 1371bfc

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

torchrec/distributed/batched_embedding_kernel.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -337,10 +337,10 @@ def _populate_zero_collision_tbe_params(
337337
table_names = [table.name for table in config.embedding_tables]
338338
l2_cache_size = tbe_params["l2_cache_size"]
339339
if "kvzch_eviction_trigger_mode" in tbe_params:
340-
eviction_tirgger_mode = tbe_params["kvzch_eviction_trigger_mode"]
340+
eviction_trigger_mode = tbe_params["kvzch_eviction_trigger_mode"]
341341
tbe_params.pop("kvzch_eviction_trigger_mode")
342342
else:
343-
eviction_tirgger_mode = 2 # 2 means mem_util based eviction
343+
eviction_trigger_mode = 2 # 2 means mem_util based eviction
344344
for i, table in enumerate(config.embedding_tables):
345345
policy_t = table.virtual_table_eviction_policy
346346
if policy_t is not None:
@@ -409,7 +409,7 @@ def _populate_zero_collision_tbe_params(
409409
f"Unsupported eviction policy {policy_t} for table {table.name}"
410410
)
411411
eviction_policy = EvictionPolicy(
412-
eviction_trigger_mode=eviction_tirgger_mode,
412+
eviction_trigger_mode=eviction_trigger_mode,
413413
eviction_mem_threshold_gb=l2_cache_size,
414414
eviction_strategy=eviction_strategy,
415415
counter_thresholds=counter_thresholds,

0 commit comments

Comments
 (0)