| .. |
|
aio
|
Add stdx utilities and rework async signature inference; tidy executable logging.
|
2023-06-21 14:45:14 +00:00 |
|
nn
|
zml: fix float8 <-> float32 conversions, support for Tensor.constant(.{}, .{ .f8 = 1.0})
|
2023-09-21 11:15:50 +00:00 |
|
tools
|
Add initial Bazel build configuration, async runtime implementation, and core MLIR dialect definitions for ZML.
|
2023-01-02 14:28:25 +00:00 |
|
aio.zig
|
Fix several compileError calls introduced by recent changes; ensure Zig compiler catches errors at comptime.
|
2023-07-17 09:10:27 +00:00 |
|
buffer.zig
|
Rework async runtime with coroutine support, rename async API (async_→asyncc, await_→awaitt), improve type inference, bump libxev (default epoll) and update related stdx and zml modules.
|
2023-08-01 11:35:04 +00:00 |
|
BUILD.bazel
|
Rework async runtime with coroutine support, rename async API (async_→asyncc, await_→awaitt), improve type inference, bump libxev (default epoll) and update related stdx and zml modules.
|
2023-08-01 11:35:04 +00:00 |
|
context.zig
|
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 |
|
dtype.zig
|
zml: fix float8 <-> float32 conversions, support for Tensor.constant(.{}, .{ .f8 = 1.0})
|
2023-09-21 11:15:50 +00:00 |
|
floats.zig
|
zml: fix float8 <-> float32 conversions, support for Tensor.constant(.{}, .{ .f8 = 1.0})
|
2023-09-21 11:15:50 +00:00 |
|
helpers.zig
|
Add stdx utilities and rework async signature inference; tidy executable logging.
|
2023-06-21 14:45:14 +00:00 |
|
hostbuffer.zig
|
Add stdx utilities and rework async signature inference; tidy executable logging.
|
2023-06-21 14:45:14 +00:00 |
|
meta.zig
|
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 |
|
mlir.zig
|
zml: fix float8 <-> float32 conversions, support for Tensor.constant(.{}, .{ .f8 = 1.0})
|
2023-09-21 11:15:50 +00:00 |
|
module.zig
|
Update module.zig to donate all buffers except the token_index buffer for the Llama+Neuron example.
|
2023-10-06 10:10:56 +00:00 |
|
nn.zig
|
Refactor rope implementation to compute only required offsets, eliminating full cos/sin matrix generation in module, nn, and tensor code.
|
2023-09-27 11:45:33 +00:00 |
|
ops.zig
|
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 |
|
pjrtx.zig
|
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 |
|
platform.zig
|
zml/platform: increase maximum device limit to support up to 32 devices per platform.
|
2023-08-24 12:23:07 +00:00 |
|
posix.h
|
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 |
|
posix.zig
|
Add initial Bazel build configuration, async runtime implementation, and core MLIR dialect definitions for ZML.
|
2023-01-02 14:28:25 +00:00 |
|
quantization.zig
|
Add multi‑axis, batched gatherValues support to tensor, shape, nn, quantization, and torch modules.
|
2023-01-18 12:03:48 +00:00 |
|
shape.zig
|
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 |
|
tensor.zig
|
Refactor rope implementation to compute only required offsets, eliminating full cos/sin matrix generation in module, nn, and tensor code.
|
2023-09-27 11:45:33 +00:00 |
|
test_runner.zig
|
Add stdx utilities and rework async signature inference; tidy executable logging.
|
2023-06-21 14:45:14 +00:00 |
|
testing.zig
|
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 |
|
tokenizer.zig
|
Add stdx utilities and rework async signature inference; tidy executable logging.
|
2023-06-21 14:45:14 +00:00 |
|
torch.zig
|
Add stdx utilities and rework async signature inference; tidy executable logging.
|
2023-06-21 14:45:14 +00:00 |
|
zml.zig
|
zml/nn: fix resize implementations (resizeBilinear and resizeBicubic) and expand refAllDecl usage; all tests pass
|
2023-01-27 14:35:11 +00:00 |
