Searched refs:parallelLoopRanges (Results 1 – 3 of 3) sorted by relevance
| /llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Utils/ |
| H A D | Utils.cpp | 514 SmallVector<Range, 2> parallelLoopRanges; in doit() local
517 parallelLoopRanges.push_back(loopRanges[iteratorType.index()]); in doit()
521 if (distributionMethod.size() < parallelLoopRanges.size()) in doit()
522 parallelLoopRanges.resize(distributionMethod.size()); in doit()
523 procInfo = distributionOptions->procInfo(b, loc, parallelLoopRanges); in doit()
755 SmallVector<Range, 2> parallelLoopRanges; in doit() local
758 parallelLoopRanges.push_back(loopRanges[iteratorType.index()]); in doit()
760 if (distributionMethod.size() < parallelLoopRanges.size()) in doit()
761 parallelLoopRanges.resize(distributionMethod.size()); in doit()
763 options.procInfo(b, loc, parallelLoopRanges); in doit()
|
| /llvm-project-15.0.7/mlir/test/lib/Dialect/Linalg/ |
| H A D | TestLinalgTransforms.cpp | 252 getGpuProcIds(OpBuilder &b, Location loc, ArrayRef<Range> parallelLoopRanges) { in getGpuProcIds() argument
253 size_t count = std::min<size_t>(3, parallelLoopRanges.size()); in getGpuProcIds()
|
| /llvm-project-15.0.7/mlir/include/mlir/Dialect/Linalg/Utils/ |
| H A D | Utils.h | 344 OpBuilder &b, Location loc, ArrayRef<Range> parallelLoopRanges)>;
|