-
Notifications
You must be signed in to change notification settings - Fork 628
/
makefile
14 lines (11 loc) · 597 Bytes
/
makefile
1
2
3
4
5
6
7
8
9
10
11
12
13
14
# Modify PREFIX and NVCC_PREFIX based on your machine environment
PREFIX=/home/ubuntu/torch/install
NVCC_PREFIX=/usr/local/cuda-8.0/bin
CFLAGS=-I$(PREFIX)/include/THC -I$(PREFIX)/include/TH -I$(PREFIX)/include
LDFLAGS_NVCC=-L$(PREFIX)/lib -Xlinker -rpath,$(PREFIX)/lib -lluaT -lTHC -lTH -lpng -lluajit -lgomp
all: libcuda_utils.so
libcuda_utils.so: cuda_utils.cu
$(NVCC_PREFIX)/nvcc -arch sm_35 -O3 -DNDEBUG -Xcompiler -fopenmp --compiler-options '-fPIC' -o libcuda_utils.so --shared cuda_utils.cu $(CFLAGS) $(LDFLAGS_NVCC)
clean:
find . -type f | xargs -n 5 touch
rm -f libcuda_utils.so