Skip to content

Commit 639c7cf

Browse files
authored
Carries the revert of llvm/llvm-project@b4c31dc Fixups: - llvm/llvm-project@32a5adb - Rename `createLinalgNamedOpConversionPass` to `createSimplifyDepthwiseConvPass` - llvm/llvm-project@003cbbd - Add `std::nullopt` arg for `populateGpuPromoteShuffleToAMDGPUPatterns` - llvm/llvm-project@613ec4c - Rename `getIndexVec()` to `getIndices()` - Rename `getIndices()` to `getOffsets()` --------- Signed-off-by: Ian Wood <[email protected]>
1 parent 44b9780 commit 639c7cf

File tree

6 files changed

+7
-7
lines changed

6 files changed

+7
-7
lines changed

compiler/src/iree/compiler/Codegen/Common/GPU/GPUDistributionPatterns.cpp

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -286,7 +286,7 @@ struct DistributeGather final : OpDistributionPattern<vector::GatherOp> {
286286
DistributionSignature &signature,
287287
PatternRewriter &rewriter) const override {
288288
VectorValue result = gatherOp.getResult();
289-
VectorValue indexVec = gatherOp.getIndexVec();
289+
VectorValue indexVec = gatherOp.getIndices();
290290
VectorValue mask = gatherOp.getMask();
291291
VectorValue passThru = gatherOp.getPassThru();
292292

@@ -318,7 +318,7 @@ struct DistributeGather final : OpDistributionPattern<vector::GatherOp> {
318318
// Simply distribute all operands and results.
319319
VectorValue distributed = rewriter.create<vector::GatherOp>(
320320
gatherOp.getLoc(), distributedType, gatherOp.getBase(),
321-
gatherOp.getIndices(),
321+
gatherOp.getOffsets(),
322322
getDistributed(rewriter, indexVec, indicesLayout),
323323
getDistributed(rewriter, mask, maskLayout),
324324
getDistributed(rewriter, passThru, passThruLayout));

compiler/src/iree/compiler/Codegen/LLVMGPU/ConvertToROCDL.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -290,7 +290,7 @@ struct ConvertToROCDLPass final
290290
{
291291
RewritePatternSet patterns(&getContext());
292292
populateGpuRewritePatterns(patterns);
293-
populateGpuPromoteShuffleToAMDGPUPatterns(patterns);
293+
populateGpuPromoteShuffleToAMDGPUPatterns(patterns, std::nullopt);
294294
populateGpuSubgroupIdPatterns(patterns);
295295
if (failed(applyPatternsGreedily(m, std::move(patterns)))) {
296296
return signalPassFailure();

compiler/src/iree/compiler/GlobalOptimization/Passes.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -115,7 +115,7 @@ void buildGlobalOptimizationPassPipeline(
115115
.addPass(IREE::Flow::createCanonicalizePass)
116116
.addPass(createRemoveZeroExtentTensorsPass)
117117
.addPass(createDetachElementwiseFromNamedOpsPass)
118-
.addPass(mlir::createLinalgNamedOpConversionPass);
118+
.addPass(mlir::createSimplifyDepthwiseConvPass);
119119
mainPassManager.addPass(createEraseUnusedLinalgOperandsPass());
120120

121121
// Expand tensor shapes into SSA values and optimize the whole program.

compiler/src/iree/compiler/GlobalOptimization/QuantizedConvToConv.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -346,7 +346,7 @@ class LinalgQuantizedConvToConvPass final
346346
Operation *op = getOperation();
347347
MLIRContext *context = op->getContext();
348348
RewritePatternSet patterns(context);
349-
linalg::populateLinalgNamedOpConversionPatterns(patterns);
349+
linalg::populateSimplifyDepthwiseConvPatterns(patterns);
350350
patterns.add<QuantizedConvToConv, QuantizedDepthwiseConvToDepthwiseConv>(
351351
context);
352352
memref::populateResolveRankedShapedTypeResultDimsPatterns(patterns);

compiler/src/iree/compiler/Preprocessing/Passes.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -119,7 +119,7 @@ buildTransposeConvolutionPassPipeline(OpPassManager &passManager,
119119
const TransformOptions &options) {
120120
FunctionLikeNest(passManager)
121121
.addPass(GlobalOptimization::createDetachElementwiseFromNamedOpsPass)
122-
.addPass(mlir::createLinalgNamedOpConversionPass)
122+
.addPass(mlir::createSimplifyDepthwiseConvPass)
123123
.addPass(createConvertConvToChannelsLastPass)
124124
.addPass(GlobalOptimization::createConvert1X1FilterConv2DToMatmulPass)
125125
.addPass(createConvertConvFilterToChannelsLastPass);

third_party/llvm-project

Submodule llvm-project updated 2485 files

0 commit comments

Comments
 (0)