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

Rename RAFT_IVF_FLAT/RAFT_IVF_PQ to GPU_IVF_FLAT/GPU_IVF_PQ (#23194) 

Signed-off-by: Yudong Cai <yudong.cai@zilliz.com>
pull/23189/head
Cai Yudong 2023-04-04 11:44:26 +08:00 committed by GitHub
parent c71bc19049
commit 9ad6d9f1a0
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 4 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
internal/util/indexparamcheck/index_type.go
View File

@ -16,8 +16,8 @@ type IndexType = string
// IndexType definitions
const (
IndexRaftIvfFlat IndexType = "RAFT_IVF_FLAT"
IndexRaftIvfPQ IndexType = "RAFT_IVF_PQ"
IndexRaftIvfFlat IndexType = "GPU_IVF_FLAT"
IndexRaftIvfPQ IndexType = "GPU_IVF_PQ"
IndexFaissIDMap IndexType = "FLAT" // no index is built.
IndexFaissIvfFlat IndexType = "IVF_FLAT"
IndexFaissIvfPQ IndexType = "IVF_PQ"

3
tests/python_client/common/common_func.py
View File

@ -486,8 +486,7 @@ def gen_invaild_search_params_type():
def gen_search_param(index_type, metric_type="L2"):
search_params = []
if index_type in ["FLAT", "IVF_FLAT", "IVF_SQ8", "IVF_PQ", "GPU_FLAT", "GPU_IVF_FLAT", "GPU_IVF_PQ", "GPU_IVF_SQ8",
"RAFT_IVF_FLAT", "RAFT_IVF_PQ"]:
if index_type in ["FLAT", "IVF_FLAT", "IVF_SQ8", "IVF_PQ", "GPU_IVF_FLAT", "GPU_IVF_PQ"]:
if index_type in ["GPU_FLAT"]:
ivf_search_params = {"metric_type": metric_type, "params": {}}
search_params.append(ivf_search_params)

2
tests/python_client/common/common_type.py
View File

@ -183,7 +183,7 @@ get_wrong_format_dict = [
""" Specially defined list """
all_index_types = ["FLAT", "IVF_FLAT", "IVF_SQ8", "IVF_PQ", "HNSW", "DISKANN", "BIN_FLAT", "BIN_IVF_FLAT",
"RAFT_IVF_FLAT", "RAFT_IVF_PQ"]
"GPU_IVF_FLAT", "GPU_IVF_PQ"]
default_index_params = [{"nlist": 128}, {"nlist": 128}, {"nlist": 128}, {"nlist": 128, "m": 16, "nbits": 8},
{"M": 48, "efConstruction": 500}, {}, {"nlist": 128}, {"nlist": 128},