mirror of
https://github.com/NVIDIA/nvbench.git
synced 2026-04-20 06:48:53 +00:00
Remove cupti from cuda-bench dependencies
This commit is contained in:
@@ -18,10 +18,6 @@ import importlib
|
||||
import importlib.metadata
|
||||
import warnings
|
||||
|
||||
from cuda.pathfinder import ( # type: ignore[import-not-found]
|
||||
load_nvidia_dynamic_lib,
|
||||
)
|
||||
|
||||
try:
|
||||
__version__ = importlib.metadata.version("cuda-bench")
|
||||
except Exception as e:
|
||||
@@ -65,10 +61,6 @@ except ImportError as e:
|
||||
f"Original error: {e}"
|
||||
)
|
||||
|
||||
# Load required NVIDIA libraries
|
||||
for libname in ("cupti", "nvperf_target", "nvperf_host"):
|
||||
load_nvidia_dynamic_lib(libname)
|
||||
|
||||
# Import and expose all public symbols from the CUDA-specific extension
|
||||
Benchmark = _nvbench_module.Benchmark
|
||||
CudaStream = _nvbench_module.CudaStream
|
||||
|
||||
@@ -240,12 +240,6 @@ class State:
|
||||
def set_blocking_kernel_timeout(self, duration: SupportsFloat) -> None:
|
||||
"Set time-out value for execution of blocking kernel, in seconds"
|
||||
...
|
||||
def collect_cupti_metrics(self) -> None:
|
||||
"Request NVBench to record CUPTI metrics while running benchmark for this configuration"
|
||||
...
|
||||
def is_cupti_required(self) -> bool:
|
||||
"True if (some) CUPTI metrics are being collected"
|
||||
...
|
||||
def exec(
|
||||
self,
|
||||
fn: Callable[[Launch], None],
|
||||
|
||||
Reference in New Issue
Block a user