Skip to content

Commit 32ecbf4

Browse files
chilo-mschilo-ms
authored
Create combined GPU tarball and zip file package (#8827)
* Add onnxruntime_providers_shared.dll into gpu nuget package * Modify for test * Temporarily remove for test * Modify for test * Modify for test * Test packging Windows combined GPU * Test packging Windows combined GPU * Test packging Windows combined GPU * Test packging Windows combined GPU * modify for test * modify for test * fix bug * Modify for test * Modify for test * Modify for test * Modify for test * Modify for test * Modify for test * Modify for test * Modify for test * Prepare for PR * Prepare for PR * Code refactor * Rename proper Artifact name * Rename intermediate Artifact names * Revert Artifact Names * Rename Artifact Names * Modify Artifact name * Modify Artifact name * Modify Artifact name * Update Java package * Update Java package * fix bug to change artifact name * Fix bug for the wrong file path * Fix no fetching correct artifact and test * temporarily modify for test * undo the change for test
1 parent cee7952 commit 32ecbf4

10 files changed

+253
-53
lines changed

‎tools/ci_build/github/azure-pipelines/c-api-noopenmp-packaging-pipelines.yml

Lines changed: 120 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -60,16 +60,16 @@ jobs:
6060
parameters:
6161
arch: 'linux-x64'
6262
buildConfig: 'Release'
63-
artifactName: 'onnxruntime-java-linux-gpu-x64'
63+
artifactName: 'onnxruntime-java-linux-x64-cuda'
6464
version: '$(OnnxRuntimeVersion)'
6565
libraryName: 'libonnxruntime.so'
6666
nativeLibraryName: 'libonnxruntime4j_jni.so'
6767

6868
- template: templates/c-api-artifacts-package-and-publish-steps-posix.yml
6969
parameters:
7070
buildConfig: 'Release'
71-
artifactName: 'onnxruntime-linux-x64-gpu-$(OnnxRuntimeVersion)'
72-
artifactNameNoVersionString: 'onnxruntime-linux-x64-gpu'
71+
artifactName: 'onnxruntime-linux-x64-cuda-$(OnnxRuntimeVersion)'
72+
artifactNameNoVersionString: 'onnxruntime-linux-x64-cuda'
7373
libraryName: 'libonnxruntime.so.$(OnnxRuntimeVersion)'
7474
commitId: $(OnnxRuntimeGitCommitHash)
7575

@@ -79,6 +79,11 @@ jobs:
7979
- template: templates/clean-agent-build-directory-step.yml
8080

8181
- template: linux-gpu-tensorrt-packaging-pipeline.yml
82+
parameters:
83+
artifactName: 'onnxruntime-linux-x64-tensorrt-$(OnnxRuntimeVersion)'
84+
artifactNameNoVersionString: 'onnxruntime-linux-x64-tensorrt'
85+
buildJava: true
86+
buildJavaOption: '--build_java'
8287

8388
- template: templates/win-ci.yml
8489
parameters:
@@ -89,7 +94,7 @@ jobs:
8994
EnvSetupScript: setup_env_gpu.bat
9095
buildArch: x64
9196
msbuildPlatform: x64
92-
packageName: gpu-x64
97+
packageName: x64-cuda
9398
buildparameter: --use_cuda --cuda_version=11.1 --cuda_home="C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.1" --cudnn_home="C:\local\cudnn-11.4-windows-x64-v8.2.2.26\cuda" --enable_onnx_tests --enable_wcos --build_java --cmake_extra_defines "CMAKE_CUDA_ARCHITECTURES=37;52;60;61;70;75;80"
9499
runTests: ${{ parameters.RunOnnxRuntimeTests }}
95100
buildJava: true
@@ -104,7 +109,7 @@ jobs:
104109
EnvSetupScript: setup_env_gpu.bat
105110
buildArch: x64
106111
msbuildPlatform: x64
107-
packageName: tensorrt-x64
112+
packageName: x64-tensorrt
108113
buildparameter: --use_tensorrt --tensorrt_home="C:\local\TensorRT-8.0.1.6.Windows10.x86_64.cuda-11.3.cudnn8.2" --cuda_version=11.1 --cuda_home="C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.1" --cudnn_home="C:\local\cudnn-11.4-windows-x64-v8.2.2.26\cuda" --enable_onnx_tests --enable_wcos --cmake_extra_defines "CMAKE_CUDA_ARCHITECTURES=37;52;60;61;70;75;80"
109114
runTests: ${{ parameters.RunOnnxRuntimeTests }}
110115
buildJava: false
@@ -130,16 +135,23 @@ jobs:
130135
displayName: 'Download Pipeline Artifact - Win x64'
131136
inputs:
132137
buildType: 'current'
133-
artifactName: 'drop-onnxruntime-java-win-gpu-x64'
138+
artifactName: 'drop-onnxruntime-java-win-x64-cuda'
134139
targetPath: '$(Build.BinariesDirectory)\java-artifact\onnxruntime-java-win-x64'
135140

136141
- task: DownloadPipelineArtifact@2
137142
displayName: 'Download Pipeline Artifact - Linux x64'
138143
inputs:
139144
buildType: 'current'
140-
artifactName: 'drop-onnxruntime-java-linux-gpu-x64'
145+
artifactName: 'drop-onnxruntime-java-linux-x64-cuda'
141146
targetPath: '$(Build.BinariesDirectory)\java-artifact\onnxruntime-java-linux-x64'
142147

148+
- task: DownloadPipelineArtifact@2
149+
displayName: 'Download Pipeline Artifact - Linux x64'
150+
inputs:
151+
buildType: 'current'
152+
artifactName: 'drop-onnxruntime-java-linux-x64-tensorrt'
153+
targetPath: '$(Build.BinariesDirectory)\java-artifact\onnxruntime-java-linux-x64-tensorrt'
154+
143155
- task: PowerShell@2
144156
displayName: 'PowerShell Script'
145157
inputs:
@@ -149,8 +161,6 @@ jobs:
149161
showWarnings: true
150162
workingDirectory: '$(Build.BinariesDirectory)\java-artifact'
151163

152-
153-
154164
- task: CopyFiles@2
155165
displayName: 'Copy Java Files to Artifact Staging Directory'
156166
inputs:
@@ -252,7 +262,103 @@ jobs:
252262
displayName: 'Clean Agent Directories'
253263
condition: always()
254264

265+
- job: Linux_Packaging_combined_GPU
266+
workspace:
267+
clean: all
268+
pool: 'Onnxruntime-Linux-GPU'
269+
dependsOn:
270+
- Linux_C_API_Packaging_GPU_x64
271+
- Linux_C_API_Packaging_GPU_TensorRT_x64
272+
condition: succeeded()
273+
steps:
274+
- checkout: self
275+
submodules: false
276+
- template: templates/set-version-number-variables-step.yml
277+
- task: DownloadPipelineArtifact@2
278+
displayName: 'Download Pipeline Artifact - Combined GPU'
279+
inputs:
280+
artifactName: 'onnxruntime-linux-x64-cuda'
281+
targetPath: '$(Build.BinariesDirectory)/tgz-artifacts'
282+
283+
- task: DownloadPipelineArtifact@2
284+
displayName: 'Download Pipeline Artifact - Combined GPU'
285+
inputs:
286+
artifactName: 'onnxruntime-linux-x64-tensorrt'
287+
targetPath: '$(Build.BinariesDirectory)/tgz-artifacts'
288+
289+
- task: ShellScript@2
290+
displayName: 'Shell Script'
291+
inputs:
292+
scriptPath: 'tools/ci_build/github/linux/extract_and_bundle_gpu_package.sh'
293+
args: '-a $(Build.BinariesDirectory)/tgz-artifacts'
294+
workingDirectory: '$(Build.BinariesDirectory)/tgz-artifacts'
295+
296+
- task: ArchiveFiles@2
297+
inputs:
298+
rootFolderOrFile: '$(Build.BinariesDirectory)/tgz-artifacts/onnxruntime-linux-x64-gpu'
299+
includeRootFolder: true
300+
archiveType: 'tar' # Options: zip, 7z, tar, wim
301+
tarCompression: 'gz'
302+
archiveFile: '$(Build.ArtifactStagingDirectory)/onnxruntime-linux-x64-gpu-$(OnnxRuntimeVersion).tgz'
303+
replaceExistingArchive: true
304+
305+
- task: PublishPipelineArtifact@1
306+
inputs:
307+
targetPath: '$(Build.ArtifactStagingDirectory)/onnxruntime-linux-x64-gpu-$(OnnxRuntimeVersion).tgz'
308+
artifactName: 'onnxruntime-linux-x64-gpu'
309+
310+
- job: Windows_Packaging_combined_GPU
311+
workspace:
312+
clean: all
313+
pool: 'Win-CPU-2021'
314+
dependsOn:
315+
- Windows_Packaging_gpu
316+
- Windows_Packaging_tensorrt
317+
condition: succeeded()
318+
steps:
319+
- checkout: self
320+
submodules: false
321+
- task: DownloadPipelineArtifact@2
322+
displayName: 'Download Pipeline Artifact - Combined GPU'
323+
inputs:
324+
artifactName: 'onnxruntime-win-x64-cuda'
325+
targetPath: '$(Build.BinariesDirectory)/zip-artifacts'
326+
327+
- task: DownloadPipelineArtifact@2
328+
displayName: 'Download Pipeline Artifact - Combined GPU'
329+
inputs:
330+
artifactName: 'onnxruntime-win-x64-tensorrt'
331+
targetPath: '$(Build.BinariesDirectory)/zip-artifacts'
255332

333+
- task: PowerShell@2
334+
displayName: 'PowerShell Script'
335+
inputs:
336+
targetType: filePath
337+
filePath: $(Build.SourcesDirectory)\tools\ci_build\github\windows\extract_zip_files_gpu.ps1
338+
339+
- script: |
340+
dir
341+
workingDirectory: '$(Build.BinariesDirectory)/zip-artifacts'
342+
displayName: 'List artifacts'
343+
344+
- task: BatchScript@1
345+
displayName: 'Bundle CUDA/TRT EP binaries'
346+
inputs:
347+
filename: $(Build.SourcesDirectory)\tools\ci_build\github\windows\bundle_dlls_gpu.bat
348+
workingFolder: $(Build.BinariesDirectory)\zip-artifacts
349+
350+
- task: CopyFiles@2
351+
displayName: 'Copy zip file to: $(Build.ArtifactStagingDirectory)'
352+
inputs:
353+
SourceFolder: '$(Build.BinariesDirectory)\zip-artifacts'
354+
Contents: 'onnxruntime-win-x64-gpu-*.zip'
355+
TargetFolder: '$(Build.ArtifactStagingDirectory)'
356+
357+
- task: PublishPipelineArtifact@0
358+
displayName: 'Publish Pipeline Combined GPU Package Artifact'
359+
inputs:
360+
artifactName: 'onnxruntime-win-x64-gpu'
361+
targetPath: '$(Build.ArtifactStagingDirectory)'
256362

257363
- job: NuGet_Packaging_GPU
258364
workspace:
@@ -272,25 +378,25 @@ jobs:
272378
- task: DownloadPipelineArtifact@2
273379
displayName: 'Download Pipeline Artifact - NuGet'
274380
inputs:
275-
artifactName: 'onnxruntime-win-gpu-x64'
381+
artifactName: 'onnxruntime-win-x64-cuda'
276382
targetPath: '$(Build.BinariesDirectory)/nuget-artifact'
277383

278384
- task: DownloadPipelineArtifact@2
279385
displayName: 'Download Pipeline Artifact - NuGet'
280386
inputs:
281-
artifactName: 'onnxruntime-win-tensorrt-x64'
387+
artifactName: 'onnxruntime-win-x64-tensorrt'
282388
targetPath: '$(Build.BinariesDirectory)/nuget-artifact'
283389

284390
- task: DownloadPipelineArtifact@2
285391
displayName: 'Download Pipeline Artifact - NuGet'
286392
inputs:
287-
artifactName: 'onnxruntime-linux-x64-gpu'
393+
artifactName: 'onnxruntime-linux-x64-cuda'
288394
targetPath: '$(Build.BinariesDirectory)/nuget-artifact'
289395

290396
- task: DownloadPipelineArtifact@2
291397
displayName: 'Download Pipeline Artifact - NuGet'
292398
inputs:
293-
artifactName: 'onnxruntime-linux-x64-gpu-tensorrt'
399+
artifactName: 'onnxruntime-linux-x64-tensorrt'
294400
targetPath: '$(Build.BinariesDirectory)/nuget-artifact'
295401

296402
- task: DownloadPipelineArtifact@2
@@ -357,7 +463,7 @@ jobs:
357463
- task: BatchScript@1
358464
displayName: 'Bundle Native NuGet and other binaries'
359465
inputs:
360-
filename: $(Build.SourcesDirectory)\tools\ci_build\github\windows\bundle_dlls_gpu.bat
466+
filename: $(Build.SourcesDirectory)\tools\ci_build\github\windows\bundle_dlls_gpu_nuget.bat
361467
workingFolder: $(Build.BinariesDirectory)\RelWithDebInfo\RelWithDebInfo\nuget-artifacts
362468

363469
- task: CopyFiles@2

‎tools/ci_build/github/azure-pipelines/linux-gpu-tensorrt-packaging-pipeline.yml

Lines changed: 30 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,20 @@
1+
parameters:
2+
- name: artifactName
3+
type: string
4+
default: 'onnxruntime-linux-x64-gpu-tensorrt-$(OnnxRuntimeVersion)'
5+
6+
- name: artifactNameNoVersionString
7+
type: string
8+
default: 'onnxruntime-linux-x64-gpu-tensorrt'
9+
10+
- name: buildJava
11+
type: boolean
12+
default: false
13+
14+
- name: buildJavaOption
15+
type: string
16+
default: ''
17+
118
jobs:
219

320
- job: Linux_C_API_Packaging_GPU_TensorRT_x64
@@ -22,14 +39,24 @@ jobs:
2239
docker run --gpus all -e CC=/opt/rh/devtoolset-10/root/usr/bin/cc -e CXX=/opt/rh/devtoolset-10/root/usr/bin/c++ -e CFLAGS="-Wp,-D_FORTIFY_SOURCE=2 -Wp,-D_GLIBCXX_ASSERTIONS -fstack-protector-strong -fstack-clash-protection -fcf-protection -O3 -Wl,--strip-all" -e CXXFLAGS="-Wp,-D_FORTIFY_SOURCE=2 -Wp,-D_GLIBCXX_ASSERTIONS -fstack-protector-strong -fstack-clash-protection -fcf-protection -O3 -Wl,--strip-all" -e NVIDIA_VISIBLE_DEVICES=all --rm --volume /data/onnx:/data/onnx:ro --volume $(Build.SourcesDirectory):/onnxruntime_src --volume $(Build.BinariesDirectory):/build \
2340
--volume /data/models:/build/models:ro --volume $HOME/.onnx:/home/onnxruntimedev/.onnx -e NIGHTLY_BUILD onnxruntimecuda114xtrt80build \
2441
/opt/python/cp37-cp37m/bin/python3 /onnxruntime_src/tools/ci_build/build.py --build_dir /build --config Release \
25-
--skip_submodule_sync --parallel --build_shared_lib --use_tensorrt --cuda_version=$(CUDA_VERSION) --cuda_home=/usr/local/cuda-$(CUDA_VERSION) --cudnn_home=/usr --tensorrt_home=/usr --cmake_extra_defines CMAKE_CUDA_HOST_COMPILER=/opt/rh/devtoolset-10/root/usr/bin/cc 'CMAKE_CUDA_ARCHITECTURES=37;50;52;60;61;70;75;80'
42+
--skip_submodule_sync --parallel --build_shared_lib ${{ parameters.buildJavaOption }} --use_tensorrt --cuda_version=$(CUDA_VERSION) --cuda_home=/usr/local/cuda-$(CUDA_VERSION) --cudnn_home=/usr --tensorrt_home=/usr --cmake_extra_defines CMAKE_CUDA_HOST_COMPILER=/opt/rh/devtoolset-10/root/usr/bin/cc 'CMAKE_CUDA_ARCHITECTURES=37;50;52;60;61;70;75;80'
2643
workingDirectory: $(Build.SourcesDirectory)
2744

45+
- ${{ if eq(parameters.buildJava, true) }}:
46+
- template: templates/java-api-artifacts-package-and-publish-steps-posix.yml
47+
parameters:
48+
arch: 'linux-x64'
49+
buildConfig: 'Release'
50+
artifactName: 'onnxruntime-java-linux-x64-tensorrt'
51+
version: '$(OnnxRuntimeVersion)'
52+
libraryName: 'libonnxruntime.so'
53+
nativeLibraryName: 'libonnxruntime4j_jni.so'
54+
2855
- template: templates/c-api-artifacts-package-and-publish-steps-posix.yml
2956
parameters:
3057
buildConfig: 'Release'
31-
artifactName: 'onnxruntime-linux-x64-gpu-tensorrt-$(OnnxRuntimeVersion)'
32-
artifactNameNoVersionString: 'onnxruntime-linux-x64-gpu-tensorrt'
58+
artifactName: ${{ parameters.artifactName }}
59+
artifactNameNoVersionString: ${{ parameters.artifactNameNoVersionString }}
3360
libraryName: 'libonnxruntime.so.$(OnnxRuntimeVersion)'
3461
commitId: $(OnnxRuntimeGitCommitHash)
3562

‎tools/ci_build/github/azure-pipelines/templates/c-api-artifacts-package-and-publish-steps-windows.yml

Lines changed: 5 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -48,6 +48,11 @@ steps:
4848
copy $(Build.BinariesDirectory)\${{parameters.buildConfig}}\${{parameters.buildConfig}}\onnxruntime_providers_cuda.pdb $(Build.BinariesDirectory)\${{parameters.artifactName}}\lib
4949
copy $(Build.BinariesDirectory)\${{parameters.buildConfig}}\${{parameters.buildConfig}}\onnxruntime_providers_cuda.lib $(Build.BinariesDirectory)\${{parameters.artifactName}}\lib
5050
51+
# copy trt ep libraries only when trt ep is enabled
52+
copy $(Build.BinariesDirectory)\${{parameters.buildConfig}}\${{parameters.buildConfig}}\onnxruntime_providers_tensorrt.dll $(Build.BinariesDirectory)\${{parameters.artifactName}}\lib
53+
copy $(Build.BinariesDirectory)\${{parameters.buildConfig}}\${{parameters.buildConfig}}\onnxruntime_providers_tensorrt.pdb $(Build.BinariesDirectory)\${{parameters.artifactName}}\lib
54+
copy $(Build.BinariesDirectory)\${{parameters.buildConfig}}\${{parameters.buildConfig}}\onnxruntime_providers_tensorrt.lib $(Build.BinariesDirectory)\${{parameters.artifactName}}\lib
55+
5156
copy $(Build.BinariesDirectory)\${{parameters.buildConfig}}\${{parameters.buildConfig}}\onnxruntime.pdb $(Build.BinariesDirectory)\${{parameters.artifactName}}\lib
5257
copy $(Build.BinariesDirectory)\${{parameters.buildConfig}}\${{parameters.buildConfig}}\onnxruntime.lib $(Build.BinariesDirectory)\${{parameters.artifactName}}\lib
5358
copy $(Build.SourcesDirectory)\include\onnxruntime\core\session\onnxruntime_*.h $(Build.BinariesDirectory)\${{parameters.artifactName}}\include
@@ -64,20 +69,6 @@ steps:
6469
6570
workingDirectory: '$(Build.BinariesDirectory)\${{parameters.buildConfig}}'
6671

67-
- task: CmdLine@2
68-
displayName: 'Copy TensorRT EP build artifacts for zipping'
69-
condition: and(succeeded(), eq('${{ parameters.trtEnabled}}', true))
70-
inputs:
71-
script: |
72-
mkdir $(Build.BinariesDirectory)\${{parameters.artifactName}}
73-
mkdir $(Build.BinariesDirectory)\${{parameters.artifactName}}\lib
74-
mkdir $(Build.BinariesDirectory)\${{parameters.artifactName}}\include
75-
76-
copy $(Build.BinariesDirectory)\${{parameters.buildConfig}}\${{parameters.buildConfig}}\onnxruntime_providers_tensorrt.dll $(Build.BinariesDirectory)\${{parameters.artifactName}}\lib
77-
copy $(Build.BinariesDirectory)\${{parameters.buildConfig}}\${{parameters.buildConfig}}\onnxruntime.dll $(Build.BinariesDirectory)\${{parameters.artifactName}}\lib
78-
copy $(Build.BinariesDirectory)\${{parameters.buildConfig}}\${{parameters.buildConfig}}\onnxruntime.pdb $(Build.BinariesDirectory)\${{parameters.artifactName}}\lib
79-
copy $(Build.BinariesDirectory)\${{parameters.buildConfig}}\${{parameters.buildConfig}}\onnxruntime.lib $(Build.BinariesDirectory)\${{parameters.artifactName}}\lib
80-
8172
- template: win-esrp-dll.yml
8273
parameters:
8374
FolderPath: '$(Build.BinariesDirectory)\${{parameters.artifactName}}'

‎tools/ci_build/github/linux/extract_and_bundle_gpu_package.sh

Lines changed: 28 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,28 @@
1+
#!/bin/bash
2+
set -e -o -x
3+
4+
while getopts a: parameter_Option
5+
do case "${parameter_Option}"
6+
in
7+
a) ARTIFACT_DIR=${OPTARG};;
8+
esac
9+
done
10+
11+
EXIT_CODE=1
12+
13+
uname -a
14+
15+
cd $ARTIFACT_DIR
16+
17+
mkdir -p $ARTIFACT_DIR/onnxruntime-linux-x64-tensorrt
18+
tar zxvf $ARTIFACT_DIR/onnxruntime-linux-x64-tensorrt-*.tgz -C onnxruntime-linux-x64-tensorrt
19+
rm $ARTIFACT_DIR/onnxruntime-linux-x64-tensorrt-*.tgz
20+
21+
mkdir -p $ARTIFACT_DIR/onnxruntime-linux-x64-gpu
22+
TAR_NAME=`ls $ARTIFACT_DIR/onnxruntime-linux-x64-cuda-*.tgz`
23+
tar zxvf $ARTIFACT_DIR/onnxruntime-linux-x64-cuda-*.tgz -C onnxruntime-linux-x64-gpu
24+
rm $ARTIFACT_DIR/onnxruntime-linux-x64-cuda-*.tgz
25+
26+
cp onnxruntime-linux-x64-tensorrt/*/lib/libonnxruntime.so* onnxruntime-linux-x64-gpu/*/lib
27+
cp onnxruntime-linux-x64-tensorrt/*/lib/libonnxruntime_providers_tensorrt.so onnxruntime-linux-x64-gpu/*/lib
28+
cp onnxruntime-linux-x64-tensorrt/*/lib/libonnxruntime_providers_shared.so onnxruntime-linux-x64-gpu/*/lib

‎tools/ci_build/github/linux/java_copy_strip_binary.sh

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -50,6 +50,11 @@ then
5050
cp $BINARY_DIR/$BUILD_CONFIG/libonnxruntime_providers_shared.so $BINARY_DIR/$ARTIFACT_NAME/$NATIVE_FOLDER/libonnxruntime_providers_shared.so
5151
cp $BINARY_DIR/$BUILD_CONFIG/libonnxruntime_providers_cuda.so $BINARY_DIR/$ARTIFACT_NAME/$NATIVE_FOLDER/libonnxruntime_providers_cuda.so
5252
fi
53+
# Add tensorrt provider if it exists
54+
if [[ -f "$BINARY_DIR/$BUILD_CONFIG/libonnxruntime_providers_tensorrt.so" ]]; then
55+
cp $BINARY_DIR/$BUILD_CONFIG/libonnxruntime_providers_shared.so $BINARY_DIR/$ARTIFACT_NAME/$NATIVE_FOLDER/libonnxruntime_providers_shared.so
56+
cp $BINARY_DIR/$BUILD_CONFIG/libonnxruntime_providers_tensorrt.so $BINARY_DIR/$ARTIFACT_NAME/$NATIVE_FOLDER/libonnxruntime_providers_tensorrt.so
57+
fi
5358
fi
5459

5560
find $BINARY_DIR/$ARTIFACT_NAME -ls

‎tools/ci_build/github/windows/bundle_dlls_gpu.bat

Lines changed: 15 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -1,22 +1,21 @@
11
REM Copyright (c) Microsoft Corporation. All rights reserved.
22
REM Licensed under the MIT License.
33

4-
REM for available runtime identifiers, see https://github.com/dotnet/corefx/blob/release/3.1/pkg/Microsoft.NETCore.Platforms/runtime.json
54
set PATH=%CD%;%PATH%
6-
SETLOCAL EnableDelayedExpansion
7-
FOR /R %%i IN (*.nupkg) do (
5+
SETLOCAL EnableDelayedExpansion
6+
FOR /R %%i IN (*.zip) do (
87
set filename=%%~ni
9-
IF NOT "!filename:~25,7!"=="Managed" (
10-
mkdir runtimes\linux-x64\native
11-
move onnxruntime-linux-x64-gpu\lib\libonnxruntime_providers_* runtimes\linux-x64\native
12-
move onnxruntime-linux-x64-gpu-tensorrt\lib\libonnxruntime.so.1* runtimes\linux-x64\native\libonnxruntime.so
13-
move onnxruntime-linux-x64-gpu-tensorrt\lib\libonnxruntime_providers_shared.so runtimes\linux-x64\native\libonnxruntime_providers_shared.so
14-
move onnxruntime-linux-x64-gpu-tensorrt\lib\libonnxruntime_providers_tensorrt.so runtimes\linux-x64\native\libonnxruntime_providers_tensorrt.so
15-
mkdir runtimes\win-x64\native
16-
move onnxruntime-win-tensorrt-x64\lib\onnxruntime_providers_tensorrt.dll runtimes\win-x64\native\onnxruntime_providers_tensorrt.dll
17-
move onnxruntime-win-tensorrt-x64\lib\onnxruntime.dll runtimes\win-x64\native\onnxruntime.dll
18-
move onnxruntime-win-tensorrt-x64\lib\onnxruntime.lib runtimes\win-x64\native\onnxruntime.lib
19-
move onnxruntime-win-tensorrt-x64\lib\onnxruntime.pdb runtimes\win-x64\native\onnxruntime.pdb
20-
7z a %%~ni.nupkg runtimes
8+
IF "!filename:~20,3!"=="gpu" (
9+
mkdir !filename!\lib
10+
move onnxruntime-win-x64-tensorrt\lib\onnxruntime_providers_tensorrt.dll !filename!\lib\onnxruntime_providers_tensorrt.dll
11+
move onnxruntime-win-x64-tensorrt\lib\onnxruntime_providers_tensorrt.lib !filename!\lib\onnxruntime_providers_tensorrt.lib
12+
move onnxruntime-win-x64-tensorrt\lib\onnxruntime_providers_tensorrt.pdb !filename!\lib\onnxruntime_providers_tensorrt.pdb
13+
move onnxruntime-win-x64-tensorrt\lib\onnxruntime_providers_shared.dll !filename!\lib\onnxruntime_providers_shared.dll
14+
move onnxruntime-win-x64-tensorrt\lib\onnxruntime_providers_shared.lib !filename!\lib\onnxruntime_providers_shared.lib
15+
move onnxruntime-win-x64-tensorrt\lib\onnxruntime_providers_shared.pdb !filename!\lib\onnxruntime_providers_shared.pdb
16+
move onnxruntime-win-x64-tensorrt\lib\onnxruntime.dll !filename!\lib\onnxruntime.dll
17+
move onnxruntime-win-x64-tensorrt\lib\onnxruntime.lib !filename!\lib\onnxruntime.lib
18+
move onnxruntime-win-x64-tensorrt\lib\onnxruntime.pdb !filename!\lib\onnxruntime.pdb
19+
7z a %%~ni.zip !filename!\lib
2120
)
22-
)
21+
)

0 commit comments

Comments
 (0)