habitat.profiling.operation#

Module Contents#

Classes#

OperationProfiler

Functions#

_is_potentially_inplace(fn_name)

Attributes#

logger

POTENTIALLY_INPLACE_FUNCTIONS

habitat.profiling.operation.logger[source]#
class habitat.profiling.operation.OperationProfiler(device, metrics=None, metrics_threshold_ms=0, warm_up=3, measure_for=10)[source]#
measure_operation(func, args, kwargs, record_kernels=True)[source]#
_get_args_for_profiling(args, kwargs, for_inplace=False)[source]#
_clone_tensors(argument, for_inplace)[source]#
_measure_backward(func_name, operation_outputs, record_kernels)[source]#
_measure_backward_engine_strategy(func_name, operation_outputs, record_kernels)[source]#
_measure_backward_torch_strategy(func_name, operation_outputs, record_kernels)[source]#
_to_run_time_measurement(func_name, runnable, record_kernels, run_time_ms=None)[source]#
_measure_ms(runnable)[source]#
habitat.profiling.operation.POTENTIALLY_INPLACE_FUNCTIONS[source]#
habitat.profiling.operation._is_potentially_inplace(fn_name)[source]#