Radix/runtimes/cuda
2025-03-13 12:27:21 +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 version 20250710.0-22ea002 2025-03-04 17:12:34 +00:00
cuda.zig Update rules_zig: add zig_srcs target, fix source handling bug, clean up BUILD files, adjust async/coro.zig tests, and disable nemo and yaml model loaders. 2025-03-13 12:27:21 +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