habitat.profiling.kernel#

Module Contents#

Classes#

KernelProfiler

Attributes#

logger

SKIP_METRICS

habitat.profiling.kernel.logger[source]#
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

_under_threshold(kernels)[source]#
habitat.profiling.kernel.SKIP_METRICS[source]#