Skip to content

Commit a00613a

Browse files
EddyLXJfacebook-github-bot
authored andcommitted
Change kvzch_eviction_tbe_config to kvzch_tbe_config (#5084)
Summary: X-link: meta-pytorch/torchrec#3514 X-link: facebookresearch/FBGEMM#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
1 parent be1b514 commit a00613a

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
@@ -253,7 +253,7 @@ def validate(self) -> None:
253253
), "backend_return_whole_row can only be enabled when enable_optimizer_offloading is enabled"
254254

255255

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

0 commit comments

Comments
 (0)