Guray Ozen 763109e346 [mlir][gpu] Use known_block_size to set maxntid for NVVM target (#77301) ...

Setting thread block size with `maxntid` on the kernel has great
performance benefits. In this way, downstream PTX compiler can do better
register allocation.

MLIR's `gpu.launch` and `gpu.launch_func` already has an attribute
(`known_block_size`) that keeps the thread block size when it is known.
This PR simply uses this attribute to set `maxntid`.

2024-01-08 14:49:19 +01:00

..

CMakeLists.txt

[mlir][GPU] Rename MLIRGPUOps CMake target to MLIRGPUDialect

2023-05-16 14:25:08 +02:00

GPUToNVVM.td

…

LowerGpuOpsToNVVMOps.cpp

[mlir][gpu] Use known_block_size to set maxntid for NVVM target (#77301)

2024-01-08 14:49:19 +01:00

WmmaOpsToNvvm.cpp

…