|
|
9e3cd6d616
|
bump runtimes/* code to Zig 0.15.1, restore PyTorch loader using std.fs.File, update CI zig fmt, remove stdx.io, note remaining issues with Neuron and CUDA debug builds
|
2025-08-07 15:09:27 +00:00 |
|
|
|
488a844a0f
|
Add Zig 0.15 compatibility: update BUILD files, async primitives, stdx utilities, MLIR dialects, and PJRT FFI.
|
2025-07-28 13:54:28 +00:00 |
|
|
|
c488b634fc
|
runtimes/rocm: implement zmlxrocm in Zig
Also, sandbox `amdgpu.ids` and restore safetensors json parsing.
|
2025-07-07 16:48:07 +00:00 |
|
|
|
cf00506dbb
|
Switch workspace build rules from zig_cc_binary to zig_binary, removing the hack and using the C linker directly.
|
2025-07-03 15:10:36 +00:00 |
|
|
|
e789e26008
|
Remove examples workspace and clean up related Bazel BUILD/MODULE files and Zig build scripts.
|
2025-06-19 09:30:29 +00:00 |
|
|
|
f27a524f31
|
Update rules_zig: add zig_srcs target, fix source handling bug, clean up BUILD files, adjust async/coro.zig tests, and disable nemo and yaml model loaders.
|
2025-03-13 12:27:21 +00:00 |
|
|
|
0a2ab7c8cb
|
Remove usingnamespace from MLIR.
|
2025-01-28 09:35:58 +00:00 |
|
|
|
c961d705f1
|
Set default values for operand_layouts and result_layouts in StableHLO dialect.
|
2024-12-26 09:29:45 +00:00 |
|
|
|
4a0b1cce50
|
Update Bazel workspace and XLA overlay (MODULE.bazel, BUILD files, patches) to prevent dual LLVM builds and apply migration/bump patches.
|
2024-09-27 14:00:44 +00:00 |
|
|
|
aec7072837
|
pjrt: add FFI bindings for custom calls
|
2024-09-10 09:14:28 +00:00 |
|
|
|
ac63c30e12
|
add mini-DSL for creating MLIR common attributes and types, leveraging Zig 0.14 to simplify mlir.Type and mlir.Attribute creation
|
2024-08-26 14:19:00 +00:00 |
|
|
|
cbe6e730bd
|
stablehlo: fix forwarding of optional operand and result layout attributes in custom call
|
2024-07-31 17:53:18 +00:00 |
|
|
|
42dee5d0e0
|
mlir: rework stablehlo custom call implementation and add a Triton example
|
2024-07-16 13:23:07 +00:00 |
|
|
|
aec1d96e6d
|
mlir: rework DenseElementsAttribute to correctly slice inputs and modify .as() to return a concrete value instead of an optional
|
2024-07-15 12:32:24 +00:00 |
|
|
|
201f5245c1
|
mlir: add MLIR index type
|
2024-07-11 11:14:22 +00:00 |
|
|
|
30f6be0e2f
|
Update core Zig modules (async, mlir, pjrt, stdx) and third‑party Bazel definitions for the Zig 0.14.0 release.
|
2024-07-02 14:19:04 +00:00 |
|
|
|
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 |
|
|
|
3aac788544
|
Update Bazel build configurations (zig.bzl, BUILD files) for MLIR, PJRT, Neuron, ROCm, tokenizer, and tools, fixing broken dependencies.
|
2024-05-20 11:28:25 +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 |
|
|
|
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 |
|
|
|
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 |
|
|
|
b53462b515
|
Fix crash in for_ by ensuring values are pushed to their block before opening a new block, adding asserts for block state, and guaranteeing first_step is used. Adjust padding syntax to improve usability.
|
2023-07-25 14:25:47 +00:00 |
|
|
|
b244a18621
|
zml: set iota default dtype to .i32, with fallback to .i64 for axes with many elements, simplifying usage.
|
2023-06-15 12:45:52 +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 |
|
|
|
499b0d20e5
|
pjrtx: change behavior to return an error when OpenXLA fails to serialize the new batching_dim attribute for gather/scatter, instead of panicking.
|
2023-05-29 17:18:19 +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 |
|
|
|
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 |
|
|
|
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 |
|
