diff --git a/torch_npu/csrc/core/npu/NPUGraph.cpp b/torch_npu/csrc/core/npu/NPUGraph.cpp
index fd060dcb086f911f0e2b345a18ba9672ded2c34c..a00448bd1cf547cb2f641eb47ff940c902341610 100644
--- a/torch_npu/csrc/core/npu/NPUGraph.cpp
+++ b/torch_npu/csrc/core/npu/NPUGraph.cpp
@@ -1,4 +1,3 @@
-#ifndef BUILD_LIBTORCH
 #include "torch_npu/csrc/core/npu/NPUGraph.h"
 #include "torch_npu/csrc/core/npu/NPUCachingAllocator.h"
 #include "torch_npu/csrc/core/npu/NPUFunctions.h"
@@ -10,7 +9,6 @@
 #include <vector>
 
 #include <ATen/Functions.h>
-#include <torch/csrc/Exceptions.h>
 
 namespace c10_npu {
 
@@ -261,4 +259,3 @@ NPUGraph::~NPUGraph()
 }
 
 } // namespace c10_npu
-#endif