Use relaxed memory-order in dynamic_map::find (#702)

caaatch22 · web-flow · commit d1f6040d3894 · 2025-04-27T19:22:14.000-07:00
diff --git a/include/cuco/detail/dynamic_map_kernels.cuh b/include/cuco/detail/dynamic_map_kernels.cuh
@@ -389,7 +389,7 @@ CUCO_KERNEL void find(InputIt first,
       auto submap_view = submap_views[i];
       auto found       = submap_view.find(key, hash, key_equal);
       if (found != submap_view.end()) {
-        found_value = found->second;
+        found_value = found->second.load(cuda::std::memory_order_relaxed);
         break;
       }
     }
@@ -466,7 +466,7 @@ CUCO_KERNEL void find(InputIt first,
       auto submap_view = submap_views[i];
       auto found       = submap_view.find(tile, key, hash, key_equal);
       if (found != submap_view.end()) {
-        found_value = found->second;
+        found_value = found->second.load(cuda::std::memory_order_relaxed);
         break;
       }
     }

Original file line number	Diff line number	Diff line change
`@@ -389,7 +389,7 @@ CUCO_KERNEL void find(InputIt first,`
`389`	`389`	`auto submap_view = submap_views[i];`
`390`	`390`	`auto found = submap_view.find(key, hash, key_equal);`
`391`	`391`	`if (found != submap_view.end()) {`
`392`		`- found_value = found->second;`
	`392`	`+ found_value = found->second.load(cuda::std::memory_order_relaxed);`
`393`	`393`	`break;`
`394`	`394`	`}`
`395`	`395`	`}`
`@@ -466,7 +466,7 @@ CUCO_KERNEL void find(InputIt first,`
`466`	`466`	`auto submap_view = submap_views[i];`
`467`	`467`	`auto found = submap_view.find(tile, key, hash, key_equal);`
`468`	`468`	`if (found != submap_view.end()) {`
`469`		`- found_value = found->second;`
	`469`	`+ found_value = found->second.load(cuda::std::memory_order_relaxed);`
`470`	`470`	`break;`
`471`	`471`	`}`
`472`	`472`	`}`