From 425a91282d8573710518208902cf7c80b23d740b Mon Sep 17 00:00:00 2001 From: brandon-b-miller Date: Tue, 21 Mar 2023 12:16:48 -0700 Subject: [PATCH] denote groupby apply udfs in cache key --- python/cudf/cudf/core/udf/groupby_utils.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/python/cudf/cudf/core/udf/groupby_utils.py b/python/cudf/cudf/core/udf/groupby_utils.py index 669db76db9a..ad20034c255 100644 --- 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