mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-17 20:36:05 +00:00
Code Issues Packages Projects Releases Wiki Activity
rocm_jax/jax_plugins
History
jax authors e5eaff84bd Replace pjrt_c_api_gpu_plugin.so symlink with XLA dependency. 
The runfiles of the original targets were lost when the symlinked files were used.

This change is needed for future Hermetic CUDA implementation. Bazel will download CUDA distributives in cache, and CUDA executables and libraries will be added in the runfiles of the targets. When pjrt_c_api_gpu_plugin.so is simlinked, the content of the runfiles is lost. With proper XLA target dependency the runfiles are preserved.

PiperOrigin-RevId: 662197057
2024-08-12 13:01:18 -07:00
..
cuda
Replace pjrt_c_api_gpu_plugin.so symlink with XLA dependency.
2024-08-12 13:01:18 -07:00
rocm
Replace pjrt_c_api_gpu_plugin.so symlink with XLA dependency.
2024-08-12 13:01:18 -07:00
BUILD.bazel
[ROCM] nits and fixes
2024-06-18 20:21:23 +00:00