diff --git a/include/triton/driver/llvm.h b/include/triton/driver/llvm.h
index b3ce0d0cc..a46eb66b3 100644
--- a/include/triton/driver/llvm.h
+++ b/include/triton/driver/llvm.h
@@ -1,4 +1,5 @@
-#include "triton/driver/dispatch.h"
+#include "triton/external/CUDA/cuda.h"
+#include "triton/external/hip.h"
 #include <string>
 
 namespace llvm {
diff --git a/include/triton/external/hip.h b/include/triton/external/hip.h
index a099f857b..a295eed68 100644
--- a/include/triton/external/hip.h
+++ b/include/triton/external/hip.h
@@ -1,3 +1,6 @@
+#ifndef __external_hip_h__
+#define __external_hip_h__
+
 /*
  * @brief hipError_t
  * @enum
@@ -286,3 +289,5 @@ typedef enum hipFuncCache_t {
 #define HIP_LAUNCH_PARAM_BUFFER_POINTER ((void*)0x01)
 #define HIP_LAUNCH_PARAM_BUFFER_SIZE ((void*)0x02)
 #define HIP_LAUNCH_PARAM_END ((void*)0x03)
+
+#endif
\ No newline at end of file