|
|
f7450a2104
|
stablehlo: bump to head and use new dialect capi
This drastically reduce the number of build steps (from 3589 to 2553 steps)
|
2024-05-31 13:02:46 +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 |
|
|
|
98b512c495
|
Implement func.call emission and function caching across MLIR dialects and ZML module/ops, propagating tags and donations.
|
2023-10-19 17:01:55 +00:00 |
|
|
|
7d36913b31
|
Refactor ZML API: move compile, compileFn and related types to exe.zig, update BaseExe allocation and inline caching in compileInternal, and clean up supporting modules (func.zig, meta.zig, signature.zig, cuda.zig, testing.zig, zml.zig).
|
2023-10-13 16:08:08 +00:00 |
|
|
|
c8c99d7d5a
|
zml/pjrtx: prefer the built‑in stablehlo version when a plugin reports a newer version, ensuring artifact serialization uses the correct stablehlo version.
|
2023-09-07 17:06:19 +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 |
|
|
|
344e07fb6e
|
stablehlo: extend dot_general API to include DotAlgorithm support by merging precision and algorithm attributes into a union, aligning with spec requirements. Currently not exposed to users due to limited algorithm support.
|
2023-06-07 11:20:25 +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 |
|
|
|
05faa5021e
|
zml.tensor: add cumulativeSum operator and refactor maxPoolND. Introduce cumulative sum using reduceWindow. Simplify reduceWindow signature by merging padding_shape and padding_value. Update maxPool1D/2D to accept tuple arguments. Revise pad to use tagged or AOS syntax; remove SOA syntax.
|
2023-05-17 09:01:27 +00:00 |
|
|
|
a4f0fc96c0
|
Integrate user sharding hints and HLO sharding annotations across MLIR dialects and ZML core, and remove the now‑unused module options arguments.
|
2023-03-21 10:50:39 +00:00 |
|
|
|
fc718ab649
|
Add StableHLO bindings for versioning functions, enabling portable serialization of StableHLO.
|
2023-02-22 15:41:33 +00:00 |
|
|
|
24a7c98476
|
Implement scatterSlices functionality.
|
2023-02-14 13:52:49 +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 |
|
|
|
7dcd8b516c
|
zml/nn: fix resize implementations (resizeBilinear and resizeBicubic) and expand refAllDecl usage; all tests pass
|
2023-01-27 14:35:11 +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 |
|
