match.h

Go to the documentation of this file.

#ifndef GPUGenie_match_h
#define GPUGenie_match_h

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

#include <genie/query/query.h>
#include <genie/table/inv_table.h>
#include "match_common.h"

namespace genie
{
namespace matching
{

int
cal_max_topk(std::vector<genie::query::Query>& queries);

void
match(
    genie::table::inv_table& table,
    std::vector<genie::query::Query>& queries,
    thrust::device_vector<genie::matching::data_t>& d_data,
    thrust::device_vector<u32>& d_bitmap,
    int hash_table_size,
    int max_load,
    int bitmap_bits,
    thrust::device_vector<u32>& d_noiih,
    thrust::device_vector<u32>& d_threshold,
    thrust::device_vector<u32>& d_passCount);

void
match_MT(
    std::vector<genie::table::inv_table*>& table,
    std::vector<std::vector<genie::query::Query> >& queries,
    std::vector<thrust::device_vector<genie::matching::data_t> >& d_data,
    std::vector<thrust::device_vector<u32> >& d_bitmap,
    std::vector<int>& hash_table_size,
    std::vector<int>& max_load,
    int bitmap_bits,
    std::vector<thrust::device_vector<u32> >& d_noiih,
    std::vector<thrust::device_vector<u32> >& d_threshold,
    std::vector<thrust::device_vector<u32> >& d_passCount,
    size_t start,
    size_t finish);


int
build_queries(std::vector<genie::query::Query>& queries, genie::table::inv_table& table,
    std::vector<genie::query::Query::dim>& dims, int max_load);

} // namespace matching
} // namespace genie

#endif