Home
last modified time | relevance | path

Searched refs:numThreadDims (Results 1 – 5 of 5) sorted by relevance

/llvm-project/mlir/lib/Conversion/SCFToGPU/
H A DSCFToGPUPass.cpp37 ForLoopMapper(unsigned numBlockDims, unsigned numThreadDims) { in ForLoopMapper()
39 this->numThreadDims = numThreadDims; in ForLoopMapper()
47 numThreadDims))) in runOnOperation()
72 mlir::createAffineForToGPUPass(unsigned numBlockDims, unsigned numThreadDims) { in createAffineForToGPUPass() argument
73 return std::make_unique<ForLoopMapper>(numBlockDims, numThreadDims); in createAffineForToGPUPass()
H A DSCFToGPU.cpp138 unsigned numThreadDims) { in checkAffineLoopNestMappable() argument
139 if (numBlockDims < 1 || numThreadDims < 1) { in checkAffineLoopNestMappable()
147 if (numThreadDims > 3) { in checkAffineLoopNestMappable()
150 return checkAffineLoopNestMappableImpl(forOp, numBlockDims + numThreadDims); in checkAffineLoopNestMappable()
161 unsigned numBlockDims, unsigned numThreadDims);
214 // and to have (numBlockDims + numThreadDims) perfectly nested loops between
219 unsigned numThreadDims) { in createLaunch() argument
224 (numBlockDims < 3 || numThreadDims < 3) in createLaunch()
230 Value blockSizeX = numThreadDims > 0 ? dims[numBlockDims] : constOne; in createLaunch()
231 Value blockSizeY = numThreadDims > in createLaunch()
283 convertAffineLoopNestToGPULaunch(AffineForOp forOp,unsigned numBlockDims,unsigned numThreadDims) convertAffineLoopNestToGPULaunch() argument
299 convertAffineLoopNestToGPULaunch(AffineForOp forOp,unsigned numBlockDims,unsigned numThreadDims) convertAffineLoopNestToGPULaunch() argument
[all...]
/llvm-project/mlir/include/mlir/Conversion/SCFToGPU/
H A DSCFToGPUPass.h34 createAffineForToGPUPass(unsigned numBlockDims, unsigned numThreadDims);
H A DSCFToGPU.h44 unsigned numThreadDims);
/llvm-project/mlir/include/mlir/Conversion/
H A DPasses.td1027 Option<"numThreadDims", "gpu-thread-dims", "unsigned", /*default=*/"1u",