add GNN packages

rapidsai · Nov 13, 2024 · 974f77f · 974f77f
1 parent 9ab53a6
commit 974f77f
Showing 1 changed file with 21 additions and 18 deletions.
diff --git a/features/src/rapids-build-utils/opt/rapids-build-utils/manifest.yaml b/features/src/rapids-build-utils/opt/rapids-build-utils/manifest.yaml
@@ -187,21 +187,6 @@ repos:
         cmake: -DCMAKE_CUDA_ARCHITECTURES="${CUDAARCHS}"
         install: *rapids_build_backend_args
 
-- name: wholegraph
-  path: wholegraph
-  git: {<<: *git_defaults, repo: wholegraph}
-  cpp:
-    - name: wholegraph
-      sub_dir: cpp
-      depends: [raft]
-      args:
-        cmake: -DCMAKE_CUDA_ARCHITECTURES="${CUDAARCHS}"
-  python:
-    - name: wholegraph
-      sub_dir: python/pylibwholegraph
-      depends: [wholegraph]
-      args: {install: *rapids_build_backend_args}
-
 - name: cugraph
   path: cugraph
   git: {<<: *git_defaults, repo: cugraph}
@@ -224,9 +209,6 @@ repos:
       sub_dir: python/cugraph
       depends: [cugraph]
       args: {cmake: -DFIND_CUGRAPH_CPP=ON, install: *rapids_build_backend_args}
-    - name: cugraph-dgl
-      sub_dir: python/cugraph-dgl
-      args: {install: *rapids_build_backend_args}
     - name: cugraph-equivariant
       sub_dir: python/cugraph-equivariant
       args: {install: *rapids_build_backend_args}
@@ -240,6 +222,27 @@ repos:
       sub_dir: python/cugraph-service/server
       args: {install: *rapids_build_backend_args}
 
+- name: cugraph-gnn
+  path: cugraph-gnn
+  git: {<<: *git_defaults, repo: cugraph-gnn}
+  cpp:
+    - name: wholegraph
+      sub_dir: cpp
+      depends: [raft]
+      args:
+        cmake: -DCMAKE_CUDA_ARCHITECTURES="${CUDAARCHS}"
+  python:
+    - name: wholegraph
+      sub_dir: python/pylibwholegraph
+      depends: [wholegraph]
+      args: {install: *rapids_build_backend_args}
+    - name: cugraph-dgl
+      sub_dir: python/cugraph-dgl
+      args: {install: *rapids_build_backend_args}
+    - name: cugraph_pyg
+      sub_dir: python/cugraph-pyg
+      args: {install: *rapids_build_backend_args}
+
 - name: nx-cugraph
   path: nx-cugraph
   git: {<<: *git_defaults, repo: nx-cugraph}