Commit Graph

  • c9c81d0584 Added additional missing unordered_map include that fixes build on FreeBSD (#1755) Jesper Stemann Andersen 2025-01-07 17:27:55 +01:00
  • 545f84d905 Refactor distributed backend (#1752) Angelos Katharopoulos 2025-01-06 17:33:15 -08:00
  • d5ec172c95 Allow boolean mask in sdpa (#1753) Awni Hannun 2025-01-06 16:57:07 -08:00
  • f14b4d72de Remove unnecessary copy from winograd winograd_qupdate Jagrit Digani 2025-01-06 14:06:03 -08:00
  • 25b3a3e541 Optionally specify names for arrays when exporting (#1749) Angelos Katharopoulos 2025-01-06 13:07:46 -08:00
  • 058d6ce683 mpi send use input as output (#1750) Awni Hannun 2025-01-06 06:08:43 -08:00
  • eab93985b8 Update custom function docs (#1748) Angelos Katharopoulos 2025-01-03 16:35:25 -08:00
  • b51d70a83c export docs (#1747) Awni Hannun 2025-01-03 15:04:17 -08:00
  • 259025100e Fix nd ternary on GPU (#1746) Awni Hannun 2025-01-03 11:52:17 -08:00
  • c9d30aa6ac MLX in C++ example (#1736) Awni Hannun 2025-01-02 19:09:04 -08:00
  • 8544b42007 Add namespace (#1745) Angelos Katharopoulos 2025-01-02 16:49:23 -08:00
  • 6fa0501387 Fix concatenate/slice_update vjp + reduce binary size (#1735) Awni Hannun 2025-01-02 16:36:33 -08:00
  • ae69cb15e9 shapeless compile in docs and partially shapeless reshape (#1742) Awni Hannun 2025-01-02 16:24:42 -08:00
  • a64a8dfe45 fix extension (#1740) Awni Hannun 2025-01-02 16:16:16 -08:00
  • 491fa95b1f Added Kronecker Product (#1728) Venkata Naga Aditya Datta Chivukula 2025-01-02 17:00:34 -07:00
  • 92ec632ad5 Fix Distributed Communication documentation (#1731) Danilo Peixoto 2025-01-02 19:08:38 -03:00
  • 8ecdfb718b Fix export.cpp compilation with MSVC (#1737) Cheng 2024-12-29 23:56:30 +09:00
  • 4ba0c24a8f Export / import functions to / from a file (#1642) Awni Hannun 2024-12-24 11:19:13 -08:00
  • 935c8c4bb1 Make mx.compile work on Windows (#1697) Cheng 2024-12-25 00:02:33 +09:00
  • 88f993da38 Explicit parentheses around some logical operators (#1732) Valentin Roussellet 2024-12-24 07:02:20 -08:00
  • ebfe64b92d shapeless slice update and broadcast when possible (#1727) Awni Hannun 2024-12-23 11:25:15 -08:00
  • 0308e9af71 Allow offset to be an mx.array for mx.fast.rope (#1724) Awni Hannun 2024-12-19 15:51:44 -08:00
  • c3628eea49 Add mx.finfo and use it when making causal mask (#1726) Awni Hannun 2024-12-19 14:52:41 -08:00
  • e03f0372b1 More shape type (#1705) Awni Hannun 2024-12-19 08:08:20 -08:00
  • f17536af9c More lenient mask type check in SDPA (#1723) Alex Barron 2024-12-18 19:41:38 -08:00
  • ed4ec81bca Link python extension with mlx statically on Windows (#1716) Cheng 2024-12-19 12:26:04 +09:00
  • 7480059306 track resource limit and throw if exceeded (#1718) Awni Hannun 2024-12-18 18:45:58 -08:00
  • 8bae22b0fa fix deletion of non-evaled arrays with siblings (#1714) Awni Hannun 2024-12-18 18:45:36 -08:00
  • 49c34c4161 check mask type (#1721) Alex Barron 2024-12-18 14:25:18 -08:00
  • 5548fcc96d fix synch race (#1719) Awni Hannun 2024-12-18 12:25:16 -08:00
  • c02e14c264 Add the 3bit packed qmm_t packed-quants Angelos Katharopoulos 2024-12-17 22:16:30 -08:00
  • 070bd433ab Shorter kernel name for Windows (#1701) Cheng 2024-12-18 11:51:38 +09:00
  • c8fb54951a Define NOMINMAX before windows.h (#1715) Cheng 2024-12-18 11:51:24 +09:00
  • f110357aaa Bump nanobind to 2.4 + fix (#1710) Awni Hannun 2024-12-17 10:57:54 -08:00
  • d75a509234 Add 3bit packed quants Angelos Katharopoulos 2024-12-17 10:08:47 -08:00
  • a6b426422e add cubic to type hinting for upsample (#1709) Tomohiro Oga 2024-12-17 15:30:23 +00:00
  • 14420949d2 Fix the optional in gather_qmm python binding Angelos Katharopoulos 2024-12-16 22:14:19 -08:00
  • 4847199ec6 Add the quantization type option to quantizable layers Angelos Katharopoulos 2024-12-16 22:11:23 -08:00
  • fb7be036af Add packed_affine_qmm_t Angelos Katharopoulos 2024-12-16 21:49:14 -08:00
  • d03c01dfbc fix unflatten vjp (#1708) Awni Hannun 2024-12-16 18:37:57 -08:00
  • 410ccdbed5 Change the argument name to quantization_type Angelos Katharopoulos 2024-12-16 13:31:34 -08:00
  • f5da489a3c Add some error reporting Angelos Katharopoulos 2024-12-16 13:22:05 -08:00
  • c2e6d58441 Revert the change in packing order Angelos Katharopoulos 2024-12-16 13:20:01 -08:00
  • a82996e9fb io/load: Enabled pread implementation for mingw32 (#1706) Jesper Stemann Andersen 2024-12-16 16:20:45 +01:00
  • 17a1fa2f0b Improve the benchmark Angelos Katharopoulos 2024-12-14 23:04:29 -08:00
  • fd161aa31f Change order in weight packing Angelos Katharopoulos 2024-12-14 22:51:41 -08:00
  • af5a614aad Eval before cleanup so model file is unlocked (#1702) Cheng 2024-12-15 14:41:49 +09:00
  • bf6dc54110 Add the 2 bit vectorized reads Angelos Katharopoulos 2024-12-14 21:19:02 -08:00
  • d7ed624502 Vectorized reads Angelos Katharopoulos 2024-12-14 15:08:24 -08:00
  • 05cb54ae3f Another packing Angelos Katharopoulos 2024-12-13 23:48:25 -08:00
  • cb358dbdda Revert "Attempt different packing" Angelos Katharopoulos 2024-12-13 23:23:21 -08:00
  • f9640e049d Install mlx.dll into the same dir with python bindings on Windows (#1690) Cheng 2024-12-14 12:50:39 +09:00
  • 4768c61b57 Make sure gguf_ctx is closed when error happens (#1699) Cheng 2024-12-14 12:50:19 +09:00
  • dfccd17ab9 Use psutil to get memory info on Windows (#1700) Cheng 2024-12-14 12:50:13 +09:00
  • e4b587819c Attempt different packing Angelos Katharopoulos 2024-12-13 18:36:36 -08:00
  • 635117c5d4 Read/write files in binary mode (#1698) Cheng 2024-12-14 10:37:05 +09:00
  • a06c968f4d Add a small benchmark Angelos Katharopoulos 2024-12-13 16:27:27 -08:00
  • 651c510940 Working packed qmv Angelos Katharopoulos 2024-12-13 16:26:55 -08:00
  • 11ec07ff9d Initial python binding Angelos Katharopoulos 2024-12-12 11:29:38 -08:00
  • bdd68bd893 Add a quantization type in the ops Angelos Katharopoulos 2024-12-12 01:30:38 -08:00
  • 50f3535693 Use expand_dims / unflatten / etc in more places (#1696) Awni Hannun 2024-12-12 17:00:44 -08:00
  • 9111999af3 Fix small sort with metal validation (#1695) Awni Hannun 2024-12-12 09:21:45 -08:00
  • 6bd28d246e Allow no copy negative strides in as_strided and slice (#1688) Awni Hannun 2024-12-12 08:59:45 -08:00
  • 4d595a2a39 Make compiled preamble work in MSVC (#1675) Cheng 2024-12-13 01:55:49 +09:00
  • 3a21f61772 Fix build (#1693) Awni Hannun 2024-12-11 23:56:25 -08:00
  • 4e1e9520e1 Flatten and unflatten (#1692) Awni Hannun 2024-12-11 21:51:37 -08:00
  • 0bf19037ca Remove "using namespace mlx::core" in python/src (#1689) Cheng 2024-12-12 08:45:39 +09:00
  • f3dfa36a3a Fix x86 tests (#1691) Awni Hannun 2024-12-11 07:47:18 -08:00
  • 4f9b60dd53 Remove "using namespace mlx::core" in benchmarks/examples (#1685) Cheng 2024-12-12 00:08:29 +09:00
  • f76a49e555 ExpandDims primitive (#1687) Awni Hannun 2024-12-10 16:39:07 -08:00
  • 310ad8d9db Build OpenBLAS from source code for MSVC (#1674) Cheng 2024-12-11 09:14:44 +09:00
  • 56db268f47 Provide a pread implementation for MSVC (#1666) Cheng 2024-12-11 08:55:53 +09:00
  • 92ab6bdeb8 Fix shared library not exporting symbols on Windows (#1684) Cheng 2024-12-11 06:59:14 +09:00
  • 0070e360a1 Disable MSVC warnings (#1680) Cheng 2024-12-10 12:41:14 +09:00
  • 9df8fed046 Metal-cpp version bump (#1668) Amethyst Shen 2024-12-09 22:40:35 -05:00
  • a59fae040f Fix library output directory for MSVC (#1681) Cheng 2024-12-10 12:07:50 +09:00
  • 29a620cab2 No reshapes in quantized embedding (#1682) Awni Hannun 2024-12-09 18:57:38 -08:00
  • 87d7a2520e Use Py_ssize_t in python bindings (#1678) Cheng 2024-12-10 05:59:19 +09:00
  • 0c1155faf5 binding + tests dynamic_reshape Awni Hannun 2024-12-06 20:05:00 -08:00
  • 2b9c24c517 works Awni Hannun 2024-12-06 16:50:08 -08:00
  • ee59d50293 try dynamic reshape Awni Hannun 2024-12-06 12:09:08 -08:00
  • 40c62c1321 Use int64 stride everywhere (#1671) Awni Hannun 2024-12-09 11:09:02 -08:00
  • 35b412c099 Fix compile hasher for string constants. (#1677) Awni Hannun 2024-12-09 09:26:18 -08:00
  • d0f471cff7 Using math defines requires switch in MSVC (#1665) Cheng 2024-12-09 01:16:28 +09:00
  • 6f316b8bf5 Use int64_t instead of ssize_t (#1673) Cheng 2024-12-08 13:10:44 +09:00
  • 7c10c93a1f Convert filesystem path to std::string explicitly (#1672) Cheng 2024-12-08 13:10:06 +09:00
  • d92ea094f1 Use && instead of and (#1663) Cheng 2024-12-08 11:26:39 +09:00
  • 6ae5423b4a Do not pass integers to isnan (#1664) Cheng 2024-12-08 11:26:23 +09:00
  • 9635cffdc8 Include io.h in MSVC for IO functions (#1661) Cheng 2024-12-08 11:26:06 +09:00
  • 96986fb362 Use auto* for pointers (#1662) Cheng 2024-12-08 11:25:40 +09:00
  • 3ceb341a75 Use correct complex type for MSVC (#1660) Cheng 2024-12-08 11:25:22 +09:00
  • 50fa705125 patch bump (#1656) v0.21.1 Awni Hannun 2024-12-06 13:16:19 -08:00
  • 69a2991614 allow compiling lambdas in C++ (#1650) Awni Hannun 2024-12-06 13:13:21 -08:00
  • fd3377dd1f Support bias correction in Adam and AdamW optimizers (#1640) mt_caret 2024-12-07 05:13:34 +09:00
  • d0b6cb0425 More primitives for compiling with shapeless (#1653) Awni Hannun 2024-12-06 11:29:18 -08:00
  • 95c4a2e3af add back conditionaltype (#1655) Alex Barron 2024-12-06 11:12:01 -08:00
  • bc2a29f033 fix (#1654) Awni Hannun 2024-12-06 10:48:58 -08:00
  • 82a956c1d9 fix test q-sdpa Alex Barron 2024-12-06 10:26:54 -08:00
  • 3bb5b4a302 Chore: Add default language in pre-commit and bump hooks (#1652) Nripesh Niketan 2024-12-06 15:54:29 +00:00
  • 769704653a cpu fallback Alex Barron 2024-12-06 01:22:50 -08:00