move CUDA libs deps to wheels, ignore deprecation warning

rapidsai · Dec 6, 2024 · fce4fcf · fce4fcf
1 parent 486bf96
commit fce4fcf
Show file tree

Hide file tree

Showing 4 changed files with 6 additions and 5 deletions.
diff --git a/dependencies.yaml b/dependencies.yaml
@@ -134,6 +134,7 @@ files:
     extras:
       table: project
     includes:
+      - cuda_wheels
       - depends_on_rmm
   py_build_pylibcugraph:
     output: pyproject
@@ -161,7 +162,6 @@ files:
     extras:
       table: project
     includes:
-      - cuda_wheels
       - depends_on_libcugraph
       - depends_on_rmm
       - depends_on_pylibraft

diff --git a/python/cugraph/pytest.ini b/python/cugraph/pytest.ini
@@ -62,6 +62,7 @@ filterwarnings =
           error::FutureWarning
           error::DeprecationWarning
           # TODO
+          ignore:.*cuda..* module is deprecated.*:DeprecationWarning
           ignore:Multi is deprecated and the removal of multi edges will no longer be supported:FutureWarning
           ignore:The legacy column names:FutureWarning
           ignore:The include_hop_column flag is deprecated and will be removed:FutureWarning

diff --git a/python/libcugraph/pyproject.toml b/python/libcugraph/pyproject.toml
@@ -39,6 +39,10 @@ classifiers = [
     "Environment :: GPU :: NVIDIA CUDA",
 ]
 dependencies = [
+    "nvidia-cublas",
+    "nvidia-curand",
+    "nvidia-cusolver",
+    "nvidia-cusparse",
     "rmm==25.2.*,>=0.0.0a0",
 ] # This list was generated by `rapids-dependency-file-generator`. To make changes, edit ../../dependencies.yaml and run `rapids-dependency-file-generator`.
 

diff --git a/python/pylibcugraph/pyproject.toml b/python/pylibcugraph/pyproject.toml
@@ -24,10 +24,6 @@ license = { text = "Apache 2.0" }
 requires-python = ">=3.10"
 dependencies = [
     "libcugraph==25.2.*,>=0.0.0a0",
-    "nvidia-cublas",
-    "nvidia-curand",
-    "nvidia-cusolver",
-    "nvidia-cusparse",
     "pylibraft==25.2.*,>=0.0.0a0",
     "rmm==25.2.*,>=0.0.0a0",
 ] # This list was generated by `rapids-dependency-file-generator`. To make changes, edit ../../dependencies.yaml and run `rapids-dependency-file-generator`.