Skip to content

GitLab

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
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
tests/kernels/quant_utils.py
View file @ b37bf51e
...@@ -103,7 +103,7 @@ def ref_dynamic_per_tensor_fp8_quant( ...@@ -103,7 +103,7 @@ def ref_dynamic_per_tensor_fp8_quant(
.clamp(fp8_traits_min, fp8_traits_max) .clamp(fp8_traits_min, fp8_traits_max)
.to(FP8_DTYPE) .to(FP8_DTYPE)
) )
return ref_out, ref_scale.view((1, 1)) return ref_out, ref_scale.view(1)
def native_w8a8_block_matmul( 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