Skip to content
This repository was archived by the owner on Nov 17, 2023. It is now read-only.

Commit 136443e

Browse files
committed
Increases compute compatibility to 7.5
1 parent c01d5f7 commit 136443e

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

ci/docker/runtime_functions.sh

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -24,8 +24,8 @@ set -ex
2424

2525
NOSE_COVERAGE_ARGUMENTS="--with-coverage --cover-inclusive --cover-xml --cover-branches --cover-package=mxnet"
2626
NOSE_TIMER_ARGUMENTS="--with-timer --timer-ok 1 --timer-warning 15 --timer-filter warning,error"
27-
CI_CUDA_COMPUTE_CAPABILITIES="-gencode=arch=compute_52,code=sm_52 -gencode=arch=compute_70,code=sm_70"
28-
CI_CMAKE_CUDA_ARCH_BIN="52,70"
27+
CI_CUDA_COMPUTE_CAPABILITIES="-gencode=arch=compute_52,code=sm_52 -gencode=arch=compute_75,code=sm_75"
28+
CI_CMAKE_CUDA_ARCH_BIN="52,75"
2929

3030
clean_repo() {
3131
set -ex

0 commit comments

Comments
 (0)