From f98dfbdf83421ace44d5b590726f248f6fab0d1a Mon Sep 17 00:00:00 2001 From: sayakpaul Date: Sun, 15 Jun 2025 15:07:56 +0530 Subject: [PATCH] annotate compilation test classes with bnb --- tests/quantization/bnb/test_4bit.py | 1 + tests/quantization/bnb/test_mixed_int8.py | 1 + 2 files changed, 2 insertions(+) diff --git a/tests/quantization/bnb/test_4bit.py b/tests/quantization/bnb/test_4bit.py index 2d8b9f698bfe..4610f41699e7 100644 --- a/tests/quantization/bnb/test_4bit.py +++ b/tests/quantization/bnb/test_4bit.py @@ -862,6 +862,7 @@ def test_fp4_double_safe(self): @require_torch_version_greater("2.7.1") +@require_bitsandbytes_version_greater("0.45.5") class Bnb4BitCompileTests(QuantCompileTests): quantization_config = PipelineQuantizationConfig( quant_backend="bitsandbytes_8bit", diff --git a/tests/quantization/bnb/test_mixed_int8.py b/tests/quantization/bnb/test_mixed_int8.py index b15a9f72a8f6..f80e1954f64e 100644 --- a/tests/quantization/bnb/test_mixed_int8.py +++ b/tests/quantization/bnb/test_mixed_int8.py @@ -827,6 +827,7 @@ def test_serialization_sharded(self): @require_torch_version_greater_equal("2.6.0") +@require_bitsandbytes_version_greater("0.45.5") class Bnb8BitCompileTests(QuantCompileTests): quantization_config = PipelineQuantizationConfig( quant_backend="bitsandbytes_8bit",