Skip to content

Commit 991b541

Browse files
committed
fix dp save issue
1 parent 50bb240 commit 991b541

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

examples/sycl/sdpa_bwd/sdpa_backward.cpp

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -513,6 +513,7 @@ dq_dk_dv_1colblock(Trait &trait, Param<typename Trait::DType> &param,
513513
sycl::group_barrier(group);
514514

515515
copy(tilesavedP, tdPrdPl, tdPgdP); // save dP to buffer after P used by dV
516+
sycl::group_barrier(group);
516517

517518
if (n_block > 0)
518519
copy(tileloaddQ, tdQgdQ, tdQrdQ);

0 commit comments

Comments
 (0)