knn.h

Go to the documentation of this file.

#ifndef KNN_H_
#define KNN_H_

#include <vector>
#include <thrust/device_vector.h>

#include <genie/table/inv_table.h>
#include <genie/table/inv_compr_table.h>
#include <genie/query/query.h>

namespace genie
{
namespace matching
{
void
knn(genie::table::inv_table& table,
    std::vector<genie::query::Query>& queries,
    thrust::device_vector<int>& d_top_indexes,
    thrust::device_vector<int>& d_top_count,
    int hash_table_size,
    int max_load,
    int bitmap_bits);

void
knn_MT(
    std::vector<genie::table::inv_table*>& table,
    std::vector<std::vector<genie::query::Query> >& queries,
    std::vector<thrust::device_vector<int> >& d_top_indexes,
    std::vector<thrust::device_vector<int> >& d_top_count,
    std::vector<int>& hash_table_size,
    std::vector<int>& max_load,
    int bitmap_bits);

void
knn_bijectMap(
    genie::table::inv_table& table,
    std::vector<genie::query::Query>& queries,
    thrust::device_vector<int>& d_top_indexes,
    thrust::device_vector<int>& d_top_count,
    int hash_table_size,
    int max_load,
    int bitmap_bits);

void
knn_bijectMap_MT(
    std::vector<genie::table::inv_table*>& table,
    std::vector<std::vector<genie::query::Query> >& queries,
    std::vector<thrust::device_vector<int> >& d_top_indexes,
    std::vector<thrust::device_vector<int> >& d_top_count,
    std::vector<int>& hash_table_size,
    std::vector<int>& max_load,
    int bitmap_bits);

} // namespace matching
} // namespace genie

#endif