fix fp8 quant tests

ProExpertProg · ProExpertProg · commit a3ebf0a2e47a · 2025-10-15T12:09:48.000-04:00
Signed-off-by: Luka Govedič &lt;lgovedic@redhat.com&gt;
diff --git a/tests/kernels/quant_utils.py b/tests/kernels/quant_utils.py
@@ -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,))
+    return ref_out, ref_scale.view((1, 1))
 
 
 def native_w8a8_block_matmul(

Original file line number	Diff line number	Diff line change
`@@ -103,7 +103,7 @@ def ref_dynamic_per_tensor_fp8_quant(`
`103`	`103`	`.clamp(fp8_traits_min, fp8_traits_max)`
`104`	`104`	`.to(FP8_DTYPE)`
`105`	`105`	`)`
`106`		`- return ref_out, ref_scale.view((1,))`
	`106`	`+ return ref_out, ref_scale.view((1, 1))`
`107`	`107`
`108`	`108`
`109`	`109`	`def native_w8a8_block_matmul(`