forked from OSchip/llvm-project
e1da62910e
- Define a gpu.printf op, which can be lowered to any GPU printf() support (which is present in CUDA, HIP, and OpenCL). This op only supports constant format strings and scalar arguments - Define the lowering of gpu.pirntf to a call to printf() (which is what is required for AMD GPUs when using OpenCL) as well as to the hostcall interface present in the AMD Open Compute device library, which is the interface present when kernels are running under HIP. - Add a "runtime" enum that allows specifying which of the possible runtimes a ROCDL kernel will be executed under or that the runtime is unknown. This enum controls how gpu.printf is lowered This change does not enable lowering for Nvidia GPUs, but such a lowering should be possible in principle. And: [MLIR][AMDGPU] Always set amdgpu-implicitarg-num-bytes=56 on kernels This is something that Clang always sets on both OpenCL and HIP kernels, and failing to include it causes mysterious crashes with printf() support. In addition, revert the max-flat-work-group-size to (1, 256) to avoid triggering bugs in the AMDGPU backend. Reviewed By: mehdi_amini Differential Revision: https://reviews.llvm.org/D110448 |
||
|---|---|---|
| .. | ||
| Analysis | ||
| Bindings/Python | ||
| CAPI | ||
| Conversion | ||
| Dialect | ||
| ExecutionEngine | ||
| IR | ||
| Interfaces | ||
| Parser | ||
| Pass | ||
| Reducer | ||
| Rewrite | ||
| Support | ||
| TableGen | ||
| Target | ||
| Tools | ||
| Transforms | ||
| Translation | ||
| CMakeLists.txt | ||