Skip to content

Commit 4f5a65f

Browse files
Fix test_nvfp4_onnx_export.py
Signed-off-by: Keval Morabia <28916987+kevalmorabia97@users.noreply.github.com>
1 parent dea2eeb commit 4f5a65f

2 files changed

Lines changed: 2 additions & 1 deletion

File tree

tests/_test_utils/import_helper.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -99,5 +99,5 @@ def skip_if_onnx_version_above_1_18():
9999

100100
if version.parse(installed_version) > version.parse(required_version):
101101
pytest.skip(
102-
f"{package_name} version {installed_version} is less than required {required_version}"
102+
f"{package_name} version {installed_version} is greater than required {required_version}"
103103
)

tests/gpu/torch/quantization/test_nvfp4_onnx_export.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -98,6 +98,7 @@ def forward_loop(model, run_backward=False):
9898
output_names=["output"],
9999
export_params=True,
100100
opset_version=17,
101+
dynamo=False,
101102
)
102103

103104
onnx_model = NVFP4QuantExporter.process_model(onnx.load(onnx_path))

0 commit comments

Comments
 (0)