Skip to content

Commit eb3c973

Browse files
snnnDu Li
authored andcommitted
Fix Python Linux GPU package name (#5943)
Fix Python Linux GPU package name. I accidentally added "noopenmp" to it. (cherry picked from commit 5fdd9f0)
1 parent 8a28843 commit eb3c973

File tree

4 files changed

+4
-7
lines changed

4 files changed

+4
-7
lines changed

docs/ONNX_Runtime_for_Mobile_Platforms.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -141,7 +141,7 @@ Currently the NNAPI execution provider is the only execution provider that has s
141141
- this is located in `build/Windows/<config>/<config>/dist/<package name>.whl` on Windows, or `build/Linux/<config>/dist/<package name>.whl` on Linux.
142142
- `<config>` is the value from the `--config` parameter from the build command (e.g. Release)
143143
- the package name will differ based on your platform, python version, and build parameters
144-
- e.g. `pip install -U build\Windows\Release\Release\dist\onnxruntime_noopenmp-1.5.2-cp37-cp37m-win_amd64.whl`
144+
- e.g. `pip install -U build\Windows\Release\Release\dist\onnxruntime-1.5.2-cp37-cp37m-win_amd64.whl`
145145
- Create an ORT format model by running `tools\python\convert_onnx_models_to_ort.py` as per the above instructions, with the addition of the `--use_nnapi` parameter
146146
- the python package from your 'full' build with NNAPI enabled must be installed for `--use_nnapi` to be a valid option
147147
- this will preserve all the nodes that can be assigned to NNAPI, as well as setup the ability to fallback to CPU execution if NNAPI is not available at runtime, or if NNAPI can not run all the nodes due to device limitations.

tools/ci_build/build.py

Lines changed: 1 addition & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -2010,10 +2010,6 @@ def main():
20102010
if args.build:
20112011
if args.build_wheel:
20122012
nightly_build = bool(os.getenv('NIGHTLY_BUILD') == '1')
2013-
wheel_name_suffix = args.wheel_name_suffix
2014-
if not args.use_openmp and wheel_name_suffix is None:
2015-
wheel_name_suffix = 'noopenmp'
2016-
20172013
build_python_wheel(
20182014
source_dir,
20192015
build_dir,
@@ -2027,7 +2023,7 @@ def main():
20272023
args.use_acl,
20282024
args.use_armnn,
20292025
args.use_dml,
2030-
wheel_name_suffix,
2026+
args.wheel_name_suffix,
20312027
args.enable_training,
20322028
nightly_build=nightly_build,
20332029
featurizers_build=args.use_featurizers,

tools/ci_build/github/azure-pipelines/templates/py-packaging-stage.yml

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -190,6 +190,7 @@ stages:
190190
--enable_lto \
191191
--build_wheel \
192192
--enable_onnx_tests \
193+
--wheel_name_suffix=noopenmp \
193194
${{ parameters.build_py_parameters }} \
194195
--cmake_extra_defines PYTHON_INCLUDE_DIR=$(python.manylinux.include.dir) PYTHON_LIBRARY=/usr/lib64/librt.so
195196
workingDirectory: $(Build.SourcesDirectory)

tools/ci_build/github/linux/ort_minimal/build_full_ort_and_create_ort_files.sh

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -38,7 +38,7 @@ python3 /onnxruntime_src/tools/python/convert_onnx_models_to_ort.py \
3838
find /home/onnxruntimedev/.test_data/ort_minimal_e2e_test_data -type f -name "*.onnx" -delete
3939

4040
# Uninstall the ORT python wheel
41-
python3 -m pip uninstall -y onnxruntime_noopenmp
41+
python3 -m pip uninstall -y onnxruntime
4242

4343
# Clear the build
4444
rm -rf /build/Debug

0 commit comments

Comments
 (0)