"vscode:/vscode.git/clone" did not exist on "6dea5c96bf714610bdb55ac361fa6bf168931a5e"
Commit 504da5eb authored by rusty1s's avatar rusty1s
Browse files

add TORCH_CUDA_ARCH_LIST to building wheels

parent 76d9d051
......@@ -62,7 +62,7 @@ jobs:
if: ${{ matrix.cuda-version != 'cpu' }}
run: |
source .github/workflows/cuda/${{ matrix.cuda-version }}-${{ runner.os }}-env.sh
FORCE_CUDA=1 TORCH_CUDA_ARCH_LIST="5.2 6.0 6.1 7.0 7.5 8.0 8.6+PTX" ./conda/pytorch-scatter/build_conda.sh ${{ matrix.python-version }} ${{ matrix.torch-version }} ${{ matrix.cuda-version }}
FORCE_CUDA=1 ./conda/pytorch-scatter/build_conda.sh ${{ matrix.python-version }} ${{ matrix.torch-version }} ${{ matrix.cuda-version }}
shell:
bash -l {0}
......
......@@ -25,6 +25,7 @@ build:
script: pip install .
script_env:
- FORCE_CUDA
- TORCH_CUDA_ARCH_LIST="5.2 6.0 6.1 7.0 7.5 8.0 8.6+PTX"
test:
imports:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment