Commit e636a6e4 authored by matthijs's avatar matthijs

fix gpu makefile

parent 00c288c0
......@@ -63,6 +63,7 @@ CUOBJ= impl/BroadcastSum.o \
utils/BlockSelectHalf.o \
utils/WarpSelectFloat.o \
utils/WarpSelectHalf.o \
utils/nvidia/fp16_emu.o \
$(foreach v,$(INS), \
utils/blockselect/BlockSelectHalf$(v).o \
utils/blockselect/BlockSelectFloat$(v).o \
......@@ -102,8 +103,8 @@ test/demo_ivfpq_indexing_gpu: test/demo_ivfpq_indexing_gpu.cpp \
# extension is .so even on the mac
../python/_swigfaiss_gpu.so: ../python/swigfaiss_gpu_wrap.cxx \
$(LIBNAME).a ../libfaiss.a
$(NVCC) $(NVCCFLAGS) $(PYTHONCFLAGS) -shared -o $@ $^ \
-Xcompiler -fopenmp -lcublas -I.. \
$(NVCC) -I.. $(NVCCFLAGS) $(PYTHONCFLAGS) -shared -o $@ $^ \
-Xcompiler -fopenmp -lcublas \
$(BLASLDFLAGSSONVCC)
../_swigfaiss_gpu.so: ../python/_swigfaiss_gpu.so
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment