Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
BaiXuePrincess
milvus
提交
0fb88f73
milvus
项目概览
BaiXuePrincess
/
milvus
与 Fork 源项目一致
Fork自
milvus / milvus
通知
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,发现更多精彩内容 >>
未验证
提交
0fb88f73
编写于
6月 24, 2020
作者:
S
shengjun.li
提交者:
GitHub
6月 24, 2020
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
fix too many copies (#2660)
Signed-off-by:
N
shengjun.li
<
shengjun.li@zilliz.com
>
上级
ed012f29
变更
3
隐藏空白更改
内联
并排
Showing
3 changed file
with
7 addition
and
18 deletion
+7
-18
core/src/index/knowhere/knowhere/index/vector_index/gpu/IndexGPUIVF.cpp
.../knowhere/knowhere/index/vector_index/gpu/IndexGPUIVF.cpp
+4
-7
core/src/index/knowhere/knowhere/index/vector_index/gpu/IndexGPUIVFPQ.cpp
...nowhere/knowhere/index/vector_index/gpu/IndexGPUIVFPQ.cpp
+1
-4
core/src/index/knowhere/knowhere/index/vector_index/gpu/IndexIVFSQHybrid.cpp
...here/knowhere/index/vector_index/gpu/IndexIVFSQHybrid.cpp
+2
-7
未找到文件。
core/src/index/knowhere/knowhere/index/vector_index/gpu/IndexGPUIVF.cpp
浏览文件 @
0fb88f73
...
...
@@ -40,14 +40,11 @@ GPUIVF::Train(const DatasetPtr& dataset_ptr, const Config& config) {
idx_config
.
device
=
gpu_id_
;
int32_t
nlist
=
config
[
IndexParams
::
nlist
];
faiss
::
MetricType
metric_type
=
GetMetricType
(
config
[
Metric
::
TYPE
].
get
<
std
::
string
>
());
faiss
::
gpu
::
GpuIndexIVFFlat
device_index
(
gpu_res
->
faiss_res
.
get
(),
dim
,
nlist
,
metric_type
,
idx_config
);
device_index
.
train
(
rows
,
(
float
*
)
p_data
);
auto
device_index
=
new
faiss
::
gpu
::
GpuIndexIVFFlat
(
gpu_res
->
faiss_res
.
get
(),
dim
,
nlist
,
metric_type
,
idx_config
);
device_index
->
train
(
rows
,
(
float
*
)
p_data
);
std
::
shared_ptr
<
faiss
::
Index
>
host_index
=
nullptr
;
host_index
.
reset
(
faiss
::
gpu
::
index_gpu_to_cpu
(
&
device_index
));
auto
device_index1
=
faiss
::
gpu
::
index_cpu_to_gpu
(
gpu_res
->
faiss_res
.
get
(),
gpu_id_
,
host_index
.
get
());
index_
.
reset
(
device_index1
);
index_
.
reset
(
device_index
);
res_
=
gpu_res
;
}
else
{
KNOWHERE_THROW_MSG
(
"Build IVF can't get gpu resource"
);
...
...
core/src/index/knowhere/knowhere/index/vector_index/gpu/IndexGPUIVFPQ.cpp
浏览文件 @
0fb88f73
...
...
@@ -38,11 +38,8 @@ GPUIVFPQ::Train(const DatasetPtr& dataset_ptr, const Config& config) {
config
[
IndexParams
::
m
],
config
[
IndexParams
::
nbits
],
GetMetricType
(
config
[
Metric
::
TYPE
].
get
<
std
::
string
>
()));
// IP not support
device_index
->
train
(
rows
,
(
float
*
)
p_data
);
std
::
shared_ptr
<
faiss
::
Index
>
host_index
=
nullptr
;
host_index
.
reset
(
faiss
::
gpu
::
index_gpu_to_cpu
(
device_index
));
auto
device_index1
=
faiss
::
gpu
::
index_cpu_to_gpu
(
gpu_res
->
faiss_res
.
get
(),
gpu_id_
,
host_index
.
get
());
index_
.
reset
(
device_index1
);
index_
.
reset
(
device_index
);
res_
=
gpu_res
;
}
else
{
KNOWHERE_THROW_MSG
(
"Build IVFPQ can't get gpu resource"
);
...
...
core/src/index/knowhere/knowhere/index/vector_index/gpu/IndexIVFSQHybrid.cpp
浏览文件 @
0fb88f73
...
...
@@ -45,19 +45,14 @@ IVFSQHybrid::Train(const DatasetPtr& dataset_ptr, const Config& config) {
auto
device_index
=
faiss
::
gpu
::
index_cpu_to_gpu
(
gpu_res
->
faiss_res
.
get
(),
gpu_id_
,
build_index
);
device_index
->
train
(
rows
,
(
float
*
)
p_data
);
std
::
shared_ptr
<
faiss
::
Index
>
host_index
=
nullptr
;
host_index
.
reset
(
faiss
::
gpu
::
index_gpu_to_cpu
(
device_index
));
delete
device_index
;
delete
build_index
;
device_index
=
faiss
::
gpu
::
index_cpu_to_gpu
(
gpu_res
->
faiss_res
.
get
(),
gpu_id_
,
host_index
.
get
());
index_
.
reset
(
device_index
);
res_
=
gpu_res
;
gpu_mode_
=
2
;
}
else
{
KNOWHERE_THROW_MSG
(
"Build IVFSQHybrid can't get gpu resource"
);
}
delete
build_index
;
}
VecIndexPtr
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录