From f34ebbc8ffdb077151f6a4d9b743f34dbda262a2 Mon Sep 17 00:00:00 2001 From: mindest <30493312+mindest@users.noreply.github.com> Date: Sat, 3 Dec 2022 18:50:47 +0800 Subject: [PATCH] fix a wrong assert condition in benchmark_helper (#13821) ### Description fix a wrong assert condition in benchmark_helper.py (introduced in #13455) --- onnxruntime/python/tools/transformers/benchmark_helper.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/onnxruntime/python/tools/transformers/benchmark_helper.py b/onnxruntime/python/tools/transformers/benchmark_helper.py index 2d49418f8d..8d22e0862a 100644 --- a/onnxruntime/python/tools/transformers/benchmark_helper.py +++ b/onnxruntime/python/tools/transformers/benchmark_helper.py @@ -164,7 +164,7 @@ def prepare_environment(cache_dir, output_dir, use_gpu, provider=None): ), "Please install onnxruntime-directml package to test GPU inference." else: - assert set(onnxruntime.get_available_providers()).isdisjoint( + assert not set(onnxruntime.get_available_providers()).isdisjoint( ["CUDAExecutionProvider", "ROCMExecutionProvider", "MIGraphXExecutionProvider"] ), "Please install onnxruntime-gpu package, or install ROCm support, to test GPU inference."