diff --git a/onnx_neural_compressor/utility.py b/onnx_neural_compressor/utility.py index 33ff2702e..2a210ded4 100644 --- a/onnx_neural_compressor/utility.py +++ b/onnx_neural_compressor/utility.py @@ -306,7 +306,7 @@ def auto_detect_ep(): if "DnnlExecutionProvider" in eps: return "DnnlExecutionProvider" elif "DmlExecutionProvider" in eps: - return "DnnlExecutionProvider" + return "DmlExecutionProvider" elif "CUDAExecutionProvider" in eps: return "CUDAExecutionProvider" else: diff --git a/test/requirements.txt b/test/requirements.txt index 1a86491d2..e61a8cbaa 100644 --- a/test/requirements.txt +++ b/test/requirements.txt @@ -1,3 +1,3 @@ optimum pytest -torch +torch < 2.5.0