File tree Expand file tree Collapse file tree 1 file changed +6
-6
lines changed
tritonbench/operators/gdpa Expand file tree Collapse file tree 1 file changed +6
-6
lines changed Original file line number Diff line number Diff line change @@ -333,11 +333,11 @@ def _do_dots(
333
333
accum_cnt_o1 += 1
334
334
335
335
# epilogue
336
- # commit to release q0, q1? FIXME
336
+ # commit to release q0, q1
337
337
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 )
339
339
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 )
341
341
tlx .barrier_wait (
342
342
producer_o1_view ,
343
343
phase_o_outer , # , first
@@ -1043,11 +1043,11 @@ def gdpa_kernel_tma_ws_blackwell(
1043
1043
accum_cnt_o1 += 1
1044
1044
1045
1045
# epilogue
1046
- # commit to release q0, q1? FIXME
1046
+ # commit to release q0, q1
1047
1047
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 )
1049
1049
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 )
1051
1051
tlx .barrier_wait (
1052
1052
producer_o1_view ,
1053
1053
phase_o_outer , # , first
You can’t perform that action at this time.
0 commit comments