From 4a6769053ab2616f7f490e6ec5b8241e76ef0c2a Mon Sep 17 00:00:00 2001 From: Michael Goin Date: Sun, 16 Jun 2024 10:07:34 -0400 Subject: [PATCH] [CI][BugFix] Flip is_quant_method_supported condition (#5577) --- tests/quantization/utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/quantization/utils.py b/tests/quantization/utils.py index 0c92d565d0ddd..29085916afb4d 100644 --- a/tests/quantization/utils.py +++ b/tests/quantization/utils.py @@ -10,5 +10,5 @@ def is_quant_method_supported(quant_method: str) -> bool: capability = torch.cuda.get_device_capability() capability = capability[0] * 10 + capability[1] - return (capability < + return (capability >= QUANTIZATION_METHODS[quant_method].get_min_capability())