mirror of
https://github.com/saymrwulf/onnxruntime.git
synced 2026-06-02 23:39:58 +00:00
1c5b15c2b8
* remove memory copy between CUDA and TRT * add info to RegisterExecutionProvider input * use new IDeviceAllocator for trt allocator * remove SetDefaultInputsMemoryType from TRT EP * remove onnx-tensorrt 5.0 * add submodule onnx-tensorrt branch 5.1 * remove redundancy * Update transformer_memcpy.cc * Update tensorrt_execution_provider.cc * switch to TensorRT 5.1.5.0 * update python binding * disable failed test case on TensorRT * Update activation_op_test.cc * upgrade to TensorRT container 19.06 * update according to feedback * add comments * remove tensorrt allocator and use cuda(gpu) allocator * update onnx-tensorrt submodule * change ci build cuda directory name
168 lines
6.7 KiB
YAML
168 lines
6.7 KiB
YAML
jobs:
|
|
|
|
- job: Linux_C_API_Packaging_CPU_x64
|
|
pool: Linux-CPU
|
|
steps:
|
|
- template: templates/linux-set-variables-and-download.yml
|
|
- template: templates/set-version-number-variables-step.yml
|
|
|
|
- script: 'tools/ci_build/github/linux/run_dockerbuild.sh -o ubuntu16.04 -d cpu -r $(Build.BinariesDirectory)'
|
|
displayName: 'Build and Test Linux on Docker'
|
|
- template: templates/c-api-artifacts-package-and-publish-steps-posix.yml
|
|
parameters:
|
|
buildConfig: 'Release'
|
|
artifactName: 'onnxruntime-linux-x64-$(OnnxRuntimeVersion)'
|
|
libraryName: 'libonnxruntime.so.$(OnnxRuntimeVersion)'
|
|
commitId: $(OnnxRuntimeGitCommitHash)
|
|
- template: templates/clean-agent-build-directory-step.yml
|
|
|
|
- job: Linux_C_API_Packaging_CPU_x86
|
|
pool: Linux-CPU
|
|
steps:
|
|
- template: templates/linux-set-variables-and-download.yml
|
|
- template: templates/set-version-number-variables-step.yml
|
|
|
|
- script: 'tools/ci_build/github/linux/run_dockerbuild.sh -o ubuntu16.04 -d cpu -r $(Build.BinariesDirectory) -a x86 -x "--x86"'
|
|
displayName: 'Build and Test Linux on Docker'
|
|
- template: templates/c-api-artifacts-package-and-publish-steps-posix.yml
|
|
parameters:
|
|
buildConfig: 'Release'
|
|
artifactName: 'onnxruntime-linux-x86-$(OnnxRuntimeVersion)'
|
|
libraryName: 'libonnxruntime.so.$(OnnxRuntimeVersion)'
|
|
commitId: $(OnnxRuntimeGitCommitHash)
|
|
- template: templates/clean-agent-build-directory-step.yml
|
|
|
|
- job: Linux_C_API_Packaging_GPU_x64
|
|
pool: Linux-GPU
|
|
steps:
|
|
- template: templates/linux-set-variables-and-download.yml
|
|
- template: templates/set-version-number-variables-step.yml
|
|
|
|
- script: 'tools/ci_build/github/linux/run_dockerbuild.sh -o ubuntu16.04 -d gpu -r $(Build.BinariesDirectory)'
|
|
displayName: 'Build and Test Linux on Docker'
|
|
- template: templates/c-api-artifacts-package-and-publish-steps-posix.yml
|
|
parameters:
|
|
buildConfig: 'Release'
|
|
artifactName: 'onnxruntime-linux-x64-gpu-$(OnnxRuntimeVersion)'
|
|
libraryName: 'libonnxruntime.so.$(OnnxRuntimeVersion)'
|
|
commitId: $(OnnxRuntimeGitCommitHash)
|
|
- template: templates/clean-agent-build-directory-step.yml
|
|
|
|
|
|
- job: MacOS_C_API_Packaging_CPU_x64
|
|
pool:
|
|
vmImage: 'macOS-10.13'
|
|
steps:
|
|
|
|
- template: templates/mac-set-variables-and-download.yml
|
|
- template: templates/set-version-number-variables-step.yml
|
|
|
|
- script: |
|
|
sudo xcode-select --switch /Applications/Xcode_10.app/Contents/Developer
|
|
python3 $(Build.SourcesDirectory)/tools/ci_build/build.py --use_openmp --build_dir $(Build.BinariesDirectory) --skip_submodule_sync --parallel --build_shared_lib --config RelWithDebInfo --enable_onnx_tests
|
|
displayName: 'Build and Test MacOS'
|
|
- template: templates/c-api-artifacts-package-and-publish-steps-posix.yml
|
|
parameters:
|
|
buildConfig: 'RelWithDebInfo'
|
|
artifactName: 'onnxruntime-osx-x64-$(OnnxRuntimeVersion)'
|
|
libraryName: 'libonnxruntime.$(OnnxRuntimeVersion).dylib'
|
|
commitId: $(OnnxRuntimeGitCommitHash)
|
|
|
|
- template: templates/clean-agent-build-directory-step.yml
|
|
|
|
|
|
- job: Windows_Packaging_CPU_x64
|
|
pool: 'Win-CPU'
|
|
variables:
|
|
buildDirectory: '$(Build.BinariesDirectory)'
|
|
buildConfig: 'RelWithDebInfo'
|
|
buildArch: 'x64'
|
|
|
|
steps:
|
|
- template: templates/set-test-data-variables-step.yml
|
|
- template: templates/set-version-number-variables-step.yml
|
|
|
|
- template: templates/windows-build-tools-setup-steps.yml
|
|
parameters:
|
|
EnvSetupScript: 'setup_env.bat'
|
|
buildArch: 'amd64' # amd64 is needed for vcvars target arch
|
|
setVcvars: false
|
|
|
|
- template: templates/windows-build-and-test-steps.yml
|
|
parameters:
|
|
buildAdditionalParams: ' --use_openmp '
|
|
buildArch: $(buildArch)
|
|
msbuildPlatform: $(buildArch)
|
|
buildConfig: $(buildConfig)
|
|
|
|
- template: templates/c-api-artifacts-package-and-publish-steps-windows.yml
|
|
parameters:
|
|
buildConfig: $(buildConfig)
|
|
artifactName: 'onnxruntime-win-$(buildArch)-$(OnnxRuntimeVersion)'
|
|
commitId: $(OnnxRuntimeGitCommitHash)
|
|
- template: templates/clean-agent-build-directory-step.yml
|
|
|
|
|
|
- job: Windows_Packaging_CPU_x86
|
|
pool: 'Win-CPU'
|
|
variables:
|
|
buildDirectory: '$(Build.BinariesDirectory)'
|
|
buildConfig: 'RelWithDebInfo'
|
|
buildArch: 'x86'
|
|
|
|
steps:
|
|
- template: templates/set-test-data-variables-step.yml
|
|
- template: templates/set-version-number-variables-step.yml
|
|
|
|
- template: templates/windows-build-tools-setup-steps.yml
|
|
parameters:
|
|
EnvSetupScript: 'setup_env.bat'
|
|
buildArch: $(buildArch)
|
|
setVcVars: false
|
|
|
|
- template: templates/windows-build-and-test-steps.yml
|
|
parameters:
|
|
buildAdditionalParams: ' --use_openmp --x86 '
|
|
buildArch: $(buildArch)
|
|
msbuildPlatform: 'Win32'
|
|
buildConfig: $(buildConfig)
|
|
|
|
- template: templates/c-api-artifacts-package-and-publish-steps-windows.yml
|
|
parameters:
|
|
buildConfig: $(buildConfig)
|
|
artifactName: 'onnxruntime-win-$(buildArch)-$(OnnxRuntimeVersion)'
|
|
commitId: $(OnnxRuntimeGitCommitHash)
|
|
- template: templates/clean-agent-build-directory-step.yml
|
|
|
|
|
|
- job: Windows_Packaging_GPU_x64
|
|
pool: 'Win-GPU'
|
|
variables:
|
|
buildDirectory: '$(Build.BinariesDirectory)'
|
|
buildConfig: 'RelWithDebInfo'
|
|
buildArch: 'x64'
|
|
|
|
steps:
|
|
- template: templates/set-test-data-variables-step.yml
|
|
- template: templates/set-version-number-variables-step.yml
|
|
|
|
- template: templates/windows-build-tools-setup-steps.yml
|
|
parameters:
|
|
EnvSetupScript: 'setup_env.bat'
|
|
buildArch: 'amd64' # amd64 is needed for vcvars target arch
|
|
setVcvars: true
|
|
|
|
- task: CmdLine@2
|
|
displayName: 'Build and Test OnnxRuntime'
|
|
inputs:
|
|
script: |
|
|
$(Build.BinariesDirectory)\packages\python\python.exe $(Build.SourcesDirectory)\tools\ci_build\build.py --config $(buildConfig) --build_dir $(Build.BinariesDirectory) --skip_submodule_sync --build_shared_lib --cmake_path $(Build.BinariesDirectory)\cmake\bin\cmake.exe --ctest_path $(Build.BinariesDirectory)\cmake\bin\ctest.exe --enable_onnx_tests --use_openmp --msvc_toolset=14.11 --use_cuda --cuda_home="C:\local\cuda_10.0.130_win10_trt515dll" --cudnn_home="C:\local\cudnn-10.0-windows10-x64-v7.3.1.20\cuda"
|
|
workingDirectory: '$(Build.BinariesDirectory)'
|
|
|
|
- template: templates/c-api-artifacts-package-and-publish-steps-windows.yml
|
|
parameters:
|
|
buildConfig: $(buildConfig)
|
|
artifactName: 'onnxruntime-win-$(buildArch)-gpu-$(OnnxRuntimeVersion)'
|
|
commitId: $(OnnxRuntimeGitCommitHash)
|
|
|
|
- template: templates/clean-agent-build-directory-step.yml
|