Add smoke test exit(0)

Author: gs-olive

.github/workflows/build-test.yml

@@ -35,7 +35,7 @@ jobs:
             pre-script: packaging/pre_build_script.sh
             env-var-script: packaging/env_vars.txt
             post-script: ""
-            smoke-test-script: ""
+            smoke-test-script: "packaging/smoke_test.py"
             package-name: torch_tensorrt
     name: Build torch-tensorrt whl package
     uses: pytorch/test-infra/.github/workflows/build_wheels_linux.yml@release/2.2
@@ -74,7 +74,7 @@ jobs:
       test-infra-repository: pytorch/test-infra
       test-infra-ref: release/2.2
       build-matrix: ${{ needs.generate-matrix.outputs.matrix }}
-      pre-script: ${{ matrix.pre-script }}
+      pre-script: packaging/pre_build_script_exit.sh
       script: |
         export USE_HOST_DEPS=1
         export LD_LIBRARY_PATH=/usr/lib64:$LD_LIBRARY_PATH
@@ -110,7 +110,7 @@ jobs:
       test-infra-repository: pytorch/test-infra
       test-infra-ref: release/2.2
       build-matrix: ${{ needs.generate-matrix.outputs.matrix }}
-      pre-script: ${{ matrix.pre-script }}
+      pre-script: packaging/pre_build_script_exit.sh
       script: |
         export USE_HOST_DEPS=1
         pushd .
@@ -138,7 +138,7 @@ jobs:
       test-infra-repository: pytorch/test-infra
       test-infra-ref: release/2.2
       build-matrix: ${{ needs.generate-matrix.outputs.matrix }}
-      pre-script: ${{ matrix.pre-script }}
+      pre-script: packaging/pre_build_script_exit.sh
       script: |
         export USE_HOST_DEPS=1
         pushd .
@@ -168,7 +168,7 @@ jobs:
       test-infra-repository: pytorch/test-infra
       test-infra-ref: release/2.2
       build-matrix: ${{ needs.generate-matrix.outputs.matrix }}
-      pre-script: ${{ matrix.pre-script }}
+      pre-script: packaging/pre_build_script_exit.sh
       script: |
         export USE_HOST_DEPS=1
         pushd .
@@ -197,7 +197,7 @@ jobs:
       test-infra-repository: pytorch/test-infra
       test-infra-ref: release/2.2
       build-matrix: ${{ needs.generate-matrix.outputs.matrix }}
-      pre-script: ${{ matrix.pre-script }}
+      pre-script: packaging/pre_build_script_exit.sh
       script: |
         export USE_HOST_DEPS=1
         pushd .

packaging/pre_build_script.sh

@@ -1,5 +1,7 @@
 #!/bin/bash
 
+if python -V | grep 3.12; then exit 0; fi
+
 # Install dependencies
 python3 -m pip install pyyaml
 TRT_VERSION=$(python3 -c "import versions; versions.tensorrt_version()")

packaging/pre_build_script_exit.sh

@@ -0,0 +1,3 @@
+#!/bin/bash
+
+if python -V | grep 3.12; then exit 0; fi

packaging/smoke_test.py

@@ -0,0 +1,4 @@
+import sys
+
+if sys.version.startswith("3.12"):
+    exit(0)