Skip to content

GitLab

Unverified Commit 4a676905 authored by Michael Goin's avatar Michael Goin Committed by GitHub
Browse files

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

parent f31c1f90
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
tests/quantization/utils.py
View file @ 4a676905
...@@ -10,5 +10,5 @@ def is_quant_method_supported(quant_method: str) -> bool: ...@@ -10,5 +10,5 @@ def is_quant_method_supported(quant_method: str) -> bool:
capability = torch.cuda.get_device_capability() capability = torch.cuda.get_device_capability()
capability = capability[0] * 10 + capability[1] capability = capability[0] * 10 + capability[1]
return (capability < return (capability >=
QUANTIZATION_METHODS[quant_method].get_min_capability()) QUANTIZATION_METHODS[quant_method].get_min_capability())
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment