From 463dea27229ca60faeb782d6ec27b650a981ad0b Mon Sep 17 00:00:00 2001 From: Olatunji Ruwase Date: Wed, 20 Sep 2023 15:26:50 -0400 Subject: [PATCH] Fix min torch version (#4375) --- tests/unit/inference/quantization/test_int4_quantization.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/unit/inference/quantization/test_int4_quantization.py b/tests/unit/inference/quantization/test_int4_quantization.py index a30d34d4e249..56a5a7d48382 100644 --- a/tests/unit/inference/quantization/test_int4_quantization.py +++ b/tests/unit/inference/quantization/test_int4_quantization.py @@ -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)