[Mlir-commits] [mlir] [mlir][PartialReductionTilingInterface] Generalize implementation of `tileUsingSCF` for `ReductionTilingStrategy::PartialOuterReduction`. (PR #143467)
llvmlistbot at llvm.org
llvmlistbot at llvm.org
Sat Jun 21 16:41:53 PDT 2025
================
@@ -115,13 +108,10 @@ struct SCFTilingOptions {
return *this;
}
- /// Specify mapping of loops to devices. This is only respected when the loop
- /// constructs support such a mapping (like `scf.forall`). Will be ignored
- /// when using loop constructs that dont support such a mapping (like
- /// `scf.for`)
- SmallVector<Attribute> mappingVector = {};
- SCFTilingOptions &setMapping(ArrayRef<Attribute> mapping) {
- mappingVector = llvm::to_vector(mapping);
+ /// Specify the reduction dimensions to be tiled.
+ SetVector<unsigned> reductionDims;
+ SCFTilingOptions &setReductionDims(ArrayRef<unsigned> dims) {
+ reductionDims.insert(dims.begin(), dims.end());
----------------
MaheshRavishankar wrote:
It says here https://github.com/llvm/llvm-project/pull/143467/files#diff-3de157757f965faf44ff1ea880201c88a70b6d0b68f0e10bf6bc9032b2ee7669R1770 . You can use it to specify a subset of reduction dimensions to tile. If you dont specify, then actually nothing happens. Added the documentation for that.
https://github.com/llvm/llvm-project/pull/143467
More information about the Mlir-commits
mailing list