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