shawn-peng / cudict

cuDict

A Dictionary Library Runs on GPU

Use case example

const uint32_t Sentinel = -1;

typedef Tuple<uint32_t, uint32_t, uint16_t> Tuple3;

int main() {
    /// Saving data to GPU
    auto dict_data = std::vector<Tuple<Tuple3, uint32_t>>{
        {{1, 2, 4}, 3},
        {{2, 3, 4}, 4},
        {{3, 4, 5}, 5},
        {{4, 5, 6}, 6},
        {{5, 5, 7}, 7},
        {{6, 9, 8}, 8}};
    auto d = CUDA_Static_Dict(dict_data, {Sentinel, Sentinel, (uint16_t)Sentinel}, Sentinel);
    std::cout << d;

    /// Checking whether items in the dictionary
    auto flags = d.contains_items({
        {1, 2, 3},
        {2, 3, 4},
        {3, 4, 5},
        {4, 5, 6},
    });
    std::cout << flags << std::endl;

    /// Get values associated with keys
    auto vals = d.get_items({
        {1, 2, 4},
        {2, 4, 4},
        {3, 4, 5},
        {4, 5, 6},
    });
    std::cout << vals << std::endl;

    return 0;
}