Skip to content

Commit

Permalink
Adds initial (untested) updates to add conda and pip packaging for cu…
Browse files Browse the repository at this point in the history
…graph-nx.
  • Loading branch information
rlratzel committed Aug 15, 2023
1 parent 0917af8 commit d99c385
Show file tree
Hide file tree
Showing 6 changed files with 316 additions and 142 deletions.
71 changes: 0 additions & 71 deletions conda/environments/all_cuda-118_arch-x86_64.yaml

This file was deleted.

70 changes: 0 additions & 70 deletions conda/environments/all_cuda-120_arch-x86_64.yaml

This file was deleted.

7 changes: 7 additions & 0 deletions conda/recipes/cugraph-nx/build.sh
Original file line number Diff line number Diff line change
@@ -0,0 +1,7 @@
#!/usr/bin/env bash

# Copyright (c) 2023, NVIDIA CORPORATION.

# This assumes the script is executed from the root of the repo directory

./build.sh cugraph-nx
38 changes: 38 additions & 0 deletions conda/recipes/cugraph-nx/meta.yaml
Original file line number Diff line number Diff line change
@@ -0,0 +1,38 @@
# Copyright (c) 2023, NVIDIA CORPORATION.

{% set version = environ.get('GIT_DESCRIBE_TAG', '0.0.0.dev').lstrip('v') %}
{% set minor_version = version.split('.')[0] + '.' + version.split('.')[1] %}
{% set py_version = environ['CONDA_PY'] %}
{% set date_string = environ['RAPIDS_DATE_STRING'] %}

package:
name: cugraph-nx
version: {{ version }}

source:
git_url: ../../..

build:
number: {{ GIT_DESCRIBE_NUMBER }}
build:
number: {{ GIT_DESCRIBE_NUMBER }}
string: py{{ py_version }}_{{ date_string }}_{{ GIT_DESCRIBE_HASH }}_{{ GIT_DESCRIBE_NUMBER }}

requirements:
host:
- python
run:
- pylibcugraph ={{ version }}
- networkx >=3.1
- cupy >=12.0.0
- python

tests:
imports:
- cugraph_nx

about:
home: https://rapids.ai/
license: Apache-2.0
license_file: ../../../LICENSE
summary: cuGraph backend for GPU-accelerated NetworkX
53 changes: 52 additions & 1 deletion dependencies.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -14,12 +14,14 @@ files:
- python_build_cythonize
- python_run_cugraph
- python_run_pylibcugraph
- python_run_cugraph_nx
- python_run_cugraph_dgl
- python_run_cugraph_pyg
- test_notebook
- test_python_common
- test_python_cugraph
- test_python_pylibcugraph
- test_python_cugraph_nx
checks:
output: none
includes:
Expand Down Expand Up @@ -103,6 +105,33 @@ files:
includes:
- test_python_common
- test_python_pylibcugraph


py_build_cugraph_nx:
output: pyproject
pyproject_dir: python/cugraph-nx
extras:
table: build-system
includes:
- python_build_wheel
py_run_cugraph_nx:
output: pyproject
pyproject_dir: python/cugraph-nx
extras:
table: project
includes:
- python_run_cugraph_nx
py_test_cugraph_nx:
output: pyproject
pyproject_dir: python/cugraph-nx
extras:
table: project.optional-dependencies
key: test
includes:
- test_python_common
- test_python_cugraph_nx


py_build_cugraph_dgl:
output: pyproject
pyproject_dir: python/cugraph-dgl
Expand Down Expand Up @@ -340,13 +369,29 @@ dependencies:
- output_types: pyproject
packages:
- &cupy_pip cupy-cuda11x>=12.0.0
- pylibcugraph==23.10.*
- &pylibcugraph pylibcugraph==23.10.*
python_run_pylibcugraph:
common:
- output_types: [conda, pyproject]
packages:
- *pylibraft
- *rmm


python_run_cugraph_nx:
common:
- output_types: [conda, pyproject]
packages:
- networkx>=3.1
- output_types: conda
packages:
- *cupy
- output_types: pyproject
packages:
- *cupy_pip
- *pylibcugraph


python_run_cugraph_dgl:
common:
- output_types: [conda, pyproject]
Expand Down Expand Up @@ -446,6 +491,12 @@ dependencies:
- output_types: [conda, pyproject]
packages:
- *cudf
test_python_cugraph_nx:
common:
- output_types: [conda, pyproject]
packages:
# required for running networkx tests
- pytest-mpl
cugraph_dgl_dev:
common:
- output_types: [conda]
Expand Down
Loading

0 comments on commit d99c385

Please sign in to comment.