[JIT] add GRAPH_DEBUG for setGraphExecutorOptimize (#153549)

davidberard98 · pytorchmergebot · commit 5e6e52e7c9e2 · 2025-05-14T20:07:25.000Z
Summary: Optionally log when setGraphExecutorOptimize is called, so we can get insight into the GraphExecutor behavior. Differential Revision: D74692508 Pull Request resolved: #153549 Approved by: https://github.com/PaulZhang12, https://github.com/SamGinzburg
diff --git a/torch/csrc/jit/jit_log.h b/torch/csrc/jit/jit_log.h
@@ -1,4 +1,5 @@
 #pragma once
+#include <c10/util/StringUtil.h>
 #include <torch/csrc/Export.h>
 #include <memory>
 #include <ostream>
diff --git a/torch/csrc/jit/python/update_graph_executor_opt.cpp b/torch/csrc/jit/python/update_graph_executor_opt.cpp
@@ -1,10 +1,12 @@
+#include <torch/csrc/jit/jit_log.h>
 #include <torch/csrc/jit/python/update_graph_executor_opt.h>
 
 namespace torch::jit {
 
 static thread_local bool kOptimize = true;
 void setGraphExecutorOptimize(bool o) {
   kOptimize = o;
+  GRAPH_DEBUG("GraphExecutorOptimize set to ", o);
 }
 bool getGraphExecutorOptimize() {
   return kOptimize;

Original file line number	Diff line number	Diff line change
`@@ -1,4 +1,5 @@`
`1`	`1`	`#pragma once`
	`2`	`+#include <c10/util/StringUtil.h>`
`2`	`3`	`#include <torch/csrc/Export.h>`
`3`	`4`	`#include <memory>`
`4`	`5`	`#include <ostream>`