提交 66c3f988 编写于 作者: J jinhai

Merge branch 'branch-0.5.0' into 'branch-0.5.0'

clang-format

See merge request megasearch/milvus!734

Former-commit-id: 8c70c92aa02043599145e9e29af1c20ff6c6e187
...@@ -292,8 +292,7 @@ DBImpl::CreateIndex(const std::string& table_id, const TableIndex& index) { ...@@ -292,8 +292,7 @@ DBImpl::CreateIndex(const std::string& table_id, const TableIndex& index) {
std::vector<int> file_types; std::vector<int> file_types;
if (index.engine_type_ == static_cast<int32_t>(EngineType::FAISS_IDMAP)) { if (index.engine_type_ == static_cast<int32_t>(EngineType::FAISS_IDMAP)) {
file_types = { file_types = {
static_cast<int32_t>(meta::TableFileSchema::NEW), static_cast<int32_t>(meta::TableFileSchema::NEW), static_cast<int32_t>(meta::TableFileSchema::NEW_MERGE),
static_cast<int32_t>(meta::TableFileSchema::NEW_MERGE),
}; };
} else { } else {
file_types = { file_types = {
......
...@@ -55,9 +55,9 @@ MemTableFile::CreateTableFile() { ...@@ -55,9 +55,9 @@ MemTableFile::CreateTableFile() {
Status Status
MemTableFile::Add(const VectorSourcePtr& source, IDNumbers& vector_ids) { MemTableFile::Add(const VectorSourcePtr& source, IDNumbers& vector_ids) {
if (table_file_schema_.dimension_ <= 0) { if (table_file_schema_.dimension_ <= 0) {
std::string err_msg = std::string err_msg = "MemTableFile::Add: table_file_schema dimension = " +
"MemTableFile::Add: table_file_schema dimension = " + std::to_string(table_file_schema_.dimension_) + std::to_string(table_file_schema_.dimension_) + ", table_id = " +
", table_id = " + table_file_schema_.table_id_; table_file_schema_.table_id_;
ENGINE_LOG_ERROR << err_msg; ENGINE_LOG_ERROR << err_msg;
return Status(DB_ERROR, "Not able to create table file"); return Status(DB_ERROR, "Not able to create table file");
} }
......
...@@ -148,18 +148,15 @@ static const MetaSchema TABLES_SCHEMA(META_TABLES, { ...@@ -148,18 +148,15 @@ static const MetaSchema TABLES_SCHEMA(META_TABLES, {
}); });
// TableFiles schema // TableFiles schema
static const MetaSchema TABLEFILES_SCHEMA(META_TABLEFILES, { static const MetaSchema TABLEFILES_SCHEMA(
MetaField("id", "BIGINT", "PRIMARY KEY AUTO_INCREMENT"), META_TABLEFILES,
MetaField("table_id", "VARCHAR(255)", "NOT NULL"), {
MetaField("engine_type", "INT", "DEFAULT 1 NOT NULL"), MetaField("id", "BIGINT", "PRIMARY KEY AUTO_INCREMENT"), MetaField("table_id", "VARCHAR(255)", "NOT NULL"),
MetaField("file_id", "VARCHAR(255)", "NOT NULL"), MetaField("engine_type", "INT", "DEFAULT 1 NOT NULL"), MetaField("file_id", "VARCHAR(255)", "NOT NULL"),
MetaField("file_type", "INT", "DEFAULT 0 NOT NULL"), MetaField("file_type", "INT", "DEFAULT 0 NOT NULL"), MetaField("file_size", "BIGINT", "DEFAULT 0 NOT NULL"),
MetaField("file_size", "BIGINT", "DEFAULT 0 NOT NULL"), MetaField("row_count", "BIGINT", "DEFAULT 0 NOT NULL"), MetaField("updated_time", "BIGINT", "NOT NULL"),
MetaField("row_count", "BIGINT", "DEFAULT 0 NOT NULL"), MetaField("created_on", "BIGINT", "NOT NULL"), MetaField("date", "INT", "DEFAULT -1 NOT NULL"),
MetaField("updated_time", "BIGINT", "NOT NULL"), });
MetaField("created_on", "BIGINT", "NOT NULL"),
MetaField("date", "INT", "DEFAULT -1 NOT NULL"),
});
} // namespace } // namespace
......
...@@ -76,7 +76,7 @@ ConvertToDataset(std::vector<SPTAG::QueryResult> query_results) { ...@@ -76,7 +76,7 @@ ConvertToDataset(std::vector<SPTAG::QueryResult> query_results) {
auto p_id = (int64_t*)malloc(sizeof(int64_t) * elems); auto p_id = (int64_t*)malloc(sizeof(int64_t) * elems);
auto p_dist = (float*)malloc(sizeof(float) * elems); auto p_dist = (float*)malloc(sizeof(float) * elems);
// TODO: throw if malloc failed. // TODO: throw if malloc failed.
#pragma omp parallel for #pragma omp parallel for
for (auto i = 0; i < query_results.size(); ++i) { for (auto i = 0; i < query_results.size(); ++i) {
......
...@@ -151,16 +151,15 @@ class IVFTest : public DataGen, public TestWithParam<::std::tuple<std::string, P ...@@ -151,16 +151,15 @@ class IVFTest : public DataGen, public TestWithParam<::std::tuple<std::string, P
knowhere::IVFIndexPtr index_ = nullptr; knowhere::IVFIndexPtr index_ = nullptr;
}; };
INSTANTIATE_TEST_CASE_P(IVFParameters, IVFTest, INSTANTIATE_TEST_CASE_P(IVFParameters, IVFTest, Values(std::make_tuple("IVF", ParameterType::ivf),
Values(std::make_tuple("IVF", ParameterType::ivf), std::make_tuple("GPUIVF", ParameterType::ivf),
std::make_tuple("GPUIVF", ParameterType::ivf), std::make_tuple("IVFPQ", ParameterType::ivfpq),
std::make_tuple("IVFPQ", ParameterType::ivfpq), std::make_tuple("GPUIVFPQ", ParameterType::ivfpq),
std::make_tuple("GPUIVFPQ", ParameterType::ivfpq), std::make_tuple("IVFSQ", ParameterType::ivfsq),
std::make_tuple("IVFSQ", ParameterType::ivfsq),
#ifdef CUSTOMIZATION #ifdef CUSTOMIZATION
std::make_tuple("IVFSQHybrid", ParameterType::ivfsq), std::make_tuple("IVFSQHybrid", ParameterType::ivfsq),
#endif #endif
std::make_tuple("GPUIVFSQ", ParameterType::ivfsq))); std::make_tuple("GPUIVFSQ", ParameterType::ivfsq)));
void void
AssertAnns(const knowhere::DatasetPtr& result, const int& nq, const int& k) { AssertAnns(const knowhere::DatasetPtr& result, const int& nq, const int& k) {
......
...@@ -88,8 +88,8 @@ XBuildIndexTask::Load(milvus::scheduler::LoadType type, uint8_t device_id) { ...@@ -88,8 +88,8 @@ XBuildIndexTask::Load(milvus::scheduler::LoadType type, uint8_t device_id) {
size_t file_size = to_index_engine_->PhysicalSize(); size_t file_size = to_index_engine_->PhysicalSize();
std::string info = "Load file id:" + std::to_string(file_->id_) + std::string info = "Load file id:" + std::to_string(file_->id_) + " file type:" +
" file type:" + std::to_string(file_->file_type_) + " size:" + std::to_string(file_size) + std::to_string(file_->file_type_) + " size:" + std::to_string(file_size) +
" bytes from location: " + file_->location_ + " totally cost"; " bytes from location: " + file_->location_ + " totally cost";
double span = rc.ElapseFromBegin(info); double span = rc.ElapseFromBegin(info);
......
...@@ -511,8 +511,8 @@ InsertTask::OnExecute() { ...@@ -511,8 +511,8 @@ InsertTask::OnExecute() {
} }
// step 6: update table flag // step 6: update table flag
user_provide_ids ? table_info.flag_ |= engine::meta::FLAG_MASK_HAS_USERID user_provide_ids ? table_info.flag_ |= engine::meta::FLAG_MASK_HAS_USERID : table_info.flag_ |=
: table_info.flag_ |= engine::meta::FLAG_MASK_NO_USERID; engine::meta::FLAG_MASK_NO_USERID;
status = DBWrapper::DB()->UpdateTableFlag(insert_param_->table_name(), table_info.flag_); status = DBWrapper::DB()->UpdateTableFlag(insert_param_->table_name(), table_info.flag_);
#ifdef MILVUS_ENABLE_PROFILING #ifdef MILVUS_ENABLE_PROFILING
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册