Skip to content

Commit 874549c

Browse files
committed
vulkan: use 4 rows for scalar FA large tile size
1 parent 5266379 commit 874549c

File tree

1 file changed

+1
-3
lines changed

1 file changed

+1
-3
lines changed

ggml/src/ggml-vulkan/ggml-vulkan.cpp

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -2537,10 +2537,8 @@ static constexpr uint32_t scalar_flash_attention_num_small_rows = 1;
25372537
static uint32_t get_fa_scalar_num_large_rows(uint32_t hsk, uint32_t hsv) {
25382538
if (hsv >= 192) {
25392539
return 2;
2540-
} else if ((hsv | hsk) & 8) {
2541-
return 4;
25422540
} else {
2543-
return 8;
2541+
return 4;
25442542
}
25452543
}
25462544

0 commit comments

Comments
 (0)