7 Commits

Author	SHA1	Message	Date
Tarry Singh	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
Tarry Singh	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
Tarry Singh	acc492454f	Add operator name to source locations and introduce QoL enhancements: remove bias from sdpa, support shape literals in gatherSlices, add Shape.outer, Tensor.all, and infer argMax dtype.	2024-01-01 15:31:41 +00:00
Tarry Singh	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
Tarry Singh	b5c4fb7c58	zml: fix float8 <-> float32 conversions, support for Tensor.constant(.{}, .{ .f8 = 1.0}) ... Mostly: * fix float8 <-> float32 conversions * support for `Tensor.constant(.{}, .{ .f8 = 1.0})` Misc: * fix small inconsistencies between different versions of sdpa * better error message for broadcast * bazelrc: --config=debug	2023-09-21 11:15:50 +00:00
Tarry Singh	0709b1b32f	zml: reduce memory usage of sdpaMemEfficient by using zml.ops.while instead of zml.ops.for, avoiding concatenation of intermediate results.	2023-08-14 14:24:11 +00:00
Tarry Singh	266da6d4be	Add initial Bazel build configuration, async runtime implementation, and core MLIR dialect definitions for ZML.	2023-01-02 14:28:25 +00:00