Skip to content

Commit 1c4878e

Browse files
EddyLXJmeta-codesync[bot]
authored andcommitted
Change kvzch_eviction_tbe_config to kvzch_tbe_config (#5084)
Summary: Pull Request resolved: #5084 X-link: meta-pytorch/torchrec#3514 X-link: https://github.com/facebookresearch/FBGEMM/pull/2092 Change tbe config name from kvzch_eviction_tbe_config to kvzch_tbe_config, as it may use not only for eviction but also for some other processes like st publish. Reviewed By: emlin Differential Revision: D86212643 fbshipit-source-id: 1cf7b8c1c023e2fc7c77964b2b3b3a1fea40a607
1 parent ee26ed4 commit 1c4878e

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

fbgemm_gpu/fbgemm_gpu/split_table_batched_embeddings_ops_common.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -260,7 +260,7 @@ def validate(self) -> None:
260260
), "backend_return_whole_row can only be enabled when enable_optimizer_offloading is enabled"
261261

262262

263-
class KVZCHEvictionTBEConfig(NamedTuple):
263+
class KVZCHTBEConfig(NamedTuple):
264264
# Eviction trigger model for kvzch table: 0: disabled, 1: iteration, 2: mem_util, 3: manual, 4: id count, 5: free_mem
265265
kvzch_eviction_trigger_mode: int = 2 # mem_util
266266
# Minimum free memory (in GB) required before triggering eviction when using free_mem trigger mode.

0 commit comments

Comments
 (0)