-
Notifications
You must be signed in to change notification settings - Fork 9
/
Copy pathbuild.sh
39 lines (33 loc) · 1.21 KB
/
build.sh
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
#!/bin/bash
# Configuration
# -gencode=arch=compute_70,code=sm_70 \
# CUDA_GENCODE="\
# -gencode=arch=compute_61,code=sm_61 \
# -gencode=arch=compute_60,code=sm_60 \
# -gencode=arch=compute_52,code=sm_52 \
# -gencode=arch=compute_50,code=sm_50"
CUDA_GENCODE="-arch=sm_30 \
-gencode=arch=compute_30,code=sm_30 \
-gencode=arch=compute_50,code=sm_50 \
-gencode=arch=compute_52,code=sm_52 \
-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_70,code=compute_70"
# CUDA_GENCODE="-arch=sm_30 \
# -gencode=arch=compute_30,code=sm_30 \
# -gencode=arch=compute_50,code=sm_50 \
# -gencode=arch=compute_52,code=sm_52 \
# -gencode=arch=compute_60,code=sm_60 \
# -gencode=arch=compute_61,code=sm_61 \
# -gencode=arch=compute_62,code=sm_62 "
# CUDA_GENCODE="-arch=sm_30 \
# -gencode=arch=compute_30,code=sm_30 \
# -gencode=arch=compute_50,code=sm_50 \
# -gencode=arch=compute_52,code=sm_52 \
# -gencode=arch=compute_60,code=sm_60 \
# -gencode=arch=compute_61,code=sm_61"
cd src
nvcc -I/usr/local/cuda/include --expt-extended-lambda -O3 -c -o bn.o bn.cu -x cu -Xcompiler -fPIC -std=c++11 ${CUDA_GENCODE}
cd ..