Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
milvus
milvus
提交
4c491471
M
milvus
项目概览
milvus
/
milvus
大约 1 年 前同步成功
通知
261
Star
22476
Fork
2472
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
0
列表
看板
标记
里程碑
合并请求
0
DevOps
流水线
流水线任务
计划
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
M
milvus
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
0
Issue
0
列表
看板
标记
里程碑
合并请求
0
合并请求
0
Pages
DevOps
DevOps
流水线
流水线任务
计划
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
流水线任务
提交
Issue看板
体验新版 GitCode,发现更多精彩内容 >>
提交
4c491471
编写于
2月 24, 2021
作者:
X
xige-16
提交者:
yefu.chen
2月 24, 2021
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
Add release collection and release partition interface for query node
Signed-off-by:
N
xige-16
<
xi.ge@zilliz.com
>
上级
b02a9d6b
变更
12
隐藏空白更改
内联
并排
Showing
12 changed file
with
206 addition
and
2 deletion
+206
-2
docs/developer_guides/chap07_query_service.md
docs/developer_guides/chap07_query_service.md
+3
-0
internal/core/CMakeLists.txt
internal/core/CMakeLists.txt
+1
-0
internal/core/bench/CMakeLists.txt
internal/core/bench/CMakeLists.txt
+18
-0
internal/core/bench/bench_naive.cpp
internal/core/bench/bench_naive.cpp
+28
-0
internal/core/bench/bench_search.cpp
internal/core/bench/bench_search.cpp
+118
-0
internal/core/build-support/lint_exclusions.txt
internal/core/build-support/lint_exclusions.txt
+1
-0
internal/core/run_clang_format.sh
internal/core/run_clang_format.sh
+1
-0
internal/core/src/segcore/SegmentGrowing.h
internal/core/src/segcore/SegmentGrowing.h
+3
-0
internal/core/src/segcore/SegmentGrowingImpl.cpp
internal/core/src/segcore/SegmentGrowingImpl.cpp
+3
-2
internal/core/src/segcore/SegmentGrowingImpl.h
internal/core/src/segcore/SegmentGrowingImpl.h
+8
-0
internal/core/thirdparty/CMakeLists.txt
internal/core/thirdparty/CMakeLists.txt
+1
-0
internal/core/thirdparty/google_benchmark/CMakeLists.txt
internal/core/thirdparty/google_benchmark/CMakeLists.txt
+21
-0
未找到文件。
docs/developer_guides/chap07_query_service.md
浏览文件 @
4c491471
...
...
@@ -225,6 +225,9 @@ type QueryNode interface {
LoadSegments
(
req
LoadSegmentRequest
)
error
ReleaseSegments
(
req
ReleaseSegmentRequest
)
error
//DescribeParition(req DescribeParitionRequest) (PartitionDescriptions, error)
ReleaseCollection
(
req
ReleaseCollectionRequest
)
error
ReleasePartitions
(
req
ReleasePartitionRequest
)
error
}
```
...
...
internal/core/CMakeLists.txt
浏览文件 @
4c491471
...
...
@@ -185,6 +185,7 @@ if ( BUILD_UNIT_TEST STREQUAL "ON" )
append_flags
(
CMAKE_CXX_FLAGS FLAGS
"-DELPP_DISABLE_LOGS"
)
add_subdirectory
(
unittest
)
add_subdirectory
(
bench
)
endif
()
add_custom_target
(
Clean-All COMMAND
${
CMAKE_BUILD_TOOL
}
clean
)
...
...
internal/core/bench/CMakeLists.txt
0 → 100644
浏览文件 @
4c491471
include_directories
(
${
CMAKE_HOME_DIRECTORY
}
/src
)
include_directories
(
${
CMAKE_HOME_DIRECTORY
}
/unittest
)
include_directories
(
${
CMAKE_HOME_DIRECTORY
}
/src/index/knowhere
)
set
(
bench_srcs
bench_naive.cpp
bench_search.cpp
)
add_executable
(
all_bench
${
bench_srcs
}
)
target_link_libraries
(
all_bench
milvus_segcore
milvus_indexbuilder
log
pthread
)
target_link_libraries
(
all_bench benchmark::benchmark_main
)
internal/core/bench/bench_naive.cpp
0 → 100644
浏览文件 @
4c491471
// Copyright (C) 2019-2020 Zilliz. All rights reserved.
//
// Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance
// with the License. You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software distributed under the License
// is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express
// or implied. See the License for the specific language governing permissions and limitations under the License
#include <benchmark/benchmark.h>
#include <string>
static
void
BN_Naive_StringCreation
(
benchmark
::
State
&
state
)
{
for
(
auto
_
:
state
)
std
::
string
empty_string
;
}
// Register the function as a benchmark
BENCHMARK
(
BN_Naive_StringCreation
);
// Define another benchmark
static
void
BN_Naive_StringCopy
(
benchmark
::
State
&
state
)
{
std
::
string
x
=
"hello"
;
for
(
auto
_
:
state
)
std
::
string
copy
(
x
);
}
BENCHMARK
(
BN_Naive_StringCopy
);
internal/core/bench/bench_search.cpp
0 → 100644
浏览文件 @
4c491471
// Copyright (C) 2019-2020 Zilliz. All rights reserved.
//
// Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance
// with the License. You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software distributed under the License
// is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express
// or implied. See the License for the specific language governing permissions and limitations under the License
#include <cstdint>
#include <benchmark/benchmark.h>
#include <string>
#include "segcore/SegmentGrowing.h"
#include "segcore/SegmentSealed.h"
#include "test_utils/DataGen.h"
using
namespace
milvus
;
using
namespace
milvus
::
query
;
using
namespace
milvus
::
segcore
;
static
int
dim
=
128
;
static
int64_t
N
=
1024
*
1024
*
1
;
const
auto
schema
=
[]()
{
auto
schema
=
std
::
make_shared
<
Schema
>
();
schema
->
AddDebugField
(
"fakevec"
,
DataType
::
VECTOR_FLOAT
,
dim
,
MetricType
::
METRIC_L2
);
return
schema
;
}();
const
auto
dataset_
=
[]
{
auto
dataset_
=
DataGen
(
schema
,
N
);
return
dataset_
;
}();
const
auto
plan
=
[]
{
std
::
string
dsl
=
R"({
"bool": {
"must": [
{
"vector": {
"fakevec": {
"metric_type": "L2",
"params": {
"nprobe": 4
},
"query": "$0",
"topk": 5
}
}
}
]
}
})"
;
auto
plan
=
CreatePlan
(
*
schema
,
dsl
);
return
plan
;
}();
auto
ph_group
=
[]
{
auto
num_queries
=
5
;
auto
ph_group_raw
=
CreatePlaceholderGroup
(
num_queries
,
dim
,
1024
);
auto
ph_group
=
ParsePlaceholderGroup
(
plan
.
get
(),
ph_group_raw
.
SerializeAsString
());
return
ph_group
;
}();
static
void
Search_SmallIndex
(
benchmark
::
State
&
state
)
{
// schema->AddDebugField("age", DataType::FLOAT);
auto
is_small_index
=
state
.
range
(
0
);
auto
chunk_size
=
state
.
range
(
1
)
*
1024
;
auto
segment
=
CreateGrowingSegment
(
schema
,
chunk_size
);
if
(
!
is_small_index
)
{
segment
->
debug_disable_small_index
();
}
segment
->
PreInsert
(
N
);
ColumnBasedRawData
raw_data
;
raw_data
.
columns_
=
dataset_
.
cols_
;
raw_data
.
count
=
N
;
segment
->
Insert
(
0
,
N
,
dataset_
.
row_ids_
.
data
(),
dataset_
.
timestamps_
.
data
(),
raw_data
);
Timestamp
time
=
10000000
;
std
::
vector
<
const
PlaceholderGroup
*>
ph_group_arr
=
{
ph_group
.
get
()};
for
(
auto
_
:
state
)
{
auto
qr
=
segment
->
Search
(
plan
.
get
(),
ph_group_arr
.
data
(),
&
time
,
1
);
}
}
BENCHMARK
(
Search_SmallIndex
)
->
MinTime
(
5
)
->
ArgsProduct
({{
true
,
false
},
{
8
,
16
,
32
,
64
,
128
}});
static
void
Search_Sealed
(
benchmark
::
State
&
state
)
{
auto
segment
=
CreateSealedSegment
(
schema
);
SealedLoader
(
dataset_
,
*
segment
);
auto
choice
=
state
.
range
(
0
);
if
(
choice
==
0
)
{
// Brute Force
}
else
if
(
choice
==
1
)
{
// ivf
auto
vec
=
(
const
float
*
)
dataset_
.
cols_
[
0
].
data
();
auto
indexing
=
GenIndexing
(
N
,
dim
,
vec
);
LoadIndexInfo
info
;
info
.
index
=
indexing
;
info
.
index_params
[
"index_type"
]
=
"IVF"
;
info
.
index_params
[
"index_mode"
]
=
"CPU"
;
info
.
index_params
[
"metric_type"
]
=
MetricTypeToName
(
MetricType
::
METRIC_L2
);
segment
->
LoadIndex
(
info
);
}
Timestamp
time
=
10000000
;
std
::
vector
<
const
PlaceholderGroup
*>
ph_group_arr
=
{
ph_group
.
get
()};
for
(
auto
_
:
state
)
{
auto
qr
=
segment
->
Search
(
plan
.
get
(),
ph_group_arr
.
data
(),
&
time
,
1
);
}
}
BENCHMARK
(
Search_Sealed
)
->
MinTime
(
5
)
->
Arg
(
1
)
->
Arg
(
0
);
internal/core/build-support/lint_exclusions.txt
浏览文件 @
4c491471
...
...
@@ -5,4 +5,5 @@
*src/grpc*
*output*
*unittest*
*bench*
*src/pb*
internal/core/run_clang_format.sh
浏览文件 @
4c491471
...
...
@@ -12,5 +12,6 @@ formatThis() {
formatThis
"
${
CorePath
}
/src"
formatThis
"
${
CorePath
}
/unittest"
formatThis
"
${
CorePath
}
/bench"
${
CorePath
}
/build-support/add_license.sh
${
CorePath
}
/build-support/cpp_license.txt
${
CorePath
}
internal/core/src/segcore/SegmentGrowing.h
浏览文件 @
4c491471
...
...
@@ -50,6 +50,9 @@ class SegmentGrowing : public SegmentInternalInterface {
};
public:
virtual
void
debug_disable_small_index
()
=
0
;
virtual
int64_t
PreInsert
(
int64_t
size
)
=
0
;
...
...
internal/core/src/segcore/SegmentGrowingImpl.cpp
浏览文件 @
4c491471
...
...
@@ -188,9 +188,10 @@ SegmentGrowingImpl::do_insert(int64_t reserved_begin,
// NOTE: this must be the last step, cannot be put above
uid2offset_
.
insert
(
std
::
make_pair
(
row_id
,
reserved_begin
+
i
));
}
record_
.
ack_responder_
.
AddSegment
(
reserved_begin
,
reserved_begin
+
size
);
indexing_record_
.
UpdateResourceAck
(
record_
.
ack_responder_
.
GetAck
()
/
size_per_chunk_
,
record_
);
if
(
!
debug_disable_small_index_
)
{
indexing_record_
.
UpdateResourceAck
(
record_
.
ack_responder_
.
GetAck
()
/
size_per_chunk_
,
record_
);
}
}
Status
...
...
internal/core/src/segcore/SegmentGrowingImpl.h
浏览文件 @
4c491471
...
...
@@ -112,6 +112,11 @@ class SegmentGrowingImpl : public SegmentGrowing {
}
public:
void
debug_disable_small_index
()
override
{
debug_disable_small_index_
=
true
;
}
ssize_t
get_row_count
()
const
override
{
return
record_
.
ack_responder_
.
GetAck
();
...
...
@@ -204,6 +209,9 @@ class SegmentGrowingImpl : public SegmentGrowing {
SealedIndexingRecord
sealed_indexing_record_
;
tbb
::
concurrent_unordered_multimap
<
idx_t
,
int64_t
>
uid2offset_
;
private:
bool
debug_disable_small_index_
=
false
;
};
}
// namespace milvus::segcore
internal/core/thirdparty/CMakeLists.txt
浏览文件 @
4c491471
...
...
@@ -41,6 +41,7 @@ find_package( Threads REQUIRED )
# ****************************** Thirdparty googletest ***************************************
if
(
MILVUS_BUILD_TESTS
)
add_subdirectory
(
gtest
)
add_subdirectory
(
google_benchmark
)
endif
()
...
...
internal/core/thirdparty/google_benchmark/CMakeLists.txt
0 → 100644
浏览文件 @
4c491471
include
(
FetchContent
)
FetchContent_Declare
(
google_benchmark
URL https://github.com/google/benchmark/archive/v1.5.2.tar.gz
URL_MD5 084b34aceaeac11a6607d35220ca2efa
DOWNLOAD_DIR
${
THIRDPARTY_DOWNLOAD_PATH
}
SOURCE_DIR
${
CMAKE_CURRENT_BINARY_DIR
}
/google_benchmark
BINARY_DIR
${
CMAKE_CURRENT_BINARY_DIR
}
/google_benchmark
)
FetchContent_GetProperties
(
google_benchmark
)
if
(
NOT google_benchmark_POPULATED
)
FetchContent_Populate
(
google_benchmark
)
# Adding the following targets:
# gtest, gtest_main, gmock, gmock_main
message
(
"gb=
${
google_benchmark_SOURCE_DIR
}
"
)
add_subdirectory
(
${
google_benchmark_SOURCE_DIR
}
${
google_benchmark_BINARY_DIR
}
EXCLUDE_FROM_ALL
)
endif
()
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录