[pre-commit.ci] auto code formatting

Author: pre-commit-ci[bot]

benchmarks/dynamic_map/retrieve_all_bench.cu

@@ -72,21 +72,21 @@ std::enable_if_t<(sizeof(Key) != sizeof(Value)), void> dynamic_map_retrieve_all(
 }
 
 NVBENCH_BENCH_TYPES(dynamic_map_retrieve_all,
-    NVBENCH_TYPE_AXES(defaults::KEY_TYPE_RANGE,
-                      defaults::VALUE_TYPE_RANGE,
-                      nvbench::type_list<distribution::unique>))
-.set_name("dynamic_map_retrieve_all_unique_capacity")
-.set_type_axes_names({"Key", "Value", "Distribution"})
-.set_max_noise(defaults::MAX_NOISE)
-.add_int64_axis("NumInputs", defaults::N_RANGE)
-.add_int64_axis("InitSize", {defaults::INITIAL_SIZE});
+                    NVBENCH_TYPE_AXES(defaults::KEY_TYPE_RANGE,
+                                      defaults::VALUE_TYPE_RANGE,
+                                      nvbench::type_list<distribution::unique>))
+  .set_name("dynamic_map_retrieve_all_unique_capacity")
+  .set_type_axes_names({"Key", "Value", "Distribution"})
+  .set_max_noise(defaults::MAX_NOISE)
+  .add_int64_axis("NumInputs", defaults::N_RANGE)
+  .add_int64_axis("InitSize", {defaults::INITIAL_SIZE});
 
 NVBENCH_BENCH_TYPES(dynamic_map_retrieve_all,
-    NVBENCH_TYPE_AXES(defaults::KEY_TYPE_RANGE,
-                      defaults::VALUE_TYPE_RANGE,
-                      nvbench::type_list<distribution::unique>))
-.set_name("dynamic_map_retrieve_all_fixed_capacity")
-.set_type_axes_names({"Key", "Value", "Distribution"})
-.set_max_noise(defaults::MAX_NOISE)
-.add_int64_axis("NumInputs", {defaults::N})
-.add_int64_axis("InitSize", {defaults::INITIAL_SIZE});
+                    NVBENCH_TYPE_AXES(defaults::KEY_TYPE_RANGE,
+                                      defaults::VALUE_TYPE_RANGE,
+                                      nvbench::type_list<distribution::unique>))
+  .set_name("dynamic_map_retrieve_all_fixed_capacity")
+  .set_type_axes_names({"Key", "Value", "Distribution"})
+  .set_max_noise(defaults::MAX_NOISE)
+  .add_int64_axis("NumInputs", {defaults::N})
+  .add_int64_axis("InitSize", {defaults::INITIAL_SIZE});

include/cuco/detail/dynamic_map.inl

@@ -245,8 +245,8 @@ std::pair<KeyOut, ValueOut> dynamic_map<Key, Value, Scope, Allocator>::retrieve_
   auto constexpr block_size = 128;
   auto constexpr stride     = 1;
 
-  auto const capacity       = get_capacity();
-  auto grid_size            = (capacity + stride * block_size - 1) / (stride * block_size);
+  auto const capacity = get_capacity();
+  auto grid_size      = (capacity + stride * block_size - 1) / (stride * block_size);
 
   std::vector<size_t> submap_cap_prefix(submaps_.size());
   std::inclusive_scan(

include/cuco/detail/dynamic_map_kernels.cuh

@@ -487,7 +487,6 @@ CUCO_KERNEL void find(InputIt first,
   }
 }
 
-
 /**
  * @brief Retrieves all of the keys and their associated values.
  *
@@ -511,7 +510,8 @@ CUCO_KERNEL void find(InputIt first,
  * perform `retrieve_all` operations on each underlying `static_map`
  * @param num_submaps The number of submaps in the map
  * @param capacity The total number of slots of all submaps
- * @param d_num_out Pointer to the device memory location where the number of keys/vals retrieved are stored
+ * @param d_num_out Pointer to the device memory location where the number of keys/vals retrieved
+ * are stored
  * @param cap_prefix_sum Array of prefix sums of the number of slots in each submap
  * @return Pair of iterators indicating the last elements in the output
  */

include/cuco/dynamic_map.cuh

@@ -27,9 +27,9 @@
 
 #include <cstddef>
 #include <memory>
+#include <numeric>
 #include <type_traits>
 #include <vector>
-#include <numeric>
 
 namespace cuco {
 
@@ -447,7 +447,7 @@ class dynamic_map {
   std::pair<KeyOut, ValueOut> retrieve_all(KeyOut keys_out,
                                            ValueOut values_out,
                                            cudaStream_t stream = 0) const;
-            
+
   /**
    * @brief Indicates whether the keys in the range `[first, last)` are contained in the map.
    *

tests/dynamic_map/retrieve_all_test.cu

@@ -20,14 +20,13 @@
 
 #include <cuda/std/functional>
 #include <thrust/device_vector.h>
-#include <thrust/sort.h>
 #include <thrust/equal.h>
-#include <thrust/sequence.h>
 #include <thrust/iterator/zip_iterator.h>
+#include <thrust/sequence.h>
+#include <thrust/sort.h>
 
 #include <catch2/catch_template_test_macros.hpp>
 
-
 TEMPLATE_TEST_CASE_SIG("dynamic_map retrieve_all tests",
                        "",
                        ((typename Key, typename Value), Key, Value),
@@ -60,8 +59,8 @@ TEMPLATE_TEST_CASE_SIG("dynamic_map retrieve_all tests",
     thrust::device_vector<Key> retrieved_keys(num_keys);
     thrust::device_vector<Value> retrieved_values(num_keys);
 
-    auto [keys_out, values_out] = map.retrieve_all(retrieved_keys.begin(),
-                                                   retrieved_values.begin());
+    auto [keys_out, values_out] =
+      map.retrieve_all(retrieved_keys.begin(), retrieved_values.begin());
 
     REQUIRE(keys_out == retrieved_keys.end());
     REQUIRE(values_out == retrieved_values.end());
@@ -71,7 +70,8 @@ TEMPLATE_TEST_CASE_SIG("dynamic_map retrieve_all tests",
     thrust::sort(thrust::device, retrieved_values.begin(), retrieved_values.end());
 
     REQUIRE(thrust::equal(thrust::device, d_keys.begin(), d_keys.end(), retrieved_keys.begin()));
-    REQUIRE(thrust::equal(thrust::device, d_values.begin(), d_values.end(), retrieved_values.begin()));
+    REQUIRE(
+      thrust::equal(thrust::device, d_values.begin(), d_values.end(), retrieved_values.begin()));
   }
 
   SECTION("retrieve_all after partial erase")
@@ -93,8 +93,8 @@ TEMPLATE_TEST_CASE_SIG("dynamic_map retrieve_all tests",
     thrust::device_vector<Key> retrieved_keys(num_keys / 2);
     thrust::device_vector<Value> retrieved_values(num_keys / 2);
 
-    auto [keys_out, values_out] = map.retrieve_all(retrieved_keys.begin(),
-                                                   retrieved_values.begin());
+    auto [keys_out, values_out] =
+      map.retrieve_all(retrieved_keys.begin(), retrieved_values.begin());
 
     REQUIRE(std::distance(retrieved_keys.begin(), keys_out) == num_keys / 2);
     REQUIRE(std::distance(retrieved_values.begin(), values_out) == num_keys / 2);
@@ -114,8 +114,8 @@ TEMPLATE_TEST_CASE_SIG("dynamic_map retrieve_all tests",
     thrust::device_vector<Key> retrieved_keys(0);
     thrust::device_vector<Value> retrieved_values(0);
 
-    auto [keys_out, values_out] = map.retrieve_all(retrieved_keys.begin(),
-                                                   retrieved_values.begin());
+    auto [keys_out, values_out] =
+      map.retrieve_all(retrieved_keys.begin(), retrieved_values.begin());
 
     REQUIRE(keys_out == retrieved_keys.begin());
     REQUIRE(values_out == retrieved_values.begin());