mirror of
https://github.com/saymrwulf/onnxruntime.git
synced 2026-05-17 21:10:43 +00:00
3f03c12986
### Description 1. Update donwload-artifacts to flex-downloadartifacts to make it eaiser to debug. 2. Move the native files into Gpu.Windows and Gpu-linux packages. Onnxruntime-Gpu has dependency on them. 3. update the package validation as well 4. Add 2 stages to run E2E test for GPU.Windows and GPU.Linux for example:  ### Motivation and Context Single Onnxruntime.Gpu Package size has already excceded the Nuget size limit. We split the package into some smaller packages to make them can be published. For compatibility, the user can install or upgrade Onnxruntime.Gpu, which will install Gpu.Windows and Gpu.Linux automatically. And the user can only install Gpu.Windows and Gpu.Linux directly. ### Test Link 1. In ORT_NIGHTLY 2. Install the preview version in nuget-int. (nuget source: https://apiint.nugettest.org/v3/index.json) --------- Co-authored-by: Scott McKay <skottmckay@gmail.com>
56 lines
2.6 KiB
Bash
Executable file
56 lines
2.6 KiB
Bash
Executable file
#!/bin/bash
|
|
# Copyright (c) Microsoft Corporation. All rights reserved.
|
|
# Licensed under the MIT License.
|
|
|
|
LocalNuGetRepo=$1
|
|
export CurrentOnnxRuntimeVersion=$2
|
|
IsMacOS=${3:-false}
|
|
PACKAGENAME=${PACKAGENAME:-Microsoft.ML.OnnxRuntime}
|
|
RunTestCsharp=${RunTestCsharp:-true}
|
|
RunTestNative=${RunTestNative:-true}
|
|
|
|
set -x -e
|
|
|
|
pushd .
|
|
cd $BUILD_SOURCESDIRECTORY
|
|
|
|
echo "Current NuGet package version is $CurrentOnnxRuntimeVersion"
|
|
|
|
if [ $RunTestCsharp = "true" ]; then
|
|
if [[ $IsMacOS == "True" || $IsMacOS == "true" ]]; then
|
|
# TODO(#12040): The test should figure out the opset version from the model file. Remove it from the path.
|
|
ONNX_DIR="${BUILD_SOURCESDIRECTORY}/cmake/external/onnx"
|
|
ONNX_VERSION_NUMBER=$(cat "${ONNX_DIR}/VERSION_NUMBER" | sed -E 's/([0-9]+\.[0-9]+\.[0-9]+).*/\1/')
|
|
OPSET_VERSION=$(grep "${ONNX_VERSION_NUMBER}" "${ONNX_DIR}/docs/Versioning.md" | sed -E "s/${ONNX_VERSION_NUMBER}\|[^|]+\|([0-9]+)\|.*/\1/")
|
|
mkdir -p "${BUILD_BINARIESDIRECTORY}/models"
|
|
ln -s "${ONNX_DIR}/onnx/backend/test/data/node" "${BUILD_BINARIESDIRECTORY}/models/opset${OPSET_VERSION}"
|
|
fi
|
|
# Run C# tests
|
|
dotnet restore $BUILD_SOURCESDIRECTORY/csharp/test/Microsoft.ML.OnnxRuntime.EndToEndTests/Microsoft.ML.OnnxRuntime.EndToEndTests.csproj -s $LocalNuGetRepo -s https://api.nuget.org/v3/index.json
|
|
if [ $? -ne 0 ]; then
|
|
echo "Failed to restore nuget packages for the test project"
|
|
exit 1
|
|
fi
|
|
|
|
if [ $PACKAGENAME = "Microsoft.ML.OnnxRuntime.Gpu" ] || [ $PACKAGENAME = "Microsoft.ML.OnnxRuntime.Gpu.Linux" ]; then
|
|
export TESTONGPU=ON
|
|
dotnet test -p:DefineConstants=USE_CUDA $BUILD_SOURCESDIRECTORY/csharp/test/Microsoft.ML.OnnxRuntime.EndToEndTests/Microsoft.ML.OnnxRuntime.EndToEndTests.csproj --no-restore --verbosity detailed
|
|
if [ $? -ne 0 ]; then
|
|
echo "Failed to build or execute the end-to-end test"
|
|
exit 1
|
|
fi
|
|
dotnet test -p:DefineConstants=USE_TENSORRT $BUILD_SOURCESDIRECTORY/csharp/test/Microsoft.ML.OnnxRuntime.EndToEndTests/Microsoft.ML.OnnxRuntime.EndToEndTests.csproj --no-restore --verbosity detailed
|
|
elif [ $PACKAGENAME = "Microsoft.ML.OnnxRuntime.ROCm" ]; then
|
|
export TESTONGPU=ON
|
|
dotnet test -p:DefineConstants=USE_ROCM $BUILD_SOURCESDIRECTORY/csharp/test/Microsoft.ML.OnnxRuntime.EndToEndTests/Microsoft.ML.OnnxRuntime.EndToEndTests.csproj --no-restore --verbosity detailed
|
|
else
|
|
dotnet test $BUILD_SOURCESDIRECTORY/csharp/test/Microsoft.ML.OnnxRuntime.EndToEndTests/Microsoft.ML.OnnxRuntime.EndToEndTests.csproj --no-restore --verbosity detailed
|
|
fi
|
|
if [ $? -ne 0 ]; then
|
|
echo "Failed to build or execute the end-to-end test"
|
|
exit 1
|
|
fi
|
|
fi
|
|
|
|
cd $OldDir
|
|
popd
|