|
|
2d321d232d
|
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 |
|
|
|
9488672d4b
|
workspace: bump xla to version 20250710.0-22ea002
Also:
- Bump XLA deps : `com_github_grpc_grpc` and `com_google_protobuf`
- Inject `rules_ml_toolchain`
- Fix `zig_proto_library` rule
|
2025-03-04 17:12:34 +00:00 |
|
|
|
fa0ed045ef
|
runtimes/cuda: downgrade cuda and cudnn
This commit reverts part of https://github.com/zml/zml/pull/238/files
This is required because XLA has a strong dependencies on CUDA 12.8 and
upgrading to 12.9 is impossible due to
https://github.com/NVIDIA/cccl/issues/4967
|
2025-02-28 17:36:12 +00:00 |
|
|
|
1cafcc3c60
|
Workspace: bump XLA to newer version.
|
2025-02-05 17:35:27 +00:00 |
|
|
|
95453c7242
|
Update XLA dependency to version 20250527.0‑cb67f2f and refresh related Bazel BUILD, MODULE, overlay and patch files.
|
2024-11-22 16:50:20 +00:00 |
|
|
|
ea3ce685a9
|
runtimes/neuron: bump runtime version and expose nrt.h header to Zig.
|
2024-11-14 13:37:47 +00:00 |
|
|
|
47a4eda5f6
|
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 |
|
|
|
63ef78efcc
|
zml: add support for NVTX tracing
|
2024-08-21 14:41:40 +00:00 |
|
|
|
967eeb928f
|
Update Bazel workspace and runtime configs: rework sandboxing, bump PJRT to 7.0.0, and upgrade CUDA (12.8), cuDNN (9.8), and ROCm (6.3.4).
|
2024-06-25 11:00:29 +00:00 |
|
|
|
f5ab6ff2c6
|
Update XLA to version 20250204.0-6789523 and adjust Bazel module and runtime files for Bazel 8 compatibility.
|
2024-05-03 15:57:56 +00:00 |
|
|
|
8a25b1eb74
|
Revert CUDA PJRT plugin version to 0.4.38 to address performance regression on XLA master.
|
2024-03-05 17:04:42 +00:00 |
|
|
|
7e6103d876
|
Upgrade XLA to version 20250122.0-cc075be, switch to nvptx compiler and nvlink with nvjitlink support, add warning for CUDA path in LD_LIBRARY_PATH, and revert the previous CUDA sandbox fix.
|
2024-02-06 09:31:48 +00:00 |
|
|
|
434cee3a6c
|
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 |
|
|
|
145e60b4dd
|
workspace: Update LLVM, XLA, StableHLO, and PJRT plugins to latest versions.
|
2023-12-13 10:10:32 +00:00 |
|
|
|
455bb3877f
|
runtimes/cuda: obtain NCCL from the pip package, matching XLA behavior.
|
2023-09-20 17:41:44 +00:00 |
|
|
|
0d5389ceda
|
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 |
|
|
|
01eff33fa0
|
Update workspace dependencies to newer LLVM, XLA, StableHLO, and PJRT versions and expose new pjrt plugin attribute APIs and stablehlo version APIs in build and runtime configurations.
|
2023-08-07 12:28:36 +00:00 |
|
|
|
833ff5f28d
|
Upgrade PJRT CUDA Plugin to version 0.2.3, adding NCCL support for correct sharding.
|
2023-04-12 15:47:06 +00:00 |
|
|
|
70d40208a2
|
runtimes/cuda: Fix version variable definitions in the build script to enable successful CUDA builds.
|
2023-03-09 11:31:02 +00:00 |
|
|
|
0606ea1d7c
|
Update Bazel workspace and runtime BUILD files to newer XLA, StableHLO, and LLVM versions, enabling batching‑dims support for the gather operator.
|
2023-02-01 15:58:30 +00:00 |
|
|
|
266da6d4be
|
Add initial Bazel build configuration, async runtime implementation, and core MLIR dialect definitions for ZML.
|
2023-01-02 14:28:25 +00:00 |
|
