habitat.profiling.kernel
#
Module Contents#
Classes#
Attributes#
- class habitat.profiling.kernel.KernelProfiler(device, metrics=None, metrics_threshold_ms=0)[source]#
- measure_kernels(runnable, func_name=None)[source]#
Uses CUPTI to measure the kernels launched by runnable.
- Returns:
A list of MeasuredKernels
- _measure_kernels_raw(runnable, func_name)[source]#
Uses CUPTI to measure the kernels launched by runnable.
- Returns:
- A list of tuples, where
tuple[0] is the raw kernel measurement that should be used for the kernel’s run time
tuple[1] is a list of the raw kernel measurements that contain the metrics requested