From db9738713df6ed67446666a92a519fe3b59933af Mon Sep 17 00:00:00 2001 From: Fabian Grob Date: Thu, 1 Feb 2024 04:40:30 -0800 Subject: [PATCH] Fix (tests): add decorator to resolve jit/hypothesis clash --- tests/brevitas/core/test_float_quant.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/tests/brevitas/core/test_float_quant.py b/tests/brevitas/core/test_float_quant.py index 4fe505bd2..890102561 100644 --- a/tests/brevitas/core/test_float_quant.py +++ b/tests/brevitas/core/test_float_quant.py @@ -46,7 +46,6 @@ def test_minifloat(minifloat_format): @given(inp=float_tensor_random_shape_st(), minifloat_format=random_minifloat_format()) -@jit_disabled_for_mock() def test_float_to_quant_float(inp, minifloat_format): bit_width, exponent_bit_width, mantissa_bit_width, signed = minifloat_format if exponent_bit_width == 0 or mantissa_bit_width == 0: @@ -70,6 +69,7 @@ def test_float_to_quant_float(inp, minifloat_format): @given(inp=float_tensor_random_shape_st(), minifloat_format=random_minifloat_format()) +@jit_disabled_for_mock() def test_scaling_impls_called_once(inp, minifloat_format): bit_width, exponent_bit_width, mantissa_bit_width, signed = minifloat_format scaling_impl = mock.Mock(side_effect=lambda x: 1.) @@ -101,6 +101,7 @@ def test_scaling_impls_called_once(inp, minifloat_format): inp=float_tensor_random_shape_st(), minifloat_format=random_minifloat_format(), scale=float_st()) +@jit_disabled_for_mock() def test_inner_scale(inp, minifloat_format, scale): bit_width, exponent_bit_width, mantissa_bit_width, signed = minifloat_format # set scaling_impl to scale and float_scaling_impl to 1 to use the same scale as we are here