[BUILD] Fix includes (#49)

2022-08-11 11:49:29 -07:00
parent 83ef74f248
commit 3a48ca0d4d
2 changed files with 7 additions and 1 deletions
--- 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 {
--- 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