Radix/runtimes/cuda
2025-02-05 17:35:27 +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 Workspace: bump XLA to newer version. 2025-02-05 17:35:27 +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