Multivalue Hashtable: use size_t explicitly for cub arguments, fix ra…

…ce condition in host side retrieve if parameter num_out is in pinned memory and stream != 0
sleeepyjack · Nov 19, 2020 · f7fdc8f · f7fdc8f
1 parent cdeea43
commit f7fdc8f
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/include/multi_value_hash_table.cuh b/include/multi_value_hash_table.cuh
@@ -408,7 +408,7 @@ public:
 
         if(values_out != nullptr)
         {
-            index_type required_temp_bytes = 0;
+            std::size_t required_temp_bytes = 0;
             
             cub::DeviceScan::InclusiveSum(
                 nullptr,
@@ -418,7 +418,8 @@ public:
                 num_in,
                 stream);
 
-            index_type available_temp_bytes_from_outputbuffer = num_out * sizeof(value_type);
+            cudaStreamSynchronize(stream);
+            std::size_t available_temp_bytes_from_outputbuffer = num_out * sizeof(value_type);
             
             if(available_temp_bytes_from_outputbuffer >= required_temp_bytes)
             {