@@ -2,6 +2,9 @@ name: Building Wheels
22
33on : [workflow_dispatch]
44
5+ defaults :
6+ shell: bash
7+
58jobs :
69
710 wheel:
9295 echo "New version name: $VERSION+$TORCH_VERSION$CUDA_VERSION"
9396 sed -i "s/$VERSION/$VERSION+$TORCH_VERSION$CUDA_VERSION/" setup.py
9497 sed -i "s/$VERSION/$VERSION+$TORCH_VERSION$CUDA_VERSION/" torch_sparse/__init__.py
95- shell :
96- bash
9798
9899 - name: Install torch-scatter
99100 run: |
@@ -103,26 +104,30 @@ jobs:
103104 if: ${{ runner.os != 'Windows' }}
104105 run: |
105106 bash .github/workflows/metis.sh
107+ echo WITH_METIS=1 >> $GITHUB_ENV
106108
107109 - name: Install METIS on Windows
108110 if: ${{ runner.os == 'Windows' }}
109111 run: |
110- bash .github/workflows/metis-${{ runner.os }}.sh
112+ # bash .github/workflows/metis-${{ runner.os }}.sh
113+ echo WITH_METIS=0 >> $GITHUB_ENV
111114
112115 - name: Build wheel for CPU
113116 if: ${{ matrix.cuda-version == 'cpu' }}
114117 run: |
115- FORCE_ONLY_CPU=1 WITH_METIS=1 python setup.py bdist_wheel --dist-dir=dist
116- shell :
117- bash
118+ python setup.py bdist_wheel --dist-dir=dist
119+ env :
120+ FORCE_ONLY_CPU: 1
121+ WITH_METIS: ${{ env.WITH_METIS }}
118122
119123 - name: Build wheel for GPU
120124 if: ${{ matrix.cuda-version != 'cpu' }}
121125 run: |
122126 source .github/workflows/cuda/${{ matrix.cuda-version }}-${{ runner.os }}-env.sh
123- FORCE_CUDA=1 WITH_METIS=1 python setup.py bdist_wheel --dist-dir=dist
124- shell :
125- bash
127+ python setup.py bdist_wheel --dist-dir=dist
128+ env :
129+ FORCE_CUDA: 1
130+ WITH_METIS: ${{ env.WITH_METIS }}
126131
127132 - name: Configure AWS
128133 uses: aws-actions/configure-aws-credentials@v1
0 commit comments