| File Name ↓ | File Size ↓ | Date ↓ |
|---|---|---|
| Parent directory/ | - | - |
| spinquant/ | - | 2024-Nov-06 11:28 |
| test_sdpa_with_kv_cache.py | 18.3 KiB | 2024-Nov-06 11:28 |
| test_preprocess_custom_ops.py | 1009 B | 2024-Nov-06 11:28 |
| targets.bzl | 4.3 KiB | 2024-Nov-06 11:28 |
| sdpa_with_kv_cache.py | 3.7 KiB | 2024-Nov-06 11:28 |
| preprocess_custom_ops.py | 2.3 KiB | 2024-Nov-06 11:28 |
| op_tile_crop_test.cpp | 3.1 KiB | 2024-Nov-06 11:28 |
| op_tile_crop.h | 550 B | 2024-Nov-06 11:28 |
| op_tile_crop.cpp | 3.1 KiB | 2024-Nov-06 11:28 |
| op_sdpa_with_kv_cache_test.cpp | 26.0 KiB | 2024-Nov-06 11:28 |
| op_sdpa_test.cpp | 22.5 KiB | 2024-Nov-06 11:28 |
| op_sdpa_aot.cpp | 3.2 KiB | 2024-Nov-06 11:28 |
| op_sdpa.h | 1.2 KiB | 2024-Nov-06 11:28 |
| op_sdpa.cpp | 32.8 KiB | 2024-Nov-06 11:28 |
| op_fallback.h | 493 B | 2024-Nov-06 11:28 |
| op_fallback.cpp | 1.3 KiB | 2024-Nov-06 11:28 |
| model_sharding.py | 3.8 KiB | 2024-Nov-06 11:28 |
| __init__.py | 0 B | 2024-Nov-06 11:28 |
| TARGETS | 807 B | 2024-Nov-06 11:28 |
| CMakeLists.txt | 3.4 KiB | 2024-Nov-06 11:28 |