CAGRA锟斤拷一锟街革拷锟斤拷锟杰★拷GPU锟斤拷锟劫碉拷ANN锟斤拷锟斤拷锟斤拷锟斤拷锟姐法

锟斤拷源锟斤拷锟斤拷锟斤拷锟斤拷锟斤拷锟斤拷|锟斤拷时锟戒：2024-06-17 08:57:47 |锟斤拷锟侥讹拷锟斤拷149锟斤拷|锟斤拷锟斤拷签锟斤拷锟斤拷 Gra v C 锟姐法锟斤拷锟斤拷锟斤拷锟斤拷锟捷匡拷锟斤拷 |锟斤拷锟斤拷锟斤拷锟斤拷锟斤拷

锟斤拷锟斤拷锟斤拷锟接ｏ拷https://wanger-sjtu.github.io/CARGA/ CAGRA 锟斤拷 N锟斤拷锟斤拷RAFT锟斤拷目锟斤拷锟斤拷锟铰碉拷 ANN 锟斤拷锟斤拷锟斤拷锟斤拷锟斤拷锟斤拷锟斤拷一锟街革拷锟斤拷锟杰的★拷 GPU 锟斤拷锟劫的★拷锟斤拷锟斤拷图锟侥凤拷锟斤拷锟斤拷锟斤拷锟斤拷锟斤拷锟斤拷锟叫★拷锟斤拷锟斤拷锟斤拷锟斤拷锟斤拷锟斤拷锟脚伙拷锟斤拷锟斤拷锟斤拷每锟轿诧拷锟斤拷只锟斤拷锟斤拷一锟斤拷锟津几革拷锟斤拷询锟斤拷锟斤拷锟斤拷锟斤拷锟斤拷锟斤拷锟斤拷HNSW锟斤拷S

锟斤拷锟斤拷锟斤拷锟接ｏ拷 https://wanger-sjtu.github.io/CARGA/

CAGRA锟斤拷N锟斤拷锟斤拷RAFT锟斤拷目锟斤拷锟斤拷锟铰碉拷ANN锟斤拷锟斤拷锟斤拷锟斤拷锟斤拷冢锟斤拷锟斤拷锟斤拷锟斤拷锟斤拷惴拷锟斤拷锟斤拷锟揭伙拷指锟斤拷锟斤拷艿摹锟紾PU锟斤拷锟劫的★拷锟斤拷锟斤拷图锟侥凤拷锟斤拷锟斤拷锟截憋拷锟斤拷锟叫★拷锟斤拷锟斤拷锟斤拷锟斤拷锟斤拷锟斤拷锟脚伙拷锟斤拷锟斤拷锟斤拷每锟轿诧拷锟斤拷只锟斤拷锟斤拷一锟斤拷锟津几革拷锟斤拷询锟斤拷锟斤拷锟斤拷锟斤拷锟斤拷锟斤拷锟斤拷锟斤拷图锟侥凤拷锟斤拷锟斤拷锟斤拷HNSW锟斤拷SONG锟饺ｏ拷锟斤拷锟狡ｏ拷CAGRA锟斤拷锟斤拷锟斤拷训锟斤拷锟阶段癸拷锟斤拷锟斤拷一锟斤拷锟斤拷锟斤拷锟脚伙拷锟斤拷k-锟斤拷锟斤拷冢锟絢-NN锟斤拷图锟斤拷锟斤拷锟酵硷拷弑锟斤拷锟斤拷锟斤拷锟斤拷锟斤拷锟斤拷裕锟斤拷芄锟斤拷诒锟斤拷趾锟斤拷锟斤拷倩锟斤拷实锟酵笔碉拷指锟叫э拷锟斤拷锟斤拷锟斤拷锟斤拷锟絅SW锟斤拷HNSW锟姐法锟斤拷同锟斤拷锟角ｏ拷CARGA锟姐法锟角碉拷锟斤拷锟酵硷拷锟斤拷诠锟斤拷锟斤拷筒锟窖拷锥锟斤拷锟斤拷锟斤拷锟斤拷锟斤拷锟脚伙拷锟斤拷锟斤拷应GPU锟斤拷锟斤拷锟斤拷佟锟�

锟斤拷锟斤拷示锟斤拷锟斤拷

using namespace raft::neighbors;
// use default index parameters based on shape of the dataset
ivf_pq::index_params build_params =   ivf_pq::index_params::from_dataset(dataset);
ivf_pq::search_params search_params;
auto knn_graph      = raft::make_host_matrix(dataset.extent(0), 128);

// create knn graph
cagra::build_knn_graph(res, dataset, knn_graph.view(), 2, build_params, search_params);
auto optimized_gaph = raft::make_host_matrix(dataset.extent(0), 64);
cagra::optimize(res, dataset, knn_graph.view(), optimized_graph.view());
// Construct an index from dataset and optimized knn_graph

auto index = cagra::index(res, build_params.metric(), dataset,
                                   optimized_graph.view());

CAGRA锟斤拷锟斤拷锟斤拷图锟叫硷拷锟斤拷锟斤拷同之锟斤拷锟斤拷