Revert IREE changes made too early PiperOrigin-RevId: 335776150
diff --git a/iree/compiler/Conversion/LinalgToSPIRV/LinalgTileAndFusePass.cpp b/iree/compiler/Conversion/LinalgToSPIRV/LinalgTileAndFusePass.cpp index e0cd21f..219104e 100644 --- a/iree/compiler/Conversion/LinalgToSPIRV/LinalgTileAndFusePass.cpp +++ b/iree/compiler/Conversion/LinalgToSPIRV/LinalgTileAndFusePass.cpp
@@ -137,15 +137,12 @@ benefit), launchConfig(launchConfig) {} - LogicalResult matchAndRewrite(Operation *op, - PatternRewriter &rewriter) const override { + virtual LogicalResult matchAndRewrite(Operation *op, + PatternRewriter &rewriter) const { // Find the parent FuncOp before tiling. If tiling succeeds, the op will be // erased. FuncOp funcOp = op->getParentOfType<FuncOp>(); - SmallVector<Value, 4> tensorResults; - if (!funcOp || - failed(Base::matchAndRewriteBase(op, rewriter, tensorResults)) || - !tensorResults.empty() || + if (!funcOp || failed(Base::matchAndRewrite(op, rewriter)) || failed(updateWorkGroupSize(funcOp, launchConfig.getWorkgroupSize())) || (funcOp.getAttr(getNumWorkgroupsFnAttrName()) && failed(createNumWorkgroupsFromResultShape( @@ -352,13 +349,14 @@ return tileSizesVal; }; - auto getSubgroupProcInfoFn = [&launchConfig]( - OpBuilder &builder, Location loc, - ArrayRef<Range> parallelLoopRanges) { - ArrayRef<int64_t> numSubgroups = - launchConfig.getNumSubgroups().take_front(parallelLoopRanges.size()); - return getSubgroupIdsAndCounts(builder, loc, numSubgroups); - }; + auto getSubgroupProcInfoFn = + [&launchConfig](OpBuilder &builder, Location loc, + ArrayRef<Range> parallelLoopRanges) { + ArrayRef<int64_t> numSubgroups = + launchConfig.getNumSubgroups().take_front( + parallelLoopRanges.size()); + return getSubgroupIdsAndCounts(builder, loc, numSubgroups); + }; linalg::LinalgLoopDistributionOptions subgroupDistributionOptions = { getSubgroupProcInfoFn, {linalg::DistributionMethod::CyclicNumProcsEqNumIters,