52 lines
1.5 KiB
Makefile
52 lines
1.5 KiB
Makefile
# Copyright (c) Facebook, Inc. and its affiliates.
|
|
#
|
|
# This source code is licensed under the MIT license found in the
|
|
# LICENSE file in the root directory of this source tree.
|
|
|
|
-include ../../makefile.inc
|
|
|
|
TESTS_SRC = TestGpuIndexFlat.cpp TestGpuIndexIVFPQ.cpp \
|
|
TestGpuIndexBinaryFlat.cpp TestGpuIndexIVFFlat.cpp TestGpuMemoryException.cpp
|
|
CUDA_TESTS_SRC = TestGpuSelect.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
|
|
|
|
run: $(TESTS_BIN) $(CUDA_TESTS_BIN)
|
|
for t in $(TESTS_BIN) $(CUDA_TESTS_BIN); do ./$$t || exit; done
|
|
|
|
$(CUDA_TESTS_OBJ): %.o: %.cu gtest
|
|
$(NVCC) $(NVCCFLAGS) -g -O3 -o $@ -c $< -Igtest/include
|
|
|
|
$(TESTS_OBJ): %.o: %.cpp gtest
|
|
$(CXX) $(CPPFLAGS) $(CXXFLAGS) $(CPUFLAGS) -o $@ -c $< -Igtest/include
|
|
|
|
$(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
|
|
$(CXX) -o $@ $^ $(LDFLAGS) $(LIBS)
|
|
|
|
demo_ivfpq_indexing_gpu.o: demo_ivfpq_indexing_gpu.cpp
|
|
$(CXX) $(CPPFLAGS) $(CXXFLAGS) $(CPUFLAGS) -o $@ -c $^
|
|
|
|
gtest/make/gtest.a: gtest
|
|
$(MAKE) -C gtest/make CXX="$(CXX)" CXXFLAGS="$(CXXFLAGS)" gtest.a
|
|
|
|
gtest:
|
|
curl -L https://github.com/google/googletest/archive/release-1.8.0.tar.gz | tar xz && \
|
|
mv googletest-release-1.8.0/googletest gtest && \
|
|
rm -rf googletest-release-1.8.0
|
|
|
|
clean:
|
|
rm -f *.o $(TESTS_BIN)
|
|
rm -rf gtest
|
|
rm -f demo_ivfpq_indexing_gpu
|
|
|
|
.PHONY: clean run
|