Remove the dependency on TensorDialect

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"
 

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

lib/ir/CMakeLists.txt

@@ -15,7 +15,5 @@ add_mlir_dialect_library(TritonIR
   # MLIRFunc
   # else
   MLIRStandard
-
-  MLIRTensor
   MLIRSCF
 )