Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
BaiXuePrincess
milvus
提交
118c8c90
milvus
项目概览
BaiXuePrincess
/
milvus
与 Fork 源项目一致
从无法访问的项目Fork
通知
7
Star
4
Fork
0
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
0
列表
看板
标记
里程碑
合并请求
0
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
milvus
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
0
Issue
0
列表
看板
标记
里程碑
合并请求
0
合并请求
0
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
体验新版 GitCode,发现更多精彩内容 >>
提交
118c8c90
编写于
10月 11, 2019
作者:
X
xiaojun.lin
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
MS-635 coding style check
Former-commit-id: 6803034828abc15f882b05cad19425a4b86db10c
上级
cf8931a9
变更
4
显示空白变更内容
内联
并排
Showing
4 changed file
with
7 addition
and
14 deletion
+7
-14
cpp/src/core/knowhere/knowhere/index/vector_index/IndexIVFSQHybrid.cpp
...knowhere/knowhere/index/vector_index/IndexIVFSQHybrid.cpp
+1
-8
cpp/src/core/unittest/test_ivf.cpp
cpp/src/core/unittest/test_ivf.cpp
+1
-2
cpp/src/wrapper/VecIndex.cpp
cpp/src/wrapper/VecIndex.cpp
+4
-3
cpp/src/wrapper/VecIndex.h
cpp/src/wrapper/VecIndex.h
+1
-1
未找到文件。
cpp/src/core/knowhere/knowhere/index/vector_index/IndexIVFSQHybrid.cpp
浏览文件 @
118c8c90
...
@@ -231,17 +231,14 @@ IVFSQHybrid::LoadQuantizer(const Config& conf) {
...
@@ -231,17 +231,14 @@ IVFSQHybrid::LoadQuantizer(const Config& conf) {
void
void
IVFSQHybrid
::
SetQuantizer
(
const
QuantizerPtr
&
q
)
{
IVFSQHybrid
::
SetQuantizer
(
const
QuantizerPtr
&
q
)
{
}
}
void
void
IVFSQHybrid
::
UnsetQuantizer
()
{
IVFSQHybrid
::
UnsetQuantizer
()
{
}
}
void
void
IVFSQHybrid
::
LoadData
(
const
knowhere
::
QuantizerPtr
&
q
,
const
Config
&
conf
)
{
IVFSQHybrid
::
LoadData
(
const
knowhere
::
QuantizerPtr
&
q
,
const
Config
&
conf
)
{
}
}
IndexModelPtr
IndexModelPtr
...
@@ -260,11 +257,7 @@ IVFSQHybrid::CopyCpuToGpu(const int64_t& device_id, const Config& config) {
...
@@ -260,11 +257,7 @@ IVFSQHybrid::CopyCpuToGpu(const int64_t& device_id, const Config& config) {
}
}
void
void
IVFSQHybrid
::
search_impl
(
int64_t
n
,
IVFSQHybrid
::
search_impl
(
int64_t
n
,
const
float
*
data
,
int64_t
k
,
float
*
distances
,
int64_t
*
labels
,
const
float
*
data
,
int64_t
k
,
float
*
distances
,
int64_t
*
labels
,
const
Config
&
cfg
)
{
const
Config
&
cfg
)
{
GPUIVF
::
search_impl
(
n
,
data
,
k
,
distances
,
labels
,
cfg
);
GPUIVF
::
search_impl
(
n
,
data
,
k
,
distances
,
labels
,
cfg
);
}
}
...
...
cpp/src/core/unittest/test_ivf.cpp
浏览文件 @
118c8c90
...
@@ -160,8 +160,7 @@ INSTANTIATE_TEST_CASE_P(IVFParameters, IVFTest,
...
@@ -160,8 +160,7 @@ INSTANTIATE_TEST_CASE_P(IVFParameters, IVFTest,
#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
)
{
...
...
cpp/src/wrapper/VecIndex.cpp
浏览文件 @
118c8c90
...
@@ -15,6 +15,8 @@
...
@@ -15,6 +15,8 @@
// specific language governing permissions and limitations
// specific language governing permissions and limitations
// under the License.
// under the License.
#include "wrapper/VecIndex.h"
#include "VecImpl.h"
#include "knowhere/common/Exception.h"
#include "knowhere/common/Exception.h"
#include "knowhere/index/vector_index/IndexGPUIVF.h"
#include "knowhere/index/vector_index/IndexGPUIVF.h"
#include "knowhere/index/vector_index/IndexGPUIVFPQ.h"
#include "knowhere/index/vector_index/IndexGPUIVFPQ.h"
...
@@ -26,8 +28,6 @@
...
@@ -26,8 +28,6 @@
#include "knowhere/index/vector_index/IndexIVFSQHybrid.h"
#include "knowhere/index/vector_index/IndexIVFSQHybrid.h"
#include "knowhere/index/vector_index/IndexKDT.h"
#include "knowhere/index/vector_index/IndexKDT.h"
#include "knowhere/index/vector_index/IndexNSG.h"
#include "knowhere/index/vector_index/IndexNSG.h"
#include "wrapper/VecIndex.h"
#include "VecImpl.h"
#include "utils/Log.h"
#include "utils/Log.h"
#include <cuda.h>
#include <cuda.h>
...
@@ -161,7 +161,8 @@ GetVecIndexFactory(const IndexType& type, const Config& cfg) {
...
@@ -161,7 +161,8 @@ GetVecIndexFactory(const IndexType& type, const Config& cfg) {
VecIndexPtr
VecIndexPtr
LoadVecIndex
(
const
IndexType
&
index_type
,
const
knowhere
::
BinarySet
&
index_binary
)
{
LoadVecIndex
(
const
IndexType
&
index_type
,
const
knowhere
::
BinarySet
&
index_binary
)
{
auto
index
=
GetVecIndexFactory
(
index_type
);
auto
index
=
GetVecIndexFactory
(
index_type
);
if
(
index
==
nullptr
)
return
nullptr
;
if
(
index
==
nullptr
)
return
nullptr
;
// else
// else
index
->
Load
(
index_binary
);
index
->
Load
(
index_binary
);
return
index
;
return
index
;
...
...
cpp/src/wrapper/VecIndex.h
浏览文件 @
118c8c90
...
@@ -20,10 +20,10 @@
...
@@ -20,10 +20,10 @@
#include <memory>
#include <memory>
#include <string>
#include <string>
#include "cache/DataObj.h"
#include "knowhere/common/BinarySet.h"
#include "knowhere/common/BinarySet.h"
#include "knowhere/common/Config.h"
#include "knowhere/common/Config.h"
#include "knowhere/index/vector_index/Quantizer.h"
#include "knowhere/index/vector_index/Quantizer.h"
#include "cache/DataObj.h"
#include "utils/Status.h"
#include "utils/Status.h"
namespace
milvus
{
namespace
milvus
{
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录