[CI][BugFix] Flip is_quant_method_supported condition (#5577)

4a676905 · Michael Goin · GitHub · f31c1f90 · 4a676905
Unverified Commit 4a676905 authored Jun 16, 2024 by Michael Goin Committed by GitHub Jun 16, 2024
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

tests/quantization/utils.py tests/quantization/utils.py +1 -1

No files found.
--- 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())