Skip to content
Snippets Groups Projects
Commit c2c9b1b8 authored by Paul McCarthy's avatar Paul McCarthy :mountain_bicyclist:
Browse files

RF: Added support for CUDA 10.1

parent 5f6cbab7
No related branches found
No related tags found
No related merge requests found
Pipeline #5797 passed
...@@ -99,7 +99,7 @@ ifneq ($(shell which nvcc), ) ...@@ -99,7 +99,7 @@ ifneq ($(shell which nvcc), )
NVCC ?= nvcc NVCC ?= nvcc
CUDAVER ?= $(shell nvcc --version | grep -Po "release \K[0-9\.]+") CUDAVER ?= $(shell nvcc --version | grep -Po "release \K[0-9\.]+")
GENCODEFLAGS ?= $(shell ${FSLDIR}/config/common/supportedGencodes.sh ${CUDAVER}) GENCODEFLAGS ?= $(shell ${FSLCONFDIR}/common/supportedGencodes.sh ${CUDAVER})
CUDADIR ?= ${FSLDIR} CUDADIR ?= ${FSLDIR}
# nvcc-specific tweaks: # nvcc-specific tweaks:
......
...@@ -8,14 +8,6 @@ ...@@ -8,14 +8,6 @@
default_arch="-gencode arch=compute_20,code=sm_20 -gencode arch=compute_20,code=sm_21 -gencode arch=compute_30,code=sm_30 -gencode arch=compute_35,code=sm_35" default_arch="-gencode arch=compute_20,code=sm_20 -gencode arch=compute_20,code=sm_21 -gencode arch=compute_30,code=sm_30 -gencode arch=compute_35,code=sm_35"
vn=$1 vn=$1
default_allowed=$2
if [ "$default_allowed" = default_allowed ]; then
echo $default_arch
exit 0
else
exit 1
fi
if [ "$vn" = 5.5 ] if [ "$vn" = 5.5 ]
then then
...@@ -57,4 +49,11 @@ elif [ "$vn" = 10.0 ] ...@@ -57,4 +49,11 @@ elif [ "$vn" = 10.0 ]
then then
echo "-gencode arch=compute_30,code=sm_30 -gencode arch=compute_32,code=sm_32 -gencode arch=compute_35,code=sm_35 -gencode arch=compute_37,code=sm_37 -gencode arch=compute_50,code=sm_50 -gencode arch=compute_52,code=sm_52 -gencode arch=compute_53,code=sm_53 -gencode arch=compute_60,code=sm_60 -gencode arch=compute_61,code=sm_61 -gencode arch=compute_62,code=sm_62 -gencode arch=compute_70,code=sm_70 -gencode arch=compute_72,code=sm_72 -gencode arch=compute_75,code=sm_75" echo "-gencode arch=compute_30,code=sm_30 -gencode arch=compute_32,code=sm_32 -gencode arch=compute_35,code=sm_35 -gencode arch=compute_37,code=sm_37 -gencode arch=compute_50,code=sm_50 -gencode arch=compute_52,code=sm_52 -gencode arch=compute_53,code=sm_53 -gencode arch=compute_60,code=sm_60 -gencode arch=compute_61,code=sm_61 -gencode arch=compute_62,code=sm_62 -gencode arch=compute_70,code=sm_70 -gencode arch=compute_72,code=sm_72 -gencode arch=compute_75,code=sm_75"
exit 0 exit 0
elif [ "$vn" = 10.1 ]
then
echo "-gencode arch=compute_30,code=sm_30 -gencode arch=compute_32,code=sm_32 -gencode arch=compute_35,code=sm_35 -gencode arch=compute_37,code=sm_37 -gencode arch=compute_50,code=sm_50 -gencode arch=compute_52,code=sm_52 -gencode arch=compute_53,code=sm_53 -gencode arch=compute_60,code=sm_60 -gencode arch=compute_61,code=sm_61 -gencode arch=compute_62,code=sm_62 -gencode arch=compute_70,code=sm_70 -gencode arch=compute_72,code=sm_72 -gencode arch=compute_75,code=sm_75"
exit 0
else
echo $default_arch
exit 0
fi fi
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment