Unverified Commit b37bf51e authored by Lucas Wilkinson's avatar Lucas Wilkinson Committed by GitHub
Browse files

[CI/Test] Fix FP8 per-tensor quant test reference scale shape (#30352)


Signed-off-by: default avatarLucas Wilkinson <lwilkins@redhat.com>
parent 95501a70
......@@ -103,7 +103,7 @@ def ref_dynamic_per_tensor_fp8_quant(
.clamp(fp8_traits_min, fp8_traits_max)
.to(FP8_DTYPE)
)
return ref_out, ref_scale.view((1, 1))
return ref_out, ref_scale.view(1)
def native_w8a8_block_matmul(
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment