NVIDIA/nvbench
1
0
Fork 0
mirror of https://github.com/NVIDIA/nvbench.git synced 2026-04-20 06:48:53 +00:00
Code Issues Packages Projects Releases Wiki Activity

Import and reexport symbols from _nvbench one-by-one

Browse Source
This commit is contained in:
Oleksandr Pavlyk
2025-07-01 09:23:45 -05:00
parent b88cc78aeb
commit c184549cda
1 changed files with 16 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -12,7 +12,20 @@ except Exception:
for libname in ("cupti", "nvperf_target", "nvperf_host"):
_load_nvidia_dynamic_library(libname)
from ._nvbench import * # noqa: E402, F403
from ._nvbench import register, run_all_benchmarks # noqa: E402
from ._nvbench import ( # noqa: E402
Benchmark,
CudaStream,
Launch,
State,
register,
run_all_benchmarks,
)
__all__ = ["register", "run_all_benchmarks"]
__all__ = [
"register",
"run_all_benchmarks",
"CudaStream",
"Launch",
"State",
"Benchmark",
]