char-fork/llama.cpp

git clone https://git.t4t.associates/char-fork/llama.cpp

AesSedaiAdd flash attention MMA / Tiles to support MiMo-V2.5 (#22812)046e28443

char/aliceai-t5-moe
439 B12 linesraw
1// This file has been autogenerated by generate_cu_files.py, do not edit manually.
2
3#include "../fattn-mma-f16.cuh"
4
5DECL_FATTN_MMA_F16_CASE(64, 64, 1, 8);
6DECL_FATTN_MMA_F16_CASE(80, 80, 1, 8);
7DECL_FATTN_MMA_F16_CASE(96, 96, 1, 8);
8DECL_FATTN_MMA_F16_CASE(112, 112, 1, 8);
9DECL_FATTN_MMA_F16_CASE(128, 128, 1, 8);
10DECL_FATTN_MMA_F16_CASE(192, 128, 1, 8);
11DECL_FATTN_MMA_F16_CASE(256, 256, 1, 8);
12DECL_FATTN_MMA_F16_CASE(512, 512, 1, 8);