Tarry Singh c488b634fc runtimes/rocm: implement zmlxrocm in Zig ... Also, sandbox `amdgpu.ids` and restore safetensors json parsing.		2025-07-07 16:48:07 +00:00
..
BUILD.bazel	runtimes/cuda: expose cuda.h in the C namespace for CUDA runtimes, enabling custom calls to CUDA functions.	2024-11-01 13:27:24 +00:00
cuda.bzl	runtimes/cuda: sandbox CUDA dependencies by removing them from the leaf binary, sandboxing the dependency graph, marking dlopen direct dependencies as NEEDED, setting RPATH to the sandbox, loading the PJRT plugin from the sandbox, and enabling weak CUDA symbols without direct linking.	2025-03-26 11:18:29 +00:00
cuda.zig	runtimes/cuda: sandbox CUDA dependencies by removing them from the leaf binary, sandboxing the dependency graph, marking dlopen direct dependencies as NEEDED, setting RPATH to the sandbox, loading the PJRT plugin from the sandbox, and enabling weak CUDA symbols without direct linking.	2025-03-26 11:18:29 +00:00
libpjrt_cuda.BUILD.bazel	runtimes/rocm: implement zmlxrocm in Zig	2025-07-07 16:48:07 +00:00
libpjrt_cuda.h	runtimes/cuda: expose cuda.h in the C namespace for CUDA runtimes, enabling custom calls to CUDA functions.	2024-11-01 13:27:24 +00:00
packages.lock.json	runtimes/cuda: sandbox CUDA dependencies by removing them from the leaf binary, sandboxing the dependency graph, marking dlopen direct dependencies as NEEDED, setting RPATH to the sandbox, loading the PJRT plugin from the sandbox, and enabling weak CUDA symbols without direct linking.	2025-03-26 11:18:29 +00:00
packages.yaml	runtimes/cuda: sandbox CUDA dependencies by removing them from the leaf binary, sandboxing the dependency graph, marking dlopen direct dependencies as NEEDED, setting RPATH to the sandbox, loading the PJRT plugin from the sandbox, and enabling weak CUDA symbols without direct linking.	2025-03-26 11:18:29 +00:00
zmlxcuda.c	Update CUDA runtime sandboxing and dynamic symbol renaming, switch to pre‑built jax‑cuda‑pjrt plugin, and bump CUDA to 12.6.2 and cuDNN to 9.5.1.	2023-09-14 13:28:25 +00:00