Skip to content

Commit 7542f6e

Browse files
committed
update
1 parent 71ea35a commit 7542f6e

File tree

1 file changed

+14
-9
lines changed

1 file changed

+14
-9
lines changed

.github/workflows/building.yml

Lines changed: 14 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -2,6 +2,9 @@ name: Building Wheels
22

33
on: [workflow_dispatch]
44

5+
defaults:
6+
shell: bash
7+
58
jobs:
69

710
wheel:
@@ -92,8 +95,6 @@ jobs:
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

Comments
 (0)