From 28e96bbfd15220dc0e2fe5dfddce8e90562d6cde Mon Sep 17 00:00:00 2001
From: Yan Da <dyanab@connect.ust.hk>
Date: Fri, 8 Apr 2022 19:43:09 +0800
Subject: [PATCH] Remove the dependency on TensorDialect

---
 include/triton/ir/Dialect.h        |  2 --
 include/triton/ir/TritonDialect.td | 11 ++---------
 lib/ir/CMakeLists.txt              |  2 --
 3 files changed, 2 insertions(+), 13 deletions(-)

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
 )