Index

_

__getattr__() (habitat.analysis._Device method) (habitat.analysis.device._Device method) (habitat.analysis.kernels.MeasuredKernel method) __getitem__() (habitat.analysis.mlp.dataset.HabitatDataset method) __len__() (habitat.analysis.mlp.dataset.HabitatDataset method) __repr__() (habitat.analysis.operation.Operation method) _bmm_scale() (habitat.analysis.predictor.Predictor method) (habitat.Predictor method) _clone_tensors() (habitat.profiling.operation.OperationProfiler method) _conv2d_scale() (habitat.analysis.predictor.Predictor method) (habitat.Predictor method) _conv_transpose2d_scale() (habitat.analysis.predictor.Predictor method) (habitat.Predictor method) _DATA_PATH (in module habitat.data) _Device (class in habitat.analysis) (class in habitat.analysis.device) _get_args_for_profiling() (habitat.profiling.operation.OperationProfiler method) _get_metric_name() (in module habitat.analysis.metrics) _hook_creator() (habitat.OperationTracker method) (habitat.tracking.operation.OperationTracker method) _is_callable() (in module habitat.tracking.callable) _is_callable_and_public() (in module habitat.tracking.callable) _is_callable_dunder() (in module habitat.tracking.callable) _is_potentially_inplace() (in module habitat.profiling.operation)

_linear_scale() (habitat.analysis.predictor.Predictor method) (habitat.Predictor method) _load_devices() (habitat.analysis._Device method) (habitat.analysis.device._Device method) _lstm_scale() (habitat.analysis.predictor.Predictor method) (habitat.Predictor method) _maybe_store_callable() (habitat.tracking.hook_manager.HookManager method) _measure_backward() (habitat.profiling.operation.OperationProfiler method) _measure_backward_engine_strategy() (habitat.profiling.operation.OperationProfiler method) _measure_backward_torch_strategy() (habitat.profiling.operation.OperationProfiler method) _measure_kernels_raw() (habitat.profiling.kernel.KernelProfiler method) _measure_ms() (habitat.profiling.operation.OperationProfiler method) _MetricInfo (class in habitat.analysis.metrics) _process_argument() (in module habitat.analysis.arguments) _ridge_point() (in module habitat.analysis.wave_scaling.roofline) _roofline_gamma() (in module habitat.analysis.wave_scaling.roofline) _special_scale() (habitat.analysis.predictor.Predictor method) (habitat.Predictor method) _to_run_time_measurement() (habitat.profiling.operation.OperationProfiler method) _train() (habitat.analysis.mlp.mlp.RuntimePredictor method) _under_threshold() (habitat.profiling.kernel.KernelProfiler method) _validate() (habitat.analysis.mlp.mlp.RuntimePredictor method) _wave_scale() (habitat.analysis.predictor.Predictor method) (habitat.Predictor method)

A

add_common_cmd_args() (in module habitat.utils) Arguments (class in habitat.analysis.arguments) arguments (habitat.analysis.operation.MeasuredOperation property) (habitat.analysis.operation.Operation property) (habitat.analysis.operation.PredictedOperation property)

attach_hook() (habitat.tracking.hook_manager.HookManager method) attach_hooks_on_module() (habitat.tracking.hook_manager.HookManager method) attach_hooks_on_module_using() (habitat.tracking.hook_manager.HookManager method) AutogradEngine (class in habitat.profiling.autograd)

B

backward (habitat.analysis.operation.MeasuredOperation property) (habitat.analysis.operation.Operation property) (habitat.analysis.operation.PredictedOperation property) backward_available() (in module habitat.profiling.backward) BackwardHelper (class in habitat.profiling.backward)

BLACKLISTED_DUNDERS (in module habitat.tracking.callable) BLACKLISTED_TENSOR_METHODS (in module habitat.tracking.callable) BLACKLISTED_TORCH_METHODS (in module habitat.tracking.callable) BMM_PARAMS (in module habitat.analysis.predictor) BMMMLP (class in habitat.analysis.mlp.mlp)

C

calculate_wave_info() (in module habitat.analysis.wave_scaling.common) CallableTracker (class in habitat.tracking.callable) CONV2D_PARAMS (in module habitat.analysis.predictor)

Conv2DMLP (class in habitat.analysis.mlp.mlp) CONVTRANSPOSE2D_PARAMS (in module habitat.analysis.predictor) ConvTranspose2DMLP (class in habitat.analysis.mlp.mlp) cupti_name (habitat.analysis.metrics._MetricInfo property)

D

DefaultPredictor (habitat.analysis.trace.Trace attribute) device (habitat.analysis.operation.MeasuredOperation property) (habitat.analysis.operation.Operation property) (habitat.analysis.operation.PredictedOperation property) (habitat.analysis.run_time.RunTime property) (habitat.analysis.run_time.RunTimeMeasurement property) (habitat.analysis.run_time.RunTimePrediction property) (habitat.analysis.run_time.RunTimePurePrediction property) (habitat.analysis.trace.Trace property)

Device (in module habitat) (in module habitat.analysis) DRAMReadBytes (habitat.analysis.metrics.Metric attribute), [1] (habitat.Metric attribute) DRAMUtilization (habitat.analysis.metrics.Metric attribute), [1] (habitat.Metric attribute) DRAMWriteBytes (habitat.analysis.metrics.Metric attribute), [1] (habitat.Metric attribute)

F

flatten_operation_output() (in module habitat.profiling.backward) forward (habitat.analysis.operation.MeasuredOperation property) (habitat.analysis.operation.Operation property) (habitat.analysis.operation.PredictedOperation property) forward() (habitat.analysis.mlp.mlp.BMMMLP method) (habitat.analysis.mlp.mlp.Conv2DMLP method) (habitat.analysis.mlp.mlp.ConvTranspose2DMLP method) (habitat.analysis.mlp.mlp.LinearMLP method) (habitat.analysis.mlp.mlp.LSTMMLP method) (habitat.analysis.mlp.mlp.MLPBase method)

from_raw_arguments() (habitat.analysis.arguments.Arguments class method)

G

get_accumulate_grad_inputs() (in module habitat.profiling.backward) get_all_devices() (in module habitat.analysis.mlp.devices) get_dataset() (in module habitat.analysis.mlp.dataset_process) get_device_features() (in module habitat.analysis.mlp.devices)

get_grad_fn() (in module habitat.profiling.backward) get_metric() (habitat.analysis.kernels.MeasuredKernel method) get_tracked_trace() (habitat.OperationTracker method) (habitat.tracking.operation.OperationTracker method), [1]

H

habitat module habitat.analysis module habitat.analysis.arguments module habitat.analysis.device module habitat.analysis.kernels module habitat.analysis.metrics module habitat.analysis.mlp module habitat.analysis.mlp.dataset module habitat.analysis.mlp.dataset_process module habitat.analysis.mlp.devices module habitat.analysis.mlp.mlp module habitat.analysis.mlp.train module habitat.analysis.operation module habitat.analysis.predictor module habitat.analysis.run_time module habitat.analysis.trace module habitat.analysis.wave_scaling module habitat.analysis.wave_scaling.common module

habitat.analysis.wave_scaling.metadata module habitat.analysis.wave_scaling.resimplified module habitat.analysis.wave_scaling.roofline module habitat.analysis.wave_scaling.unified module habitat.data module habitat.profiling module habitat.profiling.autograd module habitat.profiling.backward module habitat.profiling.kernel module habitat.profiling.operation module habitat.profiling.run_time module habitat.tracking module habitat.tracking.base module habitat.tracking.callable module habitat.tracking.hook_manager module habitat.tracking.operation module habitat.utils module HabitatDataset (class in habitat.analysis.mlp.dataset) HookManager (class in habitat.tracking.hook_manager)

K

kernel_registers_for() (habitat.analysis.wave_scaling.metadata.MetadataManager method) KernelProfiler (class in habitat.profiling.kernel) kernels (habitat.analysis.run_time.RunTime property) (habitat.analysis.run_time.RunTimeMeasurement property) (habitat.analysis.run_time.RunTimePrediction property)

ktime_ns (habitat.analysis.operation.Operation property) (habitat.analysis.run_time.RunTime property)

L

legacy_cupti_name (habitat.analysis.metrics._MetricInfo property) LINEAR_PARAMS (in module habitat.analysis.predictor) LinearMLP (class in habitat.analysis.mlp.mlp) load_state() (habitat.analysis.mlp.mlp.RuntimePredictor method) logger (in module habitat.analysis.predictor) (in module habitat.analysis.wave_scaling.metadata) (in module habitat.profiling.kernel) (in module habitat.profiling.operation)

LSTM_PARAMS (in module habitat.analysis.predictor) LSTM_PARAMS_NO_BATCH_SIZES (in module habitat.analysis.predictor) LSTMMLP (class in habitat.analysis.mlp.mlp)

M

main() (in module habitat.analysis.mlp.train) MATMUL_PARAMS (in module habitat.analysis.predictor) measure_kernels() (habitat.profiling.kernel.KernelProfiler method) measure_ms() (habitat.profiling.run_time.RunTimeProfiler method) measure_operation() (habitat.profiling.operation.OperationProfiler method) MeasuredKernel (class in habitat.analysis.kernels) MeasuredOperation (class in habitat.analysis.operation) MetadataManager (class in habitat.analysis.wave_scaling.metadata) Metric (class in habitat) (class in habitat.analysis.metrics), [1] MLPBase (class in habitat.analysis.mlp.mlp) module habitat habitat.analysis habitat.analysis.arguments habitat.analysis.device habitat.analysis.kernels habitat.analysis.metrics habitat.analysis.mlp habitat.analysis.mlp.dataset habitat.analysis.mlp.dataset_process habitat.analysis.mlp.devices habitat.analysis.mlp.mlp habitat.analysis.mlp.train habitat.analysis.operation habitat.analysis.predictor habitat.analysis.run_time habitat.analysis.trace habitat.analysis.wave_scaling habitat.analysis.wave_scaling.common habitat.analysis.wave_scaling.metadata habitat.analysis.wave_scaling.resimplified habitat.analysis.wave_scaling.roofline habitat.analysis.wave_scaling.unified habitat.data habitat.profiling habitat.profiling.autograd habitat.profiling.backward habitat.profiling.kernel habitat.profiling.operation habitat.profiling.run_time habitat.tracking habitat.tracking.base habitat.tracking.callable habitat.tracking.hook_manager habitat.tracking.operation habitat.utils

ms_to_ns() (in module habitat.utils)

N

name (habitat.analysis.kernels.PredictedKernel property) (habitat.analysis.operation.MeasuredOperation property) (habitat.analysis.operation.Operation property) (habitat.analysis.operation.PredictedOperation property)

name_all_arguments() (in module habitat.utils) new_from() (habitat.profiling.autograd.AutogradEngine class method) (habitat.profiling.backward.BackwardHelper class method) ns_to_ms() (in module habitat.utils)

O

Operation (class in habitat.analysis.operation) OperationProfiler (class in habitat.profiling.operation)

operations (habitat.analysis.trace.Trace property) OperationTracker (class in habitat) (class in habitat.tracking.operation), [1]

P

path_to_data() (in module habitat.data) POTENTIALLY_INPLACE_FUNCTIONS (in module habitat.profiling.operation) predict() (habitat.analysis.mlp.mlp.RuntimePredictor method) predict_operation() (habitat.analysis.predictor.Predictor method) (habitat.Predictor method)

PredictedKernel (class in habitat.analysis.kernels) PredictedOperation (class in habitat.analysis.operation) Predictor (class in habitat) (class in habitat.analysis.predictor)

R

remove_hooks() (habitat.tracking.hook_manager.HookManager method) resimplified_wave_scaling() (in module habitat.analysis.wave_scaling.resimplified) resolve_metrics() (in module habitat.analysis.metrics) roofline_wave_scaling() (in module habitat.analysis.wave_scaling.roofline) run_accumulate_grad() (habitat.profiling.backward.BackwardHelper method) run_backward() (habitat.profiling.autograd.AutogradEngine method) run_time_ms (habitat.analysis.operation.Operation property) (habitat.analysis.run_time.RunTime property) (habitat.analysis.run_time.RunTimeMeasurement property) (habitat.analysis.run_time.RunTimePrediction property) (habitat.analysis.run_time.RunTimePurePrediction property) (habitat.analysis.trace.Trace property)

run_time_ns (habitat.analysis.kernels.PredictedKernel property) RunTime (class in habitat.analysis.run_time) RunTimeMeasurement (class in habitat.analysis.run_time) RunTimePrediction (class in habitat.analysis.run_time) RuntimePredictor (class in habitat.analysis.mlp.mlp) RunTimeProfiler (class in habitat.profiling.run_time) RunTimePurePrediction (class in habitat.analysis.run_time)

S

save_state() (habitat.analysis.mlp.mlp.RuntimePredictor method) set_up_logging() (in module habitat.utils) SinglePrecisionAddOps (habitat.analysis.metrics.Metric attribute), [1] (habitat.Metric attribute) SinglePrecisionFLOPEfficiency (habitat.analysis.metrics.Metric attribute), [1] (habitat.Metric attribute) SKIP_METRICS (in module habitat.profiling.kernel) SPECIAL_OPERATIONS (in module habitat.analysis)

start_tracking() (habitat.OperationTracker method) (habitat.tracking.base.TrackerBase method) (habitat.tracking.callable.CallableTracker method) (habitat.tracking.operation.OperationTracker method), [1] stop_tracking() (habitat.OperationTracker method) (habitat.tracking.base.TrackerBase method) (habitat.tracking.callable.CallableTracker method) (habitat.tracking.operation.OperationTracker method), [1]

T

to_canonical_value() (habitat.analysis.metrics._MetricInfo method) to_device() (habitat.analysis.operation.MeasuredOperation method) (habitat.analysis.operation.Operation method) (habitat.analysis.operation.PredictedOperation method) (habitat.analysis.trace.Trace method)

Trace (class in habitat.analysis.trace) track() (habitat.tracking.base.TrackerBase method) TrackerBase (class in habitat.tracking.base) train_with_dataset() (habitat.analysis.mlp.mlp.RuntimePredictor method)

U

unified_wave_scaling() (in module habitat.analysis.wave_scaling.unified)