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

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

..

kernels

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

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

mps

copyright + ack

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

allocator.cpp

Support disable metal buffer cache to prevent performance degradation caused by large memory caching (#390)

2024-01-18 08:33:34 -08:00

allocator.h

Memory allocation (#292)

2024-01-02 11:59:19 -08:00

CMakeLists.txt

Custom primitive + RoPE fat op (#676)

2024-02-14 14:04:25 -08:00

compiled_preamble.h

Kernel generation (#614)

2024-02-07 13:15:59 -08:00

compiled.cpp

Shapeless compilation for some graphs (#687)

2024-02-19 21:43:54 -08:00

conv.cpp

Remove unused variables (#706)

2024-02-18 12:50:10 -08:00

copy.cpp

Buffer Donation (#519)

2024-01-26 16:30:33 -08:00

copy.h

copyright + ack

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

device.cpp

Update gather and scatter to not use Argument Encoder (#683)

2024-02-14 13:42:13 -08:00

device.h

Kernel generation (#614)

2024-02-07 13:15:59 -08:00

fft.cpp

copyright + ack

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

indexing.cpp

Up to 10x faster scatter. (#709)

2024-02-21 11:09:30 -08:00

make_compiled_preamble.sh

quote file name (#670)

2024-02-11 10:33:30 -08:00

matmul.cpp

Custom VJP and checkpointing (#541)

2024-01-30 16:04:45 -08:00

matmul.h

Update GEMM (#424)

2024-01-17 12:42:39 -08:00

metal.cpp

Fix eval in trace bugs (#612)

2024-02-02 09:57:12 -08:00

metal.h

Support disable metal buffer cache to prevent performance degradation caused by large memory caching (#390)

2024-01-18 08:33:34 -08:00

primitives.cpp

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

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

quantized.cpp

Quantized matmul fix (#677)

2024-02-12 18:54:21 -08:00

reduce.cpp

Add GPU support for uint64/int64 reductions (#569)

2024-01-31 11:18:04 -08:00

rope.cpp

Separate fast ops and primitives (#699)

2024-02-16 19:16:39 -08:00

scan.cpp

copyright + ack

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

softmax.cpp

Fix the softmax fix (#661)

2024-02-09 17:02:13 -08:00

sort.cpp

Spelling (#342)

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

utils.h

Update gather and scatter to not use Argument Encoder (#683)

2024-02-14 13:42:13 -08:00