Tarry Singh 63ef78efcc zml: add support for NVTX tracing		2024-08-21 14:41:40 +00:00
..
BUILD.bazel	Fix CUDA and ROCm sandbox discovery, update epoll libxev patch to prevent high CPU usage, enable XLA GPU latency‑hiding scheduler, and upgrade cuDNN to 9.6.0.	2024-01-15 09:41:42 +00:00
cuda.bzl	zml: add support for NVTX tracing	2024-08-21 14:41:40 +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
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