Skip to content

Commit fff92ea

Browse files
Robert Luofacebook-github-bot
authored andcommitted
Fix type to assert group size (pytorch#682)
Summary: X-link: pytorch#3601 Pull Request resolved: facebookresearch/FBGEMM#682 Fix type to assert group size Reviewed By: jianyuh Differential Revision: D68499555 fbshipit-source-id: 58ac6869189924610d03ae7339ca3d24b1898fed
1 parent 05b3f3b commit fff92ea

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

fbgemm_gpu/fbgemm_gpu/quantize_comm.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -241,7 +241,7 @@ def calc_quantized_size(
241241
group_size = MX_GROUP_SIZE_DEFAULT
242242
assert (
243243
input_len % group_size == 0
244-
), f"input_len {input_len} needs to be multiple of group_size 32"
244+
), f"input_len {input_len} needs to be multiple of group_size {group_size}"
245245
# quantized output size = half input size + number of groups (shared exp)
246246
ctx = none_throws(ctx)
247247
return (input_len // 2) + (input_len // ctx.mx_group_size)

0 commit comments

Comments
 (0)