torchsparsegradutils 0.2.0__tar.gz → 0.2.2__tar.gz

{torchsparsegradutils-0.2.0/torchsparsegradutils.egg-info → torchsparsegradutils-0.2.2}/PKG-INFO

@@ -1,14 +1,16 @@
 Metadata-Version: 2.4
 Name: torchsparsegradutils
-Version: 0.2.0
+Version: 0.2.2
 Summary: A collection of utility functions to work with PyTorch sparse tensors
-Home-page: https://github.com/cai4cai/torchsparsegradutils
-Author: CAI4CAI research group
-Author-email: contact@cai4cai.uk
-License: Apache-2.0
-Keywords: sparse torch utility
+Author-email: CAI4CAI research group <contact@cai4cai.uk>
+License-Expression: Apache-2.0
+Project-URL: Homepage, https://github.com/cai4cai/torchsparsegradutils
+Project-URL: Documentation, https://torchsparsegradutils.readthedocs.io
+Project-URL: Repository, https://github.com/cai4cai/torchsparsegradutils
+Project-URL: Issues, https://github.com/cai4cai/torchsparsegradutils/issues
+Project-URL: Changelog, https://github.com/cai4cai/torchsparsegradutils/releases
+Keywords: sparse,torch,utility
 Classifier: Operating System :: OS Independent
-Classifier: License :: OSI Approved :: Apache Software License
 Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
@@ -17,9 +19,14 @@ Description-Content-Type: text/markdown
 License-File: LICENSE
 Requires-Dist: torch>=2.5
 Requires-Dist: scipy
-Provides-Extra: extras
-Requires-Dist: jax; extra == "extras"
-Requires-Dist: cupy; extra == "extras"
+Requires-Dist: packaging
+Provides-Extra: cupy
+Requires-Dist: cupy-cuda12x>=13.0; extra == "cupy"
+Provides-Extra: jax
+Requires-Dist: jax[cuda12]; extra == "jax"
+Provides-Extra: all
+Requires-Dist: cupy-cuda12x>=13.0; extra == "all"
+Requires-Dist: jax[cuda12]; extra == "all"
 Provides-Extra: docs
 Requires-Dist: sphinx>=7.0.0; extra == "docs"
 Requires-Dist: sphinx-rtd-theme>=1.3.0; extra == "docs"
@@ -28,23 +35,19 @@ Requires-Dist: myst-parser>=2.0.0; extra == "docs"
 Requires-Dist: sphinx-autobuild>=2021.3.14; extra == "docs"
 Requires-Dist: matplotlib>=3.5.0; extra == "docs"
 Requires-Dist: sphinx-autodoc-typehints>=1.24.0; extra == "docs"
-Dynamic: author
-Dynamic: author-email
-Dynamic: classifier
-Dynamic: description
-Dynamic: description-content-type
-Dynamic: home-page
-Dynamic: keywords
-Dynamic: license
+Requires-Dist: sphinxcontrib-bibtex>=2.5.0; extra == "docs"
+Requires-Dist: sphinx-gallery>=0.13.0; extra == "docs"
+Provides-Extra: dev
+Requires-Dist: pytest; extra == "dev"
+Requires-Dist: pytest-cov; extra == "dev"
+Requires-Dist: black; extra == "dev"
+Requires-Dist: isort; extra == "dev"
+Requires-Dist: flake8; extra == "dev"
 Dynamic: license-file
-Dynamic: provides-extra
-Dynamic: requires-dist
-Dynamic: requires-python
-Dynamic: summary
 
 # torchsparsegradutils: Sparsity-preserving gradient utility tools for PyTorch
 
-[![Python tests](https://github.com/cai4cai/torchsparsegradutils/actions/workflows/python-package.yml/badge.svg)](https://github.com/cai4cai/torchsparsegradutils/actions/workflows/python-package.yml) [![License](https://img.shields.io/github/license/cai4cai/torchsparsegradutils)](https://github.com/cai4cai/torchsparsegradutils?tab=Apache-2.0-1-ov-file#readme) [![Code Style: Black](https://img.shields.io/badge/code%20style-black-000000.svg)](https://github.com/psf/black)
+[![PyPI](https://img.shields.io/pypi/v/torchsparsegradutils.svg)](https://pypi.org/project/torchsparsegradutils/) [![Python Versions](https://img.shields.io/pypi/pyversions/torchsparsegradutils.svg)](https://pypi.org/project/torchsparsegradutils/) [![Downloads](https://img.shields.io/pypi/dm/torchsparsegradutils.svg)](https://pypi.org/project/torchsparsegradutils/) ![PyTorch 2.5+](https://img.shields.io/badge/PyTorch-2.5%2B-ee4c2c?logo=pytorch) ![Tested 2.5 / 2.11 / nightly](https://img.shields.io/badge/Tested-2.5%20|%202.11%20|%20nightly-ee4c2c?logo=pytorch) [![Build](https://github.com/cai4cai/torchsparsegradutils/actions/workflows/python-package.yml/badge.svg)](https://github.com/cai4cai/torchsparsegradutils/actions/workflows/python-package.yml) [![Docs](https://readthedocs.org/projects/torchsparsegradutils/badge/?version=latest)](https://readthedocs.org/projects/torchsparsegradutils) [![Code Style: Black](https://img.shields.io/badge/code%20style-black-000000.svg)](https://github.com/psf/black) [![License](https://img.shields.io/github/license/cai4cai/torchsparsegradutils)](LICENSE) [![status](https://joss.theoj.org/papers/6da0e92488d06f70c0a03d0a7cbfba7d/status.svg)](https://joss.theoj.org/papers/6da0e92488d06f70c0a03d0a7cbfba7d)
 
 A comprehensive collection of utility functions to work with PyTorch sparse tensors, ensuring memory efficiency and supporting various sparsity-preserving tensor operations with automatic differentiation. This package addresses fundamental gaps in PyTorch's sparse tensor ecosystem, providing essential operations that preserve sparsity in gradients during backpropagation.
 
@@ -135,17 +138,21 @@ pip install git+https://github.com/cai4cai/torchsparsegradutils
 For full functionality, install optional dependencies:
 
 ```bash
-# For CuPy sparse solver support (GPU acceleration)
-pip install cupy-cuda12x  # Replace with your CUDA version
+# For CuPy sparse solver support (GPU acceleration, requires CUDA 12.x)
+pip install torchsparsegradutils[cupy]
 
 # For JAX sparse solver support
-pip install "jax[cpu]"     # CPU version
-pip install "jax[cuda12]"  # GPU version (replace with your CUDA version)
+pip install torchsparsegradutils[jax]
+
+# Install all optional dependencies
+pip install torchsparsegradutils[all]
 
 # For benchmarking and testing
 pip install scipy matplotlib pandas tqdm pytest
 ```
 
+> **Note:** The CuPy extra installs `cupy-cuda12x>=13.0`. If you are using a different CUDA version, install the appropriate CuPy package manually (e.g. `pip install cupy-cuda11x`).
+
 ### Requirements
 
 - **Python**: ≥ 3.10
@@ -156,13 +163,13 @@ pip install scipy matplotlib pandas tqdm pytest
 
 ## 📊 Performance Benchmarks
 
-Our comprehensive benchmark suite demonstrates significant performance improvements across various sparse operations. All benchmarks were conducted on an NVIDIA GeForce RTX 4090 with PyTorch 2.8.0+cu128. Benchmarks are performed using [Rothberg/cfd2](https://suitesparse-collection-website.herokuapp.com/Rothberg/cfd2) matrix from [SuiteSparse Matrix Collection](https://suitesparse-collection-website.herokuapp.com/)
+Our comprehensive benchmark suite demonstrates significant performance improvements across various sparse operations. All benchmarks were conducted on an NVIDIA GeForce RTX 4090 with PyTorch 2.9.0+cu130. Benchmarks are performed using [Rothberg/cfd2](https://suitesparse-collection-website.herokuapp.com/Rothberg/cfd2) matrix from [SuiteSparse Matrix Collection](https://suitesparse-collection-website.herokuapp.com/)
 
 ![Sparse MM Suite Performance (int32/float32 COO)](torchsparsegradutils/benchmarks/benchmark_visualizations/sparse_mm_suite_performance_int32_float32_coo.png)
 
 ![Sparse Triangular Solve Suite Performance (int32/float32 COO)](torchsparsegradutils/benchmarks/benchmark_visualizations/triangular_solve_suitesparse_performance_int32_float32_coo.png)
 
-![Sparse Genertic Solve Suite Performance (int32/float32 COO)](torchsparsegradutils/benchmarks/benchmark_visualizations/sparse_solve_suite_performance_int32_float32_coo.png)
+![Sparse Generic Solve Suite Performance (int32/float32 COO)](torchsparsegradutils/benchmarks/benchmark_visualizations/sparse_solve_suite_performance_int32_float32_coo.png)
 
 ## 🚀 Quick Start
 
@@ -543,7 +550,7 @@ For a consistent development environment with GPU support and all dependencies p
 
 **Available Configurations:**
 
-- **`.devcontainer/Dockerfile.stable`** (default): Uses stable PyTorch with CUDA 12.8 support
+- **`.devcontainer/Dockerfile.stable`** (default): Uses stable PyTorch with CUDA 13.0 support
 - **`.devcontainer/Dockerfile.nightly`**: Uses nightly PyTorch builds for latest features
 
 To switch configurations, modify the `dockerfile` field in `.devcontainer/devcontainer.json`:
@@ -555,7 +562,7 @@ To switch configurations, modify the `dockerfile` field in `.devcontainer/devcon
 ```
 
 **What's Included:**
-- **CUDA 12.8**: Full GPU development support with NVIDIA drivers
+- **CUDA 13.0**: Full GPU development support with NVIDIA drivers
 - **Pre-installed Dependencies**: PyTorch, CuPy, JAX, SciPy, and all development tools
 - **VS Code Extensions**: Python, Pylance, Jupyter, GitHub Copilot, and code formatting tools
 - **Development Tools**: pytest, black, flake8, pre-commit hooks

{torchsparsegradutils-0.2.0 → torchsparsegradutils-0.2.2}/README.md

@@ -1,6 +1,6 @@
 # torchsparsegradutils: Sparsity-preserving gradient utility tools for PyTorch
 
-[![Python tests](https://github.com/cai4cai/torchsparsegradutils/actions/workflows/python-package.yml/badge.svg)](https://github.com/cai4cai/torchsparsegradutils/actions/workflows/python-package.yml) [![License](https://img.shields.io/github/license/cai4cai/torchsparsegradutils)](https://github.com/cai4cai/torchsparsegradutils?tab=Apache-2.0-1-ov-file#readme) [![Code Style: Black](https://img.shields.io/badge/code%20style-black-000000.svg)](https://github.com/psf/black)
+[![PyPI](https://img.shields.io/pypi/v/torchsparsegradutils.svg)](https://pypi.org/project/torchsparsegradutils/) [![Python Versions](https://img.shields.io/pypi/pyversions/torchsparsegradutils.svg)](https://pypi.org/project/torchsparsegradutils/) [![Downloads](https://img.shields.io/pypi/dm/torchsparsegradutils.svg)](https://pypi.org/project/torchsparsegradutils/) ![PyTorch 2.5+](https://img.shields.io/badge/PyTorch-2.5%2B-ee4c2c?logo=pytorch) ![Tested 2.5 / 2.11 / nightly](https://img.shields.io/badge/Tested-2.5%20|%202.11%20|%20nightly-ee4c2c?logo=pytorch) [![Build](https://github.com/cai4cai/torchsparsegradutils/actions/workflows/python-package.yml/badge.svg)](https://github.com/cai4cai/torchsparsegradutils/actions/workflows/python-package.yml) [![Docs](https://readthedocs.org/projects/torchsparsegradutils/badge/?version=latest)](https://readthedocs.org/projects/torchsparsegradutils) [![Code Style: Black](https://img.shields.io/badge/code%20style-black-000000.svg)](https://github.com/psf/black) [![License](https://img.shields.io/github/license/cai4cai/torchsparsegradutils)](LICENSE) [![status](https://joss.theoj.org/papers/6da0e92488d06f70c0a03d0a7cbfba7d/status.svg)](https://joss.theoj.org/papers/6da0e92488d06f70c0a03d0a7cbfba7d)
 
 A comprehensive collection of utility functions to work with PyTorch sparse tensors, ensuring memory efficiency and supporting various sparsity-preserving tensor operations with automatic differentiation. This package addresses fundamental gaps in PyTorch's sparse tensor ecosystem, providing essential operations that preserve sparsity in gradients during backpropagation.
 
@@ -91,17 +91,21 @@ pip install git+https://github.com/cai4cai/torchsparsegradutils
 For full functionality, install optional dependencies:
 
 ```bash
-# For CuPy sparse solver support (GPU acceleration)
-pip install cupy-cuda12x  # Replace with your CUDA version
+# For CuPy sparse solver support (GPU acceleration, requires CUDA 12.x)
+pip install torchsparsegradutils[cupy]
 
 # For JAX sparse solver support
-pip install "jax[cpu]"     # CPU version
-pip install "jax[cuda12]"  # GPU version (replace with your CUDA version)
+pip install torchsparsegradutils[jax]
+
+# Install all optional dependencies
+pip install torchsparsegradutils[all]
 
 # For benchmarking and testing
 pip install scipy matplotlib pandas tqdm pytest
 ```
 
+> **Note:** The CuPy extra installs `cupy-cuda12x>=13.0`. If you are using a different CUDA version, install the appropriate CuPy package manually (e.g. `pip install cupy-cuda11x`).
+
 ### Requirements
 
 - **Python**: ≥ 3.10
@@ -112,13 +116,13 @@ pip install scipy matplotlib pandas tqdm pytest
 
 ## 📊 Performance Benchmarks
 
-Our comprehensive benchmark suite demonstrates significant performance improvements across various sparse operations. All benchmarks were conducted on an NVIDIA GeForce RTX 4090 with PyTorch 2.8.0+cu128. Benchmarks are performed using [Rothberg/cfd2](https://suitesparse-collection-website.herokuapp.com/Rothberg/cfd2) matrix from [SuiteSparse Matrix Collection](https://suitesparse-collection-website.herokuapp.com/)
+Our comprehensive benchmark suite demonstrates significant performance improvements across various sparse operations. All benchmarks were conducted on an NVIDIA GeForce RTX 4090 with PyTorch 2.9.0+cu130. Benchmarks are performed using [Rothberg/cfd2](https://suitesparse-collection-website.herokuapp.com/Rothberg/cfd2) matrix from [SuiteSparse Matrix Collection](https://suitesparse-collection-website.herokuapp.com/)
 
 ![Sparse MM Suite Performance (int32/float32 COO)](torchsparsegradutils/benchmarks/benchmark_visualizations/sparse_mm_suite_performance_int32_float32_coo.png)
 
 ![Sparse Triangular Solve Suite Performance (int32/float32 COO)](torchsparsegradutils/benchmarks/benchmark_visualizations/triangular_solve_suitesparse_performance_int32_float32_coo.png)
 
-![Sparse Genertic Solve Suite Performance (int32/float32 COO)](torchsparsegradutils/benchmarks/benchmark_visualizations/sparse_solve_suite_performance_int32_float32_coo.png)
+![Sparse Generic Solve Suite Performance (int32/float32 COO)](torchsparsegradutils/benchmarks/benchmark_visualizations/sparse_solve_suite_performance_int32_float32_coo.png)
 
 ## 🚀 Quick Start
 
@@ -499,7 +503,7 @@ For a consistent development environment with GPU support and all dependencies p
 
 **Available Configurations:**
 
-- **`.devcontainer/Dockerfile.stable`** (default): Uses stable PyTorch with CUDA 12.8 support
+- **`.devcontainer/Dockerfile.stable`** (default): Uses stable PyTorch with CUDA 13.0 support
 - **`.devcontainer/Dockerfile.nightly`**: Uses nightly PyTorch builds for latest features
 
 To switch configurations, modify the `dockerfile` field in `.devcontainer/devcontainer.json`:
@@ -511,7 +515,7 @@ To switch configurations, modify the `dockerfile` field in `.devcontainer/devcon
 ```
 
 **What's Included:**
-- **CUDA 12.8**: Full GPU development support with NVIDIA drivers
+- **CUDA 13.0**: Full GPU development support with NVIDIA drivers
 - **Pre-installed Dependencies**: PyTorch, CuPy, JAX, SciPy, and all development tools
 - **VS Code Extensions**: Python, Pylance, Jupyter, GitHub Copilot, and code formatting tools
 - **Development Tools**: pytest, black, flake8, pre-commit hooks
@@ -658,4 +662,4 @@ dist_stable = SparseMultivariateNormal(
 - **No SPD Constraints**: Doesn't require strict positive definiteness
 - **Better Conditioning**: Diagonal component can be controlled independently
 
-**Status**: This is a known limitation of the LL^T precision formulation. LDL^T parameterization is the recommended approach for precision matrices.
+**Status**: This is a known limitation of the LL^T precision formulation. LDL^T parameterization is the recommended approach for precision matrices.

torchsparsegradutils-0.2.2/docs/source/conf.py

@@ -0,0 +1,187 @@
+# Configuration file for the Sphinx documentation builder.
+#
+# For the full list of built-in configuration values, see the documentation:
+# https://www.sphinx-doc.org/en/master/usage/configuration.html
+
+# -- Project information -----------------------------------------------------
+# https://www.sphinx-doc.org/en/master/usage/configuration.html#project-information
+
+import os
+import sys
+
+# Add the package to the Python path
+sys.path.insert(0, os.path.abspath("../../"))
+
+project = "torchsparsegradutils"
+copyright = "2026, CAI4CAI research group"
+author = "CAI4CAI research group"
+release = "0.2.2"
+version = "0.2.2"
+
+# -- General configuration ---------------------------------------------------
+# https://www.sphinx-doc.org/en/master/usage/configuration.html#general-configuration
+
+extensions = [
+    "sphinx.ext.autodoc",  # Automatic documentation from docstrings
+    "sphinx.ext.autosummary",  # Generate autodoc summaries
+    "sphinx.ext.napoleon",  # Support for NumPy and Google style docstrings
+    "sphinx.ext.viewcode",  # Add links to highlighted source code
+    "sphinx.ext.mathjax",  # Math notation support
+    "sphinx.ext.intersphinx",  # Link to other project's documentation
+    "sphinx.ext.coverage",  # Documentation coverage
+    "sphinx.ext.githubpages",  # GitHub Pages support
+    "sphinx_copybutton",  # Add copy button to code blocks
+    "myst_parser",  # Markdown support
+]
+
+# Autosummary settings
+autosummary_generate = True
+autosummary_imported_members = True
+
+# Napoleon settings for docstring parsing
+napoleon_google_docstring = True
+napoleon_numpy_docstring = True
+napoleon_include_init_with_doc = False
+napoleon_include_private_with_doc = False
+napoleon_include_special_with_doc = True
+napoleon_use_admonition_for_examples = False
+napoleon_use_admonition_for_notes = False
+napoleon_use_admonition_for_references = False
+napoleon_use_ivar = False
+napoleon_use_param = True
+napoleon_use_rtype = True
+napoleon_preprocess_types = False
+napoleon_type_aliases = None
+napoleon_attr_annotations = True
+
+# Autodoc settings
+autodoc_default_options = {
+    "members": True,
+    "member-order": "bysource",
+    "special-members": "__init__",
+    "undoc-members": True,
+    "exclude-members": "__weakref__",
+}
+
+# Intersphinx mapping
+intersphinx_mapping = {
+    "python": ("https://docs.python.org/3/", None),
+    "torch": ("https://pytorch.org/docs/stable/", None),
+    "numpy": ("https://numpy.org/doc/stable/", None),
+    "scipy": ("https://docs.scipy.org/doc/scipy/", None),
+    "jax": ("https://jax.readthedocs.io/en/latest/", None),
+    "cupy": ("https://docs.cupy.dev/en/stable/", None),
+}
+
+templates_path = ["_templates"]
+exclude_patterns = []
+
+# Source file parsers
+source_suffix = {
+    ".rst": None,
+    ".md": "myst_parser",
+}
+
+# -- Options for HTML output -------------------------------------------------
+# https://www.sphinx-doc.org/en/master/usage/configuration.html#options-for-html-output
+
+html_theme = "sphinx_rtd_theme"
+html_static_path = ["_static"]
+
+# Theme options
+html_theme_options = {
+    "canonical_url": "",
+    "analytics_id": "",
+    "logo_only": False,
+    "prev_next_buttons_location": "bottom",
+    "style_external_links": False,
+    "vcs_pageview_mode": "",
+    "style_nav_header_background": "#2980B9",
+    # Toc options
+    "collapse_navigation": True,
+    "sticky_navigation": True,
+    "navigation_depth": 4,
+    "includehidden": True,
+    "titles_only": False,
+}
+
+# Add any paths that contain custom static files (such as style sheets) here,
+# relative to this directory. They are copied after the builtin static files,
+# so a file named "default.css" will overwrite the builtin "default.css".
+html_static_path = ["_static"]
+
+# Custom CSS
+html_css_files = [
+    "custom.css",
+]
+
+# HTML title
+html_title = f"{project} v{version}"
+
+# Favicon
+# html_favicon = '_static/favicon.ico'
+
+# Logo
+# html_logo = '_static/logo.png'
+
+# -- Options for LaTeX output ------------------------------------------------
+latex_engine = "pdflatex"
+latex_elements = {
+    # The paper size ('letterpaper' or 'a4paper').
+    "papersize": "letterpaper",
+    # The font size ('10pt', '11pt' or '12pt').
+    "pointsize": "10pt",
+    # Additional stuff for the LaTeX preamble.
+    "preamble": r"""
+\usepackage{amsmath}
+\usepackage{amsfonts}
+\usepackage{amssymb}
+\usepackage{mathrsfs}
+""",
+}
+
+# Grouping the document tree into LaTeX files. List of tuples
+# (source start file, target name, title, author, documentclass [howto, manual, or own class]).
+latex_documents = [
+    ("index", "torchsparsegradutils.tex", "torchsparsegradutils Documentation", "CAI4CAI research group", "manual"),
+]
+
+# -- Options for manual page output ------------------------------------------
+# One entry per manual page. List of tuples
+# (source start file, name, description, authors, manual section).
+man_pages = [("index", "torchsparsegradutils", "torchsparsegradutils Documentation", [author], 1)]
+
+# -- Options for Texinfo output ----------------------------------------------
+# Grouping the document tree into Texinfo files. List of tuples
+# (source start file, target name, title, author,
+#  dir menu entry, description, category)
+texinfo_documents = [
+    (
+        "index",
+        "torchsparsegradutils",
+        "torchsparsegradutils Documentation",
+        author,
+        "torchsparsegradutils",
+        "PyTorch sparse tensor utilities.",
+        "Miscellaneous",
+    ),
+]
+
+# -- Extension configuration -------------------------------------------------
+# Copy button configuration
+copybutton_prompt_text = r">>> |\.\.\. |\$ |In \[\d*\]: | {2,5}\.\.\.: | {5,8}: "
+copybutton_prompt_is_regexp = True
+
+# Math notation
+mathjax3_config = {
+    "tex": {
+        "inlineMath": [["$", "$"], ["\\(", "\\)"]],
+        "displayMath": [["$$", "$$"], ["\\[", "\\]"]],
+        "processEscapes": True,
+        "processEnvironments": True,
+    },
+    "options": {
+        "ignoreHtmlClass": "tex2jax_ignore",
+        "processHtmlClass": "tex2jax_process",
+    },
+}

torchsparsegradutils-0.2.2/pyproject.toml

@@ -0,0 +1,116 @@
+[build-system]
+requires = ["setuptools>=61.0", "wheel"]
+build-backend = "setuptools.build_meta"
+
+[project]
+name = "torchsparsegradutils"
+version = "0.2.2"
+description = "A collection of utility functions to work with PyTorch sparse tensors"
+readme = "README.md"
+requires-python = ">=3.10"
+license = "Apache-2.0"
+authors = [
+    {name = "CAI4CAI research group", email = "contact@cai4cai.uk"}
+]
+keywords = ["sparse", "torch", "utility"]
+classifiers = [
+    "Operating System :: OS Independent",
+    "Programming Language :: Python :: 3.10",
+    "Programming Language :: Python :: 3.11",
+    "Programming Language :: Python :: 3.12",
+]
+dependencies = [
+    "torch>=2.5",
+    "scipy",
+    "packaging",
+]
+
+[project.optional-dependencies]
+cupy = ["cupy-cuda12x>=13.0"]
+jax = ["jax[cuda12]"]
+all = ["cupy-cuda12x>=13.0", "jax[cuda12]"]
+docs = [
+    "sphinx>=7.0.0",
+    "sphinx-rtd-theme>=1.3.0",
+    "sphinx-copybutton>=0.5.0",
+    "myst-parser>=2.0.0",
+    "sphinx-autobuild>=2021.3.14",
+    "matplotlib>=3.5.0",
+    "sphinx-autodoc-typehints>=1.24.0",
+    "sphinxcontrib-bibtex>=2.5.0",
+    "sphinx-gallery>=0.13.0",
+]
+dev = [
+    "pytest",
+    "pytest-cov",
+    "black",
+    "isort",
+    "flake8",
+]
+
+[project.urls]
+Homepage = "https://github.com/cai4cai/torchsparsegradutils"
+Documentation = "https://torchsparsegradutils.readthedocs.io"
+Repository = "https://github.com/cai4cai/torchsparsegradutils"
+Issues = "https://github.com/cai4cai/torchsparsegradutils/issues"
+Changelog = "https://github.com/cai4cai/torchsparsegradutils/releases"
+
+[tool.setuptools]
+zip-safe = false
+include-package-data = true
+
+[tool.setuptools.packages.find]
+exclude = ["tests*"]
+
+[tool.black]
+line-length = 120
+target-version = ['py310', 'py311', 'py312']
+include = '\.pyi?$'
+exclude = '''
+(
+  /(
+    # exclude a few common directories in the root of the project
+      \.eggs
+    | \.git
+    | \.hg
+    | \.mypy_cache
+    | \.tox
+    | \.venv
+    | \.pytype
+    | _build
+    | buck-out
+    | build
+    | dist
+  )/
+  # also separately exclude other files if needed
+  #| some_file
+)
+'''
+
+[tool.isort]
+profile = "black"
+multi_line_output = 3
+line_length = 120
+known_first_party = ["torchsparsegradutils"]
+force_grid_wrap = 0
+combine_as_imports = true
+include_trailing_comma = true
+
+[tool.pytest.ini_options]
+minversion = "6.0"
+addopts = ["-v"]
+testpaths = [
+    "torchsparsegradutils/tests",
+]
+pythonpath = [
+    "torchsparsegradutils/tests",
+]
+doctest_optionflags = [
+    "NORMALIZE_WHITESPACE",
+    "IGNORE_EXCEPTION_DETAIL",
+    "ELLIPSIS"
+]
+# Separate command for doctests due to potential issues
+markers = [
+    "slow: marks tests as slow (deselect with '-m \"not slow\"')",
+]

torchsparsegradutils-0.2.2/setup.py

@@ -0,0 +1,5 @@
+# Minimal setup.py for backward compatibility
+# All configuration is now in pyproject.toml
+from setuptools import setup
+
+setup()

{torchsparsegradutils-0.2.0 → torchsparsegradutils-0.2.2}/torchsparsegradutils/cupy/__init__.py

@@ -11,7 +11,16 @@ if cupy_spec is None:
 else:
     have_cupy = True
 
-from .cupy_bindings import _get_array_modules, c2t_coo, c2t_csr, t2c_coo, t2c_csr
+from .cupy_bindings import _backend_to_torch, _get_array_modules, _torch_to_backend, c2t_coo, c2t_csr, t2c_coo, t2c_csr
 from .cupy_sparse_solve import sparse_solve_c4t
 
-__all__ = ["c2t_coo", "t2c_coo", "c2t_csr", "t2c_csr", "_get_array_modules", "sparse_solve_c4t"]
+__all__ = [
+    "c2t_coo",
+    "t2c_coo",
+    "c2t_csr",
+    "t2c_csr",
+    "_get_array_modules",
+    "_torch_to_backend",
+    "_backend_to_torch",
+    "sparse_solve_c4t",
+]

{torchsparsegradutils-0.2.0 → torchsparsegradutils-0.2.2}/torchsparsegradutils/cupy/cupy_bindings.py

@@ -38,6 +38,20 @@ import scipy.sparse.linalg
 import torch
 
 
+def _torch_to_backend(tensor: torch.Tensor, xp) -> Any:
+    r"""Convert a PyTorch tensor to a backend array (NumPy or CuPy) using DLPack for CUDA."""
+    if tsgucupy.have_cupy and xp is cp:
+        return cp.from_dlpack(tensor)
+    return xp.asarray(tensor)
+
+
+def _backend_to_torch(array) -> torch.Tensor:
+    r"""Convert a backend array (NumPy or CuPy) to a PyTorch tensor using DLPack for CUDA."""
+    if tsgucupy.have_cupy and isinstance(array, cp.ndarray):
+        return torch.from_dlpack(array)
+    return torch.as_tensor(array)
+
+
 def _get_array_modules(x: Any) -> Tuple[Any, Any]:
     r"""
     Select dense & sparse array modules (NumPy/SciPy or CuPy/cupyx.scipy.sparse).
@@ -116,9 +130,9 @@ def t2c_csr(x_torch: torch.Tensor) -> Any:
     """
     xp, xsp = _get_array_modules(x_torch)
 
-    data_c = xp.asarray(x_torch.values())
-    col_idx_c = xp.asarray(x_torch.col_indices())
-    ind_ptr_c = xp.asarray(x_torch.crow_indices())
+    data_c = _torch_to_backend(x_torch.values(), xp)
+    col_idx_c = _torch_to_backend(x_torch.col_indices(), xp)
+    ind_ptr_c = _torch_to_backend(x_torch.crow_indices(), xp)
     x_cupy = xsp.csr_matrix((data_c, col_idx_c, ind_ptr_c), shape=x_torch.shape)
     return x_cupy
 
@@ -150,9 +164,9 @@ def c2t_csr(x_cupy: Any) -> torch.Tensor:
     >>> x.layout is torch.sparse_csr
     True
     """
-    data_t = torch.as_tensor(x_cupy.data)
-    idices_t = torch.as_tensor(x_cupy.indices)
-    ind_ptr_t = torch.as_tensor(x_cupy.indptr)
+    data_t = _backend_to_torch(x_cupy.data)
+    idices_t = _backend_to_torch(x_cupy.indices)
+    ind_ptr_t = _backend_to_torch(x_cupy.indptr)
     x_torch = torch.sparse_csr_tensor(ind_ptr_t, idices_t, data_t, x_cupy.shape)
     return x_torch
 
@@ -205,8 +219,8 @@ def t2c_coo(x_torch: torch.Tensor) -> Any:
             "Requested a conversion from torch to cupy/numpy on a non-coalesced tensor -> coalescing implicitly"
         )
         x_torch = x_torch.coalesce()
-    data_c = xp.asarray(x_torch.values())
-    idx_cp = xp.asarray(x_torch.indices())
+    data_c = _torch_to_backend(x_torch.values(), xp)
+    idx_cp = _torch_to_backend(x_torch.indices(), xp)
     x_cupy = xsp.coo_matrix((data_c, idx_cp), shape=x_torch.shape)
     return x_cupy
 
@@ -238,8 +252,8 @@ def c2t_coo(x_cupy: Any) -> torch.Tensor:
     >>> x.layout is torch.sparse_coo
     True
     """
-    data_t = torch.as_tensor(x_cupy.data)
-    row_t = torch.as_tensor(x_cupy.row)
-    col_t = torch.as_tensor(x_cupy.col)
+    data_t = _backend_to_torch(x_cupy.data)
+    row_t = _backend_to_torch(x_cupy.row)
+    col_t = _backend_to_torch(x_cupy.col)
     x_torch = torch.sparse_coo_tensor(torch.stack([row_t, col_t], dim=0), data_t, x_cupy.shape)
     return x_torch