add ods logging for l2 cache perf (#2335)

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

Reviewed By: q10

Differential Revision: D61417980

fbshipit-source-id: 68478b77172bf563484d2db5356977cd1c4f51b9

torchrec/distributed/batched_embedding_kernel.py

@@ -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,