Use counting_iterator in HLL unit test

NVIDIA · Jun 11, 2024 · 76d77c9 · 76d77c9
1 parent 693698d
commit 76d77c9
Showing 1 changed file with 5 additions and 8 deletions.
diff --git a/tests/distinct_count_estimator/unique_sequence_test.cu b/tests/distinct_count_estimator/unique_sequence_test.cu
@@ -19,8 +19,7 @@
 #include <cuco/distinct_count_estimator.cuh>
 #include <cuco/hash_functions.cuh>
 
-#include <thrust/device_vector.h>
-#include <thrust/sequence.h>
+#include <thrust/iterator/counting_iterator.h>
 
 #include <catch2/catch_template_test_macros.hpp>
 #include <catch2/generators/catch_generators.hpp>
@@ -50,10 +49,8 @@ TEMPLATE_TEST_CASE_SIG("distinct_count_estimator: unique sequence",
  double const relative_standard_deviation =
  1.04 / std::sqrt(static_cast<double>(1ull << hll_precision));
 
- thrust::device_vector<T> items(num_items);
-
- // Generate `num_items` distinct items
- thrust::sequence(items.begin(), items.end(), 0);
+ // Generates distinct items
+ auto const items_begin = thrust::counting_iterator<T>(0);
 
  // Initialize the estimator
  cuco::distinct_count_estimator<T, cuda::thread_scope_device, Hash> estimator{
@@ -62,12 +59,12 @@ TEMPLATE_TEST_CASE_SIG("distinct_count_estimator: unique sequence",
  REQUIRE(estimator.estimate() == 0);
 
  // Add all items to the estimator
- estimator.add(items.begin(), items.end());
+ estimator.add(items_begin, items_begin + num_items);
 
  auto const estimate = estimator.estimate();
 
  // Adding the same items again should not affect the result
- estimator.add(items.begin(), items.begin() + num_items / 2);
+ estimator.add(items_begin, items_begin + num_items / 2);
  REQUIRE(estimator.estimate() == estimate);
 
  // Clearing the estimator should reset the estimate