add more logging code in the test

microsoft · Sep 30, 2024 · 45704ae · 45704ae
1 parent 12d19ca
commit 45704ae
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/tests/torch_compile/test_compile.py b/tests/torch_compile/test_compile.py
@@ -97,3 +97,5 @@ def forward(self, data, residual):
 
 if comm.get_rank() == 0:
     print(dynamo_stats)
+    print(torch._dynamo.utils.counters["stats"]["calls_captured"])
+    print(torch._dynamo.utils.log)