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

Remove the dependency on TensorDialect

Browse Source
This commit is contained in:
Yan Da
2022-04-08 19:43:09 +08:00
parent a3d0812d27
commit 28e96bbfd1
3 changed files with 2 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
include/triton/ir/Dialect.h
View File

@@ -7,9 +7,7 @@
#include "mlir/IR/Dialect.h" #include "mlir/IR/Dialect.h"
#include "mlir/Interfaces/ControlFlowInterfaces.h" #include "mlir/Interfaces/ControlFlowInterfaces.h"
#include "mlir/Dialect/StandardOps/IR/Ops.h" #include "mlir/Dialect/StandardOps/IR/Ops.h"
#include "mlir/Dialect/Tensor/IR/Tensor.h"
#include "mlir/Dialect/SCF/SCF.h" #include "mlir/Dialect/SCF/SCF.h"
// #include "mlir/Dialect/ControlFlow/IR/ControlFlow.h"
#include "triton/ir/Dialect.h.inc" #include "triton/ir/Dialect.h.inc"

11
include/triton/ir/TritonDialect.td
View File

@@ -16,17 +16,12 @@ def Triton_Dialect : Dialect {
Dependent Dialects: Dependent Dialects:
* Arithmetic: * Arithmetic:
* addf, addi, andi, cmpf, cmpi, divf, fptosi, ... * addf, addi, andi, cmpf, cmpi, divf, fptosi, ...
* Tensor: * StructuredControlFlow:
* reshape (?) * ForOp, IfOp, WhileOp, YieldOp, ConditionOp
* ControlFlow:
* bf, cond_bf
* Func:
* call, return
}]; }];
let dependentDialects = [ let dependentDialects = [
"arith::ArithmeticDialect", "arith::ArithmeticDialect",
"tensor::TensorDialect",
"StandardOpsDialect", "StandardOpsDialect",
"scf::SCFDialect" "scf::SCFDialect"
@@ -35,8 +30,6 @@ def Triton_Dialect : Dialect {
// "func::FuncDialect" // "func::FuncDialect"
]; ];
// let useDefaultTypePrinterParser = 0;
let extraClassDeclaration = [{ let extraClassDeclaration = [{
void registerTypes(); void registerTypes();
}]; }];

2
lib/ir/CMakeLists.txt
View File

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