debug ndcg metrics qps

Differential Revision: D65689878
iamzainhuda · Nov 9, 2024 · ead7c90 · ead7c90
1 parent 9a4d8a8
commit ead7c90
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/torchrec/pt2/utils.py b/torchrec/pt2/utils.py
@@ -172,6 +172,7 @@ def inner_forward(
         **kwargs: Any,
     ) -> Any:
         if hasattr(ref, "enable_pt2_compile") and ref.enable_pt2_compile:
+            # print("##### WH: PT2 compile enabled #####")
             pt2_compiled_attr_name = f"_{f.__name__}_pt2_compiled"
             if not hasattr(ref, pt2_compiled_attr_name):
                 setattr(ref, pt2_compiled_attr_name, torch.compile(f))