Add tests

Xilinx · Dec 30, 2024 · e330cab · e330cab
1 parent e4d25a0
commit e330cab
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/tests/brevitas/proxy/test_proxy.py b/tests/brevitas/proxy/test_proxy.py
@@ -1,4 +1,5 @@
 import pytest
+import torch
 
 from brevitas.nn import QuantLinear
 from brevitas.nn.quant_activation import QuantReLU
@@ -80,3 +81,11 @@ def test_dynamic_act_proxy(self):
 
         model.act_quant.disable_quant = True
         assert model.act_quant.bit_width() is None
+
+    def test_delay_act_proxy(self):
+        model = QuantReLU(quant_delay_steps=1)
+        inp = torch.randn(1, 5)
+        o = model(inp)
+        assert torch.allclose(inp, o)
+        o = model(inp)
+        assert not torch.allclose(inp, o)