resolve minor nits

NVIDIA · Jul 31, 2024 · 550f3c9 · 550f3c9
1 parent 6ca8195
commit 550f3c9
Show file tree

Hide file tree

Showing 4 changed files with 14 additions and 11 deletions.
diff --git a/include/cuco/detail/static_map/static_map.inl b/include/cuco/detail/static_map/static_map.inl
@@ -315,7 +315,7 @@ template <typename InputIt, typename Init, typename Op>
 void static_map<Key, T, Extent, Scope, KeyEqual, ProbingScheme, Allocator, Storage>::
   insert_or_apply(InputIt first, InputIt last, Init init, Op op, cuda::stream_ref stream)
 {
-  return this->insert_or_apply_async(first, last, init, op, stream);
+  this->insert_or_apply_async(first, last, init, op, stream);
   stream.wait();
 }
 

diff --git a/include/cuco/detail/static_map/static_map_ref.inl b/include/cuco/detail/static_map/static_map_ref.inl
@@ -16,6 +16,7 @@
 
 #pragma once
 
+#include <cuco/detail/bitwise_compare.cuh>
 #include <cuco/detail/equal_wrapper.cuh>
 #include <cuco/operator.hpp>
 
@@ -703,7 +704,7 @@ class operator_impl<
   {
     ref_type& ref_ = static_cast<ref_type&>(*this);
     // if init equals sentinel value, then we can just `apply` op instead of write
-    if (init == ref_.empty_value_sentinel()) {
+    if (cuco::detail::bitwise_compare(init, ref_.empty_value_sentinel())) {
       ref_.insert_or_apply_impl<true>(value, op);
     } else {
       ref_.insert_or_apply_impl<false>(value, op);
@@ -734,7 +735,7 @@ class operator_impl<
   {
     ref_type& ref_ = static_cast<ref_type&>(*this);
     // if init equals sentinel value, then we can just `apply` op instead of write
-    if (init == ref_.empty_value_sentinel()) {
+    if (cuco::detail::bitwise_compare(init, ref_.empty_value_sentinel())) {
       ref_.insert_or_apply_impl<true>(group, value, op);
     } else {
       ref_.insert_or_apply_impl<false>(group, value, op);

diff --git a/include/cuco/utility/reduction_functors.cuh b/include/cuco/utility/reduction_functors.cuh
@@ -15,6 +15,8 @@
  */
 #pragma once
 
+#include <cuda/atomic>
+
 namespace cuco::reduce {
 
 /**

diff --git a/tests/static_map/insert_or_apply_test.cu b/tests/static_map/insert_or_apply_test.cu
@@ -113,22 +113,22 @@ TEMPLATE_TEST_CASE_SIG(
   SECTION("sentinel equals init; has_init = true")
   {
     auto map = map_type{num_keys, cuco::empty_key<Key>{-1}, cuco::empty_value<Value>{0}};
-    test_insert_or_apply<true>(map, num_keys, num_unique_keys, Key(0));
+    test_insert_or_apply<true>(map, num_keys, num_unique_keys, static_cast<Value>(0));
   }
   SECTION("sentinel equals init; has_init = false")
   {
     auto map = map_type{num_keys, cuco::empty_key<Key>{-1}, cuco::empty_value<Value>{0}};
-    test_insert_or_apply<false>(map, num_keys, num_unique_keys, Key(0));
+    test_insert_or_apply<false>(map, num_keys, num_unique_keys, static_cast<Value>(0));
   }
   SECTION("sentinel not equals init; has_init = true")
   {
     auto map = map_type{num_keys, cuco::empty_key<Key>{-1}, cuco::empty_value<Value>{0}};
-    test_insert_or_apply<true>(map, num_keys, num_unique_keys, Key(-1));
+    test_insert_or_apply<true>(map, num_keys, num_unique_keys, static_cast<Value>(-1));
   }
   SECTION("sentinel not equals init; has_init = false")
   {
     auto map = map_type{num_keys, cuco::empty_key<Key>{-1}, cuco::empty_value<Value>{0}};
-    test_insert_or_apply<false>(map, num_keys, num_unique_keys, Key(-1));
+    test_insert_or_apply<false>(map, num_keys, num_unique_keys, static_cast<Value>(-1));
   }
 }
 
@@ -151,21 +151,21 @@ TEMPLATE_TEST_CASE_SIG(
   SECTION("sentinel equals init; has_init = true")
   {
     auto map = map_type{num_keys, cuco::empty_key<Key>{-1}, cuco::empty_value<Value>{0}};
-    test_insert_or_apply<true>(map, num_keys, num_keys, Key(0));
+    test_insert_or_apply<true>(map, num_keys, num_keys, static_cast<Value>(0));
   }
   SECTION("sentinel equals init; has_init = false")
   {
     auto map = map_type{num_keys, cuco::empty_key<Key>{-1}, cuco::empty_value<Value>{0}};
-    test_insert_or_apply<false>(map, num_keys, num_keys, Key(0));
+    test_insert_or_apply<false>(map, num_keys, num_keys, static_cast<Value>(0));
   }
   SECTION("sentinel not equals init; has_init = true")
   {
     auto map = map_type{num_keys, cuco::empty_key<Key>{-1}, cuco::empty_value<Value>{0}};
-    test_insert_or_apply<true>(map, num_keys, num_keys, Key(-1));
+    test_insert_or_apply<true>(map, num_keys, num_keys, static_cast<Value>(-1));
   }
   SECTION("sentinel not equals init; has_init = false")
   {
     auto map = map_type{num_keys, cuco::empty_key<Key>{-1}, cuco::empty_value<Value>{0}};
-    test_insert_or_apply<false>(map, num_keys, num_keys, Key(-1));
+    test_insert_or_apply<false>(map, num_keys, num_keys, static_cast<Value>(-1));
   }
 }