L-electron-Rare/mlx
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity
1,832 Commits 15 Branches 83 Tags
test/free-threading
Commit Graph

202 Commits

Author SHA1 Message Date
Abhilash Shankarampeta 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
Cheng dec6b4d10f ThreadLocalStream in C++ (#3405) 2026-04-15 15:46:11 -07:00
Matias Insaurralde 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
Dan Anderson 6cef1e995e Validate safetensors data offsets (#3364) 2026-04-15 00:52:42 -07:00
Cheng 435f0b6cdb Add clear_streams API for cleanup before exit (#3395) 2026-04-14 18:41:32 +09:00
Daniil Seredkin 8332e228e4 Fix test "test get streams" missing initialization (#3376) 2026-04-09 08:29:04 +09:00
Cheng 4403165843 [CUDA] Thread safety (#3367) 2026-04-09 08:18:00 +09:00
Cheng 5e2c44259f Make CommandEncoder thread local (#3348) 2026-04-01 18:42:49 +09:00
Cheng df7f7db943 Make each thread have its own default stream (#3281) 2026-03-25 15:48:49 +09:00
Sheldon Aristide 57c813f042 Add norm parameter to FFT transforms (#3287) 
Co-authored-by: Cheng <git@zcbenz.com>
 2026-03-25 13:27:40 +09:00
Sheldon Aristide 1b1c56352a Fix moved-from shape bug in broadcast_arrays causing vmap bus error (#3310) 2026-03-24 17:02:31 -07:00
Robert Johansson e18d4e97f6 Fix vmap + floor_divide: preserve integer dtype (#3292) 
Co-authored-by: Robert Johansson <robert@Mac-Mini-KI.lan>
Co-authored-by: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
Co-authored-by: Angelos Katharopoulos <katharas@gmail.com>
Co-authored-by: Angelos Katharopoulos <a_katharopoulos@apple.com>
 2026-03-25 08:10:37 +09:00
LongYinan 604c825538 Fix stale transform copy-chain leaks (#3290) 2026-03-24 14:15:23 -07:00
Angelos Katharopoulos 7bc61cceed Slice update with operation (#3266) 2026-03-18 06:18:02 -07:00
Long Yixing 0bdbfdb838 [CUDA] Implement MaskedScatter (#3151) 2026-03-15 10:33:55 +09:00
Lucas Newman 5d1700493a [CUDA] Add FFT support (#3243) 2026-03-14 21:02:19 +09:00
Valentin Roussellet b0564a9112 Fix crashes in multi-threaded process teardown (#3167) 2026-03-12 21:45:06 -07:00
Dan Anderson 6ac5280db4 Fix assigning bool to float16/bfloat16 (#3229) 
Co-authored-by: KD2YCU <me@kd2ycu.com>
Co-authored-by: Angelos Katharopoulos <a_katharopoulos@apple.com>
 2026-03-09 21:50:05 -07:00
Dan Anderson a25399cbd4 Validate num_splits in split (#3234) 
Co-authored-by: KD2YCU <me@kd2ycu.com>
 2026-03-09 21:47:08 -07:00
Awni Hannun 43f4a74826 Manage stream placement in import function (#3127) 2026-02-15 06:17:06 -08:00
Cheng 8ef539522c Fix failing python tests on Windows (#3076) 2026-01-30 17:50:18 +09:00
Nripesh Niketan b6aa03e5b8 Update pre-commit hooks and versions for clang-format, black, and isort (#3059) 2026-01-26 06:57:04 -08:00
Cheng 617fd9cbbd Use C++20 (#3050) 2026-01-24 08:48:41 +09:00
Cheng 1650c4905a Link with prebuilt OpenBLAS and fix shared libs build on Windows (#3036) 2026-01-23 11:17:26 +09:00
Daniel Hiltgen 687508dd98 win: symbol exports and minor fixes (#3024) 
Co-authored-by: Cheng <zcbenz@gmail.com>
 2026-01-23 10:16:22 +09:00
Dan Anderson 83bb7891db Fix negative dim indexing (#2994) 
Co-authored-by: KD2YCU <me@kd2ycu.com>
Co-authored-by: Awni Hannun <awni@apple.com>
 2026-01-20 06:24:33 -08:00
MillaFleurs 4160ec10f7 Fix RandomBits::is_equivalent to include width (#2978) 
Co-authored-by: KD2YCU <me@kd2ycu.com>
Co-authored-by: Angelos Katharopoulos <katharas@gmail.com>
Co-authored-by: Awni Hannun <awni@apple.com>
 2026-01-13 12:42:37 -08:00
Ronan Collobert 1596839256 fix array allocator with user buffer and deleter (#2971) 2026-01-07 10:08:22 -08:00
Melissa Kilby ff26b00cb1 new[CI]: add linux sanitizer tests (#2860) 
Signed-off-by: Melissa Kilby <mkilby@apple.com>
 2025-12-18 12:15:26 -08:00
Cheng f9004103ca Use CUDA runtime headers from local python package (#2906) 2025-12-16 08:36:32 +09:00
Awni Hannun 667c0f3bb9 [Metal] No copy array init (#2875) 2025-12-05 13:36:45 -08:00
CCYeh b3825ac149 Add Masked Scatter (#2663) 
Co-authored-by: Awni Hannun <awni@apple.com>
Co-authored-by: Angelos Katharopoulos <katharas@gmail.com>
Co-authored-by: Angelos Katharopoulos <a_katharopoulos@apple.com>
 2025-11-19 14:53:32 -08:00
Cheng 940f4c7818 Fix building with CUDA < 12.8 (#2782) 2025-11-18 12:55:19 +09:00
Awni Hannun 66519fb348 fix slice (#2758) 2025-11-13 11:30:02 -08:00
Awni Hannun 23f81ed1c1 Linux on arm (#2751) 
* try linux on arm

* ssh

* fix
 2025-11-11 11:41:14 -08:00
CCYeh be9e2aebd6 Shapeless support for zeros/ones_like (#2726) 
* shapeless support for zeros/ones_like

* Improvements

* fix access after moved
 2025-11-06 19:12:20 -08:00
AN Long 1ff2b713b6 Check isnan in maximum / minimum with CPU backend (#2652) 
* Check isnan in maximum / minimum with CPU backend

* Add tests

* fix

---------

Co-authored-by: Awni Hannun <awni@apple.com>
 2025-11-03 08:51:14 -08:00
Awni Hannun 93d76b0f30 Fix compile multi capture (#2678) 
* fix compile when compiling multiple lambdas with the same capture

* add test
 2025-11-03 06:33:43 -08:00
Awni Hannun 969924cc69 Fp8 conversion (#2686) 
* add fp8 e4m3 converters

* add cuda

* default saturate to min/max

* fix for older OS

* fix no gpu/cpu

* fix saturate

* fix compile
 2025-10-27 16:35:50 -07:00
Ronan Collobert 8f8af61a37 fix warnings showing up with -Wall (#2692) 2025-10-24 11:43:35 -07:00
Cheng b194d65a6a Some tweaks in cmake files (#2574) 
* Do proper check of Metal lib

* Update doctest to get rid of cmake version hack
 2025-09-09 08:27:18 +09:00
Awni Hannun 70560b6bd5 Add mode parameter for quantization (#2499) 
* add mode parameter for quantization

* mxfp4 quantize/dequantize + start of optional biases

* mxfp4 works

* speedup

* cpu mxfp4

* fix

* fix test tol

* fix

* refactor

* add quant mode enum
 2025-08-28 06:45:26 -07:00
Cheng a9bac3d9e5 Run CPP tests for CUDA build in CI (#2544) 2025-08-27 08:06:46 +09:00
Abe Leininger fce53b61d6 Fix reduce sum/prod overflow (#2477) 2025-08-12 00:05:33 -07:00
Angelos Katharopoulos 8831064493 Fix arctan2 grads (#2453) 2025-08-01 21:06:04 -07:00
Cheng 31fc530c76 [CUDA] Add more ways finding CCCL headers in JIT (#2382) 2025-07-17 15:25:34 -07:00
Awni Hannun 2ba69bc8fa lower memory uniform sampling (#2361) 
* lower memory uniform

* use fp32

* fix
 2025-07-15 14:22:07 -07:00
Cheng 8347575ba1 [CUDA] Implement Scan kernel (#2347) 
* Contiguous scan

* Strided scan

* Enable tests

* Fix failing logaddexp test

* Use cexpf in Metal
 2025-07-10 16:54:12 -07:00
jhavukainen 8b9a3f3cea Align mlx::core::max op nan propagation with NumPy (#2339) 
* Make max op NaN propagation rules align with numpy

* Adding benchmarks and testing for max op nanpropagation

* Pre-commit formatting

* Fix max complex64 nan propagation and add test

* Improve the cpp unittest

* Only check nans on non-integral types in simd_reduce_impl.

* Cleanup using namespace alias

* Add cpu Max nanpropagation. Fix a small fib in cpu max dispatch data types for int8/int16.

* Make the max nanpropagation test more meaningful for integer types

* Remove tuple unpacking syntax to comply with earlier python versions. Add cuda skip to nanpropagation tests, fix cuda implementation in a separate PR.
 2025-07-09 11:26:27 -07:00
Cheng 79071bfba4 Fix out-of-bounds default value in logsumexp/softmax (#2213) 2025-05-21 07:25:16 -07:00