Fix sign conversion error

Author: ksapru

tests/static_set/atomic_storage_test.cu

@@ -69,6 +69,7 @@ TEST_CASE("atomic_storage_test", "")
                                                   cuda::proclaim_return_type<Key>(build_fn{}));
 
   set.insert_async(keys_begin, keys_begin + num_keys);
+  CUCO_CUDA_TRY(cudaDeviceSynchronize());
   auto const count = set.size();
 
   REQUIRE(count == num_keys);

tests/static_set/shared_memory_test.cu

@@ -49,6 +49,8 @@ __global__ void shared_memory_test_kernel(Ref* sets,
   auto insert_ref = sets[set_id].make_copy(g, sm_buffer, cuco::thread_scope_block);
   auto find_ref   = insert_ref.rebind_operators(cuco::op::find);
 
+  __syncthreads();
+
   for (int i = g.thread_rank(); i < number_of_elements; i += g.size()) {
     auto found_it = find_ref.find(insterted_keys[offset + i]);
 
@@ -80,10 +82,11 @@ TEMPLATE_TEST_CASE_SIG("static_set shared memory tests",
 )
 {
   // For uint8_t: sentinel = 0xFF (255), so usable key range is 0..254.
+  // For uint16_t: sentinel = 0xFFFF (65535), so usable key range is 0..65534.
   // thrust::sequence over number_of_sets*elements_in_set keys must not wrap.
-  // Use a smaller set count and element count for 1-byte types.
-  constexpr std::size_t number_of_sets  = (sizeof(Key) == 1) ? 2 : 1000;
-  constexpr std::size_t elements_in_set = (sizeof(Key) == 1) ? 100 : 500;
+  // Use smaller set count and element count for smaller types.
+  constexpr std::size_t number_of_sets  = (sizeof(Key) <= 2) ? (sizeof(Key) == 1 ? 2 : 100) : 1000;
+  constexpr std::size_t elements_in_set = (sizeof(Key) <= 2) ? (sizeof(Key) == 1 ? 100 : 600) : 500;
   constexpr std::size_t set_capacity    = 2 * elements_in_set;
 
   using extent_type = cuco::extent<std::size_t, set_capacity>;
@@ -103,7 +106,8 @@ TEMPLATE_TEST_CASE_SIG("static_set shared memory tests",
   // operator yet
   std::vector<std::unique_ptr<set_type>> sets;
   for (std::size_t set_id = 0; set_id < number_of_sets; ++set_id) {
-    sets.push_back(std::make_unique<set_type>(extent_type{}, cuco::empty_key<Key>{-1}));
+    sets.push_back(
+      std::make_unique<set_type>(extent_type{}, cuco::empty_key<Key>{static_cast<Key>(-1)}));
   }
 
   thrust::device_vector<bool> d_keys_exist(number_of_sets * elements_in_set);