Fixes

NVIDIA · Jun 10, 2024 · 35ed630 · 35ed630
1 parent 253b8fb
commit 35ed630
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/cub/cub/device/dispatch/dispatch_merge.cuh b/cub/cub/device/dispatch/dispatch_merge.cuh
@@ -174,9 +174,9 @@ struct DispatchMerge
   // Cannot check output iterators, since they could be discard iterators, which do not have the right value_type
   static_assert(::cuda::std::is_same<value_t<KeyIt2>, KeyT>::value, "");
   static_assert(::cuda::std::is_same<value_t<ValueIt2>, ValueT>::value, "");
-  static_assert(::cuda::std::is_invocable<CompareOp, KeyT, KeyT>::value, "Comparison operator cannot compare two keys");
-  static_assert(::cuda::std::is_same<::cuda::std::invoke_result_t<CompareOp, KeyT, KeyT>, bool>::value,
-                "Comparison operator cannot compare two keys");
+  static_assert(::cuda::std::__invokable<CompareOp, KeyT, KeyT>::value, "Comparison operator cannot compare two keys");
+  static_assert(::cuda::std::is_convertible<typename ::cuda::std::__invoke_of<CompareOp, KeyT, KeyT>::type, bool>::value,
+                "Comparison operator must be convertible to bool");
 
   void* d_temp_storage;
   std::size_t& temp_storage_bytes;