milvus/tests/milvus_benchmark/suites/gpu_accuracy.yaml

157 lines
5.2 KiB
YAML

accuracy:
tables:
# sift1b
-
server:
db_config.primary_path: /test/milvus/db_data_gpu/sift_1b_2048_128_l2_sq8
cache_config.cpu_cache_capacity: 150
engine_config.use_blas_threshold: 1100
engine_config.gpu_search_threshold: 1
gpu_resource_config.enable: true
gpu_resource_config.cache_capacity: 4
gpu_resource_config.search_resources:
- gpu0
- gpu1
gpu_resource_config.build_index_resources:
- gpu0
- gpu1
table_name: sift_1b_2048_128_l2
search_params:
nprobes: [1, 2, 4, 8, 16, 32, 64, 128, 256, 512]
top_ks: [64]
nqs: [1000]
-
server:
db_config.primary_path: /test/milvus/db_data_gpu/sift_1b_2048_128_l2_sq8h
cache_config.cpu_cache_capacity: 150
engine_config.use_blas_threshold: 1100
engine_config.gpu_search_threshold: 1
gpu_resource_config.enable: true
gpu_resource_config.cache_capacity: 4
gpu_resource_config.search_resources:
- gpu0
- gpu1
gpu_resource_config.build_index_resources:
- gpu0
- gpu1
table_name: sift_1b_2048_128_l2
search_params:
nprobes: [1, 2, 4, 8, 16, 32, 64, 128, 256, 512]
top_ks: [64]
nqs: [1000]
# -
# server:
# db_config.primary_path: /test/milvus/db_data_gpu/sift_1b_2048_128_l2_pq
# cache_config.cpu_cache_capacity: 150
# engine_config.use_blas_threshold: 1100
# engine_config.gpu_search_threshold: 1
# gpu_resource_config.enable: true
# gpu_resource_config.cache_capacity: 4
# gpu_resource_config.search_resources:
# - gpu0
# - gpu1
# gpu_resource_config.build_index_resources:
# - gpu0
# - gpu1
# table_name: sift_1b_2048_128_l2
# search_params:
# nprobes: [1, 2, 4, 8, 16, 32, 64, 128, 256, 512]
# top_ks: [64]
# nqs: [1000]
# sift50m
-
server:
db_config.primary_path: /test/milvus/db_data_gpu/sift_50m_1024_128_l2_ivf
cache_config.cpu_cache_capacity: 60
engine_config.use_blas_threshold: 1100
engine_config.gpu_search_threshold: 1
gpu_resource_config.enable: true
gpu_resource_config.cache_capacity: 4
gpu_resource_config.search_resources:
- gpu0
- gpu1
gpu_resource_config.build_index_resources:
- gpu0
- gpu1
table_name: sift_50m_1024_128_l2
search_params:
nprobes: [1, 2, 4, 8, 16, 32, 64, 128, 256, 512]
top_ks: [64]
nqs: [1000]
-
server:
db_config.primary_path: /test/milvus/db_data_gpu/sift_50m_1024_128_l2_sq8
cache_config.cpu_cache_capacity: 30
engine_config.use_blas_threshold: 1100
engine_config.gpu_search_threshold: 1
gpu_resource_config.enable: true
gpu_resource_config.cache_capacity: 4
gpu_resource_config.search_resources:
- gpu0
- gpu1
gpu_resource_config.build_index_resources:
- gpu0
- gpu1
table_name: sift_50m_1024_128_l2
search_params:
nprobes: [1, 2, 4, 8, 16, 32, 64, 128, 256, 512]
top_ks: [64]
nqs: [1000]
-
server:
db_config.primary_path: /test/milvus/db_data_gpu/sift_50m_1024_128_l2_sq8h
cache_config.cpu_cache_capacity: 30
engine_config.use_blas_threshold: 1100
engine_config.gpu_search_threshold: 1
gpu_resource_config.enable: true
gpu_resource_config.cache_capacity: 4
gpu_resource_config.search_resources:
- gpu0
- gpu1
gpu_resource_config.build_index_resources:
- gpu0
- gpu1
table_name: sift_50m_1024_128_l2
search_params:
nprobes: [1, 2, 4, 8, 16, 32, 64, 128, 256, 512]
top_ks: [64]
nqs: [1000]
# -
# server:
# db_config.primary_path: /test/milvus/db_data_gpu/sift_50m_1024_128_l2_pq
# cache_config.cpu_cache_capacity: 30
# engine_config.use_blas_threshold: 1100
# engine_config.gpu_search_threshold: 1
# gpu_resource_config.enable: true
# gpu_resource_config.cache_capacity: 4
# gpu_resource_config.search_resources:
# - gpu0
# - gpu1
# gpu_resource_config.build_index_resources:
# - gpu0
# - gpu1
# table_name: sift_50m_1024_128_l2
# search_params:
# nprobes: [1, 2, 4, 8, 16, 32, 64, 128, 256, 512]
# top_ks: [64]
# nqs: [1000]
-
server:
db_config.primary_path: /test/milvus/db_data_gpu/sift_50m_1024_128_l2_nsg
cache_config.cpu_cache_capacity: 100
engine_config.use_blas_threshold: 1100
engine_config.gpu_search_threshold: 1
gpu_resource_config.enable: true
gpu_resource_config.cache_capacity: 4
gpu_resource_config.search_resources:
- gpu0
- gpu1
gpu_resource_config.build_index_resources:
- gpu0
- gpu1
table_name: sift_50m_1024_128_l2
search_params:
nprobes: [1]
top_ks: [64]
nqs: [1000]