[code-reflection] RFR: [hat][cuda] 2D/3D NDrange supported for CUDA

duke duke at openjdk.org
Wed Jul 30 12:27:45 UTC 2025


On Wed, 30 Jul 2025 12:19:58 GMT, Juan Fumero <duke at openjdk.org> wrote:

> This PR extends the 2D and 3D NDRange developed for the OpenCL backend/runtime for CUDA devices. 
> In addition, it improves the thread-block selection. 
> 
> The heuristic for thread-block selection here is quite simple (static decision based on the number of dimensions). However, in the future we can improve this. My goal with this PR is to showcase 2D and 2D ranges for CUDA compatible devices. 
> 
> How to test?
> 
> 
> java @hat/run ffi-cuda matmul 1D
> 
> java @hat/run ffi-cuda matmul 2D
> 
> # Old examples should continue to work
> java @hat/run ffi-cuda squares

@jjfumero 
Your change (at version bccbbeaa3b125c6e6ad7b084b7912758cff20e29) is now ready to be sponsored by a Committer.

-------------

PR Comment: https://git.openjdk.org/babylon/pull/504#issuecomment-3136070177


More information about the babylon-dev mailing list