From 136443e2e54f4aaf913b2578874695caa2408f07 Mon Sep 17 00:00:00 2001 From: Per Goncalves da Silva Date: Fri, 29 Mar 2019 14:45:11 +0100 Subject: [PATCH] Increases compute compatibility to 7.5 --- ci/docker/runtime_functions.sh | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/ci/docker/runtime_functions.sh b/ci/docker/runtime_functions.sh index f05fc6484290..ac5737fec5fb 100755 --- a/ci/docker/runtime_functions.sh +++ b/ci/docker/runtime_functions.sh @@ -24,8 +24,8 @@ set -ex NOSE_COVERAGE_ARGUMENTS="--with-coverage --cover-inclusive --cover-xml --cover-branches --cover-package=mxnet" NOSE_TIMER_ARGUMENTS="--with-timer --timer-ok 1 --timer-warning 15 --timer-filter warning,error" -CI_CUDA_COMPUTE_CAPABILITIES="-gencode=arch=compute_52,code=sm_52 -gencode=arch=compute_70,code=sm_70" -CI_CMAKE_CUDA_ARCH_BIN="52,70" +CI_CUDA_COMPUTE_CAPABILITIES="-gencode=arch=compute_52,code=sm_52 -gencode=arch=compute_75,code=sm_75" +CI_CMAKE_CUDA_ARCH_BIN="52,75" clean_repo() { set -ex