habitat.analysis.metrics#

Module Contents#

Classes#

_MetricInfo

Metric

Generic enumeration.

Functions#

resolve_metrics(metrics, device)

Converts Metric enum values into raw metric strings that can be passed to

_get_metric_name(metric, device)

class habitat.analysis.metrics._MetricInfo(cupti_name, legacy_cupti_name, legacy_to_canonical_fn)[source]#
property cupti_name[source]#
property legacy_cupti_name[source]#
to_canonical_value(value, device)[source]#
class habitat.analysis.metrics.Metric[source]#

Bases: enum.Enum

Generic enumeration.

Derive from this class to define new enumerations.

DRAMUtilization[source]#
DRAMReadBytes[source]#
DRAMWriteBytes[source]#
SinglePrecisionFLOPEfficiency[source]#
SinglePrecisionAddOps[source]#
habitat.analysis.metrics.resolve_metrics(metrics, device)[source]#

Converts Metric enum values into raw metric strings that can be passed to CUPTI, depending on the compute capability of the given device.

This is needed because the metrics names changed after (and including) compute capability 7.0 (Volta).

If the metrics passed in are already resolved, this function will return a copy of them.

habitat.analysis.metrics._get_metric_name(metric, device)[source]#