Skip to content

Commit

Permalink
Update doeverything.ps1
Browse files Browse the repository at this point in the history
  • Loading branch information
the-database committed Nov 7, 2022
1 parent a1438d0 commit cee62c5
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions doeverything.ps1
Original file line number Diff line number Diff line change
Expand Up @@ -122,7 +122,6 @@ if ($installVsMlrt) {
}

# Copy HD ONNX and create engine
Write-Host "Creating TensorRT engine from ONNX model $hdOnnx"
$env:CUDA_MODULE_LOADING="LAZY"
$hdEngineName = "$((Get-Item -Path $hdOnnx).BaseName)-1080"
$enginePath = "$pluginPath/$hdEngineName.engine"
Expand All @@ -146,11 +145,11 @@ if (-not $createEngine) {
$createEngine = $response -eq 0
}
if ($createEngine) {
Write-Host "Creating TensorRT engine from ONNX model $hdOnnx"
& "$pluginPath\trtexec" --fp16 --onnx=$hdOnnx --minShapes=input:1x3x8x8 --optShapes=input:1x3x1080x1920 --maxShapes=input:1x3x1080x1920 --saveEngine="$enginePath" --tacticSources=+CUDNN,-CUBLAS,-CUBLAS_LT
}

# Copy SD ONNX and create engine
Write-Host "Creating TensorRT engine from ONNX model $sdOnnx"
$env:CUDA_MODULE_LOADING="LAZY"
$sdEngineName = "$((Get-Item -Path $sdOnnx).BaseName)-720"
$enginePath = "$pluginPath/$sdEngineName.engine"
Expand All @@ -174,6 +173,7 @@ if (-not $createEngine) {
$createEngine = $response -eq 0
}
if ($createEngine) {
Write-Host "Creating TensorRT engine from ONNX model $sdOnnx"
& "$pluginPath\trtexec" --fp16 --onnx=$sdOnnx --minShapes=input:1x3x8x8 --optShapes=input:1x3x720x1280 --maxShapes=input:1x3x1080x1920 --saveEngine="$enginePath" --tacticSources=+CUDNN,-CUBLAS,-CUBLAS_LT
}

Expand Down

0 comments on commit cee62c5

Please sign in to comment.