update

Xilinx · Mar 9, 2025 · c690dd5 · c690dd5
1 parent 39cbb45
commit c690dd5
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/brevitas/proxy/test_compile.py b/tests/brevitas/proxy/test_compile.py
@@ -34,7 +34,7 @@
 
 @pytest_cases.parametrize('weight_quantizer', WEIGHT_QUANTIZERS.items())
 @given(weight=float_tensor_st(shape=(8, 16)))
-@requires_pt_ge('2.0')
+@requires_pt_ge('2.3.1')
 def test_compile_weight(weight, weight_quantizer):
     name, quant = weight_quantizer
     inp = torch.randn(8, 16)
@@ -53,7 +53,7 @@ def test_compile_weight(weight, weight_quantizer):
 
 @pytest_cases.parametrize('act_quantizer', ACT_QUANTIZERS.items())
 @given(inp=float_tensor_st(shape=(8, 16)))
-@requires_pt_ge('2.0')
+@requires_pt_ge('2.3.1')
 def test_compile_act(inp, act_quantizer):
     name, quant = act_quantizer
     if 'mx' in name: