73f33e92cd
fix(ci): use branch_to_build input for checkout and artifact naming
2026-05-10 13:41:03 +02:00
765ebd7ef2
ci: add macOS arm64 wheel build workflow
...
Builds mlx wheels for Python 3.11 and 3.12 on macos-14 (arm64)
on push to main, metal-*, q-*, attn-mask-fix, fix-rope branches
or via workflow_dispatch. Wheels uploaded as artifacts (30d retention).
Tagged commits (v*) also publish a GitHub Release.
2026-05-10 13:13:33 +02:00
84961223c0
[CUDA] Separate main loop into a function in qmm ( #3443 )
2026-05-09 11:11:25 +09:00
662115c1f0
Do not use prebuilt cpu compile preamble when headers are installed ( #3463 )
2026-05-09 09:02:12 +09:00
a1c0b6f9ac
Compute contiguity from the actual occupied data ( #3475 )
2026-05-08 01:29:29 -07:00
c9aa560577
Make device_count() return 0 when there is no GPU ( #3486 )
2026-05-08 08:33:55 +09:00
ff57d875ea
Fix indexing bug in slice update with op ( #3483 )
2026-05-06 17:04:50 -07:00
80bcd1c658
[CUDA] Fix half type matmul in cutlass kernels ( #3469 )
2026-05-06 08:35:53 +09:00
1fdd4e23c2
Clearer error when shape dimension overflows int32 ( #3425 )
...
Co-authored-by: Kanishk <kanishk.chores@gmail.com >
Co-authored-by: Claude Opus 4.7 (1M context) <noreply@anthropic.com >
2026-05-05 09:53:36 +09:00
b43965925f
Define ST_F8_E8M0 ( #3448 )
2026-05-05 09:22:23 +09:00
0938db7e54
Add determinant and sign-log-determinant functions to mlx.core.linalg ( #3416 )
...
Co-authored-by: Lucas Fernandes Martins <Lucas-Fernandes-Martins@users.noreply.github.com >
2026-05-05 09:06:23 +09:00
e8ebdebeeb
Add barrier to JACCL ( #3459 )
2026-04-28 09:39:56 -07:00
d7d0992d75
Reuse nightly build's ccache for release ( #3458 )
2026-04-28 10:54:41 +09:00
bdb6ff8881
Keep gguflib input-validation asserts active in release builds ( #3436 )
2026-04-27 08:46:57 +09:00
894c948773
[CUDA] Fix qmm_naive K-tail dispatch for FP quantized kernels ( #3445 )
2026-04-27 08:40:14 +09:00
211e57be53
Bump minor ( #3438 )
2026-04-22 11:09:30 -07:00
c284e0a231
Enable swap for all CI building CUDA ( #3437 )
2026-04-22 13:13:24 +09:00
b9b1bfb9a5
Generate qmm implementaions with cmake ( #3424 )
2026-04-22 13:11:55 +09:00
68cf2fddd8
Fix mx.prod vjp for complex types ( #3433 )
2026-04-21 17:35:20 -07:00
c594e6ec38
Fix use after move in Compiled primitive ( #3427 )
2026-04-21 15:22:45 -07:00
7d40a4fd5a
Throw meaningful error when Metal device is not found ( #3428 )
2026-04-21 15:21:08 -07:00
5f519ef6f9
Fix bytes_per_key truncation in random kernels (Metal + CUDA) ( #3432 )
2026-04-21 15:15:11 -07:00
705c828feb
Fix synchronize for ThreadLocalStream ( #3429 )
2026-04-20 11:29:49 -07:00
b4ddf9b374
Fix flaky TestVmap.test_vmap_masked_scatter ( #3421 )
2026-04-20 17:19:20 +09:00
1f5a413a27
Make Scheduler::enqueue thread safe ( #3423 )
2026-04-20 14:30:05 +09:00
a6222f53d5
Speed up NAX split-K by better tuning and routing and fix NAX addmm ( #3422 )
...
I 'll merge now and comment with more benchmarks later since this also fixes two bugs so worst case we 'll do another tuning, it isn't like we won't need the functionality of this PR.
2026-04-19 18:05:39 -07:00
fa4320d5fa
[CUDA] Handle residue k in qmm_naive ( #3379 )
2026-04-18 13:30:07 +09:00
859f22fbb0
[CUDA] GatherQMM matrix-matrix sm80/naive path ( #3417 )
...
Co-authored-by: Cheng <git@zcbenz.com >
2026-04-18 10:59:47 +09:00
d142de6a20
[CUDA] gather_mm ( #3414 )
2026-04-17 16:53:44 +09:00
940ba473fe
Segmented mm nax kernel ( #3419 )
2026-04-16 17:26:29 -07:00
8e649be4d0
Fix jaccl init bug ( #3418 )
2026-04-16 01:23:35 -07:00
dec6b4d10f
ThreadLocalStream in C++ ( #3405 )
2026-04-15 15:46:11 -07:00
fd8e849e26
Document sort stability and NaN handling ( #3400 )
2026-04-15 14:32:42 -07:00
50ae31241a
Validate safetensors data offsets against file boundaries ( #3410 )
...
Co-authored-by: Angelos Katharopoulos <a_katharopoulos@apple.com >
2026-04-15 14:30:55 -07:00
6cef1e995e
Validate safetensors data offsets ( #3364 )
2026-04-15 00:52:42 -07:00
57bcced8cb
Fixes for CUDA CI ( #3413 )
2026-04-14 23:52:52 -07:00
4400504ad5
Jaccl refactor ( #3412 )
2026-04-14 23:52:21 -07:00
1fa764fbec
Update nanobind version to v2.12.0 ( #3396 )
2026-04-14 17:21:00 -07:00
435f0b6cdb
Add clear_streams API for cleanup before exit ( #3395 )
2026-04-14 18:41:32 +09:00
520cea2bec
Avoid joining threads on exit ( #3388 )
2026-04-11 09:22:34 +09:00
a33b791615
Fix int16 overflow in SDPA NAX mask indexing for KV sequences > 32K ( #3361 )
...
Co-authored-by: Angelos Katharopoulos <a_katharopoulos@apple.com >
2026-04-10 00:01:47 -07:00
d6d9b24801
Conjugate VJP and JVP support ( #3386 )
2026-04-09 15:04:46 -07:00
8332e228e4
Fix test "test get streams" missing initialization ( #3376 )
2026-04-09 08:29:04 +09:00
4403165843
[CUDA] Thread safety ( #3367 )
2026-04-09 08:18:00 +09:00
a8776b7bbd
Fix: Correct cross-attention query routing in Post-LN TransformerDecoderLayer ( #3382 )
2026-04-07 09:16:12 -07:00
b98831ad0e
fix: fail build when Metal compiler header resolution fails ( #3332 )
2026-04-06 12:49:25 -07:00
d025111b1d
[CUDA] Add GatherQMM for quantized gather matmul ( #3321 )
2026-04-06 12:48:18 -07:00
9239808225
Fix CMake finding wrong Python during pip install ( #3375 )
2026-04-06 12:32:16 -07:00
6a9a121d09
Add a convenience for making local streams in python ( #3355 )
2026-04-02 18:43:02 -07:00
befe42d303
Add printoptions ( #3333 )
2026-04-01 22:24:48 -07:00
electron-rare