mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-19 05:16:06 +00:00
Code Issues Packages Projects Releases Wiki Activity
rocm_jax/docs/cuda_custom_call
History
Peter Hawkins 6d1f51e63d Clean up BUILD files. 
PiperOrigin-RevId: 667604964
2024-08-26 09:11:17 -07:00
..
BUILD
Clean up BUILD files.
2024-08-26 09:11:17 -07:00
cuda_custom_call_test.py
Update CUDA custom call example code to use ffi_call.
2024-08-02 10:15:10 -04:00
foo.cu.cc
Fix cuda custom call example to build with updated XLA FFI API.
2024-07-10 05:29:58 -07:00
Makefile
Add CUDA custom call example as a JAX test
2024-06-17 15:21:49 -04:00