Introduce cuda.parallel module (#2276)

NVIDIA · Aug 23, 2024 · d62e979 · d62e979
1 parent 529f910
commit d62e979
Show file tree

Hide file tree

Showing 16 changed files with 604 additions and 7 deletions.
diff --git a/.github/actions/docs-build/action.yml b/.github/actions/docs-build/action.yml
@@ -38,6 +38,8 @@ runs:
         cp -rf ./docs/_build/docs/cudax/latest/* _site/cudax
         mkdir _site/cuda_cooperative
         cp -rf ./docs/_build/docs/cuda_cooperative/latest/* _site/cuda_cooperative
+        mkdir _site/cuda_parallel
+        cp -rf ./docs/_build/docs/cuda_parallel/latest/* _site/cuda_parallel
         ./docs/scrape_docs.bash ./_site
 
     # Update docs as workflow artifact:

diff --git a/c/include/cccl/types.h b/c/include/cccl/types.h
@@ -44,8 +44,8 @@ struct cccl_type_info
 
 enum class cccl_op_kind_t
 {
-  stateless,
-  stateful
+  stateless = 0,
+  stateful = 1
 };
 
 struct cccl_op_t
@@ -61,8 +61,8 @@ struct cccl_op_t
 
 enum class cccl_iterator_kind_t
 {
-  pointer,
-  iterator
+  pointer = 0,
+  iterator = 1
 };
 
 struct cccl_value_t

diff --git a/ci/update_version.sh b/ci/update_version.sh
@@ -36,7 +36,8 @@ CUB_CMAKE_VERSION_FILE="cub/cub/cmake/cub-config-version.cmake"
 LIBCUDACXX_CMAKE_VERSION_FILE="libcudacxx/lib/cmake/libcudacxx/libcudacxx-config-version.cmake"
 THRUST_CMAKE_VERSION_FILE="thrust/thrust/cmake/thrust-config-version.cmake"
 CUDAX_CMAKE_VERSION_FILE="cudax/lib/cmake/cudax/cudax-config-version.cmake"
-PYCUDA_VERSION_FILE="python/cuda_cooperative/cuda/cooperative/_version.py"
+CUDA_COOPERATIVE_VERSION_FILE="python/cuda_cooperative/cuda/cooperative/_version.py"
+CUDA_PARALLEL_VERSION_FILE="python/cuda_parallel/cuda/parallel/_version.py"
 
 # Calculated version codes
 new_cccl_version=$((major * 1000000 + minor * 1000 + patch))     # MMMmmmppp
@@ -102,7 +103,8 @@ update_file "$CUDAX_CMAKE_VERSION_FILE" "set(cudax_VERSION_MAJOR \([0-9]\+\))" "
 update_file "$CUDAX_CMAKE_VERSION_FILE" "set(cudax_VERSION_MINOR \([0-9]\+\))" "set(cudax_VERSION_MINOR $minor)"
 update_file "$CUDAX_CMAKE_VERSION_FILE" "set(cudax_VERSION_PATCH \([0-9]\+\))" "set(cudax_VERSION_PATCH $patch)"
 
-update_file "$PYCUDA_VERSION_FILE" "^__version__ = \"\([0-9.]\+\)\"" "__version__ = \"$pymajor.$pyminor.$major.$minor.$patch\""
+update_file "$CUDA_COOPERATIVE_VERSION_FILE" "^__version__ = \"\([0-9.]\+\)\"" "__version__ = \"$pymajor.$pyminor.$major.$minor.$patch\""
+update_file "$CUDA_PARALLEL_VERSION_FILE" "^__version__ = \"\([0-9.]\+\)\"" "__version__ = \"$pymajor.$pyminor.$major.$minor.$patch\""
 
 if [ "$DRY_RUN" = true ]; then
     echo "Dry run completed. No changes made."

diff --git a/docs/cuda_parallel/index.rst b/docs/cuda_parallel/index.rst
@@ -0,0 +1,13 @@
+.. _cuda_parallel-module:
+
+CUDA Parallel
+==================================================
+
+.. warning::
+  Python exposure of parallel algorithms is in public beta.
+  The API is subject to change without notice.
+
+.. automodule:: cuda.parallel.experimental
+  :members:
+  :undoc-members:
+  :imported-members:
diff --git a/docs/python.rst b/docs/python.rst
@@ -8,8 +8,12 @@ CUDA Python Core Libraries
    :maxdepth: 3
 
    cuda.cooperative <https://nvidia.github.io/cccl/cuda_cooperative/>
+   cuda.parallel <https://nvidia.github.io/cccl/cuda_parallel/>
 
 Welcome to the CUDA Core Compute Libraries (CCCL) libraries for Python.
 
 - `cuda.cooperative <https://nvidia.github.io/cccl/cuda_cooperative/>`__
   is a still-experimental library exposing cooperative algorithms to Python.
+
+- `cuda.parallel <https://nvidia.github.io/cccl/cuda_parallel/>`__
+  is a still-experimental library exposing parallel algorithms to Python.
diff --git a/docs/repo.toml b/docs/repo.toml
@@ -25,14 +25,15 @@ sphinx_exclude_patterns = [
     "VERSION.md",
 ]
 
-project_build_order = [ "libcudacxx", "cudax", "cub", "thrust", "cccl", "cuda_cooperative" ]
+project_build_order = [ "libcudacxx", "cudax", "cub", "thrust", "cccl", "cuda_cooperative", "cuda_parallel" ]
 
 # deps can be used to link to other projects' documentation
 deps = [
     [ "libcudacxx", "_build/docs/libcudacxx/latest" ],
     [ "cub", "_build/docs/cub/latest" ],
     [ "thrust", "_build/docs/thrust/latest" ],
     [ "cuda_cooperative", "_build/docs/cuda_cooperative/latest" ],
+    [ "cuda_parallel", "_build/docs/cuda_parallel/latest" ],
 ]
 
 [repo_docs.projects.libcudacxx]
@@ -304,6 +305,29 @@ python_paths = [
     "${root}/../python/cuda_cooperative"
 ]
 
+[repo_docs.projects.cuda_parallel]
+name = "cuda.parallel"
+docs_root = "cuda_parallel"
+logo = "../img/logo.png"
+
+repo_url         = "https://github.com/NVIDIA/cccl/python/cuda"
+social_media_set = ""
+social_media     = [
+    [ "github", "https://github.com/NVIDIA/cccl" ],
+]
+
+autodoc.mock_imports = [
+    "numba",
+    "pynvjitlink",
+    "cuda.nvrtc",
+    "llvmlite"
+]
+
+enhanced_search_enabled = true
+python_paths = [
+    "${root}/../python/cuda_parallel"
+]
+
 [repo_docs.projects.cudax]
 name = "Cudax: Experimental new features"
 docs_root = "cudax"

diff --git a/python/cuda_parallel/.gitignore b/python/cuda_parallel/.gitignore
@@ -0,0 +1,4 @@
+cuda/_include
+env
+*egg-info
+*so
diff --git a/python/cuda_parallel/MANIFEST.in b/python/cuda_parallel/MANIFEST.in
@@ -0,0 +1 @@
+recursive-include cuda/_include *
diff --git a/python/cuda_parallel/README.md b/python/cuda_parallel/README.md
@@ -0,0 +1,12 @@
+# `cuda.parallel`: Experimental CUDA Core Compute Library for Python
+
+## Documentation
+
+Please visit the documentation here: https://nvidia.github.io/cccl/python.html.
+
+## Local development
+
+```bash
+pip3 install -e .[test]
+pytest -v ./tests/
+```
diff --git a/python/cuda_parallel/cuda/parallel/__init__.py b/python/cuda_parallel/cuda/parallel/__init__.py
@@ -0,0 +1,6 @@
+# Copyright (c) 2024, NVIDIA CORPORATION & AFFILIATES. ALL RIGHTS RESERVED.
+#
+# SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
+
+import cuda.parallel.experimental
+from cuda.parallel._version import __version__
diff --git a/python/cuda_parallel/cuda/parallel/_version.py b/python/cuda_parallel/cuda/parallel/_version.py
@@ -0,0 +1,7 @@
+# Copyright (c) 2024, NVIDIA CORPORATION & AFFILIATES. ALL RIGHTS RESERVED.
+#
+# SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
+
+# This file is generated by ci/update_version.sh
+# Do not edit this file manually.
+__version__ = "0.1.2.6.0"