Skip to content

Commit 6717ddc

Browse files
Lluo/fix merge issue (#3162)
1 parent a662599 commit 6717ddc

File tree

1 file changed

+5
-2
lines changed

1 file changed

+5
-2
lines changed

tests/py/dynamo/models/test_models_export.py

+5-2
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,5 @@
11
# type: ignore
2+
import importlib
23
import platform
34
import unittest
45
from importlib import metadata
@@ -250,8 +251,10 @@ def calibrate_loop(model):
250251

251252

252253
@unittest.skipIf(
253-
modelopt.__version__ < "0.16.1",
254-
"Int8 quantization is supported in modelopt since 0.16.1 or later",
254+
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",
255258
)
256259
@pytest.mark.unit
257260
def test_base_int8(ir):

0 commit comments

Comments
 (0)