Home
last modified time | relevance | path

Searched refs:numProcessors (Results 1 – 3 of 3) sorted by relevance

/llvm-project-15.0.7/mlir/test/lib/Dialect/Affine/
H A DTestLoopMapping.cpp45 SmallVector<Value, 8> processorIds, numProcessors; in runOnOperation() local
46 getOperation()->walk([&processorIds, &numProcessors](Operation *op) { in runOnOperation()
50 numProcessors.push_back(op->getResult(1)); in runOnOperation()
53 getOperation()->walk([&processorIds, &numProcessors](scf::ForOp op) { in runOnOperation()
57 mapLoopToProcessorIds(op, processorIds, numProcessors); in runOnOperation()
/llvm-project-15.0.7/mlir/include/mlir/Dialect/Affine/
H A DLoopUtils.h263 ArrayRef<Value> numProcessors);
/llvm-project-15.0.7/mlir/lib/Dialect/Affine/Utils/
H A DLoopUtils.cpp1830 ArrayRef<Value> numProcessors) { in mapLoopToProcessorIds() argument
1831 assert(processorId.size() == numProcessors.size()); in mapLoopToProcessorIds()
1845 loc, mulMap, ValueRange{linearIndex, numProcessors[i]}); in mapLoopToProcessorIds()
1857 for (auto numProcs : numProcessors) in mapLoopToProcessorIds()