Add lazy registration for cudf

This is similar to how we handle cupy or pydata/sparse today
mrocklin · Jan 8, 2019 · 8d51bbe · 8d51bbe
1 parent 9ebc7f1
commit 8d51bbe
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/dask/dataframe/methods.py b/dask/dataframe/methods.py
@@ -247,6 +247,15 @@ def _get_level_values(x, n):
 concat_dispatch = Dispatch('concat')
 
 
+@concat_dispatch.register_lazy('cudf')
+def register_cudf():
+    import cudf
+    concat_dispatch.register(
+        (cudf.DataFrame, cudf.Series, cudf.Index),
+        cudf.concat
+    )
+
+
 def concat(dfs, axis=0, join='outer', uniform=False, filter_warning=True):
     """Concatenate, handling some edge cases: