Use env var instead of wheel-build-params for release flag

pytorch · May 11, 2024 · 8915f84 · 8915f84
1 parent ec6e951
commit 8915f84
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/.github/workflows/build-test-windows.yml b/.github/workflows/build-test-windows.yml
@@ -47,7 +47,6 @@ jobs:
           - repository: pytorch/tensorrt
             pre-script: packaging/pre_build_script_windows.sh
             env-script: packaging/vc_env_helper.bat
-            wheel-build-params: "--release"
             smoke-test-script: packaging/smoke_test_windows.py
             package-name: torch_tensorrt
     name: Build torch-tensorrt whl package
@@ -60,7 +59,6 @@ jobs:
       build-matrix: ${{ needs.substitute-runner.outputs.matrix }}
       pre-script: ${{ matrix.pre-script }}
       env-script: ${{ matrix.env-script }}
-      wheel-build-params: ${{ matrix.wheel-build-params }}
       smoke-test-script: ${{ matrix.smoke-test-script }}
       package-name: ${{ matrix.package-name }}
       trigger-event: ${{ github.event_name }}

diff --git a/packaging/pre_build_script_windows.sh b/packaging/pre_build_script_windows.sh
@@ -23,3 +23,5 @@ unzip TensorRT.zip -d C:/
 export CUDA_HOME="$(echo ${CUDA_PATH} | sed -e 's#\\#\/#g')"
 
 cat toolchains/ci_workspaces/WORKSPACE.win.release.tmpl | envsubst > WORKSPACE
+
+echo "RELEASE=1" >> ${GITHUB_ENV}