Skip to content

Commit 54129fe

Browse files
committed
chore: align test_quant_mxfp4 with triton kernel
1 parent a31b891 commit 54129fe

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

op_tests/triton_tests/test_quant_mxfp4.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -127,8 +127,8 @@ def torch_dynamic_mxfp4_quant(
127127

128128
# Merge results
129129
e2m1_value = torch.full_like(qx, 0x7, dtype=torch.uint8)
130-
e2m1_value = torch.where(denormal_mask, denormal_x, e2m1_value)
131130
e2m1_value = torch.where(normal_mask, normal_x, e2m1_value)
131+
e2m1_value = torch.where(denormal_mask, denormal_x, e2m1_value)
132132

133133
# add sign back
134134
sign_lp = s >> (MBITS_F32 + EBITS_F32 - MBITS_FP4 - EBITS_FP4)

0 commit comments

Comments
 (0)