admin/milvus
1
0
Fork 0
mirror of https://gitee.com/milvus-io/milvus.git synced 2025-12-07 01:28:27 +08:00
Code Issues Packages Projects Releases Wiki Activity

enhance: support load params for vector index (#44747)

Browse Source 
issue: #44746 

Support modifying vector index behavior during loading by change/add
aram knowhere.xxx.load.xxx.

Signed-off-by: xianliang.li <xianliang.li@zilliz.com>
This commit is contained in:
foxspy 2025-10-11 18:30:08 +08:00 committed by GitHub
parent 7b8ecdaad5
commit 0eb42a7870
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 17 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
pkg/util/indexparams/index_params.go
View File

@ -378,5 +378,8 @@ func AppendPrepareLoadParams(params *paramtable.ComponentParam, indexParams map[
indexParams[k] = v indexParams[k] = v
} }
} }
params.KnowhereConfig.MergeIndexParams(indexParams[common.IndexTypeKey], paramtable.LoadStage, indexParams)
return nil return nil
} }

14
pkg/util/indexparams/index_params_test.go
View File

@ -604,5 +604,19 @@ func TestAppendPrepareInfo_parse(t *testing.T) {
assert.NoError(t, err) assert.NoError(t, err)
assert.Equal(t, resultMapString["key1"], "value1") assert.Equal(t, resultMapString["key1"], "value1")
assert.Equal(t, resultMapString["key2"], "value2") assert.Equal(t, resultMapString["key2"], "value2")
params.Save(params.KnowhereConfig.Enable.Key, "true")
params.Save(params.KnowhereConfig.IndexParam.KeyPrefix+"GPU_CAGRA.load.adapt_for_cpu", "true")
indexParams := map[string]string{
"index_type": "GPU_CAGRA",
"nn_descent_niter": "20",
"build_algo": "NN_DESCENT",
}
err = AppendPrepareLoadParams(&params, indexParams)
assert.NoError(t, err)
assert.Equal(t, indexParams["nn_descent_niter"], "20")
assert.Equal(t, indexParams["build_algo"], "NN_DESCENT")
assert.Equal(t, indexParams["adapt_for_cpu"], "true")
}) })
} }