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

D1.3: Fix layout diagnostic - compute c_major outside kernel

Browse Source
This commit is contained in:
biondizzle
2026-05-23 22:17:54 +00:00
parent 6afa610b0d
commit fc4106f37e
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_3_layout_diag.py
View File

@@ -49,7 +49,7 @@ def main():
pv_a_major = a_major if use_smem_p else cute.nvgpu.OperandMajorMode.K
pv_source = tcgen05.OperandSource.SMEM if use_smem_p else tcgen05.OperandSource.TMEM
c_major = LayoutEnum.from_tensor(ct.from_dlpack(_c)).mma_major_mode()
c_major = LayoutEnum.from_tensor(ct.from_dlpack(c)).mma_major_mode()
pv_mma = utils.sm100.make_trivial_tiled_mma(
BFloat16, BFloat16, pv_a_major, c_major, Float32,