logging nit, avoid explicit global use (#1548)

Author: williamwen42

torchdynamo/logging.py

@@ -1,3 +1,4 @@
+import itertools
 import logging
 import os
 
@@ -23,7 +24,7 @@ def set_loggers_level(level):
     "version": 1,
     "formatters": {
         "torchdynamo_format": {
-            "format": "%(name)s: [%(levelname)s] [%(asctime)s] %(message)s"
+            "format": "[%(asctime)s] %(name)s: [%(levelname)s] %(message)s"
         },
     },
     "handlers": {
@@ -74,13 +75,11 @@ def init_logging(log_level, log_file_name=None):
 # def fn():
 #     _step_logger()(logging.INFO, "msg")
 
-_step_counter = 1
+_step_counter = itertools.count(1)
 
 
 def get_step_logger(logger):
-    global _step_counter
-    step = _step_counter
-    _step_counter += 1
+    step = next(_step_counter)
 
     def log(level, msg):
         logger.log(level, f"Step {step}: {msg}")

torchinductor/compile_fx.py

@@ -1,5 +1,6 @@
 import dataclasses
 import functools
+import itertools
 import logging
 from typing import List
 
@@ -304,7 +305,7 @@ def is_not_gradout(x):
     return len(static_arg_idxs)
 
 
-_graph_counter = 0
+_graph_counter = itertools.count(0)
 
 
 def compile_fx(model_: torch.fx.GraphModule, example_inputs_: List[torch.Tensor]):
@@ -323,9 +324,7 @@ def compile_fx(model_: torch.fx.GraphModule, example_inputs_: List[torch.Tensor]
     num_example_inputs = len(example_inputs_)
     cudagraphs = BoxedBool(config.triton.cudagraphs)
 
-    global _graph_counter
-    graph_id = _graph_counter
-    _graph_counter += 1
+    graph_id = next(_graph_counter)
 
     @dynamo_utils.dynamo_timed
     def fw_compiler(model: torch.fx.GraphModule, example_inputs):