Skip to content

Commit 3419911

Browse files
authored
Merge pull request #94 from teddykoker/fix-cuda-ci
Ensure cuda compilation in CI build
2 parents 3b3d154 + 7e10fc8 commit 3419911

1 file changed

Lines changed: 2 additions & 1 deletion

File tree

.github/workflows/release.yml

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -126,8 +126,9 @@ jobs:
126126
env:
127127
TORCHSORT_VERSION_SUFFIX: "+${{ matrix.include-combo.cuda-version }}-torch${{ matrix.include-combo.torch-version }}"
128128
run: |
129+
export PATH=/usr/local/cuda/bin:$PATH
129130
export FORCE_CUDA=1
130-
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"
131132
TORCH_VERSION=`echo "pt${{ matrix.include-combo.torch-version }}" | sed "s/..$//" | sed "s/\.//g"`
132133
CUDA_VERSION=`echo ${{ matrix.include-combo.cuda-version }}`
133134
export TORCHSORT_VERSION_SUFFIX="+$TORCH_VERSION$CUDA_VERSION"

0 commit comments

Comments
 (0)