| .. |
|
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 |
