From 672016021159bc8cfb92bc7bbbee859f71e18fd0 Mon Sep 17 00:00:00 2001 From: "Wang, Mengni" Date: Wed, 13 Nov 2024 09:03:54 +0800 Subject: [PATCH 1/2] Fix typo for auto_detect_ep function Signed-off-by: Wang, Mengni --- onnx_neural_compressor/utility.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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: From 15b02e9923910df6edbab6cb1184fcf0c91bc06c Mon Sep 17 00:00:00 2001 From: "Wang, Mengni" Date: Mon, 18 Nov 2024 13:45:13 +0800 Subject: [PATCH 2/2] Update requirements.txt Signed-off-by: Wang, Mengni --- test/requirements.txt | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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