Skip to content

Commit 4c20624

Browse files
authored
[tests] annotate compilation test classes with bnb (#11715)
annotate compilation test classes with bnb
1 parent 0454fbb commit 4c20624

File tree

2 files changed

+2
-0
lines changed

2 files changed

+2
-0
lines changed

tests/quantization/bnb/test_4bit.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -872,6 +872,7 @@ def test_fp4_double_safe(self):
872872

873873

874874
@require_torch_version_greater("2.7.1")
875+
@require_bitsandbytes_version_greater("0.45.5")
875876
class Bnb4BitCompileTests(QuantCompileTests):
876877
@property
877878
def quantization_config(self):

tests/quantization/bnb/test_mixed_int8.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -837,6 +837,7 @@ def test_serialization_sharded(self):
837837

838838

839839
@require_torch_version_greater_equal("2.6.0")
840+
@require_bitsandbytes_version_greater("0.45.5")
840841
class Bnb8BitCompileTests(QuantCompileTests):
841842
@property
842843
def quantization_config(self):

0 commit comments

Comments
 (0)