From 45704ae4857786355e296a110372f589caad7821 Mon Sep 17 00:00:00 2001 From: Guokai Ma Date: Mon, 30 Sep 2024 11:48:48 +0800 Subject: [PATCH] add more logging code in the test --- tests/torch_compile/test_compile.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/tests/torch_compile/test_compile.py b/tests/torch_compile/test_compile.py index 529ca56ae0a8..c054d9d37c1b 100644 --- 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)