Fix Makefile in gpu/test (#843)

pull/849/head
Lucas Hosseini 2019-05-29 14:32:05 +02:00 committed by GitHub
parent dacd5a9f51
commit ae09cabd6a
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 8 additions and 6 deletions

View File

@ -9,8 +9,10 @@ TESTS_SRC = TestGpuIndexFlat.cpp TestGpuIndexIVFPQ.cpp \
TestGpuIndexBinaryFlat.cpp TestGpuIndexIVFFlat.cpp TestGpuMemoryException.cpp
CUDA_TESTS_SRC = TestGpuSelect.cu
TESTS_BIN = $(TESTS_SRC:.cpp=)
CUDA_TESTS_BIN = $(CUDA_TESTS_SRC:.cu=)
TESTS_OBJ = $(TESTS_SRC:.cpp=.o)
CUDA_TESTS_OBJ = $(CUDA_TESTS_SRC:.cu=.o)
TESTS_BIN = $(TESTS_OBJ:.o=) $(CUDA_TESTS_OBJ:.o=)
# test_gpu_index.py test_pytorch_faiss.py
@ -18,13 +20,13 @@ CUDA_TESTS_BIN = $(CUDA_TESTS_SRC:.cu=)
run: $(TESTS_BIN) $(CUDA_TESTS_BIN)
for t in $(TESTS_BIN) $(CUDA_TESTS_BIN); do ./$$t || exit; done
$(CUDA_TESTS_SRC:.cu=.o): %.o: %.cu gtest
$(CUDA_TESTS_OBJ): %.o: %.cu gtest
$(NVCC) $(NVCCFLAGS) -g -O3 -o $@ -c $< -Igtest/include
$(CUDA_TESTS_SRC:.cpp=.o): %.o: %.cpp gtest
$(TESTS_OBJ): %.o: %.cpp gtest
$(CXX) $(CPPFLAGS) $(CXXFLAGS) $(CPUFLAGS) -o $@ -c $< -Igtest/include
%: %.o TestUtils.o ../../libfaiss.a gtest/make/gtest.a
$(TESTS_BIN): %: %.o TestUtils.o ../../libfaiss.a gtest/make/gtest.a
$(CXX) -o $@ $^ $(LDFLAGS) $(LIBS)
demo_ivfpq_indexing_gpu: demo_ivfpq_indexing_gpu.o ../../libfaiss.a
@ -46,4 +48,4 @@ clean:
rm -rf gtest
rm -f demo_ivfpq_indexing_gpu
.PHONY: clean
.PHONY: clean run