gaspersic/triton
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Move dependentDialects from .cpp to .td

Browse Source
This commit is contained in:
Yan Da
2022-05-01 13:06:51 +08:00
parent d9017f8593
commit 4ece9fd1f3
2 changed files with 6 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
include/triton/Conversion/Passes.td
View File

@@ -9,6 +9,11 @@ def ConvertTritonToTritonGPU: Pass<"convert-triton-to-tritongpu", "mlir::ModuleO
}]; }];
let constructor = "mlir::triton::createConvertTritonToTritonGPUPass()"; let constructor = "mlir::triton::createConvertTritonToTritonGPUPass()";
let dependentDialects = ["mlir::arith::ArithmeticDialect",
"mlir::StandardOpsDialect",
// TODO: Does this pass depend on SCF?
"mlir::scf::SCFDialect"];
} }
#endif #endif

11
lib/Conversion/TritonToTritonGPU/TritonToTritonGPU.cpp
View File

@@ -51,15 +51,6 @@ class ConvertTritonToTritonGPU:
public ConvertTritonToTritonGPUBase<ConvertTritonToTritonGPU> { public ConvertTritonToTritonGPUBase<ConvertTritonToTritonGPU> {
public: 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 { void runOnOperation() override {
MLIRContext *context = &getContext(); MLIRContext *context = &getContext();
ConversionTarget target(*context); ConversionTarget target(*context);
@@ -83,4 +74,4 @@ public:
std::unique_ptr<OperationPass<ModuleOp>> std::unique_ptr<OperationPass<ModuleOp>>
mlir::triton::createConvertTritonToTritonGPUPass() { mlir::triton::createConvertTritonToTritonGPUPass() {
return std::make_unique<::ConvertTritonToTritonGPU>(); return std::make_unique<::ConvertTritonToTritonGPU>();
} }