Skip to content

Commit 0ce3cec

Browse files
committed
Review suggestions
1 parent dbd8b51 commit 0ce3cec

File tree

1 file changed

+4
-12
lines changed

1 file changed

+4
-12
lines changed

mlir/tools/mlir-tblgen/RewriterGen.cpp

Lines changed: 4 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -1819,6 +1819,8 @@ void PatternEmitter::createAggregateLocalVarsForOpArgs(
18191819
"if (auto tmpAttr = {1}) {\n"
18201820
" tblgen_attrs.emplace_back(rewriter.getStringAttr(\"{0}\"), "
18211821
"tmpAttr);\n}\n";
1822+
const char *setterCmd = (useProperties) ? setPropCmd : addAttrCmd;
1823+
18221824
int numVariadic = 0;
18231825
bool hasOperandSegmentSizes = false;
18241826
std::vector<std::string> sizes;
@@ -1833,22 +1835,12 @@ void PatternEmitter::createAggregateLocalVarsForOpArgs(
18331835
PrintFatalError(loc, "only NativeCodeCall allowed in nested dag node "
18341836
"for creating attribute");
18351837

1836-
if (useProperties) {
1837-
os << formatv(setPropCmd, opArgName, childNodeNames.lookup(argIndex));
1838-
} else {
1839-
os << formatv(addAttrCmd, opArgName, childNodeNames.lookup(argIndex));
1840-
}
1838+
os << formatv(setterCmd, opArgName, childNodeNames.lookup(argIndex));
18411839
} else {
18421840
auto leaf = node.getArgAsLeaf(argIndex);
18431841
// The argument in the result DAG pattern.
18441842
auto patArgName = node.getArgName(argIndex);
1845-
if (useProperties) {
1846-
os << formatv(setPropCmd, opArgName,
1847-
handleOpArgument(leaf, patArgName));
1848-
} else {
1849-
os << formatv(addAttrCmd, opArgName,
1850-
handleOpArgument(leaf, patArgName));
1851-
}
1843+
os << formatv(setterCmd, opArgName, handleOpArgument(leaf, patArgName));
18521844
}
18531845
continue;
18541846
}

0 commit comments

Comments
 (0)