Fix Makefile in gpu/test (#843)
parent
dacd5a9f51
commit
ae09cabd6a
|
@ -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
|
||||
|
|
Loading…
Reference in New Issue