Cheng d34f887abc Add Primitive::name and remove Primitive::print (#2365)

2025-07-14 14:06:35 -07:00

..

gemms

add a half simd gemm fallback (#2046)

2025-04-07 09:31:29 -07:00

simd

Complex scan (#2094)

2025-04-22 18:56:28 -07:00

arange.h

redesign for faster cpu/gpu synch (#1869)

2025-03-06 19:23:38 -08:00

arg_reduce.cpp

Add remove_index utility (#2173)

2025-05-13 17:09:56 -07:00

available.cpp

Generalize gpu backend (#2138)

2025-04-30 09:08:17 -07:00

available.h

Generalize gpu backend (#2138)

2025-04-30 09:08:17 -07:00

binary_ops.h

Refactor common into cpu specific and truly common (#1817)

2025-02-03 15:58:02 -08:00

binary_two.h

reduce binary size (#1952)

2025-03-11 06:30:44 -07:00

binary.cpp

Complex scan (#2094)

2025-04-22 18:56:28 -07:00

binary.h

reduce binary size (#1952)

2025-03-11 06:30:44 -07:00

cholesky.cpp

redesign for faster cpu/gpu synch (#1869)

2025-03-06 19:23:38 -08:00

CMakeLists.txt

non-symmetric eig and eigh (#2188)

2025-05-15 13:01:44 -07:00

compiled_preamble.h

Refactor common into cpu specific and truly common (#1817)

2025-02-03 15:58:02 -08:00

compiled.cpp

Add Primitive::name and remove Primitive::print (#2365)

2025-07-14 14:06:35 -07:00

conv.cpp

fix: conv_general differences between gpu, cpu (#2070)

2025-05-09 10:26:52 -07:00

copy.cpp

reduce binary size (#1952)

2025-03-11 06:30:44 -07:00

copy.h

redesign for faster cpu/gpu synch (#1869)

2025-03-06 19:23:38 -08:00

distributed.cpp

Fix the test and add custom min/max reductions for uncommon MPI types (#2060)

2025-04-10 17:01:17 -07:00

eig.cpp

non-symmetric eig and eigh (#2188)

2025-05-15 13:01:44 -07:00

eigh.cpp

Add complex eigh (#2191)

2025-05-18 00:18:43 -07:00

encoder.cpp

redesign for faster cpu/gpu synch (#1869)

2025-03-06 19:23:38 -08:00

encoder.h

Fix multistream GPU deadlock (#1969)

2025-03-20 07:19:47 -07:00

eval.cpp

Fix multistream GPU deadlock (#1969)

2025-03-20 07:19:47 -07:00

eval.h

redesign for faster cpu/gpu synch (#1869)

2025-03-06 19:23:38 -08:00

fft.cpp

fix malloc or wait deadlock (#1976)

2025-03-20 16:48:43 -07:00

gemm.h

redesign for faster cpu/gpu synch (#1869)

2025-03-06 19:23:38 -08:00

hadamard.cpp

redesign for faster cpu/gpu synch (#1869)

2025-03-06 19:23:38 -08:00

indexing.cpp

Add remove_index utility (#2173)

2025-05-13 17:09:56 -07:00

inverse.cpp

fix malloc or wait deadlock (#1976)

2025-03-20 16:48:43 -07:00

jit_compiler.cpp

Fix compilation error on Windows (#1844)

2025-02-10 19:53:05 -08:00

jit_compiler.h

Refactor common into cpu specific and truly common (#1817)

2025-02-03 15:58:02 -08:00

lapack.h

Add complex eigh (#2191)

2025-05-18 00:18:43 -07:00

logsumexp.cpp

Custom logsumexp (#2028)

2025-03-31 07:36:55 -07:00

luf.cpp

fix malloc or wait deadlock (#1976)

2025-03-20 16:48:43 -07:00

make_compiled_preamble.ps1

Refactor common into cpu specific and truly common (#1817)

2025-02-03 15:58:02 -08:00

make_compiled_preamble.sh

fix cpu compile (#1897)

2025-02-24 14:10:30 -08:00

masked_mm.cpp

MoE backward improvements (#2335)

2025-07-07 17:59:53 -07:00

matmul.cpp

Close a couple edge case bugs: hadamard and addmm on empty inputs (#2177)

2025-05-12 10:48:57 -07:00

primitives.cpp

Distributed layers (#1270)

2025-03-21 13:52:17 -07:00

qrf.cpp

fix malloc or wait deadlock (#1976)

2025-03-20 16:48:43 -07:00

quantized.cpp

5bit quants (#2226)

2025-05-30 12:12:10 -07:00

reduce.cpp

Align mlx::core::min op nan propagation with NumPy (#2346)

2025-07-10 06:20:43 -07:00

scan.cpp

Complex scan (#2094)

2025-04-22 18:56:28 -07:00

select.cpp

reduce binary size (#1952)

2025-03-11 06:30:44 -07:00

slicing.h

Refactor common into cpu specific and truly common (#1817)

2025-02-03 15:58:02 -08:00

softmax.cpp

Custom logsumexp (#2028)

2025-03-31 07:36:55 -07:00

sort.cpp

fix malloc or wait deadlock (#1976)

2025-03-20 16:48:43 -07:00

svd.cpp

fix malloc or wait deadlock (#1976)

2025-03-20 16:48:43 -07:00

ternary.h

reduce binary size (#1952)

2025-03-11 06:30:44 -07:00

threefry.cpp

Refactor common into cpu specific and truly common (#1817)

2025-02-03 15:58:02 -08:00

threefry.h

Refactor common into cpu specific and truly common (#1817)

2025-02-03 15:58:02 -08:00

unary_ops.h

Fix CPU sign for unsigned ints (#2024)

2025-03-30 17:56:59 -07:00

unary.cpp

Fix MSVC build due to use of M_LN2 (#2058)

2025-04-10 07:41:41 -07:00

unary.h

CUDA backend: unary ops (#2158)

2025-06-09 06:45:08 -07:00