tc

torch/testing/_internal/common_utils.py

@@ -115,7 +115,6 @@ CI_FUNCTORCH_ROOT = ""
 CI_PT_ROOT = ""
 CI_TEST_PREFIX = ""
 DISABLED_TESTS_FILE = ""
-GRAPH_EXECUTOR : Optional[ProfilingMode] = None
 LOG_SUFFIX = ""
 PYTEST_SINGLE_TEST = ""
 REPEAT_COUNT = 0
@@ -878,6 +877,10 @@ def cppProfilingFlagsToProfilingMode():
     else:
         return ProfilingMode.LEGACY
 
+# Set default value for the internal test runner which imports files/modules
+# directly
+GRAPH_EXECUTOR : ProfilingMode = cppProfilingFlagsToProfilingMode()
+
 @contextmanager
 def enable_profiling_mode_for_profiling_tests():
     old_prof_exec_state = False