[CUDA][CUDA Graphs] Fix debug mode warning message (pytorch#145996)

eqy · pytorchmergebot · commit 5d7532140f37 · 2025-02-07T08:04:49.000Z
The real method is `enable_debug_mode()`, `_cuda_enable_graphs_debug_mode` does not exist. Pull Request resolved: pytorch#145996 Approved by: https://github.com/ptrblck, https://github.com/eellison
diff --git a/aten/src/ATen/cuda/CUDAGraph.cpp b/aten/src/ATen/cuda/CUDAGraph.cpp
@@ -257,7 +257,7 @@ void CUDAGraph::debug_dump(const std::string& debug_path) {
       has_graph_ = false;
     }
   } else {
-    TORCH_WARN("CUDA Graphs debug not enabled, set with torch._C._cuda_enable_graphs_debug_mode");
+    TORCH_WARN("CUDA Graphs debug not enabled, set with [graph].enable_debug_mode()");
   }
 #else
   TORCH_CHECK(false, "CUDA graphs may only be used in Pytorch built with CUDA >= 11.3 or ROCM >= 5.6");

Original file line number	Diff line number	Diff line change
`@@ -257,7 +257,7 @@ void CUDAGraph::debug_dump(const std::string& debug_path) {`
`257`	`257`	`has_graph_ = false;`
`258`	`258`	`}`
`259`	`259`	`} else {`
`260`		`- TORCH_WARN("CUDA Graphs debug not enabled, set with torch._C._cuda_enable_graphs_debug_mode");`
	`260`	`+ TORCH_WARN("CUDA Graphs debug not enabled, set with [graph].enable_debug_mode()");`
`261`	`261`	`}`
`262`	`262`	`#else`
`263`	`263`	`TORCH_CHECK(false, "CUDA graphs may only be used in Pytorch built with CUDA >= 11.3 or ROCM >= 5.6");`