| .. |
|
steel
|
Remove masks from BlockLoader and clear out load case for invalid thread (#634)
|
2024-02-05 16:00:17 -08:00 |
|
arange.metal
|
copyright + ack
|
2023-11-30 11:12:53 -08:00 |
|
arg_reduce.metal
|
Add GPU support for uint64/int64 reductions (#569)
|
2024-01-31 11:18:04 -08:00 |
|
atomic.h
|
Fix eye for larger matrices (#463)
|
2024-01-16 00:51:24 -08:00 |
|
bf16_math.h
|
copyright + ack
|
2023-11-30 11:12:53 -08:00 |
|
bf16.h
|
copyright + ack
|
2023-11-30 11:12:53 -08:00 |
|
binary_two.metal
|
Multi output primitives (#330)
|
2024-01-08 16:39:08 -08:00 |
|
binary.h
|
Kernel generation (#614)
|
2024-02-07 13:15:59 -08:00 |
|
binary.metal
|
Kernel generation (#614)
|
2024-02-07 13:15:59 -08:00 |
|
CMakeLists.txt
|
Update GEMM (#424)
|
2024-01-17 12:42:39 -08:00 |
|
compiled_preamble.h
|
Kernel generation (#614)
|
2024-02-07 13:15:59 -08:00 |
|
complex.h
|
Spelling (#342)
|
2024-01-01 21:08:17 -08:00 |
|
conv_params.h
|
copyright + ack
|
2023-11-30 11:12:53 -08:00 |
|
conv.h
|
Update GEMM (#424)
|
2024-01-17 12:42:39 -08:00 |
|
conv.metal
|
Update GEMM (#424)
|
2024-01-17 12:42:39 -08:00 |
|
copy.metal
|
copyright + ack
|
2023-11-30 11:12:53 -08:00 |
|
defines.h
|
copyright + ack
|
2023-11-30 11:12:53 -08:00 |
|
erf.h
|
copyright + ack
|
2023-11-30 11:12:53 -08:00 |
|
gemv.metal
|
Fix oob reads in gemv kernel (#523)
|
2024-01-22 12:06:04 -08:00 |
|
indexing.metal
|
Faster gather (#626)
|
2024-02-04 17:25:44 -08:00 |
|
quantized.metal
|
Quantize with groups of 32 (#511)
|
2024-01-21 06:19:05 -08:00 |
|
random.metal
|
copyright + ack
|
2023-11-30 11:12:53 -08:00 |
|
reduce.h
|
Fix eye for larger matrices (#463)
|
2024-01-16 00:51:24 -08:00 |
|
reduce.metal
|
Add GPU support for uint64/int64 reductions (#569)
|
2024-01-31 11:18:04 -08:00 |
|
scan.metal
|
copyright + ack
|
2023-11-30 11:12:53 -08:00 |
|
softmax.metal
|
copyright + ack
|
2023-11-30 11:12:53 -08:00 |
|
sort.metal
|
Spelling (#342)
|
2024-01-01 21:08:17 -08:00 |
|
unary.h
|
Kernel generation (#614)
|
2024-02-07 13:15:59 -08:00 |
|
unary.metal
|
Kernel generation (#614)
|
2024-02-07 13:15:59 -08:00 |
|
utils.h
|
Kernel generation (#614)
|
2024-02-07 13:15:59 -08:00 |
