Skip to content

Commit bb87e43

Browse files
cthifacebook-github-bot
authored andcommitted
Fix typo in FP4 quantize (#4933)
Summary: X-link: facebookresearch/FBGEMM#1956 Pull Request resolved: #4933 as title Differential Revision: D83083612 fbshipit-source-id: 8c6b065f05f1b202e1fd7b3746565c00e6b1d0bd
1 parent 947e4e5 commit bb87e43

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

fbgemm_gpu/experimental/gemm/triton_gemm/fp4_quantize.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1461,7 +1461,7 @@ def triton_scale_nvfp4_quant(
14611461
stochastic_casting (bool): Whether to use stochastic casting.
14621462
14631463
Returns:
1464-
torch.Tensor: [M / 2] nvfp4 scaled tensor packed into in8
1464+
torch.Tensor: [M / 2] nvfp4 scaled tensor packed into int8
14651465
torch.Tensor: [M / group_size] nvfp4 shared exponents into int8
14661466
14671467
eg.

0 commit comments

Comments
 (0)