Skip to content

Commit

Permalink
fixes
Browse files Browse the repository at this point in the history
  • Loading branch information
h-vetinari committed May 12, 2020
1 parent c028135 commit 5e8fb54
Showing 1 changed file with 9 additions and 9 deletions.
18 changes: 9 additions & 9 deletions recipe/build-lib.sh
Original file line number Diff line number Diff line change
Expand Up @@ -5,17 +5,17 @@ if [ ${cuda_compiler_version} != "None" ]; then
# docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html#major-components__table-cuda-toolkit-driver-versions
# docs.nvidia.com/cuda/cuda-compiler-driver-nvcc/index.html#gpu-feature-list
if [ ${cuda_compiler_version} == "9.2" ]; then
CUDA_ARCH="-gencode=arch=compute_35,code=compute_35"
CUDA_ARCH="-gencode arch=compute_35,code=compute_35 "
fi
CUDA_ARCH="${CUDA_ARCH} -gencode=arch=compute_50,code=compute_50"
CUDA_ARCH="${CUDA_ARCH} -gencode=arch=compute_52,code=compute_52"
CUDA_ARCH="${CUDA_ARCH} -gencode=arch=compute_60,code=compute_60"
CUDA_ARCH="${CUDA_ARCH} -gencode=arch=compute_61,code=compute_61"
CUDA_ARCH="${CUDA_ARCH} -gencode=arch=compute_70,code=compute_70"
CUDA_ARCH="${CUDA_ARCH} -gencode=arch=compute_72,code=compute_72"
#CUDA_ARCH="${CUDA_ARCH}-gencode=arch=compute_50,code=compute_50 "
#CUDA_ARCH="${CUDA_ARCH}-gencode=arch=compute_52,code=compute_52 "
CUDA_ARCH="${CUDA_ARCH}-gencode=arch=compute_60,code=compute_60 "
CUDA_ARCH="${CUDA_ARCH}-gencode=arch=compute_61,code=compute_61 "
CUDA_ARCH="${CUDA_ARCH}-gencode=arch=compute_70,code=compute_70 "
CUDA_ARCH="${CUDA_ARCH}-gencode=arch=compute_72,code=compute_72 "
# compare part of cuda_compiler_version before the dot
if [ ${cuda_compiler_version%.*} == "10" ]; then
CUDA_ARCH="${CUDA_ARCH} -gencode=arch=compute_75,code=compute_75"
CUDA_ARCH="${CUDA_ARCH}-gencode=arch=compute_75,code=compute_75"
fi

# wrap CUDA_ARCH in quotes
Expand All @@ -28,7 +28,7 @@ echo $CUDA_CONFIG_ARG
# Build vanilla version (no avx)
./configure --prefix=${PREFIX} --exec-prefix=${PREFIX} \
--with-blas=-lblas --with-lapack=-llapack \
${CUDA_CONFIG_ARG}
${CUDA_CONFIG_ARG} && exit 1

echo $CUDA_CONFIG_ARG
# make sets SHAREDEXT correctly for linux/osx
Expand Down

0 comments on commit 5e8fb54

Please sign in to comment.