onnxruntime-extensions/.pipelines/ci.yml

534 строки
18 KiB
YAML

# IMPORTANT: READ ME BEFORE MAKING ANY CHANGES
#
# Supported matrix should follow these conventions strictly and
# for any exceptions to the rules, please document them in comments.
#
# * When adding new configurations to the matrix, update all platforms simultaneously.
# * Onnxruntime version should be one release ahead of torch, torchvision, and torchaudio.
#
# The rationale is that the onnxruntime release process would have verified the
# last available version of its dependencies (including the three listed above)
# and so it's likely to be _better_ supported relative to any older versions.
jobs:
#######
# Linux
#######
- job: Linux
pool:
vmImage: 'ubuntu-latest'
strategy:
matrix:
py311-1150:
python.version: '3.11'
torch.version: 'torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu'
ort.version: '1.15.0'
py310-1141:
python.version: '3.10'
torch.version: 'torch==1.13.1+cpu torchvision==0.14.1+cpu torchaudio==0.13.1 --index-url https://download.pytorch.org/whl/cpu'
ort.version: '1.14.1'
py39-1131:
python.version: '3.9'
torch.version: 'torch==1.12.1+cpu torchvision==0.13.1+cpu torchaudio==0.12.1+cpu -f https://download.pytorch.org/whl/cpu/torch_stable.html'
ort.version: '1.13.1'
py38-1121:
python.version: '3.8'
torch.version: 'torch==1.11.0+cpu torchvision==0.12.0+cpu torchaudio==0.11.0+cpu -f https://download.pytorch.org/whl/cpu/torch_stable.html'
ort.version: '1.12.1'
steps:
- task: DownloadGitHubRelease@0
inputs:
connection: 'microsoft'
userRepository: 'microsoft/onnxruntime'
defaultVersionType: 'specificTag'
version: 'v$(ort.version)'
itemPattern: '*-linux-x64-$(ort.version)*'
downloadPath: '$(Build.SourcesDirectory)'
displayName: Download the ONNXRuntime prebuilt package.
- task: ExtractFiles@1
inputs:
archiveFilePatterns: '**/*.tgz'
destinationFolder: '$(Build.SourcesDirectory)'
cleanDestinationFolder: false
overwriteExistingFiles: true
displayName: Unpack ONNXRuntime package.
- script: |
wget https://github.com/Tencent/rapidjson/archive/refs/tags/v1.1.0.tar.gz
tar zxvf v1.1.0.tar.gz
cd rapidjson-1.1.0
mkdir build
cd build
cmake ..
sudo cmake --install .
cd ../..
git clone https://github.com/triton-inference-server/client.git --branch r23.05 ~/client
sudo ln -s ~/client/src/c++/library/libhttpclient.ldscript /lib/libhttpclient.ldscript
displayName: install deps for azure invokers
- script: |
CPU_NUMBER=2 sh ./build.sh -DOCOS_ENABLE_CTEST=ON -DONNXRUNTIME_PKG_DIR=$(Build.SourcesDirectory)/onnxruntime-linux-x64-$(ort.version) -DOCOS_ENABLE_AZURE=ON
displayName: build the customop library with onnxruntime
- script: |
cd out/Linux/RelWithDebInfo
ctest -C RelWithDebInfo --output-on-failure
displayName: Run C++ native tests
- task: UsePythonVersion@0
inputs:
versionSpec: '$(python.version)'
addToPath: true
- script: |
python -m pip install --upgrade pip
python -m pip install --upgrade setuptools
python -m pip install onnxruntime==$(ort.version)
python -m pip install -r requirements.txt
displayName: Install requirements.txt
- script: |
CPU_NUMBER=2 python -m pip install -e .
displayName: Build the library and tests
- script: python -m pip install $(torch.version)
displayName: Install pytorch
- script: |
python -m pip install -r requirements-dev.txt
displayName: Install requirements-dev.txt
- script: python -m pytest test --verbose
displayName: Run python test
###########
# macOS C++
###########
- job: MacOSX
pool:
vmImage: 'macOS-13'
strategy:
matrix:
ort-1150:
ort.version: '1.15.0'
ort.dirname: 'onnxruntime-osx-x86_64-$(ort.version)'
ort-1141:
ort.version: '1.14.1'
ort.dirname: 'onnxruntime-osx-x86_64-$(ort.version)'
ort-1131:
ort.version: '1.13.1'
ort.dirname: 'onnxruntime-osx-x86_64-$(ort.version)'
ort-1121:
ort.version: '1.12.1'
ort.dirname: 'onnxruntime-osx-x86_64-$(ort.version)'
steps:
- template: templates/use-xcode-version.yml
# needed for onnxruntime
- script: brew install libomp
displayName: 'Install omp'
- task: DownloadGitHubRelease@0
inputs:
connection: 'microsoft'
userRepository: 'microsoft/onnxruntime'
defaultVersionType: 'specificTag'
version: 'v$(ort.version)'
itemPattern: '$(ort.dirname)*'
downloadPath: '$(Build.SourcesDirectory)'
displayName: Download the ONNXRuntime prebuilt package.
- task: ExtractFiles@1
inputs:
archiveFilePatterns: '**/*.tgz'
destinationFolder: '$(Build.SourcesDirectory)'
cleanDestinationFolder: false
overwriteExistingFiles: true
displayName: Unpack ONNXRuntime package.
- script: |
sh ./build.sh -DOCOS_ENABLE_CTEST=ON -DONNXRUNTIME_PKG_DIR=$(Build.SourcesDirectory)/$(ort.dirname)
displayName: build the customop library with onnxruntime
- script: |
cd out/Darwin/RelWithDebInfo
ctest -C RelWithDebInfo --output-on-failure
displayName: Run C++ native tests
#############
# macOS Python
#############
- job: MacOSPython
pool:
vmImage: 'macOS-13'
strategy:
matrix:
py311-1150:
python.version: '3.11'
torch.version: 'torch torchvision torchaudio'
ort.version: '1.15.0'
py310-1141:
python.version: '3.10'
torch.version: 'torch==1.13.1 torchvision==0.14.1 torchaudio==0.13.1 --index-url https://download.pytorch.org/whl/cpu'
ort.version: '1.14.1'
py39-1131:
python.version: '3.9'
torch.version: 'torch==1.12.1 torchvision==0.13.1 torchaudio==0.12.1 -f https://download.pytorch.org/whl/torch_stable.html'
ort.version: '1.13.1'
py38-1120:
python.version: '3.8'
torch.version: 'torch==1.11.0 torchvision==0.12.0 torchaudio==0.11.0 -f https://download.pytorch.org/whl/torch_stable.html'
ort.version: '1.12.0'
steps:
- template: templates/use-xcode-version.yml
# needed for onnxruntime
- script: brew install libomp
displayName: 'Install omp'
- task: UsePythonVersion@0
inputs:
versionSpec: '$(python.version)'
addToPath: true
- script: |
python -m pip install --upgrade pip
python -m pip install --upgrade setuptools
python -m pip install --upgrade wheel
python -m pip install onnxruntime==$(ort.version)
python -m pip install -r requirements.txt
displayName: Install requirements.txt
- script: |
python -c "import onnxruntime;print(onnxruntime.__version__)"
displayName: Check installation
- script: |
python -m pip install -e .
displayName: Build and install the wheel
- script: python -m pip install -r requirements-dev.txt
displayName: Install requirements-dev.txt
- script: python -m pip install $(torch.version)
displayName: Install pytorch
- script: cd test && python -m pytest . --verbose
displayName: Run python test
#########
# Windows C++
#########
- job: WindowsC
pool:
vmImage: 'windows-latest'
strategy:
matrix:
ort-1150:
ort.version: '1.15.0'
ort-1141:
ort.version: '1.14.1'
ort-1131:
ort.version: '1.13.1'
ort-1121:
ort.version: '1.12.1'
steps:
- task: DownloadGitHubRelease@0
inputs:
connection: 'microsoft'
userRepository: 'microsoft/onnxruntime'
defaultVersionType: 'specificTag'
version: 'v$(ort.version)'
itemPattern: '*-win-x64-$(ort.version)*'
downloadPath: '$(Build.SourcesDirectory)'
displayName: Download the ONNXRuntime prebuilt package.
- task: ExtractFiles@1
inputs:
archiveFilePatterns: '**/*.zip'
destinationFolder: '$(Build.SourcesDirectory)'
cleanDestinationFolder: false
overwriteExistingFiles: true
displayName: Unpack ONNXRuntime package.
- script: |
@echo off
set vswherepath="%ProgramFiles(x86)%\Microsoft Visual Studio\Installer\vswhere.exe"
for /f "usebackq delims=" %%i in (`%vswherepath% -latest -property installationPath`) do (
if exist "%%i\Common7\Tools\vsdevcmd.bat" (
set vsdevcmd="%%i\Common7\Tools\vsdevcmd.bat"
)
)
@echo %vsdevcmd% will be used as the VC compiler
@echo ##vso[task.setvariable variable=vsdevcmd]%vsdevcmd%
displayName: 'locate vsdevcmd via vswhere'
- script: |
call $(vsdevcmd)
call .\build.bat -DOCOS_ENABLE_CTEST=ON -DONNXRUNTIME_PKG_DIR=.\onnxruntime-win-x64-$(ort.version)
displayName: build the customop library with onnxruntime
- script: |
cd out/Windows
ctest -C RelWithDebInfo --output-on-failure
displayName: Run C++ native tests
- job: WindowsStaticVC
pool:
vmImage: 'windows-latest'
steps:
- script: |
call .\build.bat -DOCOS_ENABLE_CTEST=ON -DCMAKE_MSVC_RUNTIME_LIBRARY="MultiThreaded$<$<CONFIG:Debug>:Debug>"
cd out/Windows
ctest -C RelWithDebInfo --output-on-failure
displayName: build and test ort-extensions with VC static runtime.
################
# Windows Python
################
- job: WindowsPython
pool:
vmImage: 'windows-latest'
strategy:
matrix:
py311-1150:
python.version: '3.11'
torch.version: 'torch torchvision torchaudio'
ort.version: '1.15.0'
py310-1141:
python.version: '3.10'
torch.version: 'torch torchvision torchaudio'
ort.version: '1.14.1'
py39-1131:
python.version: '3.9'
torch.version: 'torch==1.12.1+cpu torchvision==0.13.1+cpu torchaudio==0.12.1+cpu -f https://download.pytorch.org/whl/cpu/torch_stable.html'
ort.version: '1.13.1'
py38-1121:
python.version: '3.8'
torch.version: 'torch==1.11.0+cpu torchvision==0.12.0+cpu torchaudio==0.11.0+cpu -f https://download.pytorch.org/whl/cpu/torch_stable.html'
ort.version: '1.12.1'
steps:
- powershell: Write-Host "##vso[task.prependpath]$env:CONDA\Scripts"
displayName: Add conda to PATH
- script: conda create --yes --quiet --name pyenv -c conda-forge python=$(python.version) numpy
displayName: Create Anaconda environment
- script: |
call activate pyenv
python -m pip install --upgrade pip
python -m pip install onnxruntime==$(ort.version)
python -m pip install -r requirements.txt
python -m pip install -r requirements-dev.txt
displayName: Install requirements.txt
- script: |
call activate pyenv
python -m pip install .
displayName: Build the wheel
- script: |
call activate pyenv
python -m pip install $(torch.version)
displayName: Install pytorch
- script: |
call activate pyenv
pytest test
displayName: Run python test
#############
# WebAssembly
#############
- job: WebAssembly
pool:
vmImage: 'ubuntu-latest'
steps:
- script: |
cd $(Build.BinariesDirectory)
git clone https://github.com/emscripten-core/emsdk
emsdk/emsdk install latest
emsdk/emsdk activate latest
displayName: Setup emscripten pipeline
- script: |
bash ./build.sh \
-DCMAKE_TOOLCHAIN_FILE=$(Build.BinariesDirectory)/emsdk/upstream/emscripten/cmake/Modules/Platform/Emscripten.cmake \
-DOCOS_ENABLE_SPM_TOKENIZER=ON \
-DOCOS_BUILD_PYTHON=OFF \
-DOCOS_ENABLE_CV2=OFF \
-DOCOS_ENABLE_VISION=OFF
displayName: build the customop library with onnxruntime
# TODO add unittest for webassembly
#############
# Android
#############
- job: AndroidPackage
pool:
vmImage: 'macOS-13'
timeoutInMinutes: 120
steps:
- task: UsePythonVersion@0
inputs:
versionSpec: "3.9"
addToPath: true
architecture: "x64"
displayName: "Use Python 3.9"
- task: JavaToolInstaller@0
displayName: Use jdk 11
inputs:
versionSpec: '11'
jdkArchitectureOption: 'x64'
jdkSourceOption: 'PreInstalled'
- script: brew install coreutils ninja
displayName: Install coreutils and ninja
- template: templates/run-with-android-emulator-steps.yml
parameters:
steps:
- bash: |
set -e -x
_BUILD_CFG="x86_64 $(Build.BinariesDirectory)/android_aar" ./build.android
VERSION=$(cat ./version.txt)
AAR_PATH="$(Build.BinariesDirectory)/android_aar/aar_out/com/microsoft/onnxruntime/onnxruntime-extensions-android/${VERSION}/onnxruntime-extensions-android-${VERSION}.aar"
# Do not output ##vso[] commands with `set -x` or they may be parsed again and include a trailing quote.
set +x
echo "##vso[task.setvariable variable=ORT_EXTENSIONS_AAR_PATH]${AAR_PATH}"
displayName: Build onnxruntime-extensions AAR package
- bash: |
set -e -x
cp -r $(Build.SourcesDirectory)/java/src/test/android $(Build.BinariesDirectory)/android_test
cd $(Build.BinariesDirectory)/android_test
./gradlew connectedDebugAndroidTest --no-daemon -DortExtensionsAarLocalPath="${ORT_EXTENSIONS_AAR_PATH}"
displayName: Build and run onnxruntime-extensions Android test with Android Emulator
#############
# iOS
#############
- job: IosPackage
pool:
vmImage: 'macOS-13'
timeoutInMinutes: 120
steps:
- template: templates/use-xcode-version.yml
- task: UsePythonVersion@0
inputs:
versionSpec: '3.9'
addToPath: true
architecture: 'x64'
displayName: "Use Python 3.9"
# iOS xcframework build doesn't work with CMake 3.25.1, pin to 3.25.0
- script: |
python -m pip install cmake==3.25.0
displayName: "Install CMake 3.25.0"
- template: templates/set-package-version-variable-step.yml
parameters:
PackageVersionVariableName: ORT_EXTENSIONS_POD_VERSION
- script: |
python ./tools/ios/build_xcframework.py \
--output_dir $(Build.BinariesDirectory)/xcframework_out \
--platform_arch iphonesimulator x86_64 \
--config Release
displayName: Build xcframework for iphonesimulator x86_64
- script: |
python ./tools/ios/assemble_pod_package.py \
--staging-dir $(Build.BinariesDirectory)/pod_staging \
--xcframework-output-dir $(Build.BinariesDirectory)/xcframework_out \
--pod-version ${ORT_EXTENSIONS_POD_VERSION}
displayName: Assemble pod
- script: |
ORT_EXTENSIONS_LOCAL_POD_PATH=$(Build.BinariesDirectory)/pod_staging \
pod install
displayName: Install pods for OrtExtensionsUsage
workingDirectory: $(Build.SourcesDirectory)/test/ios/OrtExtensionsUsage
- template: templates/run-with-ios-simulator-steps.yml
parameters:
steps:
- script: |
xcrun xcodebuild \
-sdk iphonesimulator \
-configuration Debug \
-workspace $(Build.SourcesDirectory)/test/ios/OrtExtensionsUsage/OrtExtensionsUsage.xcworkspace \
-scheme OrtExtensionsUsage \
-destination "platform=iOS Simulator,id=${ORT_EXTENSIONS_BUILD_SIMULATOR_ID}" \
test CODE_SIGNING_ALLOWED=NO
displayName: Build and test OrtExtensionsUsage
#####################################
# Linux prevent exception propagation
#####################################
- job: Linux_Prevent_Exception_Propagation
pool:
vmImage: 'ubuntu-latest'
steps:
# Simulate an embedded build as part of ORT with exceptions disabled by manually setting CMAKE_CXX_FLAGS and
# using _OCOS_PREVENT_EXCEPTION_PROPAGATION_OVERRIDE. The build should re-enable exceptions within ort-ext
# but prevent them from propagating. Unit tests are run to validate this.
- script: '
./build_lib.sh --enable_cxx_tests --onnxruntime_version 1.14.0 --config RelWithDebInfo
--cmake_extra_defines
_OCOS_PREVENT_EXCEPTION_PROPAGATION_OVERRIDE=ON OCOS_ENABLE_CPP_EXCEPTIONS=OFF
CMAKE_CXX_FLAGS="-fno-exceptions -fno-unwind-tables -fno-asynchronous-unwind-tables"
'
displayName: Build ort-ext with exception propagation disabled
# As an extra validation check CMakeCache.txt as well
- script: |
grep "^_OCOS_PREVENT_EXCEPTION_PROPAGATION.*ON$" build/Linux/RelWithDebInfo/CMakeCache.txt
if [ $? -ne 0 ]; then
echo "Exception propogation was not enabled correctly."
exit 1
fi
##############################
# Linux for selected_ops build
##############################
- job: Linux_SelectedOpsBuild
pool:
vmImage: 'ubuntu-latest'
steps:
# compiled as only one operator selected.
- bash: |
set -e -x -u
echo 'set (OCOS_ENABLE_BERT_TOKENIZER ON CACHE BOOL "" FORCE)' > cmake/_selectedoplist.cmake
./build.sh -DOCOS_ENABLE_CPP_EXCEPTIONS=OFF -DOCOS_ENABLE_SELECTED_OPLIST=ON
displayName: Build ort-extensions with only one operator was selected