Skip to content

Commit

Permalink
Fix min torch version (#4375)
Browse files Browse the repository at this point in the history
  • Loading branch information
tjruwase authored Sep 20, 2023
1 parent 1795772 commit 463dea2
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions tests/unit/inference/quantization/test_int4_quantization.py
Original file line number Diff line number Diff line change
Expand Up @@ -20,8 +20,8 @@

device = get_accelerator().device_name() if get_accelerator().is_available() else 'cpu'

if not required_torch_version(min_version=1.10):
pytest.skip("torch.Tensor.bitwise_left_shift in INT4 quantizer needs torch 1.10 or above.",
if not required_torch_version(min_version=1.11):
pytest.skip("torch.Tensor.bitwise_left_shift in INT4 quantizer needs torch 1.11 or above.",
allow_module_level=True)


Expand Down

0 comments on commit 463dea2

Please sign in to comment.