shift_corr.so: shift_corr.cc shift_corr.cu.o
g++ $(CFLAGS) -o shift_corr.so shift_corr.cc shift_corr.cu.o $(LDFLAGS) -D RUN_GPU=$(USE_GPU)
shift_corr.cu.o: shift_corr.cu.cc
nvcc -arch=sm_60 -std=c++11 -c -o shift_corr.cu.o shift_corr.cu.cc $(TF_CFLAGS) -x cu -Xcompiler -fPIC --expt-relaxed-constexpr -D GOOGLE_CUDA=1