saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-07-01 03:45:06 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Fix nuget gpu ci test error (#2164)

Browse source 
* fix nuget version extraction script for Gpu packages
* fix cuda version in gpu end-to-end test
This commit is contained in:
shahasad 2019-10-18 23:01:26 -07:00 committed by GitHub
parent ecf5ae8b76
commit 35dae992f1
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 9 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
csharp/test/Microsoft.ML.OnnxRuntime.EndToEndTests/runtest-docker-gpu.sh
View file

@ -14,7 +14,7 @@ PackageName=${PACKAGENAME:-Microsoft.ML.OnnxRuntime.Gpu}
RunTestCsharp=${RunTestCsharp:-true}
RunTestNative=${RunTestNative:-true}
#CUDA_VER=cuda10.0-cudnn7.3, cuda9.1-cudnn7.1, cuda10.0-cudnn7.3
CUDA_VER=${4:-cuda10.0-cudnn7.3}
CUDA_VER=${5:-cuda10.0-cudnn7.3}
PYTHON_VER=3.5
IMAGE="ubuntu16.04-$CUDA_VER"

8
tools/ci_build/github/azure-pipelines/nuget/templates/get-nuget-package-version-as-variable.yml
View file

@ -12,6 +12,10 @@ steps:
FOR /R %%i IN (Microsoft.ML.OnnxRuntime*.nupkg) do (
set filename=%%~ni
set ortversion=!filename:~25!
IF "!filename:~25,3!"=="Gpu" (
set ortversion=!filename:~29!
)
@echo ortversion is !ortversion!
@echo ##vso[task.setvariable variable=NuGetPackageVersionNumber;]!ortversion!
)
@ -25,4 +29,8 @@ steps:
filenamewithext=$(ls Microsoft.ML.OnnxRuntime*.nupkg)
filename=${filenamewithext%.*}
ortversion=${filename:25}
if [ "${filename:25:3}" = "Gpu" ]
then
ortversion=${filename:29}
fi
echo "##vso[task.setvariable variable=NuGetPackageVersionNumber;]$ortversion"