Skip to content

[FEA] CAGRA ACE Follow up tracker #1486

@tfeher

Description

@tfeher

#1404 implements a new build method for CAGRA indices. Here we list follow up issues that are out of scope of the original PR.

  • Default build config should select ACE build when regular build would go OOM
  • cuvs-bench param parser to accept graph_build_algo: "ACE" (and set other params automatically if needed)
  • Set ace_params.npartitions default value 0 for auto select?
  • do we need to expose build_knn_graph for ivf_pq method only?
  • does get_device_for_address (used in src/neighbors/cagra.cuh) work as intended on systems with AST or HMM?
  • hnsw::from_cagra does only serialize the index when it is on_disk. Is this the right behavior?
  • python API tests
  • [BUG] sample_rows + balanced k-means leads to imbalanced clusters on BIGANN 1B #1461
  • writing numpy headers for files is repetitive in cagra_build.cuh is repetitive, create better helpers.

Metadata

Metadata

Assignees

No one assigned

    Labels

    Type

    No type

    Projects

    Status

    Todo

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions