a25399cbd4
Validate num_splits in split ( #3234 )
...
Co-authored-by: KD2YCU <me@kd2ycu.com >
2026-03-09 21:47:08 -07:00
43f4a74826
Manage stream placement in import function ( #3127 )
2026-02-15 06:17:06 -08:00
8ef539522c
Fix failing python tests on Windows ( #3076 )
2026-01-30 17:50:18 +09:00
b6aa03e5b8
Update pre-commit hooks and versions for clang-format, black, and isort ( #3059 )
2026-01-26 06:57:04 -08:00
617fd9cbbd
Use C++20 ( #3050 )
2026-01-24 08:48:41 +09:00
1650c4905a
Link with prebuilt OpenBLAS and fix shared libs build on Windows ( #3036 )
2026-01-23 11:17:26 +09:00
687508dd98
win: symbol exports and minor fixes ( #3024 )
...
Co-authored-by: Cheng <zcbenz@gmail.com >
2026-01-23 10:16:22 +09:00
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
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
1596839256
fix array allocator with user buffer and deleter ( #2971 )
2026-01-07 10:08:22 -08:00
ff26b00cb1
new[CI]: add linux sanitizer tests ( #2860 )
...
Signed-off-by: Melissa Kilby <mkilby@apple.com >
2025-12-18 12:15:26 -08:00
f9004103ca
Use CUDA runtime headers from local python package ( #2906 )
2025-12-16 08:36:32 +09:00
667c0f3bb9
[Metal] No copy array init ( #2875 )
2025-12-05 13:36:45 -08:00
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
940f4c7818
Fix building with CUDA < 12.8 ( #2782 )
2025-11-18 12:55:19 +09:00
66519fb348
fix slice ( #2758 )
2025-11-13 11:30:02 -08:00
23f81ed1c1
Linux on arm ( #2751 )
...
* try linux on arm
* ssh
* fix
2025-11-11 11:41:14 -08:00
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
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
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
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
8f8af61a37
fix warnings showing up with -Wall ( #2692 )
2025-10-24 11:43:35 -07:00
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
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
a9bac3d9e5
Run CPP tests for CUDA build in CI ( #2544 )
2025-08-27 08:06:46 +09:00
fce53b61d6
Fix reduce sum/prod overflow ( #2477 )
2025-08-12 00:05:33 -07:00
8831064493
Fix arctan2 grads ( #2453 )
2025-08-01 21:06:04 -07:00
31fc530c76
[CUDA] Add more ways finding CCCL headers in JIT ( #2382 )
2025-07-17 15:25:34 -07:00
2ba69bc8fa
lower memory uniform sampling ( #2361 )
...
* lower memory uniform
* use fp32
* fix
2025-07-15 14:22:07 -07:00
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
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
79071bfba4
Fix out-of-bounds default value in logsumexp/softmax ( #2213 )
2025-05-21 07:25:16 -07:00
cf6c939e86
Fix some complex vjps ( #2178 )
2025-05-14 23:37:12 -07:00
0cae0bdac8
CUDA backend: backbone ( #2075 )
2025-05-06 21:26:46 -07:00
9c5e7da507
fix compile merging ( #2150 )
2025-05-02 15:08:50 -07:00
ea890d8710
Remove metal-only tests ( #2139 )
2025-04-30 09:08:39 -07:00
bb6565ef14
add fftshift and ifftshift fft helpers ( #2135 )
...
* add fftshift and ifftshift fft helpers
* address comments
* axes have to be iterable
* fix fp error in roll + add test
---------
Co-authored-by: Aashiq Dheeraj <aashiq@aashiq-mbp-m4.local >
2025-04-29 22:13:45 -07:00
600e87e03c
Added output_padding parameters in conv_transpose ( #2092 )
2025-04-23 09:26:33 -07:00
dc4eada7f0
Use unordered map for kwargs in export/import ( #2087 )
...
* use unordered map for kwargs in export/import
* comment
2025-04-21 07:17:22 -07:00
5f04c0f818
Fixed shift operations issue ( #2080 )
...
* Fixed shift operations issue
* Added tests and fixes
* Fixed loop syntax error
* Added tests for bool
* Fixed typo
2025-04-18 14:28:33 -07:00
ba09f01ce8
Remove test of converting negative float to uint ( #2048 )
2025-04-06 06:21:46 -07:00
5f5770e3a2
Fix CPU sign for unsigned ints ( #2024 )
...
Co-authored-by: Angelos Katharopoulos <a_katharopoulos@apple.com >
2025-03-30 17:56:59 -07:00
5580b47291
iinfo and scalar overflow detection ( #2009 )
2025-03-27 19:54:56 -07:00
a6b5d6e759
revise cmake minimum for doctest ( #2014 )
2025-03-27 19:30:58 -07:00
4e1994e9d7
move memory APIs into top level mlx.core ( #1982 )
2025-03-21 07:25:12 -07:00
c4230747a1
redesign for faster cpu/gpu synch ( #1869 )
...
* redesign for faster cpu/gpu synch
* load + more async CPU
* use command encoder API and move more ops to use it
* make fence back-end generic + CPU only fence
* faster build
* fix async eval
* fixes + handle temporaries
* fix / improve cpu conv
* remove unused status, fix siblings
* fix extensions
* fix
* fix no cpu build
* format
* comments
* fix perf regression, remove unecessary abort
* fix events, task limit cpu
* fix waiting
* fix donation / temporaries in normalization
2025-03-06 19:23:38 -08:00
3835a428c5
Adds nuclear norm support ( #1894 )
...
* adjust norm unit test tolerance
2025-03-04 13:26:02 -08:00
a5ededf1c3
CPU LU factorization and linear solvers ( #1451 )
...
* linalg solve backend
* nits
* more nits + fix
* luf primitive and lu, solve, and solve_triangular backends
* changes / nits
---------
Co-authored-by: Awni Hannun <awni@apple.com >
2025-02-10 12:32:24 -08:00
f6c0499b8d
Resolved ambiguity in mlx::core::take_along_axis ( #1822 )
...
* Resolved ambiguity in mlx::core::take_along_axis
Detected by GCC 10 on riscv64-linux-gnu.
* Formatted
* Removed superfluous parentheses in random_tests.cpp
2025-02-04 06:06:17 -08:00
2d8e667400
MinGW support ( #1806 )
...
* Changed /bin/bash to bash for generating compiling preamble
* Fix wrt jit_compiler mingw like msvc wrt. WEXITSTATUS
* Solved ambiguity wrt. bernoulli test shape
* Disabled distributed/ring on Windows
* Fixed jit_compiler command wrt. MinGW
* Extended jit_compiler patch wrt. WEXITSTATUS to FreeBSD
2025-02-01 12:40:06 -08:00
Dan Anderson