Tarry Singh ea3ce685a9 runtimes/neuron: bump runtime version and expose nrt.h header to Zig.		2024-11-14 13:37:47 +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/neuron: bump runtime version and expose nrt.h header to Zig.	2024-11-14 13:37:47 +00:00
cuda.zig	Revert CUDA PJRT plugin version to 0.4.38 to address performance regression on XLA master.	2024-03-05 17:04:42 +00:00
libpjrt_cuda.BUILD.bazel	zml: add support for NVTX tracing	2024-08-21 14:41:40 +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
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