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

Rename NVBenchRuntimeException to NVBenchRuntimeError

Browse Source 
Added exception to __init__.pyi
This commit is contained in:
Oleksandr Pavlyk
2025-07-25 16:58:02 -05:00
parent b97e27cbf2
commit b6821b7624
2 changed files with 6 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
python/cuda/nvbench/__init__.pyi
View File

@@ -261,3 +261,8 @@ def run_all_benchmarks(argv: Sequence[str]) -> None:
Sequence of CLI arguments controlling NVBench. Usually, it is `sys.argv`.
"""
...
class NVBenchRuntimeError(RuntimeError):
"""An exception raised if running benchmarks encounters an error"""
...

2
python/src/py_nvbench.cpp
View File

@@ -595,7 +595,7 @@ PYBIND11_MODULE(_nvbench, m)
// Use handle to take a memory leak here, since this object's destructor may be called after
// interpreter has shut down
benchmark_exc =
py::exception<nvbench_run_error>(m, "NVBenchRuntimeException", PyExc_RuntimeError).release();
py::exception<nvbench_run_error>(m, "NVBenchRuntimeError", PyExc_RuntimeError).release();
// == STEP 6
// ATTN: nvbench::benchmark_manager is a singleton