Home
last modified time | relevance | path

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

/llvm-project-15.0.7/mlir/lib/Conversion/SCFToGPU/
H A DSCFToGPUPass.cpp31 ForLoopMapper(unsigned numBlockDims, unsigned numThreadDims) { in ForLoopMapper()
33 this->numThreadDims = numThreadDims; in ForLoopMapper()
41 numThreadDims))) in runOnOperation()
66 mlir::createAffineForToGPUPass(unsigned numBlockDims, unsigned numThreadDims) { in createAffineForToGPUPass() argument
67 return std::make_unique<ForLoopMapper>(numBlockDims, numThreadDims); in createAffineForToGPUPass()
H A DSCFToGPU.cpp135 unsigned numThreadDims) { in checkAffineLoopNestMappable() argument
136 if (numBlockDims < 1 || numThreadDims < 1) { in checkAffineLoopNestMappable()
144 if (numThreadDims > 3) { in checkAffineLoopNestMappable()
147 return checkAffineLoopNestMappableImpl(forOp, numBlockDims + numThreadDims); in checkAffineLoopNestMappable()
157 unsigned numBlockDims, unsigned numThreadDims);
221 unsigned numThreadDims) { in createLaunch() argument
226 (numBlockDims < 3 || numThreadDims < 3) in createLaunch()
232 Value blockSizeX = numThreadDims > 0 ? dims[numBlockDims] : constOne; in createLaunch()
233 Value blockSizeY = numThreadDims > 1 ? dims[numBlockDims + 1] : constOne; in createLaunch()
234 Value blockSizeZ = numThreadDims > 2 ? dims[numBlockDims + 2] : constOne; in createLaunch()
[all …]
/llvm-project-15.0.7/mlir/include/mlir/Conversion/SCFToGPU/
H A DSCFToGPUPass.h30 createAffineForToGPUPass(unsigned numBlockDims, unsigned numThreadDims);
H A DSCFToGPU.h42 unsigned numThreadDims);
/llvm-project-15.0.7/mlir/include/mlir/Conversion/
H A DPasses.td681 Option<"numThreadDims", "gpu-thread-dims", "unsigned", /*default=*/"1u",