disable vitis test for ort 1.6.0

examples/test/test_resnet_vitis_ai_ptq_cpu.py

@@ -6,6 +6,8 @@
 from pathlib import Path
 
 import pytest
+from onnxruntime import __version__ as OrtVersion
+from packaging import version
 from utils import check_output, patch_config
 
 from olive.common.utils import retry_func, run_subprocess
@@ -30,6 +32,9 @@ def setup():
 @pytest.mark.parametrize("execution_order", ["pass-by-pass"])
 @pytest.mark.parametrize("system", ["local_system", "aml_system"])
 @pytest.mark.parametrize("olive_json", ["resnet_vitis_ai_ptq_cpu.json"])
+@pytest.mark.skipif(
+    OrtVersion >= version.parse("1.6.0"), reason="VitisAIQuantization is not supported in ORT 1.6.0 with TensorsData"
+)
 def test_resnet(search_algorithm, execution_order, system, olive_json):
     # TODO: add gpu e2e test
     from olive.workflows import run as olive_run

olive/passes/onnx/quantization.py

@@ -451,7 +451,9 @@ def _quant_preprocess(self, model: ONNXModel, output_model_path: Union[str, Path
             # there are some problems with the path to where the external data is saved
             # need to find out why before enabling this
 
-            logger.warning(f"Failed to run quantization preprocessing with error of {e}. Using original model.")
+            logger.warning(
+                f"Failed to run quantization preprocessing with error of {e}. Using original model.", exc_info=True
+            )
             # save original model to output path
             onnx_model = onnx.load(model.model_path)
             model_proto_to_file(

test/unit_test/passes/vitis_ai/test_vitis_ai_quantization.py

@@ -7,7 +7,10 @@
 from test.unit_test.utils import get_onnx_model
 
 import numpy as np
+import pytest
+from onnxruntime import __version__ as OrtVersion
 from onnxruntime.quantization.calibrate import CalibrationDataReader
+from packaging import version
 
 from olive.passes.olive_pass import create_pass_from_dict
 from olive.passes.onnx.vitis_ai_quantization import VitisAIQuantization
@@ -34,6 +37,9 @@ def dummy_calibration_reader(data_dir=None, batch_size=1, *args, **kwargs):
     return RandomDataReader()
 
 
+@pytest.mark.skipif(
+    OrtVersion >= version.parse("1.6.0"), reason="VitisAIQuantization is not supported in ORT 1.6.0 with TensorsData"
+)
 def test_vitis_ai_quantization_pass(tmp_path):
     # setup
     input_model = get_onnx_model()