[TRITONGPU] Added template for Triton -> TritonGPU conversion

This commit is contained in:
Phil Tillet
2022-04-30 14:31:18 -07:00
parent 2239ac1998
commit 2c6a213131
16 changed files with 146 additions and 2 deletions

View File

@@ -0,0 +1,18 @@
add_mlir_conversion_library(TritonToTritonGPU
TritonToTritonGPU.cpp
ADDITIONAL_HEADER_DIRS
${PROJECT_SOURCE_DIR}/include/triton/Conversion/TritonToTritonGPU
DEPENDS
TritonConversionPassIncGen
LINK_COMPONENTS
Core
LINK_LIBS PUBLIC
MLIRIR
MLIRPass
TritonIR
TritonGPUIR
)

View File

@@ -0,0 +1,36 @@
#include "mlir/Transforms/DialectConversion.h"
#include "triton/Dialect/Triton/IR/Dialect.h"
#include "triton/Conversion/TritonToTritonGPU/TritonToTritonGPU.h"
#include "../PassDetail.h"
using namespace mlir;
using namespace mlir::triton;
namespace {
class ConvertTritonToTritonGPU:
public ConvertTritonToTritonGPUBase<ConvertTritonToTritonGPU> {
public:
void getDependentDialects(DialectRegistry& registry) const override {
registry.insert<arith::ArithmeticDialect>();
registry.insert<StandardOpsDialect>();
registry.insert<scf::SCFDialect>();
// LLVM15
// registry.insert<cf::ControlFlowDialect>()
// registry.insert<func::FuncDialect>()
}
void runOnOperation() override {
MLIRContext *context = &getContext();
ConversionTarget target(*context);
std::cout << "Converting" << std::endl;
}
};
}
std::unique_ptr<OperationPass<ModuleOp>>
mlir::triton::createConvertTritonToTritonGPUPass() {
return std::make_unique<::ConvertTritonToTritonGPU>();
}