Factor out the core FA loop into flash_atten_f16_one_chunk and add an outter loop on top that handles the chunks. |
||
|---|---|---|
| .. | ||
| cmake | ||
| include | ||
| src | ||
| .gitignore | ||
| CMakeLists.txt | ||
Factor out the core FA loop into flash_atten_f16_one_chunk and add an outter loop on top that handles the chunks. |
||
|---|---|---|
| .. | ||
| cmake | ||
| include | ||
| src | ||
| .gitignore | ||
| CMakeLists.txt | ||