Merge commit 'e9596228ff7f6ddb68fbd2f0f9e964cfb6af61cf' into develop

This commit is contained in:
assistant-librarian[bot]
2025-10-31 18:15:38 +00:00
parent 3d78c17295
commit 96199abfbe

View File

@@ -578,6 +578,9 @@ struct BlockFmhaPipelineQRKSVS
if constexpr(kHasDropout)
{
// K and dropout use the same address in LDS, finish loading from k_lds_window by
// gemm_0 to reuse LDS.
block_sync_lds();
dropout.template Run<decltype(gemm_0), SMPLComputeDataType, RandValOutputDataType>(
smem_ptr, seqlen_k_start + i_total_loops * kN0, p_compute, randval_dram_window);
}