diff --git a/python/src/ir.h b/include/triton/Tools/OpBuilder.h
similarity index 100%
rename from python/src/ir.h
rename to include/triton/Tools/OpBuilder.h
diff --git a/include/triton/Tools/PluginUtils.h b/include/triton/Tools/PluginUtils.h
index 86536948105b..09e1126abdba 100644
--- a/include/triton/Tools/PluginUtils.h
+++ b/include/triton/Tools/PluginUtils.h
@@ -3,7 +3,7 @@
 
 #include "mlir/Pass/PassManager.h"
 #include "mlir/Tools/Plugins/DialectPlugin.h"
-#include "python/src/ir.h"
+#include "triton/Tools/OpBuilder.h"
 #include "llvm/Support/DynamicLibrary.h"
 #include "llvm/Support/Error.h"
 #include <cstdint>
diff --git a/python/src/gluon_ir.cc b/python/src/gluon_ir.cc
index 35f358b165c9..f42a8d59011e 100644
--- a/python/src/gluon_ir.cc
+++ b/python/src/gluon_ir.cc
@@ -1,4 +1,3 @@
-#include "ir.h"
 #include "pybind11/pybind11.h"
 #include <pybind11/stl.h>
 
@@ -22,6 +21,7 @@
 #include "triton/Tools/GenericSwizzling.h"
 #include "triton/Tools/LayoutUtils.h"
 #include "triton/Tools/LinearLayout.h"
+#include "triton/Tools/OpBuilder.h"
 #include "llvm/ADT/StringSwitch.h"
 #include "llvm/Support/MathExtras.h"
 
diff --git a/python/src/ir.cc b/python/src/ir.cc
index 243410935e5b..0bfcb78e3918 100644
--- a/python/src/ir.cc
+++ b/python/src/ir.cc
@@ -1,5 +1,3 @@
-#include "ir.h"
-
 #include <optional>
 #include <pybind11/cast.h>
 #include <pybind11/functional.h>
@@ -36,6 +34,7 @@
 #include "triton/Dialect/TritonInstrument/IR/Dialect.h"
 #include "triton/Dialect/TritonNvidiaGPU/IR/Dialect.h"
 #include "triton/Dialect/TritonNvidiaGPU/Transforms/TMAUtilities.h"
+#include "triton/Tools/OpBuilder.h"
 #include "triton/Tools/PluginUtils.h"
 #include "triton/Tools/Sys/Dump.hpp"
 #include "triton/Tools/Sys/GetEnv.hpp"
diff --git a/third_party/proton/Dialect/triton_proton.cc b/third_party/proton/Dialect/triton_proton.cc
index ecc721eddb92..1e30b6cc72a3 100644
--- a/third_party/proton/Dialect/triton_proton.cc
+++ b/third_party/proton/Dialect/triton_proton.cc
@@ -6,10 +6,10 @@
 #include "Dialect/Proton/IR/Dialect.h"
 #include "Dialect/ProtonGPU/IR/Dialect.h"
 #include "Dialect/ProtonGPU/Transforms/Passes.h"
-#include "ir.h"
 #include "mlir/IR/Builders.h"
 #include "mlir/Pass/PassManager.h"
 #include "passes.h"
+#include "triton/Tools/OpBuilder.h"
 #include <pybind11/pybind11.h>
 #include <pybind11/stl.h>
 #include <pybind11/stl_bind.h>