more progress on TritonGPU
This commit is contained in:
@@ -1 +1,2 @@
|
||||
add_subdirectory(Triton)
|
||||
add_subdirectory(TritonGPU)
|
||||
|
1
lib/Dialect/TritonGPU/CMakeLists.txt
Normal file
1
lib/Dialect/TritonGPU/CMakeLists.txt
Normal file
@@ -0,0 +1 @@
|
||||
add_subdirectory(IR)
|
10
lib/Dialect/TritonGPU/IR/CMakeLists.txt
Normal file
10
lib/Dialect/TritonGPU/IR/CMakeLists.txt
Normal file
@@ -0,0 +1,10 @@
|
||||
add_mlir_dialect_library(TritonGPUIR
|
||||
Dialect.cpp
|
||||
|
||||
DEPENDS
|
||||
TritonGPUTableGen
|
||||
TritonGPUAttrDefsIncGen
|
||||
|
||||
LINK_LIBS PUBLIC
|
||||
TritonIR
|
||||
)
|
12
lib/Dialect/TritonGPU/IR/Dialect.cpp
Normal file
12
lib/Dialect/TritonGPU/IR/Dialect.cpp
Normal file
@@ -0,0 +1,12 @@
|
||||
#include "triton/Dialect/TritonGPU/IR/Dialect.h"
|
||||
|
||||
#include "triton/Dialect/TritonGPU/IR/Dialect.cpp.inc"
|
||||
|
||||
using namespace mlir::triton::gpu;
|
||||
|
||||
void TritonGPUDialect::initialize() {
|
||||
addOperations<
|
||||
#define GET_OP_LIST
|
||||
#include "triton/Dialect/TritonGPU/IR/Ops.cpp.inc"
|
||||
>();
|
||||
}
|
Reference in New Issue
Block a user