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

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

This is a preparation step to remove those methods from OpState.

PiperOrigin-RevId: 348010582
This commit is contained in:
Christian Sigg 2020-12-17 06:28:06 -08:00 committed by TensorFlow MLIR Team
parent 1f244c3e2c
commit dc7e63f74c
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -185,8 +185,8 @@ struct HloToLhloCustomCallOpConverter
// for args and outputs. // for args and outputs.
const int32_t segments[2] = {static_cast<int32_t>(operands.size()), const int32_t segments[2] = {static_cast<int32_t>(operands.size()),
static_cast<int32_t>(op->getNumResults())}; static_cast<int32_t>(op->getNumResults())};
lhloOp.setAttr(lhloOp.getOperandSegmentSizeAttr(), lhloOp->setAttr(lhloOp.getOperandSegmentSizeAttr(),
rewriter.getI32VectorAttr(segments)); rewriter.getI32VectorAttr(segments));
rewriter.replaceOp(op, ArrayRef<Value>(buffer_args).slice(operands.size())); rewriter.replaceOp(op, ArrayRef<Value>(buffer_args).slice(operands.size()));
return success(); return success();