Alex Barron 99bb7d3a58 GPU mx.sign for complex64 (#1326)

2024-08-14 07:54:53 -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

JIT compile option for binary minimization (#1091)

2024-05-22 12:57:13 -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

Fix cpu compile (#934)

2024-04-01 17:37:12 -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 donation condition for compilation (#1237)

2024-06-26 09:04:05 -07:00

compiled.h

Fix cpu compile (#934)

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

conv.cpp

Add groups to 2-D convolutions (#1129)

2024-05-22 20:01:44 -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

Spelling (#342)

2024-01-01 21:08:17 -08:00

make_compiled_preamble.sh

Fix a couple bugs (#1161)

2024-05-28 15:18:18 -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

Fix reshape copy bug (#1253)

2024-07-07 21:37:00 -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

smaller CPU binary (#1203)

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

reduce.cpp

Add isfinite (#1318)

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

reduce.h

smaller CPU binary (#1203)

2024-06-14 09:46:55 -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

Add some internal GPU apis (#1177)

2024-06-04 09:24:26 -07:00

slicing.h

Add some internal GPU apis (#1177)

2024-06-04 09:24:26 -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

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

2024-02-22 15:10:48 -08: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

Refactor CPU compile preamble (#708)

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

utils.h

Fix reshape copy bug (#1253)

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