onnxruntime/tools/ci_build/github/windows/extract_zip_files_gpu.ps1
Chi Lo eb8f84e2a2
Fix issue of GPU tarball/zip/java package (#8850)
* modify for test

* modify for test

* modify for test

* modify for test

* modify for test

* modify for test

* prepare for PR

* Rename cuda directory to gpu directory in tarball

* Fix gpu java package

* fix bug

* fix small bug
2021-08-26 10:16:16 -07:00

25 lines
1.3 KiB
PowerShell

# Copyright (c) Microsoft Corporation. All rights reserved.
# Licensed under the MIT License.
# extract *-cuda-*.zip and *-tensorrt-*.zip
Get-ChildItem $Env:BUILD_BINARIESDIRECTORY\zip-artifacts -Filter *.zip |
Foreach-Object {
$cmd = "7z.exe x $($_.FullName) -y -o$Env:BUILD_BINARIESDIRECTORY\zip-artifacts"
Write-Output $cmd
Invoke-Expression -Command $cmd
}
# Rename tensorrt directory for later use in bundle_dlls_gpu.bat
Get-ChildItem $Env:BUILD_BINARIESDIRECTORY\zip-artifacts | Where-Object { $_.Name -match 'onnxruntime-win-x64-tensorrt-\d{1,}\.\d{1,}\.\d{1,}$' } | Rename-Item -NewName $Env:BUILD_BINARIESDIRECTORY\zip-artifacts\onnxruntime-win-x64-tensorrt
Remove-Item $Env:BUILD_BINARIESDIRECTORY\zip-artifacts\*.zip
# Rename cuda directory to gpu directory and re-compress it for later use in bundle_dlls_gpu.bat
Get-ChildItem $Env:BUILD_BINARIESDIRECTORY\zip-artifacts -Filter *cuda* |
Foreach-Object {
$($_.FullName) -match '.*onnxruntime-win-x64-cuda-(.*)'
$version=$matches[1]
Rename-Item -Path $($_.FullName) -NewName onnxruntime-win-x64-gpu-$version
$cmd = "7z.exe a $Env:BUILD_BINARIESDIRECTORY\zip-artifacts\onnxruntime-win-x64-gpu-$version.zip $Env:BUILD_BINARIESDIRECTORY\zip-artifacts\onnxruntime-win-x64-gpu-$version"
Write-Output $cmd
Invoke-Expression -Command $cmd
}