Skip to content

Commit

Permalink
add ods logging for l2 cache perf (#2335)
Browse files Browse the repository at this point in the history
Summary:
Pull Request resolved: #2335

X-link: pytorch/FBGEMM#3031

X-link: facebookresearch/FBGEMM#129

collect performance related metrics from KV store and export them to ODS

Differential Revision: D61417980
  • Loading branch information
Joe Wang authored and facebook-github-bot committed Aug 29, 2024
1 parent e35cb48 commit 73990f4
Showing 1 changed file with 4 additions and 0 deletions.
4 changes: 4 additions & 0 deletions torchrec/distributed/batched_embedding_kernel.py
Original file line number Diff line number Diff line change
Expand Up @@ -1156,6 +1156,10 @@ def __init__(
**ssd_tbe_params,
).to(device)

logger.info(
f"tbe_unique_id:{self._emb_module.tbe_unique_id} => table name to count dict:{self.table_name_to_count}"
)

self._optim: KeyValueEmbeddingFusedOptimizer = KeyValueEmbeddingFusedOptimizer(
config,
self._emb_module,
Expand Down

0 comments on commit 73990f4

Please sign in to comment.