ONNX Runtime training cpu package name for ADO (#17109)

This commit is contained in:
Baiju Meswani 2023-08-14 11:32:35 -07:00 committed by GitHub
parent be2200c00b
commit 289600b47d
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 6 additions and 0 deletions

View file

@ -582,6 +582,10 @@ if enable_training or enable_training_apis:
else:
# cpu version for documentation
local_version = "+cpu"
else:
if not (cuda_version or rocm_version):
# Training CPU package for ADO feeds is called onnxruntime-training-cpu
package_name = "onnxruntime-training-cpu"
if package_name == "onnxruntime-tvm":
packages += ["onnxruntime.providers.tvm"]

View file

@ -76,6 +76,8 @@ stages:
--volume $HOME/.onnx:/home/onnxruntimedev/.onnx \
-e NIGHTLY_BUILD \
-e BUILD_BUILDNUMBER \
-e ORT_DISABLE_PYTHON_PACKAGE_LOCAL_VERSION \
-e DEFAULT_TRAINING_PACKAGE_DEVICE \
onnxruntimetrainingcpubuild \
$(PythonManylinuxDir)/bin/python3 /onnxruntime_src/tools/ci_build/build.py \
--build_dir /build --cmake_generator Ninja \