diff --git a/include/cuco/bloom_filter_policies.cuh b/include/cuco/bloom_filter_policies.cuh index cf9ddb371..a433e7cca 100644 --- a/include/cuco/bloom_filter_policies.cuh +++ b/include/cuco/bloom_filter_policies.cuh @@ -18,6 +18,7 @@ #include #include +#include #include @@ -28,9 +29,12 @@ namespace cuco { * fingerprint. * * @tparam Key The type of the values to generate a fingerprint for. + * @tparam XXHash64 Custom (64 bit) XXHash hasher to generate a key's fingerprint. + * By default, cuco::xxhash_64 hasher will be used. + * */ -template -using arrow_filter_policy = detail::arrow_filter_policy; +template > +using arrow_filter_policy = detail::arrow_filter_policy; /** * @brief The default policy that defines how a Blocked Bloom Filter generates and stores a key's diff --git a/include/cuco/detail/bloom_filter/arrow_filter_policy.cuh b/include/cuco/detail/bloom_filter/arrow_filter_policy.cuh index 23b95793e..e1f1b641a 100644 --- a/include/cuco/detail/bloom_filter/arrow_filter_policy.cuh +++ b/include/cuco/detail/bloom_filter/arrow_filter_policy.cuh @@ -78,12 +78,13 @@ namespace cuco::detail { * @endcode * * @tparam Key The type of the values to generate a fingerprint for. + * @tparam XXHash64 64-bit XXHash hasher implementation for fingerprint generation. */ -template +template class arrow_filter_policy { public: - using hasher = cuco::xxhash_64; ///< xxhash_64 hasher for Arrow bloom filter policy - using word_type = std::uint32_t; ///< uint32_t for Arrow bloom filter policy + using hasher = XXHash64; ///< 64-bit XXHash hasher for Arrow bloom filter policy + using word_type = std::uint32_t; ///< uint32_t for Arrow bloom filter policy using hash_argument_type = typename hasher::argument_type; ///< Hash function input type using hash_result_type = decltype(std::declval()( std::declval())); ///< hash function output type diff --git a/tests/bloom_filter/unique_sequence_test.cu b/tests/bloom_filter/unique_sequence_test.cu index c5970b17c..b23564270 100644 --- a/tests/bloom_filter/unique_sequence_test.cu +++ b/tests/bloom_filter/unique_sequence_test.cu @@ -35,8 +35,8 @@ void test_unique_sequence(Filter& filter, size_type num_keys) { using Key = typename Filter::key_type; + // Generate keys thrust::device_vector keys(num_keys); - thrust::sequence(thrust::device, keys.begin(), keys.end()); thrust::device_vector contained(num_keys, false); @@ -119,4 +119,4 @@ TEMPLATE_TEST_CASE_SIG("bloom_filter arrow policy tests", auto filter = filter_type{1000}; test_unique_sequence(filter, num_keys); -} +} \ No newline at end of file