|
|
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 |
|
|
|
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 |
|
|
|
266da6d4be
|
Add initial Bazel build configuration, async runtime implementation, and core MLIR dialect definitions for ZML.
|
2023-01-02 14:28:25 +00:00 |
|
