diff --git a/include/triton/ir/Dialect.h b/include/triton/ir/Dialect.h index 7234627e7..4c225a7bc 100644 --- a/include/triton/ir/Dialect.h +++ b/include/triton/ir/Dialect.h @@ -7,9 +7,7 @@ #include "mlir/IR/Dialect.h" #include "mlir/Interfaces/ControlFlowInterfaces.h" #include "mlir/Dialect/StandardOps/IR/Ops.h" -#include "mlir/Dialect/Tensor/IR/Tensor.h" #include "mlir/Dialect/SCF/SCF.h" -// #include "mlir/Dialect/ControlFlow/IR/ControlFlow.h" #include "triton/ir/Dialect.h.inc" diff --git a/include/triton/ir/TritonDialect.td b/include/triton/ir/TritonDialect.td index 100982365..c481b54f5 100644 --- a/include/triton/ir/TritonDialect.td +++ b/include/triton/ir/TritonDialect.td @@ -16,17 +16,12 @@ def Triton_Dialect : Dialect { Dependent Dialects: * Arithmetic: * addf, addi, andi, cmpf, cmpi, divf, fptosi, ... - * Tensor: - * reshape (?) - * ControlFlow: - * bf, cond_bf - * Func: - * call, return + * StructuredControlFlow: + * ForOp, IfOp, WhileOp, YieldOp, ConditionOp }]; let dependentDialects = [ "arith::ArithmeticDialect", - "tensor::TensorDialect", "StandardOpsDialect", "scf::SCFDialect" @@ -35,8 +30,6 @@ def Triton_Dialect : Dialect { // "func::FuncDialect" ]; - // let useDefaultTypePrinterParser = 0; - let extraClassDeclaration = [{ void registerTypes(); }]; diff --git a/lib/ir/CMakeLists.txt b/lib/ir/CMakeLists.txt index 83c61e047..4878b5621 100644 --- a/lib/ir/CMakeLists.txt +++ b/lib/ir/CMakeLists.txt @@ -15,7 +15,5 @@ add_mlir_dialect_library(TritonIR # MLIRFunc # else MLIRStandard - - MLIRTensor MLIRSCF )