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

Update comments to reflect the new names of `GenericOp` constructor parameters 

PiperOrigin-RevId: 321795872
This commit is contained in:
A. Unique TensorFlower 2020-07-17 16:43:03 +00:00 committed by Mehdi Amini
parent c44e08351d
commit 30fa3db949
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
lib/Dialect/mhlo/transforms/legalize_to_linalg.cc
View File

@ -298,8 +298,8 @@ class DataMovementOpConverter : public OpConversionPattern<OpTy> {
auto nloops = resultType.getRank(); auto nloops = resultType.getRank();
auto loc = op.getLoc(); auto loc = op.getLoc();
auto linalgOp = rewriter.create<linalg::GenericOp>( auto linalgOp = rewriter.create<linalg::GenericOp>(
loc, isLHLO ? ArrayRef<Type>{} : resultType, args, /*inputCount=*/1, loc, isLHLO ? ArrayRef<Type>{} : resultType, args, /*argsIn=*/1,
/*outputCount=*/1, indexing_maps, GetNParallelLoopsAttrs(nloops), /*argsOut=*/1, indexing_maps, GetNParallelLoopsAttrs(nloops),
[&](OpBuilder& nestedBuilder, Location nestedLoc, ValueRange args) { [&](OpBuilder& nestedBuilder, Location nestedLoc, ValueRange args) {
nestedBuilder.create<linalg::YieldOp>(loc, *args.begin()); nestedBuilder.create<linalg::YieldOp>(loc, *args.begin());
}); });
@ -420,7 +420,7 @@ class LhloBroadcastInDimConverter
rewriter.create<LoadOp>(loc, operand, llvm::makeArrayRef({zero})); rewriter.create<LoadOp>(loc, operand, llvm::makeArrayRef({zero}));
rewriter.create<linalg::GenericOp>( rewriter.create<linalg::GenericOp>(
loc, llvm::None, llvm::makeArrayRef(operand_adaptor.output()), loc, llvm::None, llvm::makeArrayRef(operand_adaptor.output()),
/*inputCount=*/0, /*outputCount=*/1, /*argsIn=*/0, /*argsOut=*/1,
llvm::makeArrayRef(rewriter.getMultiDimIdentityMap(nloops)), llvm::makeArrayRef(rewriter.getMultiDimIdentityMap(nloops)),
GetNParallelLoopsAttrs(nloops), GetNParallelLoopsAttrs(nloops),
[&](OpBuilder& nestedBuilder, Location nestedLoc, ValueRange args) { [&](OpBuilder& nestedBuilder, Location nestedLoc, ValueRange args) {
@ -433,7 +433,7 @@ class LhloBroadcastInDimConverter
rewriter.create<linalg::GenericOp>( rewriter.create<linalg::GenericOp>(
loc, llvm::None, loc, llvm::None,
llvm::makeArrayRef({operand, operand_adaptor.output()}), llvm::makeArrayRef({operand, operand_adaptor.output()}),
/*inputCount=*/1, /*outputCount=*/1, indexing_maps, /*argsIn=*/1, /*argsOut=*/1, indexing_maps,
GetNParallelLoopsAttrs(nloops), GetNParallelLoopsAttrs(nloops),
[&](OpBuilder& nestedBuilder, Location nestedLoc, ValueRange args) { [&](OpBuilder& nestedBuilder, Location nestedLoc, ValueRange args) {
nestedBuilder.create<linalg::YieldOp>(loc, *args.begin()); nestedBuilder.create<linalg::YieldOp>(loc, *args.begin());