HIP: fix template skip for DKQ > 256 mfma kernels (llama/29559)

This commit is contained in:
uvos
2026-10-06 10:38:06 +03:00
committed by Georgi Gerganov
parent 0e537510d6
commit 0c17f7d6bb
+1 -1
View File
@@ -1875,7 +1875,7 @@ static __global__ void flash_attn_ext_f16(
#endif // defined(AMD_WMMA_AVAILABLE)
#if defined(AMD_MFMA_AVAILABLE)
if (ncols1*ncols2 < 16 || (DKQ > 256 && ncols1*ncols2 < 64)) {
if (ncols1*ncols2 < 16 || (DKQ > 256 && ncols1*ncols2 < 32)) {
NO_DEVICE_CODE;
return;
}