onnxruntime/tools/ci_build/github/azure-pipelines/py-package-build-pipeline.yml
leqiao-1 8d06e5a9df
Add openvino base image option (#10581)
* add selectable python package build pipeline

* update tensorrt version

* update tensorrt version

* Update Dockerfile.ubuntu_openvino

* Update install_ubuntu.sh

* add parameters for openvino base image

* fix syntax error
2022-02-17 17:10:01 +08:00

54 lines
1.6 KiB
YAML

parameters:
- name: enable_linux_cpu
displayName: 'Whether Linux CPU package is built.'
type: boolean
default: true
- name: enable_linux_gpu
displayName: 'Whether Linux GPU package is built.'
type: boolean
default: true
- name: enable_windows_cpu
displayName: 'Whether Windows CPU package is built.'
type: boolean
default: true
- name: enable_windows_gpu
displayName: 'Whether Windows GPU package is built.'
type: boolean
default: true
- name: python_version
displayName: 'Python version used for build'
type: object
default: ['3.7', '3.8', '3.9']
- name: cpu_build_py_parameters
displayName: 'Extra parameters to pass to build.py for CPU package.'
type: string
default: '--use_dnnl --use_openvino CPU_FP32'
- name: gpu_build_py_parameters
displayName: 'Extra parameters to pass to build.py for GPU package.'
type: string
default: ' '
- name: ubuntu_version_linux_cpu
displayName: 'Ubuntu Version for Linux CPU package.'
type: string
default: '20.04'
trigger: none
stages:
- template: templates/py-packaging-selectable-stage.yml
parameters:
enable_linux_gpu: ${{ parameters.enable_linux_gpu }}
enable_linux_cpu: ${{ parameters.enable_linux_cpu }}
enable_windows_cpu: ${{ parameters.enable_windows_cpu }}
enable_windows_gpu: ${{ parameters.enable_windows_gpu }}
python_version: ${{ parameters.python_version }}
cpu_build_py_parameters: ${{ parameters.cpu_build_py_parameters }}
gpu_build_py_parameters: ${{ parameters.gpu_build_py_parameters }}
ubuntu_version_linux_cpu: ${{ parameters.ubuntu_version_linux_cpu }}