Searched refs:gridSizeZ (Results 1 – 5 of 5) sorted by relevance
| /llvm-project-15.0.7/mlir/lib/Conversion/SCFToGPU/ |
| H A D | SCFToGPU.cpp | 231 Value gridSizeZ = numBlockDims > 2 ? dims[2] : constOne; in createLaunch() local
239 rootForOp.getLoc(), gridSizeX, gridSizeY, gridSizeZ, blockSizeX, in createLaunch()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/GPU/IR/ |
| H A D | GPUDialect.cpp | 393 Value gridSizeX, Value gridSizeY, Value gridSizeZ, in build() argument
403 {gridSizeX, gridSizeY, gridSizeZ, blockSizeX, blockSizeY, blockSizeZ}); in build()
669 constPropIdUses(op.getBlockIds().z, op.gridSizeZ()); in matchAndRewrite()
|
| /llvm-project-15.0.7/mlir/include/mlir/Dialect/GPU/IR/ |
| H A D | GPUOps.td | 338 Index:$gridSizeX, Index:$gridSizeY, Index:$gridSizeZ,
466 `blocks` `in` ` ` `(`$gridSizeX`,` $gridSizeY`,` $gridSizeZ`)`
478 Index:$gridSizeX, Index:$gridSizeY, Index:$gridSizeZ,
563 "Value":$gridSizeZ, "Value":$blockSizeX, "Value":$blockSizeY,
|
| /llvm-project-15.0.7/llvm/docs/ |
| H A D | NVPTXUsage.rst | 721 unsigned gridSizeZ = 1;
729 checkCudaErrors(cuLaunchKernel(function, gridSizeX, gridSizeY, gridSizeZ,
|
| /llvm-project-15.0.7/mlir/lib/Conversion/GPUCommon/ |
| H A D | GPUToLLVMConversion.cpp | 777 adaptor.gridSizeZ(), adaptor.blockSizeX(), adaptor.blockSizeY(), in matchAndRewrite()
|