| .. |
|
jit
|
Use int64 stride everywhere (#1671)
|
2024-12-09 11:09:02 -08:00 |
|
kernels
|
Allow no copy negative strides in as_strided and slice (#1688)
|
2024-12-12 08:59:45 -08:00 |
|
allocator.cpp
|
Fix some leaks and races (#1629)
|
2024-11-27 20:01:20 -08:00 |
|
allocator.h
|
Wired (#1510)
|
2024-10-25 09:35:33 -07:00 |
|
binary.cpp
|
Allow no copy negative strides in as_strided and slice (#1688)
|
2024-12-12 08:59:45 -08:00 |
|
binary.h
|
Fixes for large arrays with a few ops (#1299)
|
2024-07-30 17:18:39 -07:00 |
|
CMakeLists.txt
|
Use osx deployment target to pick Metal version (#1595)
|
2024-11-18 19:16:49 -08:00 |
|
compiled.cpp
|
Use int64 stride everywhere (#1671)
|
2024-12-09 11:09:02 -08:00 |
|
conv.cpp
|
Use int64 stride everywhere (#1671)
|
2024-12-09 11:09:02 -08:00 |
|
copy.cpp
|
Use int64 stride everywhere (#1671)
|
2024-12-09 11:09:02 -08:00 |
|
copy.h
|
Use int64 stride everywhere (#1671)
|
2024-12-09 11:09:02 -08:00 |
|
custom_kernel.cpp
|
fix dispatch threads for a few kernels (#1594)
|
2024-11-18 08:35:25 -08:00 |
|
device.cpp
|
docs: update device.cpp (#1632)
|
2024-11-27 20:58:26 -08:00 |
|
device.h
|
Fully wrap the command encoder (#1572)
|
2024-11-08 11:50:21 -08:00 |
|
distributed.cpp
|
Adds send/recv ops in distributed (#1366)
|
2024-08-26 23:01:37 -07:00 |
|
event.cpp
|
Fix array is_available race cases (#1468)
|
2024-10-07 19:13:50 -07:00 |
|
fft.cpp
|
Use int64 stride everywhere (#1671)
|
2024-12-09 11:09:02 -08:00 |
|
hadamard.cpp
|
Fully wrap the command encoder (#1572)
|
2024-11-08 11:50:21 -08:00 |
|
indexing.cpp
|
Allow no copy negative strides in as_strided and slice (#1688)
|
2024-12-12 08:59:45 -08:00 |
|
jit_kernels.cpp
|
Allow no copy negative strides in as_strided and slice (#1688)
|
2024-12-12 08:59:45 -08:00 |
|
kernels.h
|
Reduce specializations (#1607)
|
2024-11-21 19:53:00 -08:00 |
|
make_compiled_preamble.sh
|
Dispatch bf16 at run time when using the JIT (#1584)
|
2024-11-15 16:54:36 -08:00 |
|
matmul.cpp
|
Use int64 stride everywhere (#1671)
|
2024-12-09 11:09:02 -08:00 |
|
matmul.h
|
Use int64 stride everywhere (#1671)
|
2024-12-09 11:09:02 -08:00 |
|
metal_impl.h
|
Add synchronize function (#1006)
|
2024-04-22 08:25:46 -07:00 |
|
metal.cpp
|
Fix some leaks and races (#1629)
|
2024-11-27 20:01:20 -08:00 |
|
metal.h
|
Added missing unordered_map includes (#1635)
|
2024-12-02 07:03:03 -08:00 |
|
nojit_kernels.cpp
|
Reduce specializations (#1607)
|
2024-11-21 19:53:00 -08:00 |
|
normalization.cpp
|
Fully wrap the command encoder (#1572)
|
2024-11-08 11:50:21 -08:00 |
|
primitives.cpp
|
Flatten and unflatten (#1692)
|
2024-12-11 21:51:37 -08:00 |
|
quantized.cpp
|
3 and 6 bit quantization (#1613)
|
2024-11-22 10:22:13 -08:00 |
|
reduce.cpp
|
Allow no copy negative strides in as_strided and slice (#1688)
|
2024-12-12 08:59:45 -08:00 |
|
reduce.h
|
Reductions update (#1351)
|
2024-11-04 22:25:16 -08:00 |
|
resident.cpp
|
Fix some leaks and races (#1629)
|
2024-11-27 20:01:20 -08:00 |
|
resident.h
|
Wired (#1510)
|
2024-10-25 09:35:33 -07:00 |
|
rope.cpp
|
Fully wrap the command encoder (#1572)
|
2024-11-08 11:50:21 -08:00 |
|
scaled_dot_product_attention.cpp
|
Use int64 stride everywhere (#1671)
|
2024-12-09 11:09:02 -08:00 |
|
scan.cpp
|
Fully wrap the command encoder (#1572)
|
2024-11-08 11:50:21 -08:00 |
|
slicing.cpp
|
Allow no copy negative strides in as_strided and slice (#1688)
|
2024-12-12 08:59:45 -08:00 |
|
slicing.h
|
Use int64 stride everywhere (#1671)
|
2024-12-09 11:09:02 -08:00 |
|
softmax.cpp
|
Fully wrap the command encoder (#1572)
|
2024-11-08 11:50:21 -08:00 |
|
sort.cpp
|
Fix small sort with metal validation (#1695)
|
2024-12-12 09:21:45 -08:00 |
|
ternary.cpp
|
Allow no copy negative strides in as_strided and slice (#1688)
|
2024-12-12 08:59:45 -08:00 |
|
ternary.h
|
Add some internal GPU apis (#1177)
|
2024-06-04 09:24:26 -07:00 |
|
unary.cpp
|
Allow no copy negative strides in as_strided and slice (#1688)
|
2024-12-12 08:59:45 -08:00 |
|
unary.h
|
Add some internal GPU apis (#1177)
|
2024-06-04 09:24:26 -07:00 |
|
utils.cpp
|
Use int64 stride everywhere (#1671)
|
2024-12-09 11:09:02 -08:00 |
|
utils.h
|
Use int64 stride everywhere (#1671)
|
2024-12-09 11:09:02 -08:00 |
