Add Windows build to workflow

.github/workflows/build.yaml

@@ -39,8 +39,8 @@ jobs:
     
     strategy:
       matrix:
-        os: [ubuntu-20.04]
-        pyver: ["3.8", "3.9", "3.10"]
+        os: [ubuntu-20.04, windows-latest]
+        pyver: ["3.8", "3.9", "3.10", "3.11"]
         cuda: ["11.8"]
     defaults:
       run:
@@ -73,6 +73,11 @@ jobs:
           conda install pytorch "pytorch-cuda=${env:CUDA_VERSION}" -c pytorch -c nvidia
           python -m pip install --upgrade build setuptools wheel ninja
 
+          # Environment variables
+          Add-Content $env:GITHUB_ENV "CUDA_PATH=$env:CONDA_PREFIX"
+          Add-Content $env:GITHUB_ENV "CUDA_HOME=$env:CONDA_PREFIX"
+          if ($IsLinux) {$env:LD_LIBRARY_PATH = $env:CONDA_PREFIX + '/lib:' + $env:LD_LIBRARY_PATH}
+
           # Print version information
           python --version
           python -c "import torch; print('PyTorch:', torch.__version__)"
@@ -81,11 +86,6 @@ jobs:
 
       - name: Build Wheel
         run: |
-          $env:CUDA_PATH = $env:CONDA_PREFIX
-          $env:CUDA_HOME = $env:CONDA_PREFIX
-          if ($IsLinux) {$env:LD_LIBRARY_PATH = $env:CONDA_PREFIX + '/lib:' + $env:LD_LIBRARY_PATH}
-          $env:TORCH_CUDA_ARCH_LIST = '8.0 8.6 8.9 9.0+PTX'
-
           python setup.py sdist bdist_wheel
       
       - name: Upload Assets