feat: First pass at Windows CI

- Update `setup.py` to enable `bdist_wheel` build
- Add GHA tooling for Windows

.github/workflows/build-test.yml → .github/workflows/build-test-windows.yml

@@ -1,24 +1,13 @@
-name: Build and test linux wheels
+name: Build and test Windows wheels
 
-on:
-  pull_request:
-  push:
-    branches:
-      - main
-      - nightly
-      - release/*
-    tags:
-        # NOTE: Binary build pipelines should only get triggered on release candidate builds
-        # Release candidate tags look like: v1.11.0-rc1
-        - v[0-9]+.[0-9]+.[0-9]+-rc[0-9]+
-  workflow_dispatch:
+on: [pull_request_target]
 
 jobs:
   generate-matrix:
     uses: pytorch/test-infra/.github/workflows/generate_binary_build_matrix.yml@main
     with:
       package-type: wheel
-      os: linux
+      os: windows
       test-infra-repository: pytorch/test-infra
       test-infra-ref: main
       with-rocm: false
@@ -31,13 +20,13 @@ jobs:
       matrix:
         include:
           - repository: pytorch/tensorrt
-            pre-script: packaging/pre_build_script.sh
+            pre-script: ""
             env-var-script: packaging/env_vars.txt
             post-script: ""
             smoke-test-script: ""
             package-name: torch_tensorrt
     name: Build torch-tensorrt whl package
-    uses: pytorch/test-infra/.github/workflows/build_wheels_linux.yml@main
+    uses: pytorch/test-infra/.github/workflows/build_wheels_windows.yml@main
     with:
       repository: ${{ matrix.repository }}
       ref: ""
@@ -54,41 +43,6 @@ jobs:
       AWS_PYTORCH_UPLOADER_ACCESS_KEY_ID: ${{ secrets.AWS_PYTORCH_UPLOADER_ACCESS_KEY_ID }}
       AWS_PYTORCH_UPLOADER_SECRET_ACCESS_KEY: ${{ secrets.AWS_PYTORCH_UPLOADER_SECRET_ACCESS_KEY }}
 
-  tests-py-torchscript-fe:
-    name: Test torchscript frontend [Python]
-    needs: [generate-matrix, build]
-    strategy:
-      fail-fast: false
-      matrix:
-        include:
-          - repository: pytorch/tensorrt
-            package-name: torch_tensorrt
-            pre-script: packaging/pre_build_script.sh
-    uses: pytorch/tensorrt/.github/workflows/linux-test.yml@main
-    with:
-      job-name: tests-py-torchscript-fe
-      repository: "pytorch/tensorrt"
-      ref: ""
-      test-infra-repository: pytorch/test-infra
-      test-infra-ref: main
-      build-matrix: ${{ needs.generate-matrix.outputs.matrix }}
-      pre-script: ${{ matrix.pre-script }}
-      script: |
-        export USE_HOST_DEPS=1
-        export LD_LIBRARY_PATH=/usr/lib64:$LD_LIBRARY_PATH
-        pushd .
-        cd tests/modules
-        ${CONDA_RUN} python -m pip install --pre -r requirements.txt --use-deprecated=legacy-resolver
-        ${CONDA_RUN} python hub.py
-        popd
-        pushd .
-        cd tests/py/ts
-        ${CONDA_RUN} python -m pip install --pre pytest timm transformers parameterized expecttest --use-deprecated=legacy-resolver
-        ${CONDA_RUN} python -m pytest --junitxml=${RUNNER_TEST_RESULTS_DIR}/ts_api_test_results.xml api/
-        ${CONDA_RUN} python -m pytest --junitxml=${RUNNER_TEST_RESULTS_DIR}/ts_models_test_results.xml models/
-        ${CONDA_RUN} python -m pytest --junitxml=${RUNNER_TEST_RESULTS_DIR}/ts_integrations_test_results.xml integrations/
-        popd
-
   tests-py-dynamo-converters:
     name: Test dynamo converters [Python]
     needs: [generate-matrix, build]
@@ -98,16 +52,14 @@ jobs:
         include:
           - repository: pytorch/tensorrt
             package-name: torch_tensorrt
-            pre-script: packaging/pre_build_script.sh
-    uses: pytorch/tensorrt/.github/workflows/linux-test.yml@main
+    uses: pytorch/tensorrt/.github/workflows/windows-test.yml@main
     with:
       job-name: tests-py-dynamo-converters
       repository: "pytorch/tensorrt"
       ref: ""
       test-infra-repository: pytorch/test-infra
       test-infra-ref: main
       build-matrix: ${{ needs.generate-matrix.outputs.matrix }}
-      pre-script: ${{ matrix.pre-script }}
       script: |
         export USE_HOST_DEPS=1
         pushd .
@@ -125,16 +77,14 @@ jobs:
         include:
           - repository: pytorch/tensorrt
             package-name: torch_tensorrt
-            pre-script: packaging/pre_build_script.sh
-    uses: pytorch/tensorrt/.github/workflows/linux-test.yml@main
+    uses: pytorch/tensorrt/.github/workflows/windows-test.yml@main
     with:
       job-name: tests-py-dynamo-fe
       repository: "pytorch/tensorrt"
       ref: ""
       test-infra-repository: pytorch/test-infra
       test-infra-ref: main
       build-matrix: ${{ needs.generate-matrix.outputs.matrix }}
-      pre-script: ${{ matrix.pre-script }}
       script: |
         export USE_HOST_DEPS=1
         pushd .
@@ -154,16 +104,14 @@ jobs:
         include:
           - repository: pytorch/tensorrt
             package-name: torch_tensorrt
-            pre-script: packaging/pre_build_script.sh
-    uses: pytorch/tensorrt/.github/workflows/linux-test.yml@main
+    uses: pytorch/tensorrt/.github/workflows/windows-test.yml@main
     with:
       job-name: tests-py-torch-compile-be
       repository: "pytorch/tensorrt"
       ref: ""
       test-infra-repository: pytorch/test-infra
       test-infra-ref: main
       build-matrix: ${{ needs.generate-matrix.outputs.matrix }}
-      pre-script: ${{ matrix.pre-script }}
       script: |
         export USE_HOST_DEPS=1
         pushd .
@@ -182,16 +130,14 @@ jobs:
         include:
           - repository: pytorch/tensorrt
             package-name: torch_tensorrt
-            pre-script: packaging/pre_build_script.sh
-    uses: pytorch/tensorrt/.github/workflows/linux-test.yml@main
+    uses: pytorch/tensorrt/.github/workflows/windows-test.yml@main
     with:
       job-name: tests-py-dynamo-core
       repository: "pytorch/tensorrt"
       ref: ""
       test-infra-repository: pytorch/test-infra
       test-infra-ref: main
       build-matrix: ${{ needs.generate-matrix.outputs.matrix }}
-      pre-script: ${{ matrix.pre-script }}
       script: |
         export USE_HOST_DEPS=1
         pushd .