NVIDIA/nvbench
1
0
Fork 0
mirror of https://github.com/NVIDIA/nvbench.git synced 2026-04-29 03:01:14 +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 from cuda.nvbench._nvbench import ( # noqa: E402
Launch as Launch, Launch as Launch,
) )
from cuda.nvbench._nvbench import ( # noqa: E402
NVBenchRuntimeError as NVBenchRuntimeError,
)
from cuda.nvbench._nvbench import ( # noqa: E402 from cuda.nvbench._nvbench import ( # noqa: E402
State as State, State as State,
) )

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

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