test/milvus
1
0
Fork 0
mirror of https://gitee.com/milvus-io/milvus.git synced 2024-11-29 18:38:44 +08:00
Code Issues Actions 11 Packages Projects Releases Wiki Activity

enhance: change gpu default mem pool size (#34159)

Browse Source 
Signed-off-by: yusheng.ma <yusheng.ma@zilliz.com>
This commit is contained in:
presburger 2024-08-06 10:24:15 +08:00 committed by GitHub
parent dea73ef288
commit a9352a073b
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 12 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
configs/milvus.yaml
View File

@ -979,5 +979,5 @@ trace:
#milvus will automatically initialize half of the available GPU memory,
#maxMemSize will the whole available GPU memory.
gpu:
initMemSize: # Gpu Memory Pool init size
maxMemSize: # Gpu Memory Pool Max size
initMemSize: 2048 # Gpu Memory Pool init size
maxMemSize: 4096 # Gpu Memory Pool Max size

18
pkg/util/paramtable/component_param.go
View File

@ -885,18 +885,20 @@ type gpuConfig struct {
func (t *gpuConfig) init(base *BaseTable) {
t.InitSize = ParamItem{
Key: "gpu.initMemSize",
Version: "2.3.4",
Doc: `Gpu Memory Pool init size`,
Export: true,
Key: "gpu.initMemSize",
Version: "2.3.4",
Doc: `Gpu Memory Pool init size`,
Export: true,
DefaultValue: "2048",
}
t.InitSize.Init(base.mgr)
t.MaxSize = ParamItem{
Key: "gpu.maxMemSize",
Version: "2.3.4",
Doc: `Gpu Memory Pool Max size`,
Export: true,
Key: "gpu.maxMemSize",
Version: "2.3.4",
Doc: `Gpu Memory Pool Max size`,
Export: true,
DefaultValue: "4096",
}
t.MaxSize.Init(base.mgr)
}