[REFACTOR][CI] Clean up GitHub Actions platform build workflow

Author: mshr-h

.github/actions/setup/action.yml

@@ -35,70 +35,83 @@ concurrency:
   cancel-in-progress: true
 
 jobs:
-  MacOS:
+  Platform:
     if: ${{ github.repository == 'apache/tvm' }}
-    runs-on: macOS-latest
+    strategy:
+      matrix:
+        include:
+          - name: MacOS
+            os: macOS-latest
+          - name: Windows
+            os: windows-latest
+    runs-on: ${{ matrix.os }}
+    name: ${{ matrix.name }}
+    env:
+      UV_PYTHON: ${{ matrix.os == 'windows-latest' && '.venv/Scripts/python.exe' || '.venv/bin/python' }}
     steps:
       - uses: actions/checkout@de0fac2e4500dabe0009e67214ff5f5447ce83dd # v6.0.2
         with:
           submodules: 'recursive'
-      - name: Set up environment
-        uses: ./.github/actions/setup
-      - name: Install LLVM dependencies
-        shell: bash -l {0}
+      - uses: astral-sh/setup-uv@v7
+        with:
+          python-version: 3.10
+          enable-cache: true
+          cache-python: true
+
+      - name: Create venv
+        shell: pwsh
         run: |
-          conda install -c conda-forge llvmdev cmake ninja zlib
-      - name: Build TVM wheel
-        shell: bash -l {0}
+          uv venv --python 3.10
+
+      - name: Install tvm-ffi package
         run: |
-          pip install scikit-build-core
-          export CMAKE_ARGS="-DUSE_LLVM=ON -DBUILD_TESTING=OFF"
-          pip wheel --no-deps -w dist . -v
-      - name: Install TVM from wheel
+          uv pip install --verbose ./3rdparty/tvm-ffi
+
+      - name: Install LLVM dependencies (MacOS)
+        if: matrix.os == 'macOS-latest'
         shell: bash -l {0}
         run: |
-          pip install dist/*.whl
+          brew install llvm cmake ninja zlib
 
-      - name: Test
-        shell: bash -l {0}
-        run: >-
-          python -m pytest -v tests/python/all-platform-minimal-test
-      - name: Minimal Metal Compile-Only
-        shell: bash -l {0}
+      - name: Install LLVM dependencies (Windows)
+        if: matrix.os == 'windows-latest'
+        shell: pwsh
         run: |
-          python -m pytest -v -s 'tests/python/codegen/test_gpu_codegen_allreduce.py::test_allreduce_sum_compile'
-          python -m pytest -v -s 'tests/python/codegen/test_target_codegen_metal.py::test_func_with_trailing_pod_params'
-      - name: Minimal Metal Compile-and-Run
-        shell: bash -l {0}
-        run: >-
-          python -m pytest -v -s 'tests/python/codegen/test_target_codegen_metal.py'
-          python -m pytest -v -s 'tests/python/codegen/test_target_codegen_gpu_common.py'
-          python -m pytest -v -s 'tests/python/codegen/test_gpu_codegen_allreduce.py::test_allreduce_sum[dims0-metal]'
+          choco install llvm cmake ninja libxml2 -y --no-progress
 
-  Windows:
-    if: ${{ github.repository == 'apache/tvm' }}
-    runs-on: windows-latest
-    steps:
-      - uses: actions/checkout@de0fac2e4500dabe0009e67214ff5f5447ce83dd # v6.0.2
-        with:
-          submodules: 'recursive'
-      - name: Set up environment
-        uses: ./.github/actions/setup
-      - name: Install LLVM dependencies
-        shell: cmd /C call {0}
+      - name: Install TVM (MacOS)
+        if: matrix.os == 'macOS-latest'
+        shell: bash -l {0}
         run: |
-          conda install -c conda-forge llvmdev cmake ninja zlib libxml2-devel
-      - name: Install TVM
-        shell: cmd /C call {0}
+          uv pip install --verbose scikit-build-core
+          export CMAKE_ARGS="-G Ninja -DUSE_LLVM=ON -DBUILD_TESTING=OFF"
+          uv pip install --no-deps . --verbose
+
+      - name: Install TVM (Windows)
+        if: matrix.os == 'windows-latest'
+        shell: pwsh
         run: |
-          pip install scikit-build-core
-          set CMAKE_ARGS=-DUSE_LLVM=ON -DBUILD_TESTING=OFF
-          pip install --no-deps . -v
+          uv pip install --verbose scikit-build-core
+          $env:CMAKE_ARGS = "-G Ninja -DUSE_LLVM=ON -DBUILD_TESTING=OFF"
+          uv pip install --no-deps . --verbose
+
       - name: Install test dependencies
-        shell: cmd /C call {0}
         run: |
-          pip install psutil cloudpickle ml_dtypes numpy packaging scipy tornado typing_extensions
-      - name: Test
-        shell: cmd /C call {0}
+          uv pip install --verbose psutil cloudpickle ml_dtypes numpy packaging scipy tornado typing_extensions
+
+      - name: Platform minimal test
+        run: >-
+          uv run pytest -v tests/python/all-platform-minimal-test
+
+      - name: Minimal Metal Compile-Only
+        if: matrix.os == 'macOS-latest'
+        run: |
+          uv run pytest -v -s 'tests/python/codegen/test_gpu_codegen_allreduce.py::test_allreduce_sum_compile'
+          uv run pytest -v -s 'tests/python/codegen/test_target_codegen_metal.py::test_func_with_trailing_pod_params'
+
+      - name: Minimal Metal Compile-and-Run
+        if: matrix.os == 'macOS-latest'
         run: >-
-          python -m pytest -v tests/python/all-platform-minimal-test
+          uv run pytest -v -s 'tests/python/codegen/test_target_codegen_metal.py'
+          uv run pytest -v -s 'tests/python/codegen/test_target_codegen_gpu_common.py'
+          uv run pytest -v -s 'tests/python/codegen/test_gpu_codegen_allreduce.py::test_allreduce_sum[dims0-metal]'