Searched refs:gridSizeX (Results 1 – 5 of 5) sorted by relevance
| /llvm-project-15.0.7/mlir/lib/Conversion/SCFToGPU/ |
| H A D | SCFToGPU.cpp | 229 Value gridSizeX = numBlockDims > 0 ? dims[0] : 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()
667 constPropIdUses(op.getBlockIds().x, op.gridSizeX()); 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,
562 OpBuilder<(ins "Value":$gridSizeX, "Value":$gridSizeY,
|
| /llvm-project-15.0.7/llvm/docs/ |
| H A D | NVPTXUsage.rst | 719 unsigned gridSizeX = 1;
729 checkCudaErrors(cuLaunchKernel(function, gridSizeX, gridSizeY, gridSizeZ,
|
| /llvm-project-15.0.7/mlir/lib/Conversion/GPUCommon/ |
| H A D | GPUToLLVMConversion.cpp | 776 {function.getResult(0), adaptor.gridSizeX(), adaptor.gridSizeY(), in matchAndRewrite()
|