Skip to content

Commit b37bf51

Browse files
[CI/Test] Fix FP8 per-tensor quant test reference scale shape (#30352)
Signed-off-by: Lucas Wilkinson <lwilkins@redhat.com>
1 parent 95501a7 commit b37bf51

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/kernels/quant_utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -103,7 +103,7 @@ def ref_dynamic_per_tensor_fp8_quant(
103103
.clamp(fp8_traits_min, fp8_traits_max)
104104
.to(FP8_DTYPE)
105105
)
106-
return ref_out, ref_scale.view((1, 1))
106+
return ref_out, ref_scale.view(1)
107107

108108

109109
def native_w8a8_block_matmul(

0 commit comments

Comments
 (0)