From 3a48ca0d4d6cada8f62897e6b35031e22eabbb9f Mon Sep 17 00:00:00 2001
From: Philippe Tillet <phil@openai.com>
Date: Thu, 11 Aug 2022 11:49:29 -0700
Subject: [PATCH] [BUILD] Fix includes (#49)

---
 include/triton/driver/llvm.h  | 3 ++-
 include/triton/external/hip.h | 5 +++++
 2 files changed, 7 insertions(+), 1 deletion(-)

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