milvus/tests/benchmark/suites/gpu_accuracy_ann.yaml

ann_accuracy:
  collections:
    -
      server:
        cache_config.cpu_cache_capacity: 16GB
        engine_config.use_blas_threshold: 1100
        engine_config.gpu_search_threshold: 1
        gpu_resource_config.enable: true
        gpu_resource_config.cache_capacity: 4GB
        gpu_resource_config.search_resources:
          - gpu0
          - gpu1
        gpu_resource_config.build_index_resources:
          - gpu0
          - gpu1
      source_file: /test/milvus/ann_hdf5/sift-128-euclidean.hdf5
      collection_name: sift_128_euclidean
      index_file_sizes: [1024]
      index_types: ['flat', 'ivf_flat', 'ivf_sq8', 'ivf_sq8h']
      index_params:
        nlist: [16384]
      top_ks: [10]
      nqs: [10000]
      search_params:
        nprobe: [1, 2, 4, 8, 16, 32, 64, 128, 256, 512, 16384]

    -
      server:
        cache_config.cpu_cache_capacity: 16GB
        engine_config.use_blas_threshold: 1100
        engine_config.gpu_search_threshold: 1
        gpu_resource_config.enable: true
        gpu_resource_config.cache_capacity: 4GB
        gpu_resource_config.search_resources:
          - gpu0
          - gpu1
        gpu_resource_config.build_index_resources:
          - gpu0
          - gpu1
      source_file: /test/milvus/ann_hdf5/sift-128-euclidean.hdf5
      collection_name: sift_128_euclidean
      index_file_sizes: [1024]
      index_types: ['ivf_pq']
      index_params:
        nlist: [16384]
        m: [32]
      top_ks: [10]
      nqs: [10000]
      search_params:
        nprobe: [1, 2, 4, 8, 16, 32, 64, 128, 256, 512]

    -
      server:
        cache_config.cpu_cache_capacity: 16GB
        engine_config.use_blas_threshold: 1100
        engine_config.gpu_search_threshold: 1
        gpu_resource_config.enable: true
        gpu_resource_config.cache_capacity: 4GB
        gpu_resource_config.search_resources:
          - gpu0
          - gpu1
        gpu_resource_config.build_index_resources:
          - gpu0
          - gpu1
      source_file: /test/milvus/ann_hdf5/sift-128-euclidean.hdf5
      collection_name: sift_128_euclidean
      index_file_sizes: [256]
      index_types: ['hnsw']
      index_params:
        M: [16]
        efConstruction: [500]
      top_ks: [10]
      nqs: [10000]
      search_params:
        ef: [16, 32, 64, 128, 256, 512]

    -
      server:
        cache_config.cpu_cache_capacity: 16
        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
      source_file: /test/milvus/ann_hdf5/glove-200-angular.hdf5
      collection_name: glove_200_angular
      index_file_sizes: [1024]
      index_types: ['flat', 'ivf_flat', 'ivf_sq8', 'ivf_sq8h']
      index_params:
        nlist: [16384]
      top_ks: [10]
      nqs: [10000]
      search_params:
        nprobe: [1, 2, 4, 8, 16, 32, 64, 128, 256, 512, 16384]

    -
      server:
        cache_config.cpu_cache_capacity: 16GB
        engine_config.use_blas_threshold: 1100
        engine_config.gpu_search_threshold: 1
        gpu_resource_config.enable: true
        gpu_resource_config.cache_capacity: 4GB
        gpu_resource_config.search_resources:
          - gpu0
          - gpu1
        gpu_resource_config.build_index_resources:
          - gpu0
          - gpu1
      source_file: /test/milvus/ann_hdf5/glove-200-angular.hdf5
      collection_name: glove_200_angular
      index_file_sizes: [256]
      index_types: ['hnsw']
      index_params:
        M: [36]
        efConstruction: [500]
      top_ks: [10]
      nqs: [10000]
      search_params:
        ef: [10, 16, 32, 64, 128, 256, 512]

    -
      server:
        cache_config.cpu_cache_capacity: 16GB
        engine_config.use_blas_threshold: 1100
        engine_config.gpu_search_threshold: 1
        gpu_resource_config.enable: false
        gpu_resource_config.cache_capacity: 4GB
        gpu_resource_config.search_resources:
          - gpu0
          - gpu1
        gpu_resource_config.build_index_resources:
          - gpu0
          - gpu1
      source_file: /test/milvus/ann_hdf5/kosarak-27983-jaccard.hdf5
      collection_name: kosarak_27984_jaccard
      index_file_sizes: [1024]
      index_types: ['flat', 'ivf_flat']
      index_params:
        nlist: [2048]
      top_ks: [10]
      nqs: [10000]
      search_params:
        nprobe: [1, 2, 4, 8, 16, 32, 64, 128, 256, 512]

    -
      server:
        cache_config.cpu_cache_capacity: 16GB
        engine_config.use_blas_threshold: 1100
        engine_config.gpu_search_threshold: 1
        gpu_resource_config.enable: false
        gpu_resource_config.cache_capacity: 4GB
        gpu_resource_config.search_resources:
          - gpu0
          - gpu1
        gpu_resource_config.build_index_resources:
          - gpu0
          - gpu1
      source_file: /test/milvus/ann_hdf5/sift-256-hamming.hdf5
      collection_name: sift_256_hamming
      index_file_sizes: [1024]
      index_types: ['flat', 'ivf_flat']
      index_params:
        nlist: [2048]
      top_ks: [100]
      nqs: [1000]
      search_params:
        nprobe: [1, 2, 4, 8, 16, 32, 64, 128, 256, 512]