NVIDIA/nvbench
1
0
Fork 0
mirror of https://github.com/NVIDIA/nvbench.git synced 2026-07-12 10:07:40 +00:00
Code Issues Packages Projects Releases Wiki Activity

Install scripts into cuda/bench.scripts in prefix

Browse Source 
Introduce optional dependency categories in the wheel metadata,
with cuda-bench[tools] encompasing all dependencies of nvbench
scripts.

Closes #393
This commit is contained in:
Oleksandr Pavlyk
2026-06-29 11:49:33 -05:00
parent 02373ec7b2
commit d682a8b0e0
1 changed files with 8 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
python/pyproject.toml
View File

@@ -46,21 +46,15 @@ test-cu13 = ["cuda-bench[cu13]", "pytest", "cupy-cuda13x", "numba"]
# Generic test dependencies (defaults to CUDA 12)
test = ["pytest", "cupy-cuda12x", "numba"]
tools = [
"colorama",
"jsondiff",
"matplotlib",
"numpy",
"pandas",
"seaborn",
"tabulate",
]
compare = ["colorama", "jsondiff", "numpy", "tabulate"]
plot = ["matplotlib", "numpy", "pandas", "seaborn"]
tools = ["cuda-bench[compare,plot]"]
[project.scripts]
nvbench-compare = "scripts.nvbench_compare:main"
nvbench-histogram = "scripts.nvbench_histogram:main"
nvbench-json-summary = "scripts.nvbench_json_summary:main"
nvbench-walltime = "scripts.nvbench_walltime:main"
nvbench-compare = "cuda.bench.scripts.nvbench_compare:main"
nvbench-histogram = "cuda.bench.scripts.nvbench_histogram:main"
nvbench-json-summary = "cuda.bench.scripts.nvbench_json_summary:main"
nvbench-walltime = "cuda.bench.scripts.nvbench_walltime:main"
[project.urls]
Homepage = "https://github.com/NVIDIA/nvbench"
@@ -94,4 +88,4 @@ fallback_version = "0.0.0"
"cuda" = "cuda"
"cuda/bench" = "cuda/bench"
"cuda/bench/results" = "cuda/bench/results"
"scripts" = "scripts"
"cuda/bench/scripts" = "scripts"