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

This commit is contained in:
Baiju Meswani 2023-08-14 11:32:35 -07:00 коммит произвёл GitHub
Родитель be2200c00b
Коммит 289600b47d
Не найден ключ, соответствующий данной подписи
Идентификатор ключа GPG: 4AEE18F83AFDEB23
2 изменённых файлов: 6 добавлений и 0 удалений

Просмотреть файл

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

Просмотреть файл

@ -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 \