Remove cupti from cuda-bench dependencies

This commit is contained in:
Nader Al Awar
2026-02-02 15:37:13 -06:00
parent 34a089f805
commit 6df5fc8c67
5 changed files with 2 additions and 27 deletions

View File

@@ -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

View File

@@ -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],