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

NVFP4-1.1: fix indentation for @cute.jit decorators

Browse Source
This commit is contained in:
biondizzle
2026-05-28 03:52:46 +00:00
parent f6f59d34cb
commit 20d5ddfa3d
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
dsv4/kernels/gemm/fp4_quant.py
View File

@@ -59,7 +59,7 @@ def half_step_to_e2m1_idx(hs: cutlass.Int32) -> cutlass.Int32:
@cute.jit
def fp8_e4m3_from_float32(val: cutlass.Float32) -> cutlass.Int32:
def fp8_e4m3_from_float32(val: cutlass.Float32) -> cutlass.Int32:
"""Convert a positive Float32 value to FP8 E4M3 bit pattern (returned as Int32).
Only handles positive values (NVFP4 scale factors are always positive).