Skip to content

Commit ba884d6

Browse files
committed
fix lint
Signed-off-by: Bill Nell <bnell@redhat.com>
1 parent 5b4a8b6 commit ba884d6

File tree

1 file changed

+3
-4
lines changed

1 file changed

+3
-4
lines changed

tests/kernels/moe/utils.py

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -156,14 +156,13 @@ def make_quantized_test_activations(
156156
assert (quant_dtype == torch.float8_e4m3fn
157157
or quant_dtype == torch.int8), "only fp8/int8 supported"
158158
a_q = torch.zeros_like(a, dtype=quant_dtype)
159-
a_scale = [None] * E
159+
a_scale_l = [None] * E
160160
for e in range(E):
161-
a_q[e], a_scale[e] = moe_kernel_quantize_input(
161+
a_q[e], a_scale_l[e] = moe_kernel_quantize_input(
162162
a[e], None, quant_dtype, per_act_token_quant, block_shape)
163-
a_scale = torch.stack(a_scale)
163+
a_scale = torch.stack(a_scale_l)
164164

165165
if not per_act_token_quant and block_shape is None:
166-
assert a_scale is not None
167166
a_scale = a_scale.view(E, 1, 1)
168167

169168
return a, a_q, a_scale

0 commit comments

Comments
 (0)