NVIDIA/nvbench
1
0
Fork 0
mirror of https://github.com/NVIDIA/nvbench.git synced 2026-04-19 22:38:52 +00:00
Code Issues Packages Projects Releases Wiki Activity

Correct propagating nvbench_main exceptions to Python

Browse Source 
python examples/cpu_only.py --run-once -d 0 --output foo.md

used to trip SystemError, returned a result with an exception set.

It now returns a clean NVBenchmarkError exception.
This commit is contained in:
Oleksandr Pavlyk
2025-07-21 13:31:34 -05:00
parent 9ab642cf69
commit e426368485
1 changed files with 3 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
python/src/py_nvbench.cpp
View File

@@ -192,16 +192,14 @@ public:
}
catch (const std::exception &e)
{
std::stringstream ss;
ss << "Caught exception while running benchmarks: ";
ss << e.what();
const std::string &exc_message = ss.str();
const std::string &exc_message = e.what();
py::set_error(benchmark_exc, exc_message.c_str());
throw py::error_already_set();
}
catch (...)
{
py::set_error(benchmark_exc, "Caught unknown exception in nvbench_main");
throw py::error_already_set();
}
}
};