revert mistake change

jiqing-feng · jiqing-feng · commit 8799041b028f · 2025-06-06T16:30:24.000Z
Signed-off-by: jiqing-feng &lt;jiqing.feng@intel.com&gt;
diff --git a/tests/test_ops.py b/tests/test_ops.py
@@ -53,6 +53,7 @@ def test_int8_vectorwise_quant(self, threshold, device):
         assert out_row.dtype == torch.int8
         assert out_row.device == A.device
         assert row_stats.shape == (10,)
+        assert row_stats.dtype == torch.float32
         assert row_stats.device == A.device
 
         if threshold > 0.0:
