Skip to content

Commit

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

X-link: pytorch/FBGEMM#3031

X-link: facebookresearch/FBGEMM#129

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

Reviewed By: q10

Differential Revision: D61417980
  • Loading branch information
duduyi2013 authored and facebook-github-bot committed Aug 29, 2024
1 parent c2d6457 commit 6fc4d25
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 6fc4d25

Please sign in to comment.