Awni Hannun d12573daa6 quote file name (#670)

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

..

kernels

Scatter optimization : Eliminate 64b integer divide. (#662)

2024-02-10 08:49:51 -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

Kernel generation (#614)

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

compiled_preamble.h

Kernel generation (#614)

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

compiled.cpp

Kernel generation (#614)

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

conv.cpp

Custom VJP and checkpointing (#541)

2024-01-30 16:04:45 -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

Kernel generation (#614)

2024-02-07 13:15:59 -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

Scatter optimization : Eliminate 64b integer divide. (#662)

2024-02-10 08:49:51 -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

Custom VJP and checkpointing (#541)

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

quantized.cpp

Custom VJP and checkpointing (#541)

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

reduce.cpp

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

2024-01-31 11:18:04 -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

Kernel generation (#614)

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