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

Remove cupti from cuda-bench dependencies

Browse Source
This commit is contained in:
Nader Al Awar
2026-02-02 15:37:13 -06:00
parent 34a089f805
commit 6df5fc8c67
5 changed files with 2 additions and 27 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
python/pyproject.toml
View File

@@ -21,19 +21,15 @@ classifiers = [
"Operating System :: POSIX :: Linux",
]
requires-python = ">=3.10"
dependencies = [
# pathfinder for finding CUDA libraries
"cuda-pathfinder",
]
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"]
cu12 = ["cuda-bindings>=12.0.0,<13.0.0"]
# CUDA 13.x dependencies
cu13 = ["cuda-bindings>=13.0.0,<14.0.0", "nvidia-cuda-cupti>=13.0"]
cu13 = ["cuda-bindings>=13.0.0,<14.0.0"]
# Test dependencies for CUDA 12
test-cu12 = ["cuda-bench[cu12]", "pytest", "cupy-cuda12x", "numba"]