|
|
8d795dd676
|
pjrt: profiler support std writer API
expose a more low-level function to customize where to write profile
reports
|
2024-05-09 11:09:29 +00:00 |
|
|
|
13eff4e661
|
pjrt,zml: add memory bindings
This preliminary PR binds PJRT memory endpoints and adds them to
`zml.Buffer`.
A follow up PR will properly integrate it inside `zml.Buffer`
|
2024-04-11 15:43:24 +00:00 |
|
|
|
5ddd034d2c
|
pjrt: Fix profiler by allowing i64 resource IDs and reserving memory when creating array lists.
|
2023-12-20 17:18:02 +00:00 |
|
|
|
7ef87236ce
|
Rewrite simple transpose as reshape in core ZML modules and raise default profiler event limit to 1,000,000.
|
2023-12-18 13:56:45 +00:00 |
|
|
|
37725cdaa6
|
Update PJRT, runtime, and ZML modules to use per‑target output folders and expose profiler.dumpDataAsJson for JSON profiling output.
|
2023-12-04 10:38:10 +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 |
|
|
|
9505992e00
|
workspace: log diagnostic message before returning NotFound to aid debugging.
|
2023-09-04 13:34:37 +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 |
|
|
|
4681ce2f24
|
PJRT: add conversion of profiling protobuf output to JSON format.
|
2023-07-05 13:34:05 +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 |
|
|
|
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 |
|
|
|
57130577e9
|
Add fallback for runtimes lacking PJRT_Event by using thread‑pool dispatch for buffer copies and treating operations as synchronous when events are absent.
|
2023-05-09 12:44:56 +00:00 |
|
|
|
2f129f76c9
|
Add in-process sharding support across core ZML components (platform, shape, tensor, MLIR generation, buffers, and PJRT integration)
|
2023-02-24 17:33:14 +00:00 |
|
|
|
8fa3878fc3
|
PJRT: Add handling for rank‑0 case in getDimensions to avoid null pointer usage when num_dims is zero.
|
2023-02-17 10:47:15 +00:00 |
|
|
|
be6328813d
|
zml: clean up dead and commented code; note that copyslice is currently broken and pending reimplementation
|
2023-02-08 17:13:47 +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 |
|
