We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent a662599 commit 6717ddcCopy full SHA for 6717ddc
tests/py/dynamo/models/test_models_export.py
@@ -1,4 +1,5 @@
1
# type: ignore
2
+import importlib
3
import platform
4
import unittest
5
from importlib import metadata
@@ -250,8 +251,10 @@ def calibrate_loop(model):
250
251
252
253
@unittest.skipIf(
- modelopt.__version__ < "0.16.1",
254
- "Int8 quantization is supported in modelopt since 0.16.1 or later",
+ not importlib.util.find_spec("modelopt")
255
+ or platform.system() != "Linux"
256
+ or Version(metadata.version("nvidia-modelopt")) < Version("0.17.0"),
257
+ "Int8 quantization is supported in modelopt since 0.17.0 or later for Linux",
258
)
259
@pytest.mark.unit
260
def test_base_int8(ir):
0 commit comments