milvus-io
/
milvus
mirror of https://github.com/milvus-io/milvus.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

redefine default index build type 

Former-commit-id: 1f10a7774b80b596323882bb74a46484a917344f
pull/191/head
xj.lin 2019-05-11 11:24:39 +08:00
parent 876f6a4c66
commit 364af3b1a8
3 changed files with 10 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
cpp/CMakeLists.txt
View File

@ -69,7 +69,7 @@ link_directories(${VECWISE_THIRD_PARTY_BUILD}/lib)
# WORKING_DIRECTORY ${CMAKE_CURRENT_SOURCE_DIR}/third_party)
add_subdirectory(src)
#add_subdirectory(test_client)
add_subdirectory(test_client)
if (BUILD_UNIT_TEST)
add_subdirectory(${CMAKE_CURRENT_SOURCE_DIR}/unittest)

9
cpp/src/db/DB.cpp
View File

@ -19,7 +19,14 @@ DB::~DB() {}
void DB::Open(const Options& options, DB** dbptr) {
*dbptr = nullptr;
*dbptr = DBFactory::Build(options);
#ifdef GPU_VERSION
std::string default_index_type{"Faiss,IVF"};
#else
std::string default_index_type{"Faiss,IDMap"};
#endif
*dbptr = DBFactory::Build(options, default_index_type);
return;
}

2
cpp/unittest/server/CMakeLists.txt
View File

@ -7,7 +7,7 @@ include_directories(../../src)
include_directories(/usr/include)
include_directories(/usr/local/cuda/include)
find_library(cuda_library cudart cublas HINTS /usr/local/cuda/lib64)
link_directories(/usr/local/cuda/lib64)
aux_source_directory(../../src/config config_files)
aux_source_directory(../../src/cache cache_srcs)