maxjhandsome
/
llvm-project
forked from OSchip/llvm-project
1
0
Fork 0
Code Issues Pull Requests Packages Releases Wiki Activity

[mlir] NFC - Add AffineMinMaxOpBase getDim/SymbolOperands

This commit is contained in:
Nicolas Vasilache 2020-06-02 18:04:10 -04:00
parent 16437992ca
commit 7de4587805
1 changed files with 8 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
mlir/include/mlir/Dialect/Affine/IR/AffineOps.td
View File

@ -460,6 +460,14 @@ class AffineMinMaxOpBase<string mnemonic, list<OpTrait> traits = []> :
static StringRef getMapAttrName() { return "map"; }
AffineMap getAffineMap() { return map(); }
ValueRange getMapOperands() { return operands(); }
ValueRange getDimOperands() {
return OperandRange{operands().begin(),
operands().begin() + map().getNumDims()};
}
ValueRange getSymbolOperands() {
return OperandRange{operands().begin() + map().getNumDims(),
operands().end()};
}
}];
let verifier = [{ return ::verifyAffineMinMaxOp(*this); }];
let printer = [{ return ::printAffineMinMaxOp(p, *this); }];