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

Export exception to nvbench namespace

Browse Source
This commit is contained in:
Oleksandr Pavlyk
2025-08-04 12:00:42 -05:00
parent 9dfdd8af89
commit d8b0acc8d4
2 changed files with 5 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
python/cuda/nvbench/__init__.py
View File

@@ -43,6 +43,9 @@ from cuda.nvbench._nvbench import ( # noqa: E402
from cuda.nvbench._nvbench import ( # noqa: E402
Launch as Launch,
)
from cuda.nvbench._nvbench import ( # noqa: E402
NVBenchRuntimeError as NVBenchRuntimeError,
)
from cuda.nvbench._nvbench import ( # noqa: E402
State as State,
)

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

@@ -33,7 +33,7 @@ class CudaStream:
Note
----
The class is not directly constructible.
The class is not user-constructible.
"""
def __cuda_stream__(self) -> tuple[int, int]:
"""
@@ -65,6 +65,7 @@ class Benchmark:
Note
----
The class is not user-constructible.
Use `~register` function to create Benchmark and register
it with NVBench.
"""