|
|
63ef78efcc
|
zml: add support for NVTX tracing
|
2024-08-21 14:41:40 +00:00 |
|
|
|
efcf955a4e
|
workspace, third_party/rules_zig: adjust ZLS to require --version as the first parameter and add missing keys to the BuildConfig object for code completion
|
2024-07-10 15:20:12 +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 |
|
|
|
a7b7ae0180
|
Fix async hangs by reworking the libxev epoll backend and using callBlocking for PJRT plugin loading, improving performance across async and runtime modules.
|
2024-01-16 14:13:45 +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 |
|
|
|
7d24329d0a
|
Add Bazel build rules and runtime implementation for AWS Neuron/Trainium/Inferentia support.
|
2023-08-18 17:11:27 +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 |
|
|
|
54e7eb30b4
|
Introduce a thin abstraction layer between ZML and PJRT to manage plugin loading decisions, enable compile‑time detection of linked runtimes, and handle cases such as libtpu blocking metadata access.
|
2023-05-15 09:36:41 +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 |
|
|
|
0c126c2e12
|
runtimes/cuda: Upgrade CUDA to 12.6.2 and cuDNN to 9.4.0.
|
2023-03-03 15:17:26 +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 |
|
