Merge 2032945 into f16deb5

rapidsai · Nov 3, 2022 · 5009d5c · 5009d5c
2 parents f16deb5 + 2032945
commit 5009d5c
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 2 deletions.
diff --git a/cpp/src/c_api/bfs.cpp b/cpp/src/c_api/bfs.cpp
@@ -194,6 +194,14 @@ extern "C" cugraph_error_code_t cugraph_bfs(const cugraph_resource_handle_t* han
                                             cugraph_paths_result_t** result,
                                             cugraph_error_t** error)
 {
+  CAPI_EXPECTS(
+    reinterpret_cast<cugraph::c_api::cugraph_graph_t*>(graph)->vertex_type_ ==
+      reinterpret_cast<cugraph::c_api::cugraph_type_erased_device_array_view_t const*>(sources)
+        ->type_,
+    CUGRAPH_INVALID_INPUT,
+    "vertex type of graph and sources must match",
+    *error);
+
   cugraph::c_api::bfs_functor functor(handle,
                                       graph,
                                       sources,

diff --git a/python/cugraph/cugraph/traversal/bfs.py b/python/cugraph/cugraph/traversal/bfs.py
@@ -55,7 +55,10 @@ def _ensure_args(G, start, i_start, directed):
         else:
             if not isinstance(start, cudf.DataFrame):
                 if not isinstance(start, dask_cudf.DataFrame):
-                    start = cudf.DataFrame({"starts": cudf.Series(start)})
+                    vertex_dtype = G.nodes().dtype
+                    start = cudf.DataFrame(
+                        {"starts": cudf.Series(start, dtype=vertex_dtype)}
+                    )
 
             if G.is_renumbered():
                 validlen = len(
@@ -224,7 +227,8 @@ def bfs(
         if is_dataframe:
             start = start[start.columns[0]]
         else:
-            start = cudf.Series(start, name="starts")
+            vertex_dtype = G.nodes().dtype
+            start = cudf.Series(start, dtype=vertex_dtype)
 
     distances, predecessors, vertices = pylibcugraph_bfs(
         handle=ResourceHandle(),