| /llvm-project-15.0.7/mlir/lib/Conversion/SCFToGPU/ |
| H A D | SCFToGPU.cpp | 154 Optional<AffineForOp> collectBounds(AffineForOp forOp, unsigned numLoops);
183 AffineLoopToGpuConverter::collectBounds(AffineForOp forOp, unsigned numLoops) { in collectBounds() argument
185 dims.reserve(numLoops); in collectBounds()
186 lbs.reserve(numLoops); in collectBounds()
187 ivs.reserve(numLoops); in collectBounds()
188 steps.reserve(numLoops); in collectBounds()
190 for (unsigned i = 0; i < numLoops; ++i) { in collectBounds()
208 if (i != numLoops - 1) in collectBounds()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Transforms/ |
| H A D | HoistPadding.cpp | 57 HoistingAnalysis(tensor::PadOp padOp, int numLoops);
168 HoistingAnalysis::HoistingAnalysis(tensor::PadOp padOp, int numLoops) { in HoistingAnalysis() argument
178 getAtMostNEnclosingLoops(padOp, numLoops, reverseEnclosingLoops); in HoistingAnalysis()
398 tensor::PadOp opToHoist, int numLoops, ArrayRef<int64_t> transposeVector, in hoistPaddingOnTensors() argument
400 LLVM_DEBUG(DBGS() << "Try to hoist " << *(opToHoist) << " by " << numLoops in hoistPaddingOnTensors()
402 HoistingAnalysis analysis(opToHoist, numLoops); in hoistPaddingOnTensors()
|
| H A D | TilingInterfaceImpl.cpp | 156 auto numLoops = linalgOp.getNumLoops(); in generateResultTileValue() local
158 SmallVector<OpFoldResult> iterationTileOffsets(numLoops), in generateResultTileValue()
159 iterationTileSizes(numLoops); in generateResultTileValue()
|
| /llvm-project-15.0.7/mlir/test/lib/Dialect/Affine/ |
| H A D | TestLoopFusion.cpp | 166 unsigned numLoops = loops.size(); in iterateLoops() local
167 for (unsigned j = 0; j < numLoops; ++j) { in iterateLoops()
168 for (unsigned k = 0; k < numLoops; ++k) { in iterateLoops()
|
| /llvm-project-15.0.7/mlir/include/mlir/Dialect/SparseTensor/Utils/ |
| H A D | Merger.h | 156 : outTensor(t - 1), syntheticTensor(t), numTensors(t + 1), numLoops(l), in Merger()
236 assert(t < numTensors && i < numLoops); in isDim()
297 const unsigned numLoops; variable
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Async/Transforms/ |
| H A D | AsyncParallelFor.cpp | 143 unsigned numLoops; member
155 unsigned numLoops; member
166 return args.drop_front(2).take_front(numLoops); in tripCounts()
170 return args.drop_front(2 + 1 * numLoops).take_front(numLoops); in lowerBounds()
174 return args.drop_front(2 + 2 * numLoops).take_front(numLoops); in upperBounds()
178 return args.drop_front(2 + 3 * numLoops).take_front(numLoops); in steps()
182 return args.drop_front(2 + 4 * numLoops); in captures()
|
| /llvm-project-15.0.7/mlir/include/mlir/Dialect/Linalg/Transforms/ |
| H A D | HoistPadding.h | 71 tensor::PadOp opToHoist, int numLoops, ArrayRef<int64_t> transposeVector,
|
| /llvm-project-15.0.7/mlir/unittests/Dialect/SparseTensor/ |
| H A D | MergerTest.cpp | 128 MergerTestBase(unsigned numTensors, unsigned numLoops) in MergerTestBase() argument
129 : numTensors(numTensors), numLoops(numLoops), in MergerTestBase()
130 merger(numTensors, numLoops) {} in MergerTestBase()
292 unsigned numLoops; member in __anonb1ca32140111::MergerTestBase
|
| /llvm-project-15.0.7/mlir/lib/Dialect/SparseTensor/Transforms/ |
| H A D | Sparsification.cpp | 57 CodeGen(SparsificationOptions o, unsigned numTensors, unsigned numLoops, in CodeGen()
59 : options(o), loops(numLoops), sizes(numLoops), buffers(numTensors), in CodeGen()
60 pointers(numTensors, std::vector<Value>(numLoops)), in CodeGen()
61 indices(numTensors, std::vector<Value>(numLoops)), in CodeGen()
62 highs(numTensors, std::vector<Value>(numLoops)), in CodeGen()
63 pidxs(numTensors, std::vector<Value>(numLoops)), in CodeGen()
337 unsigned numLoops = iteratorTypes.size(); in isAdmissableTensorExp() local
338 for (unsigned i = 0; i < numLoops; i++) in isAdmissableTensorExp()
354 for (unsigned i = 0; i < numLoops; i++) { in isAdmissableTensorExp()
1721 unsigned numLoops = op.iterator_types().getValue().size(); in matchAndRewrite() local
[all …]
|
| /llvm-project-15.0.7/mlir/lib/Dialect/SCF/Transforms/ |
| H A D | TileUsingInterface.cpp | 173 size_t numLoops = iterationDomain.size(); in returningMatchAndRewrite() local
174 if (numLoops == 0) { in returningMatchAndRewrite()
187 tileSizeVector.append(numLoops - tileSizeVector.size(), zero); in returningMatchAndRewrite()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Linalg/TransformOps/ |
| H A D | LinalgTransformOps.cpp | 103 unsigned numLoops, in applyTilingToAll() argument
107 SmallVector<SmallVector<Operation *>> loopOps(numLoops); in applyTilingToAll()
108 for (unsigned int i = 0; i < numLoops; ++i) in applyTilingToAll()
121 if (tiled->loops.size() != numLoops) in applyTilingToAll()
126 for (unsigned int i = 0; i < numLoops; ++i) in applyTilingToAll()
131 for (unsigned int i = 0; i < numLoops; ++i) in applyTilingToAll()
|
| /llvm-project-15.0.7/mlir/test/lib/Dialect/Vector/ |
| H A D | TestVectorTransforms.cpp | 520 int64_t numLoops = 0; in runOnOperation() local
522 numLoops = readOp.getVectorType().getRank(); in runOnOperation()
524 numLoops = writeOp.getVectorType().getRank(); in runOnOperation()
527 auto order = llvm::reverse(llvm::seq<int64_t>(0, numLoops)); in runOnOperation()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Linalg/IR/ |
| H A D | LinalgInterfaces.cpp | 664 unsigned numLoops = linalgOp.getNumLoops(); in verifyStructuredOpInterface() local
665 if (indexingMap.getNumDims() != numLoops) in verifyStructuredOpInterface()
667 << opOperand->getOperandNumber() << " to have " << numLoops in verifyStructuredOpInterface()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Utils/ |
| H A D | Utils.cpp | 740 unsigned numLoops = iteratorTypes.size(); in doit() local
741 ivs.reserve(numLoops); in doit()
742 lbsStorage.reserve(numLoops); in doit()
743 ubsStorage.reserve(numLoops); in doit()
744 stepsStorage.reserve(numLoops); in doit()
|
| /llvm-project-15.0.7/mlir/lib/Target/LLVMIR/Dialect/OpenMP/ |
| H A D | OpenMPToLLVMIRTranslation.cpp | 539 unsigned numLoops = *orderedOp.num_loops_val(); in convertOmpOrdered() local
546 storeValues.reserve(numLoops); in convertOmpOrdered()
547 for (unsigned i = 0; i < numLoops; i++) { in convertOmpOrdered()
555 ompLoc, allocaIP, numLoops, storeValues, ".cnt.addr", isDependSource)); in convertOmpOrdered()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Vector/Transforms/ |
| H A D | VectorUnrollDistribute.cpp | 166 getUnrollOrder(unsigned numLoops, Operation *op, in getUnrollOrder() argument
169 llvm::to_vector(llvm::seq<int64_t>(0, static_cast<int64_t>(numLoops))); in getUnrollOrder()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/SparseTensor/Utils/ |
| H A D | Merger.cpp | 143 assert(t < numTensors && i < numLoops); in addLat()
145 latPoints.push_back(LatPoint(numLoops * numTensors, e, numTensors * i + t)); in addLat()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Affine/Utils/ |
| H A D | LoopUtils.cpp | 378 unsigned numLoops = origLoops.size(); in checkTilingLegalityImpl() local
380 for (unsigned d = 1; d <= numLoops + 1; ++d) { in checkTilingLegalityImpl()
|