diff --git a/include/triton/Conversion/Passes.td b/include/triton/Conversion/Passes.td index 2e10e0a09..b41964657 100644 --- a/include/triton/Conversion/Passes.td +++ b/include/triton/Conversion/Passes.td @@ -9,6 +9,11 @@ def ConvertTritonToTritonGPU: Pass<"convert-triton-to-tritongpu", "mlir::ModuleO }]; let constructor = "mlir::triton::createConvertTritonToTritonGPUPass()"; + + let dependentDialects = ["mlir::arith::ArithmeticDialect", + "mlir::StandardOpsDialect", + // TODO: Does this pass depend on SCF? + "mlir::scf::SCFDialect"]; } #endif diff --git a/lib/Conversion/TritonToTritonGPU/TritonToTritonGPU.cpp b/lib/Conversion/TritonToTritonGPU/TritonToTritonGPU.cpp index 8b8ceb289..5387df5d6 100644 --- a/lib/Conversion/TritonToTritonGPU/TritonToTritonGPU.cpp +++ b/lib/Conversion/TritonToTritonGPU/TritonToTritonGPU.cpp @@ -51,15 +51,6 @@ class ConvertTritonToTritonGPU: public ConvertTritonToTritonGPUBase { public: - void getDependentDialects(DialectRegistry& registry) const override { - registry.insert(); - registry.insert(); - registry.insert(); - // LLVM15 - // registry.insert() - // registry.insert() - } - void runOnOperation() override { MLIRContext *context = &getContext(); ConversionTarget target(*context); @@ -83,4 +74,4 @@ public: std::unique_ptr> mlir::triton::createConvertTritonToTritonGPUPass() { return std::make_unique<::ConvertTritonToTritonGPU>(); -} \ No newline at end of file +}