Giters

rapidsai / raft

RAFT contains fundamental widely-used algorithms and primitives for machine learning and information retrieval. The algorithms are CUDA-accelerated and form building blocks for more easily writing high performance applications.

Home Page:https://docs.rapids.ai/api/raft/stable/

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

[FEA] `raft-ann-bench` the Python library should expose all of the CLI options in the underlying C++ binary

fy opened this issue · comments

commented

Is your feature request related to a problem? Please describe.

When using the Python library, it is not possible to override some CLI options in the underlying C++ binaries.
For example, --benchmark_min_warmup_time may be a useful option for the end users.

Describe the solution you'd like

The Python library transparently pass through CLI options while it continues to provide helpful defaults as we do currently.
This should also help reduce maintenance efforts.