Skip to content

Commit 08b87f7

Browse files
committed
use gen5_commit
Summary: Test Plan: Reviewers: Subscribers: Tasks: Tags:
1 parent 1890a9e commit 08b87f7

File tree

1 file changed

+6
-6
lines changed

1 file changed

+6
-6
lines changed

tritonbench/operators/gdpa/gdpa_blackwell_tlx.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -333,11 +333,11 @@ def _do_dots(
333333
accum_cnt_o1 += 1
334334

335335
# epilogue
336-
# commit to release q0, q1? FIXME
336+
# commit to release q0, q1
337337
release_q0_view = tlx.local_view(consumer_release_q0, bufIdx_q)
338-
# tlx.gen5_commit(release_q0_view)
338+
tlx.tcgen05_commit(release_q0_view)
339339
release_q1_view = tlx.local_view(consumer_release_q1, bufIdx_q)
340-
# tlx.gen5_commit(release_q1_view)
340+
tlx.tcgen05_commit(release_q1_view)
341341
tlx.barrier_wait(
342342
producer_o1_view,
343343
phase_o_outer, # , first
@@ -1043,11 +1043,11 @@ def gdpa_kernel_tma_ws_blackwell(
10431043
accum_cnt_o1 += 1
10441044

10451045
# epilogue
1046-
# commit to release q0, q1? FIXME
1046+
# commit to release q0, q1
10471047
release_q0_view = tlx.local_view(consumer_release_q0, bufIdx_q)
1048-
# tlx.gen5_commit(release_q0_view)
1048+
tlx.tcgen05_commit(release_q0_view)
10491049
release_q1_view = tlx.local_view(consumer_release_q1, bufIdx_q)
1050-
# tlx.gen5_commit(release_q1_view)
1050+
tlx.tcgen05_commit(release_q1_view)
10511051
tlx.barrier_wait(
10521052
producer_o1_view,
10531053
phase_o_outer, # , first

0 commit comments

Comments
 (0)