mirror of
https://github.com/saymrwulf/onnxruntime.git
synced 2026-05-14 20:48:00 +00:00
ONNX Runtime training cpu package name for ADO (#17109)
This commit is contained in:
parent
be2200c00b
commit
289600b47d
2 changed files with 6 additions and 0 deletions
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 \
|
||||
|
|
|
|||
Loading…
Reference in a new issue