Index

_

__add_per_datatype_tree() (hespas.estimator.roofline_estimator.RooflineEstimator method) __add_roofline_stats() (hespas.estimator.roofline_estimator.RooflineEstimator method) __cache_hit_hooks() (hespas.estimator.estimator.Estimator method) __cache_miss_hooks() (hespas.estimator.estimator.Estimator method) __close_executor() (hespas.estimator.estimator.Estimator method) __configure() (hespas.estimator.estimator.Estimator method) __count_cache_hits() (hespas.estimator.compute_estimator.ComputeEstimator method) __count_processed() (hespas.estimator.compute_estimator.ComputeEstimator method) __create_new_cache_entry() (hespas.estimator.estimator.Estimator method) __flush_outstanding_writes() (hespas.estimator.estimator.Estimator method) __get__module_op_times() (hespas.estimator.compute_estimator.ComputeEstimator method) __get_bytes_flops() (hespas.estimator.roofline_estimator.RooflineEstimator method) __get_cache_writer() (hespas.estimator.estimator.Estimator method) __get_cached_module_times() (hespas.estimator.compute_estimator.ComputeEstimator method) __get_datatype_str() (hespas.estimator.roofline_estimator.RooflineEstimator method) __get_datatype_str_by_op() (hespas.estimator.roofline_estimator.RooflineEstimator method) __get_estimator_cached_result() (hespas.estimator.estimator.Estimator method) __get_estimator_result() (hespas.estimator.estimator.Estimator method) __get_estimator_run_result() (hespas.estimator.estimator.Estimator method) __get_flops() (hespas.estimator.roofline_estimator.RooflineEstimator method) __get_flops_by_datatype() (hespas.estimator.roofline_estimator.RooflineEstimator method) __get_flops_by_datatype_str() (hespas.estimator.roofline_estimator.RooflineEstimator method) __get_flops_by_op() (hespas.estimator.roofline_estimator.RooflineEstimator method) __get_hash() (hespas.estimator.estimator.Estimator method) __get_hlo_file_path() (hespas.estimator.xla_estimator.XLAEstimator method) __get_metadata() (hespas.estimator.estimator.Estimator method) __get_module_bytes_flops() (hespas.estimator.roofline_estimator.RooflineEstimator method) __get_module_metadata() (hespas.estimator.estimator.Estimator method) __get_module_runtime() (hespas.estimator.compute_estimator.ComputeEstimator method) __get_module_statistics() (hespas.estimator.estimator.Estimator method) __get_op_estimate() (hespas.estimator.estimator.Estimator method) __get_op_estimate_by_name() (hespas.estimator.estimator.Estimator method) __get_op_estimate_list() (hespas.estimator.estimator.Estimator method) __get_op_estimates() (hespas.estimator.estimator.Estimator method) __get_per_op_module_result() (hespas.estimator.estimator.Estimator method) __get_per_op_runtime() (hespas.estimator.compute_estimator.ComputeEstimator method) __get_per_op_stats() (hespas.estimator.roofline_estimator.RooflineEstimator method) __get_total_bytes_flops() (hespas.estimator.roofline_estimator.RooflineEstimator method) __get_total_estimate_time() (hespas.estimator.compute_estimator.ComputeEstimator method) __get_total_runtime() (hespas.estimator.compute_estimator.ComputeEstimator method) __init__() (hespas.chakra_gen.chakra_config.ChakraGenConfig method) (hespas.chakra_gen.chakra_trace_generator.ChakraTraceGen method) (hespas.estimator.cache.CacheElement method) (hespas.estimator.check_available.EstimatorNotAvailableError method) (hespas.estimator.config_option.ConfigOption method) (hespas.estimator.estimator.Estimator method) (hespas.estimator.result.EstimatorResult method) (hespas.estimator.result.ModuleResult method) (hespas.estimator.result.OpResult method) (hespas.estimator.result.Result method) (hespas.estimator.run_config.EstimatorRunConfig method) (hespas.estimator.run_config.EstimatorSingleRunConfig method) (hespas.event_counters.CounterEvent method) (hespas.event_counters.CounterEventStream method) (hespas.event_counters.event.CounterEvent method) (hespas.event_counters.stream.CounterEventStream method) (hespas.mlir_parser.mlir_analyzer.MLIRAnalyzer method) (hespas.mlir_parser.mlir_module.MLIRModule method) (hespas.mlir_parser.mlir_parser.MLIRParser method) (hespas.mlir_parser.MLIRModule method) (hespas.mlir_parser.MLIRParser method) (hespas.statistics.statistics.AggregateStatistic method) (hespas.statistics.statistics.ContainerStatistic method) (hespas.statistics.statistics.IntervalStatistic method) (hespas.statistics.statistics.RatioStatistic method) (hespas.statistics.statistics.Statistic method) (hespas.statistics.statistics.ValueStatistic method) (hespas.utils.logging.LogFormatter method) (hespas.utils.op_info.OpInfo method) (hespas.utils.serialisable.Serialisable method) __init_hooks() (hespas.estimator.estimator.Estimator method) __load_cache() (hespas.estimator.estimator.Estimator method) __merge_lower_stats_tree() (hespas.estimator.roofline_estimator.RooflineEstimator method) __module_run_end_time() (hespas.estimator.compute_estimator.ComputeEstimator method) __module_run_start_time() (hespas.estimator.compute_estimator.ComputeEstimator method) __post_estimate_hooks() (hespas.estimator.estimator.Estimator method) __post_op_hooks() (hespas.estimator.estimator.Estimator method) __post_run_hooks() (hespas.estimator.estimator.Estimator method) __pre_estimate_hooks() (hespas.estimator.estimator.Estimator method) __pre_op_hooks() (hespas.estimator.estimator.Estimator method) __pre_run_hooks() (hespas.estimator.estimator.Estimator method) __print_cached_runtime() (hespas.estimator.compute_estimator.ComputeEstimator method) __print_run_runtime() (hespas.estimator.compute_estimator.ComputeEstimator method) __print_start_line() (hespas.estimator.compute_estimator.ComputeEstimator method) __setup_config_dict() (hespas.estimator.estimator.Estimator method) __setup_module_roofline_stats() (hespas.estimator.roofline_estimator.RooflineEstimator method)

__setup_per_module_stat_tree() (hespas.estimator.compute_estimator.ComputeEstimator method) __setup_per_op_roofline_stats() (hespas.estimator.roofline_estimator.RooflineEstimator method) __setup_per_op_stat_tree() (hespas.estimator.compute_estimator.ComputeEstimator method) __setup_per_op_tree() (hespas.estimator.compute_estimator.ComputeEstimator method) __setup_roofline_stats() (hespas.estimator.roofline_estimator.RooflineEstimator method) __to_dict() (hespas.utils.serialisable.Serialisable method) __total_estimate_start_time() (hespas.estimator.compute_estimator.ComputeEstimator method) __unknown_op_panic() (hespas.estimator.compute_estimator.ComputeEstimator method) __ver_str_to_tuple() (hespas.event_counters.CounterEventStream method) (hespas.event_counters.stream.CounterEventStream method) __ver_tuple_to_str() (hespas.event_counters.CounterEventStream method) (hespas.event_counters.stream.CounterEventStream method) __write_cache() (hespas.estimator.estimator.Estimator method) _abc_impl (hespas.utils.dict_tools.FloatDict attribute) (hespas.utils.dict_tools.PositiveFloatDict attribute) _add_event() (hespas.estimator.estimator.Estimator method) _cache_hit_hooks (hespas.estimator.cocossim_estimator.CocossimEstimator attribute) (hespas.estimator.compute_estimator.ComputeEstimator attribute) (hespas.estimator.estimator.Estimator attribute) (hespas.estimator.roofline_estimator.RooflineEstimator attribute) (hespas.estimator.xla_estimator.XLAEstimator attribute) _cache_miss_hooks (hespas.estimator.cocossim_estimator.CocossimEstimator attribute) (hespas.estimator.compute_estimator.ComputeEstimator attribute) (hespas.estimator.estimator.Estimator attribute) (hespas.estimator.roofline_estimator.RooflineEstimator attribute) (hespas.estimator.xla_estimator.XLAEstimator attribute) _default_op_handler() (hespas.estimator.cocossim_estimator.CocossimEstimator method) (hespas.estimator.compute_estimator.ComputeEstimator method) (hespas.estimator.roofline_estimator.RooflineEstimator method) (hespas.estimator.xla_estimator.XLAEstimator method) _get_whole_module_result() (hespas.estimator.estimator.Estimator method) (hespas.estimator.xla_estimator.XLAEstimator method) _init_hooks (hespas.estimator.cocossim_estimator.CocossimEstimator attribute) (hespas.estimator.compute_estimator.ComputeEstimator attribute) (hespas.estimator.estimator.Estimator attribute) (hespas.estimator.roofline_estimator.RooflineEstimator attribute) (hespas.estimator.xla_estimator.XLAEstimator attribute) _metadata_hooks (hespas.estimator.cocossim_estimator.CocossimEstimator attribute) (hespas.estimator.compute_estimator.ComputeEstimator attribute) (hespas.estimator.estimator.Estimator attribute) (hespas.estimator.roofline_estimator.RooflineEstimator attribute) (hespas.estimator.xla_estimator.XLAEstimator attribute) _module_metadata_hooks (hespas.estimator.cocossim_estimator.CocossimEstimator attribute) (hespas.estimator.compute_estimator.ComputeEstimator attribute) (hespas.estimator.estimator.Estimator attribute) (hespas.estimator.roofline_estimator.RooflineEstimator attribute) (hespas.estimator.xla_estimator.XLAEstimator attribute) _next_idx (hespas.mlir_parser.mlir_module.MLIRModule attribute) (hespas.mlir_parser.MLIRModule attribute) _op_handlers (hespas.estimator.cocossim_estimator.CocossimEstimator attribute) (hespas.estimator.compute_estimator.ComputeEstimator attribute) (hespas.estimator.estimator.Estimator attribute) (hespas.estimator.roofline_estimator.RooflineEstimator attribute) (hespas.estimator.xla_estimator.XLAEstimator attribute) _parse_benchmark_output() (hespas.estimator.xla_estimator.XLAEstimator method) _post_estimate_hooks (hespas.estimator.cocossim_estimator.CocossimEstimator attribute) (hespas.estimator.compute_estimator.ComputeEstimator attribute) (hespas.estimator.estimator.Estimator attribute) (hespas.estimator.roofline_estimator.RooflineEstimator attribute) (hespas.estimator.xla_estimator.XLAEstimator attribute) _post_op_hooks (hespas.estimator.cocossim_estimator.CocossimEstimator attribute) (hespas.estimator.compute_estimator.ComputeEstimator attribute) (hespas.estimator.estimator.Estimator attribute) (hespas.estimator.roofline_estimator.RooflineEstimator attribute) (hespas.estimator.xla_estimator.XLAEstimator attribute) _post_run_hooks (hespas.estimator.cocossim_estimator.CocossimEstimator attribute) (hespas.estimator.compute_estimator.ComputeEstimator attribute) (hespas.estimator.estimator.Estimator attribute) (hespas.estimator.roofline_estimator.RooflineEstimator attribute) (hespas.estimator.xla_estimator.XLAEstimator attribute) _pre_estimate_hooks (hespas.estimator.cocossim_estimator.CocossimEstimator attribute) (hespas.estimator.compute_estimator.ComputeEstimator attribute) (hespas.estimator.estimator.Estimator attribute) (hespas.estimator.roofline_estimator.RooflineEstimator attribute) (hespas.estimator.xla_estimator.XLAEstimator attribute) _pre_op_hooks (hespas.estimator.cocossim_estimator.CocossimEstimator attribute) (hespas.estimator.compute_estimator.ComputeEstimator attribute) (hespas.estimator.estimator.Estimator attribute) (hespas.estimator.roofline_estimator.RooflineEstimator attribute) (hespas.estimator.xla_estimator.XLAEstimator attribute) _pre_run_hooks (hespas.estimator.cocossim_estimator.CocossimEstimator attribute) (hespas.estimator.compute_estimator.ComputeEstimator attribute) (hespas.estimator.estimator.Estimator attribute) (hespas.estimator.roofline_estimator.RooflineEstimator attribute) (hespas.estimator.xla_estimator.XLAEstimator attribute) _write_comm_group_json() (hespas.chakra_gen.chakra_trace_generator.ChakraTraceGen method)

A

add_classifier() (in module hespas.mlir_parser.mlir_common) add_estimator_choice() (in module hespas.estimator.choice) add_event() (hespas.event_counters.CounterEventStream method) (hespas.event_counters.stream.CounterEventStream method) add_member() (hespas.statistics.statistics.ContainerStatistic method) add_members() (hespas.statistics.statistics.ContainerStatistic method)

AggregateStatistic (class in hespas.statistics.statistics) allow_multiprocess (hespas.estimator.estimator.Estimator attribute) (hespas.estimator.roofline_estimator.RooflineEstimator attribute) analyzer (hespas.mlir_parser.mlir_module.MLIRModule property) (hespas.mlir_parser.MLIRModule property) append_private_functions() (in module hespas.mlir_parser.mlir_common)

B

bases_order (hespas.estimator.cocossim_estimator.CocossimEstimator attribute) (hespas.estimator.compute_estimator.ComputeEstimator attribute) (hespas.estimator.estimator.Estimator attribute) (hespas.estimator.roofline_estimator.RooflineEstimator attribute) (hespas.estimator.xla_estimator.XLAEstimator attribute)

block_type (hespas.mlir_parser.mlir_module.MLIRModule property) (hespas.mlir_parser.MLIRModule property) bounded_number() (in module hespas.utils.num_tools)

C

cache_dir (hespas.estimator.estimator.Estimator attribute) CacheElement (class in hespas.estimator.cache) calculate() (hespas.statistics.statistics.AggregateStatistic method) (hespas.statistics.statistics.ContainerStatistic method) (hespas.statistics.statistics.IntervalStatistic method) (hespas.statistics.statistics.RatioStatistic method) (hespas.statistics.statistics.Statistic method) (hespas.statistics.statistics.ValueStatistic method) ChakraGenConfig (class in hespas.chakra_gen.chakra_config) ChakraTraceGen (class in hespas.chakra_gen.chakra_trace_generator) check_programs() (in module hespas.estimator.check_available) chips (hespas.estimator.cocossim_estimator.CocossimEstimator attribute) class_version (hespas.event_counters.CounterEventStream attribute) (hespas.event_counters.stream.CounterEventStream attribute) cocossim_path (hespas.estimator.cocossim_estimator.CocossimEstimator attribute) CocossimEstimator (class in hespas.estimator.cocossim_estimator) collective (hespas.mlir_parser.mlir_module.MLIRModule property) (hespas.mlir_parser.MLIRModule property) collective_to_chakra_collective() (hespas.chakra_gen.chakra_trace_generator.ChakraTraceGen method) comm_bytes (hespas.mlir_parser.mlir_module.MLIRModule property) (hespas.mlir_parser.MLIRModule property) COMM_COLL_NODE (hespas.mlir_parser.mlir_common.NodeType attribute) COMM_RECV_NODE (hespas.mlir_parser.mlir_common.NodeType attribute) COMM_SEND_NODE (hespas.mlir_parser.mlir_common.NodeType attribute) COMP_NODE (hespas.mlir_parser.mlir_common.NodeType attribute) compute_estimator (hespas.estimator.compute_estimator.ComputeEstimator attribute) (hespas.estimator.estimator.Estimator attribute) compute_runtime() (hespas.estimator.roofline_estimator.RooflineEstimator method) compute_tensor_size() (in module hespas.utils.tensor_size)

ComputeEstimator (class in hespas.estimator.compute_estimator) conditions (hespas.estimator.cocossim_estimator.CocossimEstimator attribute) (hespas.estimator.estimator.Estimator attribute) (hespas.estimator.xla_estimator.XLAEstimator attribute) config_arguments (hespas.estimator.cocossim_estimator.CocossimEstimator attribute) (hespas.estimator.compute_estimator.ComputeEstimator attribute) (hespas.estimator.estimator.Estimator attribute) (hespas.estimator.roofline_estimator.RooflineEstimator attribute) (hespas.estimator.xla_estimator.XLAEstimator attribute) config_options (hespas.estimator.cocossim_estimator.CocossimEstimator attribute) (hespas.estimator.compute_estimator.ComputeEstimator attribute) (hespas.estimator.estimator.Estimator attribute) (hespas.estimator.roofline_estimator.RooflineEstimator attribute) (hespas.estimator.xla_estimator.XLAEstimator attribute) ConfigOption (class in hespas.estimator.config_option) ContainerStatistic (class in hespas.statistics.statistics) conv_bool() (in module hespas.utils.num_tools) conv_dim_number_re (hespas.mlir_parser.mlir_parser.MLIRParser attribute) (hespas.mlir_parser.MLIRParser attribute) copy() (hespas.statistics.statistics.Statistic method) count_unique_operators() (hespas.mlir_parser.mlir_analyzer.MLIRAnalyzer method) CounterEvent (class in hespas.event_counters) (class in hespas.event_counters.event) CounterEventStream (class in hespas.event_counters) (class in hespas.event_counters.stream) create_chakra_traces() (hespas.chakra_gen.chakra_trace_generator.ChakraTraceGen method) (in module hespas.chakra_gen.chakra_trace_generator) create_digraph() (in module hespas.mlir_parser.mlir_splitter) create_digraph_with_deps() (in module hespas.mlir_parser.mlir_splitter) create_new_module_with_operations() (in module hespas.mlir_parser.mlir_common)

D

DependencyGraphValidationError disable_cache (hespas.estimator.estimator.Estimator attribute) display_name (hespas.estimator.cocossim_estimator.CocossimEstimator attribute) (hespas.estimator.compute_estimator.ComputeEstimator attribute) (hespas.estimator.estimator.Estimator attribute) (hespas.estimator.roofline_estimator.RooflineEstimator attribute) (hespas.estimator.xla_estimator.XLAEstimator attribute)

display_name_map (hespas.estimator.cocossim_estimator.CocossimEstimator attribute) (hespas.estimator.compute_estimator.ComputeEstimator attribute) (hespas.estimator.estimator.Estimator attribute) (hespas.estimator.roofline_estimator.RooflineEstimator attribute) (hespas.estimator.xla_estimator.XLAEstimator attribute)

E

END_BLOCK (hespas.mlir_parser.mlir_splitter.SeparatorPolicy attribute) ensure_dir() (hespas.mlir_parser.mlir_module.MLIRModule method) (hespas.mlir_parser.MLIRModule method) error_on_unknown_type (hespas.estimator.roofline_estimator.RooflineEstimator attribute) Estimator (class in hespas.estimator.estimator) estimator_available() (in module hespas.estimator.choice) estimator_name (hespas.estimator.cache.CacheElement attribute) EstimatorMeta (class in hespas.estimator.estimator_meta)

EstimatorNotAvailableError EstimatorResult (class in hespas.estimator.result) EstimatorRunConfig (class in hespas.estimator.run_config) EstimatorSingleRunConfig (class in hespas.estimator.run_config) event_name (hespas.event_counters.CounterEvent attribute) (hespas.event_counters.event.CounterEvent attribute) events (hespas.event_counters.CounterEventStream attribute) (hespas.event_counters.stream.CounterEventStream attribute)

F

flatten_dict() (in module hespas.utils.dict_tools) FloatDict (class in hespas.utils.dict_tools) format() (hespas.utils.logging.LogFormatter method) frequency (hespas.estimator.cocossim_estimator.CocossimEstimator attribute) from_dict() (hespas.statistics.statistics.Statistic class method) (hespas.utils.serialisable.Serialisable class method)

from_json() (hespas.utils.serialisable.Serialisable class method) from_json_str() (hespas.utils.serialisable.Serialisable class method) from_module_result() (hespas.estimator.result.EstimatorResult class method) from_op_results() (hespas.estimator.result.ModuleResult class method) from_string() (hespas.mlir_parser.mlir_splitter.SeparatorPolicy class method)

G

generic_compile() (hespas.estimator.xla_estimator.XLAEstimator method) get_abspath() (hespas.mlir_parser.mlir_module.MLIRModule static method) (hespas.mlir_parser.MLIRModule static method) get_abspath_dir() (hespas.mlir_parser.mlir_module.MLIRModule static method) (hespas.mlir_parser.MLIRModule static method) get_aligned_bytes() (in module hespas.utils.dt_sizes) get_arg_parser() (in module hespas.chakra_gen.chakra_trace_generator) get_args_from_dict() (hespas.statistics.statistics.AggregateStatistic class method) (hespas.statistics.statistics.ContainerStatistic class method) (hespas.statistics.statistics.IntervalStatistic class method) (hespas.statistics.statistics.RatioStatistic class method) (hespas.statistics.statistics.Statistic class method) (hespas.statistics.statistics.ValueStatistic class method) get_bit_width() (in module hespas.utils.dt_sizes) get_block_type() (in module hespas.chakra_gen.chakra_trace_generator) get_conditions() (in module hespas.estimator.choice) get_copy_args() (hespas.statistics.statistics.Statistic method) get_cur_log_level() (in module hespas.utils.logging) get_divided_title() (in module hespas.utils.logging) get_dot_general_dimensions() (in module hespas.mlir_parser.mlir_common) get_estimate() (hespas.estimator.estimator.Estimator method) get_estimator() (in module hespas.estimator.choice) get_exec_args() (hespas.estimator.xla_estimator.XLAEstimator method) get_expanded_operator_counts() (hespas.mlir_parser.mlir_analyzer.MLIRAnalyzer method) get_expanded_operator_counts_list() (hespas.mlir_parser.mlir_analyzer.MLIRAnalyzer method) get_external_inputs() (in module hespas.mlir_parser.mlir_common) get_external_outputs() (in module hespas.mlir_parser.mlir_common) get_input_bytes() (hespas.utils.op_info.OpInfo method) get_input_types() (hespas.utils.op_info.OpInfo method) get_ir_context() (hespas.mlir_parser.mlir_parser.MLIRParser static method) (hespas.mlir_parser.MLIRParser static method) get_iree_input_string() (in module hespas.utils.iree) get_json_metadata() (hespas.mlir_parser.mlir_module.MLIRModule method) (hespas.mlir_parser.MLIRModule method) get_largest_type() (hespas.utils.op_info.OpInfo method)

get_log_levels() (in module hespas.utils.logging) get_member() (hespas.statistics.statistics.ContainerStatistic method) get_metadata() (hespas.estimator.estimator.Estimator method) get_metadata_copy() (hespas.mlir_parser.mlir_module.MLIRModule method) (hespas.mlir_parser.MLIRModule method) get_mlir_function_inputs() (hespas.mlir_parser.mlir_parser.MLIRParser method) (hespas.mlir_parser.MLIRParser method) get_mlir_function_outputs() (hespas.mlir_parser.mlir_parser.MLIRParser method) (hespas.mlir_parser.MLIRParser method) get_mlir_to_hlo_args() (hespas.estimator.xla_estimator.XLAEstimator method) get_module() (hespas.mlir_parser.mlir_parser.MLIRParser static method) (hespas.mlir_parser.MLIRParser static method) get_modules_dep_graph() (hespas.chakra_gen.chakra_trace_generator.ChakraTraceGen method) get_next_idx() (hespas.mlir_parser.mlir_module.MLIRModule class method) (hespas.mlir_parser.MLIRModule class method) get_number_of_inputs() (hespas.utils.op_info.OpInfo method) get_op_key() (hespas.mlir_parser.mlir_analyzer.MLIRAnalyzer static method) get_operator_counts() (hespas.mlir_parser.mlir_analyzer.MLIRAnalyzer method) get_output_bytes() (hespas.utils.op_info.OpInfo method) get_output_files() (hespas.mlir_parser.mlir_module.MLIRModule method) (hespas.mlir_parser.MLIRModule method) get_output_types() (hespas.utils.op_info.OpInfo method) get_paths() (hespas.mlir_parser.mlir_module.MLIRModule method) (hespas.mlir_parser.MLIRModule method) get_private_functions_map() (hespas.mlir_parser.mlir_parser.MLIRParser method) (hespas.mlir_parser.MLIRParser method) get_private_functions_nx_tree() (hespas.mlir_parser.mlir_parser.MLIRParser method) (hespas.mlir_parser.MLIRParser method) get_private_functions_ops() (hespas.mlir_parser.mlir_parser.MLIRParser method) (hespas.mlir_parser.MLIRParser method) get_statistics() (hespas.estimator.estimator.Estimator method) get_str_divider() (in module hespas.utils.logging) get_subclasses() (hespas.statistics.statistics.Statistic class method) get_tensor_bit_width() (in module hespas.utils.dt_sizes) get_terminal_cols() (in module hespas.utils.logging) get_types() (hespas.utils.op_info.OpInfo method)

H

handle_clamp() (hespas.estimator.roofline_estimator.RooflineEstimator method) handle_concatenate() (hespas.estimator.roofline_estimator.RooflineEstimator method) handle_convolution() (hespas.estimator.roofline_estimator.RooflineEstimator method) handle_custom_call() (hespas.estimator.roofline_estimator.RooflineEstimator method) handle_dot_general() (hespas.estimator.roofline_estimator.RooflineEstimator method) handle_dot_general_cocosim() (hespas.estimator.cocossim_estimator.CocossimEstimator method) handle_dynamic_update_slice() (hespas.estimator.roofline_estimator.RooflineEstimator method) handle_elementwise_binary() (hespas.estimator.roofline_estimator.RooflineEstimator method) handle_free_ops() (hespas.estimator.roofline_estimator.RooflineEstimator method) handle_fusion() (hespas.estimator.cocossim_estimator.CocossimEstimator method) (hespas.estimator.roofline_estimator.RooflineEstimator method) handle_gather() (hespas.estimator.roofline_estimator.RooflineEstimator method) handle_noflop_ops() (hespas.estimator.roofline_estimator.RooflineEstimator method) handle_ragged_dot() (hespas.estimator.roofline_estimator.RooflineEstimator method) handle_reduce() (hespas.estimator.roofline_estimator.RooflineEstimator method) handle_reduce_window() (hespas.estimator.roofline_estimator.RooflineEstimator method) handle_scatter() (hespas.estimator.roofline_estimator.RooflineEstimator method) handle_select() (hespas.estimator.roofline_estimator.RooflineEstimator method) handle_select_and_scatter() (hespas.estimator.roofline_estimator.RooflineEstimator method) handle_slice_ops() (hespas.estimator.roofline_estimator.RooflineEstimator method) handle_sort() (hespas.estimator.roofline_estimator.RooflineEstimator method) handle_unary_elemwise() (hespas.estimator.roofline_estimator.RooflineEstimator method) has_member() (hespas.statistics.statistics.ContainerStatistic method) has_ssa() (hespas.utils.op_info.OpInfo method) hash (hespas.estimator.cache.CacheElement attribute) (hespas.mlir_parser.mlir_module.MLIRModule property) (hespas.mlir_parser.MLIRModule property) hbm_power_ratio (hespas.estimator.roofline_estimator.RooflineEstimator attribute) hespas module hespas.chakra_gen module hespas.chakra_gen.chakra_config module hespas.chakra_gen.chakra_trace_generator module hespas.estimator module hespas.estimator.cache module hespas.estimator.check_available module hespas.estimator.choice module hespas.estimator.cocossim_estimator module hespas.estimator.compute_estimator module hespas.estimator.config_option module hespas.estimator.estimator module hespas.estimator.estimator_meta module

hespas.estimator.result module hespas.estimator.roofline_estimator module hespas.estimator.run_config module hespas.estimator.xla_estimator module hespas.event_counters module hespas.event_counters.event module hespas.event_counters.stream module hespas.hespas_chakra_gen module hespas.mlir_parser module hespas.mlir_parser.mlir_analyzer module hespas.mlir_parser.mlir_common module hespas.mlir_parser.mlir_module module hespas.mlir_parser.mlir_parser module hespas.mlir_parser.mlir_splitter module hespas.statistics module hespas.statistics.statistics module hespas.utils module hespas.utils.dict_tools module hespas.utils.dt_sizes module hespas.utils.iree module hespas.utils.logging module hespas.utils.num_tools module hespas.utils.op_info module hespas.utils.serialisable module hespas.utils.tensor_size module histogram_of_operators() (hespas.mlir_parser.mlir_analyzer.MLIRAnalyzer method) hlo_runner_main_path (hespas.estimator.xla_estimator.XLAEstimator attribute) hw_config (hespas.estimator.cache.CacheElement attribute)

I

in_memory_only_cache (hespas.estimator.estimator.Estimator attribute) individual_split() (in module hespas.mlir_parser.mlir_splitter) input_dims (hespas.mlir_parser.mlir_module.MLIRModule property) (hespas.mlir_parser.MLIRModule property) input_dims_mlir (hespas.mlir_parser.mlir_module.MLIRModule property) (hespas.mlir_parser.MLIRModule property) IntervalStatistic (class in hespas.statistics.statistics) INVALID_NODE (hespas.mlir_parser.mlir_common.NodeType attribute) InvalidConfigOptionError InvalidOpError ir_context (hespas.mlir_parser.mlir_parser.MLIRParser attribute) (hespas.mlir_parser.MLIRParser attribute)

is_communication_block (hespas.mlir_parser.mlir_module.MLIRModule property) (hespas.mlir_parser.MLIRModule property) is_communication_op() (in module hespas.mlir_parser.mlir_common) is_communication_op_str() (in module hespas.mlir_parser.mlir_common) is_computation_block (hespas.mlir_parser.mlir_module.MLIRModule property) (hespas.mlir_parser.MLIRModule property) is_convolution_op() (in module hespas.mlir_parser.mlir_common) is_dot_general_op() (in module hespas.mlir_parser.mlir_common) is_reduce_op() (in module hespas.mlir_parser.mlir_common) is_return_op() (in module hespas.mlir_parser.mlir_common) ISOLATED (hespas.mlir_parser.mlir_splitter.SeparatorPolicy attribute) items() (hespas.statistics.statistics.ContainerStatistic method)

K

keys() (hespas.statistics.statistics.ContainerStatistic method)

L

linear_split() (in module hespas.mlir_parser.mlir_splitter) load_dependency_graph() (in module hespas.mlir_parser.mlir_splitter) LoadDependencyException

LogFormatter (class in hespas.utils.logging) Logger (class in hespas.utils.logging) logger_basic_config() (in module hespas.utils.logging)

M

main() (hespas.estimator.estimator.Estimator class method) (in module hespas.chakra_gen.chakra_trace_generator) (in module hespas.mlir_parser.mlir_analyzer) (in module hespas.mlir_parser.mlir_splitter) main_function (hespas.mlir_parser.mlir_parser.MLIRParser property) (hespas.mlir_parser.MLIRParser property) main_index (hespas.mlir_parser.mlir_parser.MLIRParser property) (hespas.mlir_parser.MLIRParser property) MEM_LOAD_NODE (hespas.mlir_parser.mlir_common.NodeType attribute) MEM_STORE_NODE (hespas.mlir_parser.mlir_common.NodeType attribute) memory_bandwidth (hespas.estimator.roofline_estimator.RooflineEstimator attribute) merge_split() (in module hespas.mlir_parser.mlir_splitter) metadata (hespas.estimator.result.Result attribute) (hespas.event_counters.CounterEvent attribute) (hespas.event_counters.CounterEventStream attribute) (hespas.event_counters.event.CounterEvent attribute) (hespas.event_counters.stream.CounterEventStream attribute) (hespas.mlir_parser.mlir_module.MLIRModule property) (hespas.mlir_parser.MLIRModule property) METADATA_NODE (hespas.mlir_parser.mlir_common.NodeType attribute) mlir_string (hespas.mlir_parser.mlir_module.MLIRModule property) (hespas.mlir_parser.mlir_parser.MLIRParser property) (hespas.mlir_parser.MLIRModule property) (hespas.mlir_parser.MLIRParser property) MLIRAnalyzer (class in hespas.mlir_parser.mlir_analyzer) MLIRModule (class in hespas.mlir_parser) (class in hespas.mlir_parser.mlir_module) MLIRParser (class in hespas.mlir_parser) (class in hespas.mlir_parser.mlir_parser) MLIRSplittingError module hespas hespas.chakra_gen hespas.chakra_gen.chakra_config hespas.chakra_gen.chakra_trace_generator hespas.estimator hespas.estimator.cache hespas.estimator.check_available hespas.estimator.choice hespas.estimator.cocossim_estimator hespas.estimator.compute_estimator hespas.estimator.config_option hespas.estimator.estimator hespas.estimator.estimator_meta hespas.estimator.result hespas.estimator.roofline_estimator hespas.estimator.run_config hespas.estimator.xla_estimator hespas.event_counters hespas.event_counters.event hespas.event_counters.stream hespas.hespas_chakra_gen hespas.mlir_parser hespas.mlir_parser.mlir_analyzer hespas.mlir_parser.mlir_common hespas.mlir_parser.mlir_module hespas.mlir_parser.mlir_parser hespas.mlir_parser.mlir_splitter hespas.statistics hespas.statistics.statistics hespas.utils hespas.utils.dict_tools hespas.utils.dt_sizes hespas.utils.iree hespas.utils.logging hespas.utils.num_tools hespas.utils.op_info hespas.utils.serialisable hespas.utils.tensor_size

module (hespas.estimator.result.ModuleResult attribute) module_file (hespas.estimator.result.EstimatorResult attribute) (hespas.mlir_parser.mlir_module.MLIRModule property) (hespas.mlir_parser.MLIRModule property) module_idx (hespas.estimator.result.EstimatorResult attribute) module_path (hespas.mlir_parser.mlir_module.MLIRModule property) (hespas.mlir_parser.MLIRModule property) ModuleResult (class in hespas.estimator.result)

N

NodeType (class in hespas.mlir_parser.mlir_common)

num_npus (hespas.estimator.estimator.Estimator attribute)

O

op_count (hespas.mlir_parser.mlir_analyzer.MLIRAnalyzer property) (hespas.mlir_parser.mlir_module.MLIRModule property) (hespas.mlir_parser.MLIRModule property) op_count_and_expanded (hespas.mlir_parser.mlir_analyzer.MLIRAnalyzer property) op_count_expanded (hespas.mlir_parser.mlir_analyzer.MLIRAnalyzer property) (hespas.mlir_parser.mlir_module.MLIRModule property) (hespas.mlir_parser.MLIRModule property) op_info (hespas.estimator.result.OpResult attribute) op_line (hespas.utils.op_info.OpInfo property) operations (hespas.mlir_parser.mlir_parser.MLIRParser property) (hespas.mlir_parser.MLIRParser property) operators (hespas.mlir_parser.mlir_analyzer.MLIRAnalyzer property) operators_expanded (hespas.mlir_parser.mlir_analyzer.MLIRAnalyzer property)

OpInfo (class in hespas.utils.op_info) OpResult (class in hespas.estimator.result) ops_list (hespas.mlir_parser.mlir_module.MLIRModule property) (hespas.mlir_parser.mlir_parser.MLIRParser property) (hespas.mlir_parser.MLIRModule property) (hespas.mlir_parser.MLIRParser property) other_from_dict() (hespas.statistics.statistics.Statistic class method) other_to_dict() (hespas.statistics.statistics.Statistic method) output_dims (hespas.mlir_parser.mlir_module.MLIRModule property) (hespas.mlir_parser.MLIRModule property) output_dims_mlir (hespas.mlir_parser.mlir_module.MLIRModule property) (hespas.mlir_parser.MLIRModule property) output_events() (hespas.estimator.estimator.Estimator method)

P

parent_module (hespas.mlir_parser.mlir_module.MLIRModule property) (hespas.mlir_parser.MLIRModule property) parse_and_split_mlir() (in module hespas.mlir_parser) (in module hespas.mlir_parser.mlir_splitter) parse_operation() (hespas.mlir_parser.mlir_parser.MLIRParser method) (hespas.mlir_parser.MLIRParser method) parse_replica_from_match() (in module hespas.mlir_parser.mlir_common) parse_replica_groups() (in module hespas.mlir_parser.mlir_common) peak_flops (hespas.estimator.roofline_estimator.RooflineEstimator attribute) per_datatype_flops (hespas.estimator.roofline_estimator.RooflineEstimator attribute) pos_float() (in module hespas.utils.num_tools)

pos_int() (in module hespas.utils.num_tools) PositiveFloatDict (class in hespas.utils.dict_tools) print() (hespas.statistics.statistics.Statistic method) print_divider() (in module hespas.utils.logging) print_subdict_path() (in module hespas.utils.dict_tools) print_val() (hespas.statistics.statistics.Statistic method) print_value() (hespas.statistics.statistics.ContainerStatistic method) (hespas.statistics.statistics.RatioStatistic method) (hespas.statistics.statistics.Statistic method) private_functions (hespas.mlir_parser.mlir_parser.MLIRParser property) (hespas.mlir_parser.MLIRParser property) progress() (hespas.utils.logging.Logger method)

R

RatioStatistic (class in hespas.statistics.statistics) recurse_merge() (in module hespas.utils.dict_tools) register_cache_hit_hook() (in module hespas.estimator.estimator) register_cache_miss_hook() (in module hespas.estimator.estimator) register_default_op_handler() (in module hespas.estimator.estimator) register_init_hook() (in module hespas.estimator.estimator) register_metadata_hook() (in module hespas.estimator.estimator) register_module_metadata_hook() (in module hespas.estimator.estimator) register_op_handler() (in module hespas.estimator.estimator) register_post_estimate_hook() (in module hespas.estimator.estimator) register_post_op_hook() (in module hespas.estimator.estimator) register_post_run_hook() (in module hespas.estimator.estimator) register_pre_estimate_hook() (in module hespas.estimator.estimator) register_pre_op_hook() (in module hespas.estimator.estimator) register_pre_run_hook() (in module hespas.estimator.estimator) register_split_fn() (in module hespas.mlir_parser.mlir_splitter) replica_groups (hespas.mlir_parser.mlir_module.MLIRModule property) (hespas.mlir_parser.MLIRModule property)

reset() (hespas.statistics.statistics.AggregateStatistic method) (hespas.statistics.statistics.IntervalStatistic method) (hespas.statistics.statistics.RatioStatistic method) (hespas.statistics.statistics.Statistic method) (hespas.statistics.statistics.ValueStatistic method) reset_index_counter() (hespas.mlir_parser.mlir_module.MLIRModule class method) (hespas.mlir_parser.MLIRModule class method) Result (class in hespas.estimator.result) result (hespas.estimator.cache.CacheElement attribute) results() (hespas.utils.logging.Logger method) retrieve_mlir_from_file() (hespas.mlir_parser.mlir_module.MLIRModule method) (hespas.mlir_parser.MLIRModule method) RooflineEstimator (class in hespas.estimator.roofline_estimator) RooflineMissingDatatypeError round_or_int_value() (hespas.statistics.statistics.Statistic method) run() (hespas.estimator.estimator.Estimator class method) runtime_estimate (hespas.estimator.result.Result attribute)

S

sample (hespas.estimator.xla_estimator.XLAEstimator attribute) sample() (hespas.statistics.statistics.AggregateStatistic method) (hespas.statistics.statistics.ContainerStatistic method) (hespas.statistics.statistics.IntervalStatistic method) (hespas.statistics.statistics.RatioStatistic method) (hespas.statistics.statistics.Statistic method) (hespas.statistics.statistics.SummingStatistic method) (hespas.statistics.statistics.ValueStatistic method) save_dependency_graph() (in module hespas.mlir_parser.mlir_splitter) save_json() (hespas.mlir_parser.mlir_module.MLIRModule method) (hespas.mlir_parser.MLIRModule method) save_mlir() (hespas.mlir_parser.mlir_module.MLIRModule method) (hespas.mlir_parser.MLIRModule method) save_to_files() (hespas.mlir_parser.mlir_module.MLIRModule method) (hespas.mlir_parser.MLIRModule method)

SeparatorPolicy (class in hespas.mlir_parser.mlir_splitter) Serialisable (class in hespas.utils.serialisable) set_next_idx() (hespas.mlir_parser.mlir_module.MLIRModule class method) (hespas.mlir_parser.MLIRModule class method) setup_output_dir() (in module hespas.chakra_gen.chakra_trace_generator) split_by_opregion() (in module hespas.mlir_parser.mlir_splitter) stablehlo_to_cocossim_dot_general() (hespas.estimator.cocossim_estimator.CocossimEstimator method) START_BLOCK (hespas.mlir_parser.mlir_splitter.SeparatorPolicy attribute) Statistic (class in hespas.statistics.statistics) statistics (hespas.estimator.result.EstimatorResult attribute) store_private_functions() (in module hespas.mlir_parser.mlir_common) success (hespas.estimator.result.Result attribute) SummingStatistic (class in hespas.statistics.statistics)

T

tdp_W (hespas.estimator.roofline_estimator.RooflineEstimator attribute) TENSOR_CORE_OPS (hespas.estimator.roofline_estimator.RooflineEstimator attribute) TENSOR_CORE_PROMOTIONS (hespas.estimator.roofline_estimator.RooflineEstimator attribute) timestamp (hespas.event_counters.CounterEvent attribute) (hespas.event_counters.event.CounterEvent attribute) timeunit (hespas.event_counters.CounterEvent attribute) (hespas.event_counters.event.CounterEvent attribute) to_dict() (hespas.statistics.statistics.AggregateStatistic method) (hespas.statistics.statistics.ContainerStatistic method) (hespas.statistics.statistics.IntervalStatistic method) (hespas.statistics.statistics.RatioStatistic method) (hespas.statistics.statistics.Statistic method) (hespas.statistics.statistics.ValueStatistic method) (hespas.utils.serialisable.Serialisable method)

to_hw_config() (hespas.estimator.run_config.EstimatorRunConfig method) to_json() (hespas.utils.serialisable.Serialisable method) to_json_str() (hespas.utils.serialisable.Serialisable method) to_stats_out() (hespas.statistics.statistics.ContainerStatistic method) (hespas.statistics.statistics.Statistic method) to_str() (hespas.mlir_parser.mlir_analyzer.MLIRAnalyzer method) total_number_of_operators() (hespas.mlir_parser.mlir_analyzer.MLIRAnalyzer method) translate (hespas.estimator.xla_estimator.XLAEstimator attribute) type (hespas.estimator.compute_estimator.ComputeEstimator attribute)

U

unavailable (in module hespas.estimator.choice) unit_print() (hespas.statistics.statistics.Statistic method) unsigned_int() (in module hespas.utils.num_tools)

unwrap_val() (hespas.statistics.statistics.Statistic static method) update() (hespas.mlir_parser.mlir_analyzer.MLIRAnalyzer method) update_metadata() (hespas.mlir_parser.mlir_module.MLIRModule method) (hespas.mlir_parser.MLIRModule method)

V

validate_dependency_graph() (in module hespas.mlir_parser.mlir_splitter) values() (hespas.statistics.statistics.ContainerStatistic method)

ValueStatistic (class in hespas.statistics.statistics) version (hespas.event_counters.CounterEventStream attribute) (hespas.event_counters.stream.CounterEventStream attribute)

W

warn_on_unknown_type (hespas.estimator.roofline_estimator.RooflineEstimator attribute) window_attr_re (hespas.mlir_parser.mlir_parser.MLIRParser attribute) (hespas.mlir_parser.MLIRParser attribute)

window_sub_attr_re (hespas.mlir_parser.mlir_parser.MLIRParser attribute) (hespas.mlir_parser.MLIRParser attribute) write_cache() (hespas.estimator.estimator.Estimator method)

X

xla_runtime_estimate() (hespas.estimator.xla_estimator.XLAEstimator method)

xla_translate_path (hespas.estimator.xla_estimator.XLAEstimator attribute) XLAEstimator (class in hespas.estimator.xla_estimator)