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:
1 file changed
+1
-1
@@ -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();
|
||||
}
|
||||
}
|
||||
|
||||
Reference in new issue
Block a user