We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
2 parents 3b3d154 + 7e10fc8 commit 3419911Copy full SHA for 3419911
1 file changed
.github/workflows/release.yml
@@ -126,8 +126,9 @@ jobs:
126
env:
127
TORCHSORT_VERSION_SUFFIX: "+${{ matrix.include-combo.cuda-version }}-torch${{ matrix.include-combo.torch-version }}"
128
run: |
129
+ export PATH=/usr/local/cuda/bin:$PATH
130
export FORCE_CUDA=1
- export TORCH_CUDA_ARCH_LIST="3.5;5.0+PTX;6.0;7.0;7.5;8.0;8.6"
131
+ export TORCH_CUDA_ARCH_LIST="6.0;7.0;7.5;8.0;8.6+PTX"
132
TORCH_VERSION=`echo "pt${{ matrix.include-combo.torch-version }}" | sed "s/..$//" | sed "s/\.//g"`
133
CUDA_VERSION=`echo ${{ matrix.include-combo.cuda-version }}`
134
export TORCHSORT_VERSION_SUFFIX="+$TORCH_VERSION$CUDA_VERSION"
0 commit comments