()
| 26 | |
| 27 | |
| 28 | def test_bitlinear_quantization(): |
| 29 | bitlinear = BitLinear(in_features=512, out_features=256, bias=True) |
| 30 | x = torch.randn(1, 512) |
| 31 | out = bitlinear(x) |
| 32 | assert torch.all(out <= bitlinear.beta.unsqueeze(0).expand_as(out)) |
| 33 | assert torch.all(out >= -bitlinear.beta.unsqueeze(0).expand_as(out)) |