Fix: fence_async_shared -> fence_view_async_shared

This commit is contained in:
2026-05-22 19:25:05 +00:00
parent 7806ac812a
commit 6827862b3d

View File

@@ -201,7 +201,7 @@ class FmhaV3RealSoftmax:
for kb in cutlass.range(cute.size(tOrP0, mode=[2]), unroll_full=True):
cute.gemm(pv_mma, tOtO0, tOrP0[(None,None,kb)], tCrV[(None,None,kb,kvh.index)], tOtO0)
pv_mma.set(tcgen05.Field.ACCUMULATE, True)
cute.arch.fence_async_shared()
cute.arch.fence_view_async_shared()
kvh.release()
acc_pipe.producer_commit(acc_st); acc_st.advance()
acc_pipe.producer_tail(acc_st)