biondizzle/nvfp4-megamoe-kernel
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity
Files
b7c7e9fb5041fa4f379dc03a2802e014d243e110
nvfp4-megamoe-kernel/src
History
biondizzle b7c7e9fb50 refactor: clean up slot_token handling in cutlass_grouped_nvfp4_gemm 
- Split provided_slot_token vs slot_token_out (returned to caller)
- No gather when slot_token=None (L2 path), no unnecessary alloc
- .contiguous() on gathered tensors for CUTLASS alignment
- Return slot_token_out consistently
2026-05-15 10:11:40 +00:00
..
nvfp4_megamoe_kernel
refactor: clean up slot_token handling in cutlass_grouped_nvfp4_gemm
2026-05-15 10:11:40 +00:00
nvfp4_megamoe_kernel.egg-info
Implement TileLang NVFP4 mega_moe L1/L2 kernels
2026-05-13 22:36:58 +00:00