NVIDIA/nvbench
1
0
Fork 0
mirror of https://github.com/NVIDIA/nvbench.git synced 2026-04-20 14:58:54 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add multi-cuda wheel build (#289)

Browse Source 
Co-authored-by: Ashwin Srinath <shwina@users.noreply.github.com>
Co-authored-by: Nader Al Awar <naderalawar@gmail.com>
This commit is contained in:
Ashwin Srinath
2026-01-28 10:37:55 -05:00
committed by GitHub
parent f3fa93f388
commit a681e2185d
13 changed files with 379 additions and 288 deletions
Download Patch File Download Diff File Expand all files Collapse all files

24
python/pyproject.toml
View File

@@ -22,22 +22,26 @@ classifiers = [
]
requires-python = ">=3.10"
dependencies = [
# pathfinder
# pathfinder for finding CUDA libraries
"cuda-pathfinder",
# Library expects to find shared libraries
# libcupti, libnvperf_target, libnvperf_host
# pathfinder is used to find it in the Python layout
"nvidia-cuda-cupti-cu12",
# The shared library
# libnvidia-ml must be installed system-wide
# (Debian package provider: libnvidia-compute)
]
dynamic = ["version"]
readme = { file = "README.md", content-type = "text/markdown" }
[project.optional-dependencies]
# CUDA 12.x dependencies
cu12 = ["cuda-bindings>=12.0.0,<13.0.0", "nvidia-cuda-cupti-cu12"]
# CUDA 13.x dependencies
cu13 = ["cuda-bindings>=13.0.0,<14.0.0", "nvidia-cuda-cupti>=13.0"]
# Test dependencies for CUDA 12
test-cu12 = ["pynvbench[cu12]", "pytest", "cupy-cuda12x", "numba"]
# Test dependencies for CUDA 13
test-cu13 = ["pynvbench[cu13]", "pytest", "cupy-cuda13x", "numba"]
# Generic test dependencies (defaults to CUDA 12)
test = ["pytest", "cupy-cuda12x", "numba"]
[project.urls]