We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 4a664bf commit 7576382Copy full SHA for 7576382
tests/quantization/utils.py
@@ -10,5 +10,5 @@ def is_quant_method_supported(quant_method: str) -> bool:
10
11
capability = torch.cuda.get_device_capability()
12
capability = capability[0] * 10 + capability[1]
13
- return (capability <
+ return (capability >=
14
QUANTIZATION_METHODS[quant_method].get_min_capability())
0 commit comments