Function register_symqg

python_bindings/symqg_bindings.cpp:115–139

Register Symqg bindings into combined module

                    dists_data + (idx * k)
                );
            } catch (...) {
#pragma omp critical(rabitq_symqg_search_error)
                {
                    if (!error) {
                        error = std::current_exception();
                    }
                }
                failed.store(true, std::memory_order_relaxed);
            }
        }
        if (error) {
            std::rethrow_exception(error);
        }

        return py::make_tuple(ids, dists);
    }

    void save(const std::string& path) const {
        if (!built_) {
            throw std::runtime_error("SymqgIndex must be built before save");
        }
        index_->save(path.c_str());
    }

    static SymqgIndex load(const std::string& path) {
        SymqgIndex wrapper;
        wrapper.index_ = std::make_unique<rabitqlib::symqg::QuantizedGraph<float>>();
        wrapper.index_->load(path.c_str());