saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-05-22 22:01:08 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix a wrong assert condition in benchmark_helper (#13821)

Browse source 
### Description
fix a wrong assert condition in benchmark_helper.py (introduced in
#13455)
This commit is contained in:
mindest 2022-12-03 18:50:47 +08:00 committed by GitHub
parent 335b62bde6
commit f34ebbc8ff
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
onnxruntime/python/tools/transformers/benchmark_helper.py
View file

@ -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."