Skip to content

Commit 03b9c7f

Browse files
Rename NVBenchRuntimeException to NVBenchRuntimeError
Added exception to __init__.pyi
1 parent b169604 commit 03b9c7f

File tree

2 files changed

+6
-1
lines changed

2 files changed

+6
-1
lines changed

python/cuda/nvbench/__init__.pyi

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -261,3 +261,8 @@ def run_all_benchmarks(argv: Sequence[str]) -> None:
261261
Sequence of CLI arguments controlling NVBench. Usually, it is `sys.argv`.
262262
"""
263263
...
264+
265+
class NVBenchRuntimeError(RuntimeError):
266+
"""An exception raised if running benchmarks encounters an error"""
267+
268+
...

python/src/py_nvbench.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -595,7 +595,7 @@ PYBIND11_MODULE(_nvbench, m)
595595
// Use handle to take a memory leak here, since this object's destructor may be called after
596596
// interpreter has shut down
597597
benchmark_exc =
598-
py::exception<nvbench_run_error>(m, "NVBenchRuntimeException", PyExc_RuntimeError).release();
598+
py::exception<nvbench_run_error>(m, "NVBenchRuntimeError", PyExc_RuntimeError).release();
599599
// == STEP 6
600600
// ATTN: nvbench::benchmark_manager is a singleton
601601

0 commit comments

Comments
 (0)