colin
/
mlir-hlo
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Use mlir::OpState::operator->() to get to Operation::getAttrs(). 

This is a preparation step to remove getAttrs() from OpState.

PiperOrigin-RevId: 360159716
This commit is contained in:
Christian Sigg 2021-03-01 04:51:42 -08:00 committed by TensorFlow MLIR Team
parent 7c071e8ee6
commit 70ee9369d5
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
lib/Dialect/mhlo/transforms/hlo_legalize_to_lhlo.cc
View File

@ -424,7 +424,7 @@ struct HloToLhloReduceOpConverter : public BaseOpConversion<mhlo::ReduceOp> {
buffer_args.push_back(InsertAlloc(loc, result, &rewriter)); buffer_args.push_back(InsertAlloc(loc, result, &rewriter));
} }
auto new_op = rewriter.create<lmhlo::ReduceOp>(loc, llvm::None, buffer_args, auto new_op = rewriter.create<lmhlo::ReduceOp>(loc, llvm::None, buffer_args,
op.getAttrs()); op->getAttrs());
// Copy over the operations inside the region. // Copy over the operations inside the region.
rewriter.inlineRegionBefore(op.body(), new_op.body(), new_op.body().end()); rewriter.inlineRegionBefore(op.body(), new_op.body(), new_op.body().end());