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

Update upgrade some uses of mlir::interleave API to take container argument directly. 

PiperOrigin-RevId: 273446814
This commit is contained in:
Jing Pu 2019-10-07 21:52:36 -07:00 committed by A. Unique TensorFlower
parent a8a73f0640
commit 780f107a57
2 changed files with 3 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
mlir/lib/Dialect/Linalg/IR/LinalgOps.cpp
View File

@ -632,15 +632,11 @@ static LogicalResult verify(YieldOp op) {
static void printLinalgLibraryOp(OpAsmPrinter &p, Operation *op) { static void printLinalgLibraryOp(OpAsmPrinter &p, Operation *op) {
assert(op->getAbstractOperation() && "unregistered operation"); assert(op->getAbstractOperation() && "unregistered operation");
p << op->getName().getStringRef() << "("; p << op->getName().getStringRef() << "(";
interleave( interleaveComma(op->getOperands(), p, [&](Value *v) { p << *v; });
op->getOperands().begin(), op->getOperands().end(),
[&](Value *v) { p << *v; }, [&]() { p << ", "; });
p << ")"; p << ")";
p.printOptionalAttrDict(op->getAttrs()); p.printOptionalAttrDict(op->getAttrs());
p << " : "; p << " : ";
interleave( interleaveComma(op->getOperands(), p, [&](Value *v) { p << v->getType(); });
op->getOperands().begin(), op->getOperands().end(),
[&](Value *v) { p << v->getType(); }, [&]() { p << ", "; });
} }
static ParseResult parseLinalgLibraryOp(OpAsmParser &parser, static ParseResult parseLinalgLibraryOp(OpAsmParser &parser,

2
mlir/lib/IR/AsmPrinter.cpp
View File

@ -394,7 +394,7 @@ public:
template <typename Container, typename UnaryFunctor> template <typename Container, typename UnaryFunctor>
inline void interleaveComma(const Container &c, UnaryFunctor each_fn) const { inline void interleaveComma(const Container &c, UnaryFunctor each_fn) const {
interleave(c.begin(), c.end(), each_fn, [&]() { os << ", "; }); mlir::interleaveComma(c, os, each_fn);
} }
void print(ModuleOp module); void print(ModuleOp module);