|
|
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 |
|
|
|
a5420068b1
|
pjrt: emit warning instead of panic when FFI Extension is missing (e.g., on TPU).
|
2025-03-24 09:40:44 +00:00 |
|
|
|
f8ab0d7b2a
|
Remove dead imports.
|
2025-01-22 10:45:04 +00:00 |
|
|
|
6aa9aa5a7b
|
Add preliminary implementation for custom call support.
|
2024-12-10 09:36:37 +00:00 |
|
|
|
aec7072837
|
pjrt: add FFI bindings for custom calls
|
2024-09-10 09:14:28 +00:00 |
|
|
|
57bf667c90
|
Add struct‑based client creation flags to the Zig PJRT API and update context.autoPlatform to accept a flag struct.
|
2023-11-13 12:45:17 +00:00 |
|
|
|
9f4194ad97
|
Fix test layer. Add tests to detect silent breakage of testLayer and regression in mapAlloc with zero-size struct fields. Add Python venv directory to .gitignore.
|
2023-11-06 11:25:57 +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 |
|
|
|
9505992e00
|
workspace: log diagnostic message before returning NotFound to aid debugging.
|
2023-09-04 13:34:37 +00:00 |
|
|
|
9b7eea8ac2
|
Add stdx utilities and rework async signature inference; tidy executable logging.
|
2023-06-21 14:45:14 +00:00 |
|
|
|
6d720126ac
|
Add PJRT custom call integration with generic zmlHostBufferCallback to copy tensors to host and invoke user callbacks. Introduce Tensor.print() method to output runtime tensor values (CUDA‑specific, uses a pre‑allocated host buffer).
|
2023-06-05 13:42:45 +00:00 |
|
|
|
52ef20f981
|
zml: reintroduce pjrtx to handle reactor blocking issues in async scenarios, particularly with Events.
|
2023-05-26 15:54:15 +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 |
|
|
|
74e90855ca
|
Configure the runfiles environment globally at context start to ensure Bazel-built binaries locate their runfiles correctly.
|
2023-05-12 11:40:23 +00:00 |
|
|
|
dfa71018a5
|
zml: Remove pjrtx wrapper, migrate remaining helpers to their native modules, and fix blocking issue in Event.await.
|
2023-03-06 17:05:56 +00:00 |
|
|
|
24a7c98476
|
Implement scatterSlices functionality.
|
2023-02-14 13:52:49 +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 |
|
