CUDA/HIP: fix race in flash_attn_ext_f16_process_tile when nbatch_combine != DKQ/2 (#30103)

Suggested-by: Johannes Gäßler <johannesg@5d6.de>
This commit is contained in:
uvos authored and GitHub committed 2026-10-10 09:14:51 +03:00
1 parent 2bbca8f202
commit 1bb2b9fcbe
1 file changed
+1 -1
+1 -1
View File
@@ -1772,7 +1772,7 @@ static __device__ __forceinline__ void flash_attn_ext_f16_process_tile(
}
}
}
if (np > 1) {
if (np > 1 || nbatch_combine != DKQ/2) {
__syncthreads();
}
}