Searched refs:blockSizeZ (Results 1 – 5 of 5) sorted by relevance
| /llvm-project-15.0.7/mlir/lib/Conversion/SCFToGPU/ |
| H A D | SCFToGPU.cpp | 234 Value blockSizeZ = numThreadDims > 2 ? dims[numBlockDims + 2] : constOne; in createLaunch() local
240 blockSizeY, blockSizeZ); in createLaunch()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/GPU/IR/ |
| H A D | GPUDialect.cpp | 394 Value blockSizeX, Value blockSizeY, Value blockSizeZ, in build() argument
403 {gridSizeX, gridSizeY, gridSizeZ, blockSizeX, blockSizeY, blockSizeZ}); in build()
672 constPropIdUses(op.getThreadIds().z, op.blockSizeZ()); in matchAndRewrite()
|
| /llvm-project-15.0.7/mlir/include/mlir/Dialect/GPU/IR/ |
| H A D | GPUOps.td | 339 Index:$blockSizeX, Index:$blockSizeY, Index:$blockSizeZ,
467 `threads` `in` ` ` `(`$blockSizeX`,` $blockSizeY`,` $blockSizeZ`)`
479 Index:$blockSizeX, Index:$blockSizeY, Index:$blockSizeZ,
564 "Value":$blockSizeZ,
|
| /llvm-project-15.0.7/llvm/docs/ |
| H A D | NVPTXUsage.rst | 718 unsigned blockSizeZ = 1;
730 blockSizeX, blockSizeY, blockSizeZ,
|
| /llvm-project-15.0.7/mlir/lib/Conversion/GPUCommon/ |
| H A D | GPUToLLVMConversion.cpp | 778 adaptor.blockSizeZ(), dynamicSharedMemorySize, stream, kernelParams, in matchAndRewrite()
|