more progress on TritonGPU

This commit is contained in:
Yan Da
2022-04-28 18:51:31 +08:00
parent 012e8c5b2b
commit 2239ac1998
12 changed files with 115 additions and 57 deletions

View File

@@ -1 +1,2 @@
add_subdirectory(Triton)
add_subdirectory(TritonGPU)

View File

@@ -0,0 +1 @@
add_subdirectory(IR)

View File

@@ -0,0 +1,10 @@
add_mlir_dialect_library(TritonGPUIR
Dialect.cpp
DEPENDS
TritonGPUTableGen
TritonGPUAttrDefsIncGen
LINK_LIBS PUBLIC
TritonIR
)

View 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"
>();
}