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

fix debug test

Browse Source
This commit is contained in:
biondizzle
2026-05-24 22:04:51 +00:00
parent a391aa1fd3
commit 35a3c04e8e
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tests/unit/test_d1_rescale_debug.py
View File

@@ -43,7 +43,7 @@ def test_multi_kv_debug(hd=64, s_k=256):
pv_n_tile = kernel.pv_n_tile
n_pv_tiles = kernel.n_pv_tiles
print(f" n_kv_tiles={kernel.n_kv_tiles}, pv_n_tile={pv_n_tile}, n_pv_tiles={n_pv_tiles}")
print(f" tmem_o0_offset={kernel.tmem_o0_offset}, tmem_p0_offset={kernel.tmem_p0_offset}")
# tmem_o0_offset is set in _setup, not __init__
stream = cuda.CUstream(torch.cuda.current_stream().cuda_stream)