More tests fix

Xilinx · Mar 26, 2024 · 3f222a7 · 3f222a7
1 parent 8931a87
commit 3f222a7
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/tests/brevitas/core/test_clamp.py b/tests/brevitas/core/test_clamp.py
@@ -26,9 +26,9 @@ def test_max_value(minifloat, expected_max_val):
     nan_values = minifloat.float_clamp_impl.nan_values
     saturating = minifloat.float_clamp_impl.saturating
     max_val = max_float(
-        minifloat.exponent_bit_width,
-        minifloat.mantissa_bit_width,
-        minifloat.exponent_bias,
+        torch.tensor(minifloat.exponent_bit_width, dtype=torch.float32),
+        torch.tensor(minifloat.mantissa_bit_width, dtype=torch.float32),
+        torch.tensor(minifloat.exponent_bias, dtype=torch.float32),
         nan_values,
         inf_values,
         saturating)
@@ -42,9 +42,9 @@ def test_float_clamp(inp, fp8_clamp):
     nan_values = fp8_clamp.float_clamp_impl.nan_values
     saturating = fp8_clamp.float_clamp_impl.saturating
     max_val = max_float(
-        fp8_clamp.exponent_bit_width,
-        fp8_clamp.mantissa_bit_width,
-        fp8_clamp.exponent_bias,
+        torch.tensor(fp8_clamp.exponent_bit_width, dtype=torch.float32),
+        torch.tensor(fp8_clamp.mantissa_bit_width, dtype=torch.float32),
+        torch.tensor(fp8_clamp.exponent_bias, dtype=torch.float32),
         nan_values,
         inf_values,
         saturating)