biondizzle/nvfp4-megamoe-kernel
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: fence_proxy for async global

Browse Source
This commit is contained in:
biondizzle
2026-05-24 03:44:26 +00:00
parent 3b6f19043f
commit 7b0b3ad1fc
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
dsv4/kernels/attention/fmha.py
View File

@@ -394,7 +394,8 @@ class FmhaKernel:
gP_flat = cute.make_tensor(tCgP.iterator, cute.coalesce(tCgP.layout))
for idx in cutlass.range(cute.size(rP_flat), vectorize=True):
gP_flat[idx] = rP_flat[idx]
cute.arch.fence_proxy("async.global", space="cta")
# Ensure gP writes are visible before TMA reads
cute.arch.fence_proxy("async", space="cta")
cute.arch.fence_proxy("async.shared", space="cta")
if kt > 0:
for i in range(n_corr_tiles):