denote groupby apply udfs in cache key

rapidsai · Mar 21, 2023 · 425a912 · 425a912
1 parent 197bce6
commit 425a912
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/python/cudf/cudf/core/udf/groupby_utils.py b/python/cudf/cudf/core/udf/groupby_utils.py
@@ -150,6 +150,9 @@ def jit_groupby_apply(offsets, grouped_values, function, *args):
     ngroups = len(offsets) - 1
 
     cache_key = _generate_cache_key(grouped_values, function)
+    # add a string breaking any degeneracies with standard apply UDFs
+    cache_key = (*cache_key, "__GROUPBY_APPLY_UDF")
+
     if cache_key not in precompiled:
         precompiled[cache_key] = _get_groupby_apply_kernel(
             grouped_values, function, args