Angelos Katharopoulos 881f09b2e2 Allow querying the allocator for the buffer size (#1404)

2024-09-11 21:02:16 -07:00

..

arange.h

copyright + ack

2023-11-30 11:12:53 -08:00

arg_reduce.cpp

copyright + ack

2023-11-30 11:12:53 -08:00

binary_two.h

Implement the 'where' primitive for conditional selection (#664)

2024-02-22 15:10:48 -08:00

binary.cpp

smaller CPU binary (#1203)

2024-06-14 09:46:55 -07:00

binary.h

Allow querying the allocator for the buffer size (#1404)

2024-09-11 21:02:16 -07:00

cholesky.cpp

JIT compile option for binary minimization (#1091)

2024-05-22 12:57:13 -07:00

CMakeLists.txt

Fast Hadamard Transform (#1249)

2024-07-09 20:39:01 -07:00

common.cpp

Custom transforms (#1246)

2024-07-10 18:00:01 -07:00

compiled_cpu.cpp

Read arrays from files faster (#1330)

2024-08-14 20:09:56 -07:00

compiled_nocpu.cpp

Ios compile (#784)

2024-03-04 20:02:26 -08:00

compiled_preamble.h

Refactor CPU compile preamble (#708)

2024-02-19 06:12:53 -08:00

compiled.cpp

Fix compiel with byte sized constants (#1381)

2024-08-30 17:24:35 -07:00

compiled.h

Fix cpu compile (#934)

2024-04-01 17:37:12 -07:00

conv.cpp

Transposed Convolution (#1245)

2024-09-06 19:52:38 -07:00

copy.cpp

Fix reshape copy bug (#1253)

2024-07-07 21:37:00 -07:00

copy.h

Add a SliceUpdate op and primitive (#850)

2024-03-20 10:39:25 -07:00

default_primitives.cpp

Custom transforms (#1246)

2024-07-10 18:00:01 -07:00

erf.cpp

copyright + ack

2023-11-30 11:12:53 -08:00

fft.cpp

copyright + ack

2023-11-30 11:12:53 -08:00

hadamard.cpp

Fast Hadamard Transform (#1249)

2024-07-09 20:39:01 -07:00

hadamard.h

Fast Hadamard Transform (#1249)

2024-07-09 20:39:01 -07:00

indexing.cpp

copyright + ack

2023-11-30 11:12:53 -08:00

inverse.cpp

CPU mx.linalg.cholesky_inverse and mx.linalg.tri_inv (#1307)

2024-08-08 15:18:02 -07:00

lapack_helper.h

Add SVD primitive (#809)

2024-03-12 12:30:11 -07:00

load.cpp

Async load (#1372)

2024-08-28 14:21:55 -07:00

load.h

Async load (#1372)

2024-08-28 14:21:55 -07:00

make_compiled_preamble.sh

fix compiling with space in paths (#1332)

2024-08-15 16:39:24 -07:00

masked_mm.cpp

Float mask update (#1152)

2024-05-23 17:20:44 -07:00

ops.h

GPU mx.sign for complex64 (#1326)

2024-08-14 07:54:53 -07:00

primitives.cpp

Xcode 160 (#1384)

2024-09-10 15:15:17 -07:00

qrf.cpp

fixes for building and running on iOS (#619)

2024-02-04 12:29:17 -08:00

quantized.cpp

Rename block sparse (#1149)

2024-05-22 07:48:34 -07:00

reduce_utils.cpp

Fix reduce edge case (#1389)

2024-09-01 21:37:51 -07:00

reduce.cpp

Add isfinite (#1318)

2024-08-13 14:49:28 -07:00

reduce.h

Refactor reductions and fix scatter atomics for large sizes (#1300)

2024-08-22 16:03:31 -07:00

scan.cpp

Fix scan (#1188)

2024-06-05 14:21:58 -07:00

select.cpp

Implement the 'where' primitive for conditional selection (#664)

2024-02-22 15:10:48 -08:00

slicing.cpp

Fix slice data size (#1394)

2024-09-04 19:10:43 -07:00

slicing.h

Fix slice data size (#1394)

2024-09-04 19:10:43 -07:00

softmax.cpp

Option for precise softmax (#953)

2024-04-04 08:32:35 -07:00

sort.cpp

Fix strided sort bug (#1236)

2024-06-26 14:32:11 -07:00

svd.cpp

No CPU option for binary minimization (#1105)

2024-05-13 16:08:11 -07:00

ternary.h

Allow querying the allocator for the buffer size (#1404)

2024-09-11 21:02:16 -07:00

threefry.cpp

copyright + ack

2023-11-30 11:12:53 -08:00

threefry.h

copyright + ack

2023-11-30 11:12:53 -08:00

unary.h

Allow querying the allocator for the buffer size (#1404)

2024-09-11 21:02:16 -07:00

utils.h

Allow querying the allocator for the buffer size (#1404)

2024-09-11 21:02:16 -07:00