Merge branch 'feature/diff-non-numeric' of https://github.com/Matt711…

…/cudf into feature/diff-non-numeric
rapidsai · Apr 14, 2022 · 1f729ad · 1f729ad
2 parents 1eee122 + 9e1655d
commit 1f729ad
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 22 deletions.
diff --git a/python/cudf/cudf/tests/test_dataframe.py b/python/cudf/cudf/tests/test_dataframe.py
@@ -9168,7 +9168,7 @@ def test_diff_many_dtypes():
         {
             "dates": pd.date_range("2020-01-01", "2020-01-06", freq="D"),
             "bools": [True, True, True, False, True, True],
-            "floats": [1.0, 2.0, 3.5, 4.0, 5.0, -1.7],
+            "floats": [1.0, 2.0, 3.5, np.nan, 5.0, -1.7],
             "ints": [1, 2, 3, 3, 4, 5],
             "nans_nulls": [np.nan, None, None, np.nan, np.nan, None],
         }

diff --git a/python/cudf/cudf/utils/cudautils.py b/python/cudf/cudf/utils/cudautils.py
@@ -14,27 +14,6 @@
 #
 
 
-@cuda.jit
-def gpu_diff(in_col, out_col, out_mask, N):
-    """Calculate the difference between values at positions i and i - N in an
-    array and store the output in a new array.
-    """
-    i = cuda.grid(1)
-
-    if N > 0:
-        if i < in_col.size:
-            out_col[i] = in_col[i] - in_col[i - N]
-            out_mask[i] = True
-        if i < N:
-            out_mask[i] = False
-    else:
-        if i <= (in_col.size + N):
-            out_col[i] = in_col[i] - in_col[i - N]
-            out_mask[i] = True
-        if i >= (in_col.size + N) and i < in_col.size:
-            out_mask[i] = False
-
-
 # Find segments