COMET/triton.patch at master · pnnl/COMET · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
diff --git a/CMakeLists.txt b/CMakeLists.txt
index a892a666d..5d2099b58 100644
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -55,6 +55,7 @@ endif()
 include_directories(${CMAKE_CURRENT_SOURCE_DIR}/include)
 set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -D__STDC_FORMAT_MACROS  -fPIC -std=gnu++17")

+add_compile_options(-w)

 # #########
 # LLVM
@@ -107,7 +108,7 @@ endfunction()


 # Disable warnings that show up in external code (gtest;pybind11)
-set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -Werror -Wno-covered-switch-default -fvisibility=hidden")
+set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS}  -Wno-covered-switch-default -fvisibility=hidden")

 include_directories(".")
 include_directories(${MLIR_INCLUDE_DIRS})
@@ -290,8 +291,8 @@ if(NOT TRITON_BUILD_PYTHON_MODULE)
 endif()

 add_subdirectory(third_party/f2reduce)
-add_subdirectory(bin)
-add_subdirectory(test)
+#add_subdirectory(bin)
+#add_subdirectory(test)

 if(TRITON_BUILD_UT)
   add_subdirectory(unittest)
diff --git a/third_party/amd/CMakeLists.txt b/third_party/amd/CMakeLists.txt
index 8228c3d39..ccc124c80 100644
--- a/third_party/amd/CMakeLists.txt
+++ b/third_party/amd/CMakeLists.txt
@@ -5,6 +5,6 @@ add_subdirectory(lib)
 if(TRITON_BUILD_PYTHON_MODULE)
   add_triton_plugin(TritonAMD ${CMAKE_CURRENT_SOURCE_DIR}/python/triton_amd.cc LINK_LIBS TritonAMDGPUToLLVM TritonAMDGPUTransforms TritonAMDGPUDialectToLLVM)
 endif()
-if(TRITON_BUILD_UT)
-  add_subdirectory(unittest)
-endif()
+#if(TRITON_BUILD_UT)
+#  add_subdirectory(unittest)
+#endif()
diff --git a/third_party/amd/include/Dialect/TritonAMDGPU/IR/CMakeLists.txt b/third_party/amd/include/Dialect/TritonAMDGPU/IR/CMakeLists.txt
index 25a57075b..4d23ecf7d 100644
--- a/third_party/amd/include/Dialect/TritonAMDGPU/IR/CMakeLists.txt
+++ b/third_party/amd/include/Dialect/TritonAMDGPU/IR/CMakeLists.txt
@@ -1,8 +1,8 @@
 set(MLIR_BINARY_DIR ${CMAKE_BINARY_DIR})

 set(LLVM_TARGET_DEFINITIONS TritonAMDGPUOps.td)
-mlir_tablegen(Dialect.h.inc -gen-dialect-decls -dialect=amdgpu)
-mlir_tablegen(Dialect.cpp.inc -gen-dialect-defs -dialect=amdgpu)
+mlir_tablegen(Dialect.h.inc -gen-dialect-decls -dialect=ttamdgpu)
+mlir_tablegen(Dialect.cpp.inc -gen-dialect-defs -dialect=ttamdgpu)
 mlir_tablegen(OpsConversions.inc -gen-llvmir-conversions)
 mlir_tablegen(Ops.h.inc -gen-op-decls)
 mlir_tablegen(Ops.cpp.inc -gen-op-defs)
diff --git a/third_party/amd/include/Dialect/TritonAMDGPU/IR/TritonAMDGPUDialect.td b/third_party/amd/include/Dialect/TritonAMDGPU/IR/TritonAMDGPUDialect.td
index c0c18b07e..25f56039a 100644
--- a/third_party/amd/include/Dialect/TritonAMDGPU/IR/TritonAMDGPUDialect.td
+++ b/third_party/amd/include/Dialect/TritonAMDGPU/IR/TritonAMDGPUDialect.td
@@ -27,7 +27,7 @@
 include "mlir/IR/OpBase.td"

 def TritonAMDGPU_Dialect : Dialect {
-  let name = "amdgpu";
+  let name = "ttamdgpu";
   let cppNamespace = "::mlir::triton::amdgpu";

   let description = [{
diff --git a/third_party/nvidia/include/Dialect/NVGPU/IR/CMakeLists.txt b/third_party/nvidia/include/Dialect/NVGPU/IR/CMakeLists.txt
index f8932cdc4..37fb66417 100644
--- a/third_party/nvidia/include/Dialect/NVGPU/IR/CMakeLists.txt
+++ b/third_party/nvidia/include/Dialect/NVGPU/IR/CMakeLists.txt
@@ -1,8 +1,8 @@
 set(MLIR_BINARY_DIR ${CMAKE_BINARY_DIR})

 set(LLVM_TARGET_DEFINITIONS NVGPUOps.td)
-mlir_tablegen(Dialect.h.inc -gen-dialect-decls -dialect=nvgpu)
-mlir_tablegen(Dialect.cpp.inc -gen-dialect-defs -dialect=nvgpu)
+mlir_tablegen(Dialect.h.inc -gen-dialect-decls -dialect=ttnvgpu)
+mlir_tablegen(Dialect.cpp.inc -gen-dialect-defs -dialect=ttnvgpu)
 mlir_tablegen(OpsConversions.inc -gen-llvmir-conversions)
 mlir_tablegen(Ops.h.inc -gen-op-decls)
 mlir_tablegen(Ops.cpp.inc -gen-op-defs)
diff --git a/third_party/nvidia/include/Dialect/NVGPU/IR/NVGPUDialect.td b/third_party/nvidia/include/Dialect/NVGPU/IR/NVGPUDialect.td
index 6978173d4..3b76e10c7 100644
--- a/third_party/nvidia/include/Dialect/NVGPU/IR/NVGPUDialect.td
+++ b/third_party/nvidia/include/Dialect/NVGPU/IR/NVGPUDialect.td
@@ -25,7 +25,7 @@
 include "mlir/IR/OpBase.td"

 def NVGPU_Dialect : Dialect {
-  let name = "nvgpu";
+  let name = "ttnvgpu";
   let cppNamespace = "::mlir::triton::nvgpu";

   let description = [{