ONNX Runtime training cpu package name for ADO (#17109)
This commit is contained in:
Родитель
be2200c00b
Коммит
289600b47d
4
setup.py
4
setup.py
|
@ -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 \
|
||||
|
|
Загрузка…
Ссылка в новой задаче