L-electron-Rare/mlx
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Commit Graph

  • aec67f2fa6 patch bump (#2386) Awni Hannun 2025-07-18 12:25:48 -07:00
  • deee214a95 Adding support for the Muon Optimizer (#1914) Gökdeniz Gülmez 2025-07-18 21:25:28 +02:00
  • 45adec102c Add contiguous_copy_gpu util for copying array (#2379) Cheng 2025-07-18 22:44:25 +09:00
  • 31fc530c76 [CUDA] Add more ways finding CCCL headers in JIT (#2382) Cheng 2025-07-18 07:25:34 +09:00
  • fbb3f65a1a fix resource leaks in matmul and graph (#2383) Awni Hannun 2025-07-17 06:50:15 -07:00
  • 6b1b8ea91b [CUDA] Add work per thread to compile (#2368) Angelos Katharopoulos 2025-07-17 06:47:52 -07:00
  • b2273733ea Test with CUDA 12.2 (#2375) Awni Hannun 2025-07-16 13:00:37 -07:00
  • f409b229a4 fix ring distributed test (#2380) Awni Hannun 2025-07-16 11:25:24 -07:00
  • 30571e2326 Rename the copy util in cpu/copy.h to copy_cpu (#2378) Cheng 2025-07-16 23:34:24 +09:00
  • d7734edd9f fix complex reduce + nan propagation in min and max (#2377) Awni Hannun 2025-07-15 18:19:47 -07:00
  • 2ba69bc8fa lower memory uniform sampling (#2361) Awni Hannun 2025-07-15 14:22:07 -07:00
  • cb349a291c [CUDA] Use cuda::std::complex in place of cuComplex (#2372) Cheng 2025-07-15 16:36:13 +09:00
  • f0a0b077a0 Install linux with mlx[cuda] and mlx[cpu] (#2356) Awni Hannun 2025-07-14 17:17:33 -07:00
  • 49114f28ab fix flaky test (#2371) Awni Hannun 2025-07-14 17:16:18 -07:00
  • e7d2ebadd2 [CUDA] Affine quantize (#2354) Awni Hannun 2025-07-14 15:45:44 -07:00
  • e569803d7c update linux build (#2370) Awni Hannun 2025-07-14 15:13:56 -07:00
  • d34f887abc Add Primitive::name and remove Primitive::print (#2365) Cheng 2025-07-15 06:06:35 +09:00
  • 5201df5030 Fix imag() vjp (#2367) Angelos Katharopoulos 2025-07-14 13:11:16 -07:00
  • 2d3c26c565 [CUDA] Do not put kernels in annoymous namespace (#2362) Cheng 2025-07-13 06:24:45 +09:00
  • 6325f60d52 [CUDA] Bundle CCCL for JIT compilation (#2357) Cheng 2025-07-12 10:45:37 +09:00
  • a9c720e8cd Improve the ring backend initialization ring-init Angelos Katharopoulos 2025-07-11 15:31:28 -07:00
  • 42cc9cfbc7 fix copy dispatch (#2360) Awni Hannun 2025-07-11 10:59:35 -07:00
  • 8347575ba1 [CUDA] Implement Scan kernel (#2347) Cheng 2025-07-11 08:54:12 +09:00
  • b6eec20260 Fix edge check in qmm_n QuantizedLoader (#2355) Angelos Katharopoulos 2025-07-10 16:28:50 -07:00
  • 0eb035b4b1 Fix type promotion in Adam with bias correction (#2350) Angelos Katharopoulos 2025-07-10 11:14:42 -07:00
  • afb9817599 [CUDA] Put version in ptx cache dir path (#2352) Cheng 2025-07-10 23:24:21 +09:00
  • 8fb3e7a26c [CUDA] Set current device before cudaGraphLaunch (#2351) Cheng 2025-07-10 23:24:02 +09:00
  • 8c7bc30ce4 Align mlx::core::min op nan propagation with NumPy (#2346) jhavukainen 2025-07-10 06:20:43 -07:00
  • 85873cb162 [CUDA] Do vectorized store/load in contiguous elementwise ops (#2342) Cheng 2025-07-10 10:48:43 +09:00
  • e14ee12491 add zero for argsort vjp (#2345) Awni Hannun 2025-07-09 14:37:14 -07:00
  • 8b9a3f3cea Align mlx::core::max op nan propagation with NumPy (#2339) jhavukainen 2025-07-09 11:26:27 -07:00
  • fb4e8b896b patch bump (#2343) v0.26.3 Awni Hannun 2025-07-08 14:26:07 -07:00
  • 2ca533b279 Fix compilation with CUDA 11 (#2331) Cheng 2025-07-08 12:00:43 +09:00
  • 4a9b29a875 MoE backward improvements (#2335) Angelos Katharopoulos 2025-07-07 17:59:53 -07:00
  • a4fcc893cd auto build linux release (#2341) Awni Hannun 2025-07-07 09:29:23 -07:00
  • 9d10239af7 [CUDA] Do vectorized store/load in binary ops (#2330) Cheng 2025-07-08 00:44:14 +09:00
  • 19facd4b20 Build with all cpu cores by default (#2336) Cheng 2025-07-07 22:06:45 +09:00
  • f5299f72cd Fix layernorm race condition (#2340) Angelos Katharopoulos 2025-07-07 06:06:01 -07:00
  • 0e0d9ac522 [CUDA] Add MLX_CUDA_GRAPH_CACHE_SIZE env for setting graph cache size (#2329) Cheng 2025-07-06 00:33:29 +09:00
  • 8917022deb fix graphs for older cuda (#2328) Awni Hannun 2025-07-02 19:37:58 -07:00
  • ec0d5db67b [CUDA] Switch to CUDA graphs (#2317) Awni Hannun 2025-07-02 15:59:13 -07:00
  • e76e9b87f0 Fix compilation error from integral_constant (#2326) Cheng 2025-07-02 22:04:38 +09:00
  • cfb6a244ea allow parameters to be deleted (#2325) Awni Hannun 2025-07-01 21:27:23 -07:00
  • 58f3860306 patch bump (#2324) v0.26.2 Awni Hannun 2025-07-01 12:12:16 -07:00
  • dd4f53db63 use fp32 for testing, add more complex ops (#2322) Awni Hannun 2025-07-01 07:30:00 -07:00
  • 3d5e17e507 MLX_SWITCH macros to templates (#2320) Angelos Katharopoulos 2025-07-01 01:33:44 -07:00
  • 33bf1a244b Fix module update in strict mode (#2321) Awni Hannun 2025-06-29 11:12:29 -07:00
  • 772f471ff2 [CUDA] Fix reductions (#2314) Angelos Katharopoulos 2025-06-27 12:59:20 -07:00
  • 2c11d10f8d Split broadcast so it is always fused in compile (#2318) Angelos Katharopoulos 2025-06-26 22:08:18 -07:00
  • 656ed7f780 Fix get 2d grid dims (#2316) Angelos Katharopoulos 2025-06-25 13:03:09 -07:00
  • 81bb9a2a9e Compile float64 functions on CPU (#2311) Awni Hannun 2025-06-24 10:18:52 -07:00
  • 5adf185f86 Fix update_modules() when providing a subset (#2308) Angelos Katharopoulos 2025-06-20 17:19:46 -07:00
  • c9a9180584 Cuda perf tuning (#2307) Awni Hannun 2025-06-20 14:50:57 -07:00
  • 76831ed83d Build CUDA release in Circle (#2306) Awni Hannun 2025-06-19 15:26:36 -07:00
  • b3d7b85376 Make ptx cache settable by environment variable (#2304) Angelos Katharopoulos 2025-06-17 23:55:56 -07:00
  • cad5c0241c [CUDA] synch properly waits for all tasks to finish and clear (#2303) Awni Hannun 2025-06-17 12:03:25 -07:00
  • b8022c578a divmod, partition, sort fixes (#2302) Awni Hannun 2025-06-16 18:49:32 -07:00
  • bc53f8293f Cuda bug fixes 2 (#2298) Awni Hannun 2025-06-16 13:14:46 -07:00
  • c552ff2451 [CUDA] Fix back-end bugs and enable corresponding tests (#2296) Awni Hannun 2025-06-16 08:45:40 -07:00
  • 4fda5fbdf9 add python testing for cuda with ability to skip list of tests (#2295) Awni Hannun 2025-06-15 10:56:48 -07:00
  • 580776559b RoPE for CUDA (#2293) Angelos Katharopoulos 2025-06-15 06:08:07 -07:00
  • a14aaa7c9d Fix cuda arg reduce (#2291) Awni Hannun 2025-06-14 17:54:00 -07:00
  • a6d780154f fix cuda gemm for bf16 (#2288) Awni Hannun 2025-06-13 22:10:46 -07:00
  • 6871e2eeb7 fix cuda jit (#2287) Awni Hannun 2025-06-13 19:21:46 -07:00
  • 8402a2acf4 Fix complex power and print (#2286) Awni Hannun 2025-06-13 11:13:00 -07:00
  • fddb6933e1 Collection of refactors (#2274) Jagrit Digani 2025-06-13 10:44:56 -07:00
  • c8b4787e4e CUDA backend: indexing ops (#2277) Cheng 2025-06-13 13:44:19 +09:00
  • 2188199ff8 [CUDA] ternary with select op (#2283) Awni Hannun 2025-06-12 20:24:43 -07:00
  • aa07429bad Fix cuda build (#2284) Awni Hannun 2025-06-12 17:48:05 -07:00
  • 918761a25a [CUDA] RMSNorm and VJP (#2280) Awni Hannun 2025-06-12 17:09:49 -07:00
  • a4fc671d3e CUDA backend: compile (#2276) Cheng 2025-06-13 09:08:39 +09:00
  • f5f65ef48c Make sliceUpdate general (#2282) Awni Hannun 2025-06-12 16:48:54 -07:00
  • c2dd81a8aa Fix warnings from latest CUDA toolkit (#2275) Cheng 2025-06-12 22:03:01 +09:00
  • d7e680ffe4 CUDA backend: layernorm (#2271) Cheng 2025-06-12 07:48:32 +09:00
  • c371baf53a CUDA backend: softmax (#2272) Cheng 2025-06-12 05:55:22 +09:00
  • ccf78f566c CUDA backend: argreduce (#2270) Cheng 2025-06-12 05:26:17 +09:00
  • c9fa68664a CUDA backend: reduce (#2269) Cheng 2025-06-12 03:22:25 +09:00
  • c35f4d089a start cuda circle config (#2256) Awni Hannun 2025-06-10 21:19:47 -07:00
  • 8590c0941e Add load_safe to the general conv loaders (#2258) Angelos Katharopoulos 2025-06-10 20:58:16 -07:00
  • 095163b8d1 Fix building cpp benchmarks on Linux (#2268) Cheng 2025-06-11 09:10:24 +09:00
  • 99c33d011d rebase + nit (#2260) Cheng 2025-06-11 02:51:51 +09:00
  • 62fecf3e13 fix conv export (#2265) Awni Hannun 2025-06-10 09:34:01 -07:00
  • 7c4eb5d03e CUDA backend: random (#2261) Cheng 2025-06-11 00:59:56 +09:00
  • bae9a6b404 CUDA backend: sort (#2262) Cheng 2025-06-11 00:59:47 +09:00
  • 004c1d8ef2 Report number of missing parameters (#2264) Christopher Fleetwood 2025-06-10 14:37:50 +01:00
  • 7ebb2e0193 CUDA backend: binary ops (#2259) Cheng 2025-06-10 22:37:40 +09:00
  • 9ce77798b1 fix export to work with gather/scatter axis (#2263) Awni Hannun 2025-06-09 20:37:27 -07:00
  • f8bad60609 CUDA backend: unary ops (#2158) Cheng 2025-06-09 22:45:08 +09:00
  • 5866b3857b Refactor the lu test (#2250) Emmanuel Ferdman 2025-06-07 16:12:08 +03:00
  • 1ca616844b Fix unintuitive metal kernel caching (#2242) Awni Hannun 2025-06-06 20:08:15 -07:00
  • 2e8cf0b450 Change layernorms to two pass algorithm (#2246) Angelos Katharopoulos 2025-06-06 13:34:56 -07:00
  • 24f89173d1 CUDA backend: matmul (#2241) Cheng 2025-06-07 04:24:04 +09:00
  • c6a20b427a Improve metal elementwise kernels (#2247) Awni Hannun 2025-06-06 11:37:40 -07:00
  • a5ac9244c4 fix linux linking error (#2248) Awni Hannun 2025-06-06 10:41:51 -07:00
  • c763fe1be0 default strict mode for module update and update_modules (#2239) Awni Hannun 2025-06-05 15:27:02 -07:00
  • 52dc8c8cd5 Add profiler annotations in common primitives for CUDA backend (#2244) Cheng 2025-06-05 11:55:12 +09:00
  • aede70e81d Perf regression fix (#2243) v0.26.1 Angelos Katharopoulos 2025-06-03 17:55:12 -07:00
  • 85a8beb5e4 Avoid atomic updates across CPU/GPU in CUDA event (#2231) Cheng 2025-06-04 08:49:06 +09:00
  • 0bb89e9e5f Share more common code in Compiled (#2240) Cheng 2025-06-04 08:48:50 +09:00
  • 5685ceb3c7 Avoid invoking allocator::malloc when creating CUDA event (#2232) Cheng 2025-06-04 08:48:40 +09:00