Fix dynamo benchmark config --print-graph-breaks (#108584)

yanboliang · pytorchmergebot · commit ff28b4b908e6 · 2023-09-05T23:31:43.000Z
Fixes #ISSUE_NUMBER Pull Request resolved: #108584 Approved by: https://github.com/anijain2305
diff --git a/benchmarks/dynamo/common.py b/benchmarks/dynamo/common.py
@@ -3191,7 +3191,7 @@ def run(runner, args, original_dir=None):
         torch._logging.set_logs(dynamo=logging.DEBUG)
 
     if args.print_graph_breaks:
-        torch._dynamo.config.print_graph_breaks = True
+        torch._logging.set_logs(graph_breaks=True)
 
     if args.quiet:
         torch._logging.set_logs(dynamo=logging.ERROR)
