diff --git a/cpp/src/groupby/hash/groupby.cu b/cpp/src/groupby/hash/groupby.cu index 49ed0b7fc1d..44df981f5bf 100644 --- a/cpp/src/groupby/hash/groupby.cu +++ b/cpp/src/groupby/hash/groupby.cu @@ -645,10 +645,14 @@ bool can_use_hash_groupby(table_view const& keys, host_spankind); + std::all_of(r.aggregations.begin(), r.aggregations.end(), [v_type](auto const& a) { + return cudf::has_atomic_support(cudf::detail::target_type(v_type, a->kind)) and + is_hash_aggregation(a->kind); }); }); }