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

fix: indentation error in diag disable

Browse Source
This commit is contained in:
biondizzle
2026-05-23 01:11:25 +00:00
parent a9cace316d
commit dc44fa187a
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
tests/unit/test_fmha_v3_stage_c.py
View File

@@ -375,8 +375,9 @@ class FmhaV3StageCMulti:
cute.arch.fence_view_async_tmem_store() cute.arch.fence_view_async_tmem_store()
# === Per-tile O rescale: O *= acc_scale for kt > 0 === # === Per-tile O rescale: O *= acc_scale for kt > 0 ===
# TEMP: disabled for diagnosis # TEMP: disabled for diagnosis — O rescale not applied
if False and kt > 0: if kt > 0:
pass # O rescale disabled
if kt > 0: if kt > 0:
for i in range(n_corr_tiles): for i in range(n_corr_tiles):
tTMEM_LOADtO_i = cute.make_tensor( tTMEM_LOADtO_i = cute.make_tensor(