File tree Expand file tree Collapse file tree 1 file changed +10
-10
lines changed
mlir/lib/Conversion/Normalize Expand file tree Collapse file tree 1 file changed +10
-10
lines changed Original file line number Diff line number Diff line change @@ -285,19 +285,19 @@ void NormalizePass::foldOperation(mlir::Operation *op) {
285285 Operands.push_back (name);
286286 }
287287 } else if (auto ba = dyn_cast<mlir::BlockArgument>(operand)) {
288- mlir::Block *ownerBlock = ba.getOwner ();
289- unsigned argIndex = ba.getArgNumber ();
290- if (auto func =
291- dyn_cast<mlir::func::FuncOp>(ownerBlock->getParentOp ())) {
292- if (&func.front () == ownerBlock) {
293- Operands.push_back (std::string (" funcArg" + std::to_string (argIndex)));
294- } else {
295- Operands.push_back (std::string (" blockArg" + std::to_string (argIndex)));
296- }
288+ mlir::Block *ownerBlock = ba.getOwner ();
289+ unsigned argIndex = ba.getArgNumber ();
290+ if (auto func = dyn_cast<mlir::func::FuncOp>(ownerBlock->getParentOp ())) {
291+ if (&func.front () == ownerBlock) {
292+ Operands.push_back (std::string (" funcArg" + std::to_string (argIndex)));
297293 } else {
298- Operands.push_back (std::string (" blockArg" + std::to_string (argIndex)));
294+ Operands.push_back (
295+ std::string (" blockArg" + std::to_string (argIndex)));
299296 }
297+ } else {
298+ Operands.push_back (std::string (" blockArg" + std::to_string (argIndex)));
300299 }
300+ }
301301 }
302302
303303 if (op->hasTrait <OpTrait::IsCommutative>())
You can’t perform that action at this time.
0 commit comments