| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / __init__.py |
__getattr__ |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / __init__.py |
__dir__ |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / __init__.py |
(no function) |
|
23 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / __init__.py |
(no function) |
|
0 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / _hf_model_spec.py |
is_auto_models_schema |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / _hf_model_spec.py |
_first_attr |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / _hf_model_spec.py |
_text_tower |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / _hf_model_spec.py |
_spec_from_aliases |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / _hf_model_spec.py |
_derive_shared_experts |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / _hf_model_spec.py |
_visual_seq_len |
|
11 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / _hf_model_spec.py |
_get_hf_config |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / _hf_model_spec.py |
_explicit_overrides |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / _hf_model_spec.py |
resolve_hf_model_spec |
|
26 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / _hf_model_spec.py |
(no function) |
|
17 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / config_adapter / __init__.py |
(no function) |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / config_adapter / _config_loader.py |
_normalize_model_spec |
|
4 |
1 |
0 |
|
75% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / config_adapter / _config_loader.py |
_get_dict |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / config_adapter / _config_loader.py |
_load_auto_models_model_spec |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / config_adapter / _config_loader.py |
_load_yaml |
|
17 |
3 |
0 |
|
82% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / config_adapter / _config_loader.py |
_parse_unified_parallelism |
|
14 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / config_adapter / _config_loader.py |
_build_config_from_search_yaml |
|
31 |
1 |
0 |
|
97% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / config_adapter / _config_loader.py |
read_search_config |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / config_adapter / _config_loader.py |
_build_config_from_auto_models_yaml |
|
30 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / config_adapter / _config_loader.py |
_build_config_from_hp_yaml |
|
34 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / config_adapter / _config_loader.py |
read_hp_yaml_config |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / config_adapter / _config_loader.py |
(no function) |
|
24 |
1 |
1 |
|
96% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / config_adapter / _constraint_checker.py |
_err |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / config_adapter / _constraint_checker.py |
_warn |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / config_adapter / _constraint_checker.py |
_check_divisibility |
|
7 |
2 |
0 |
|
71% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / config_adapter / _constraint_checker.py |
_get_fixed_value |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / config_adapter / _constraint_checker.py |
_resolve_candidates |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / config_adapter / _constraint_checker.py |
_candidates_or_default |
|
4 |
1 |
0 |
|
75% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / config_adapter / _constraint_checker.py |
_total_cards |
|
5 |
1 |
0 |
|
80% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / config_adapter / _constraint_checker.py |
validate |
|
20 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / config_adapter / _constraint_checker.py |
validate_strict |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / config_adapter / _constraint_checker.py |
_check_required_fields |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / config_adapter / _constraint_checker.py |
_check_batch_size_relationships |
|
14 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / config_adapter / _constraint_checker.py |
_check_tp_divisibility |
|
19 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / config_adapter / _constraint_checker.py |
_check_cp_divisibility |
|
9 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / config_adapter / _constraint_checker.py |
_check_ep_divisibility |
|
9 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / config_adapter / _constraint_checker.py |
_check_fixed_dims_vs_search_space |
|
8 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / config_adapter / _constraint_checker.py |
_check_pipeline_constraints |
|
27 |
1 |
0 |
|
96% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / config_adapter / _constraint_checker.py |
_check_layer_offset |
|
12 |
1 |
0 |
|
92% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / config_adapter / _constraint_checker.py |
_check_layer_recompute |
|
9 |
1 |
0 |
|
89% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / config_adapter / _constraint_checker.py |
_check_device_product_limit |
|
22 |
3 |
0 |
|
86% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / config_adapter / _constraint_checker.py |
_check_memory_limit |
|
7 |
1 |
0 |
|
86% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / config_adapter / _constraint_checker.py |
_check_dense_model_ep_cp_warning |
|
9 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / config_adapter / _constraint_checker.py |
_check_fsdp_hsdp_device_product |
|
28 |
3 |
0 |
|
89% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / config_adapter / _constraint_checker.py |
(no function) |
|
24 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / config_adapter / _normalized_config.py |
NormalizedConfig.to_dict |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / config_adapter / _normalized_config.py |
(no function) |
|
18 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / config_adapter / _search_runner.py |
_get_dim_module |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / config_adapter / _search_runner.py |
_get_machine_mod |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / config_adapter / _search_runner.py |
_search_dim_map |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / config_adapter / _search_runner.py |
_validate_before_search |
|
11 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / config_adapter / _search_runner.py |
_build_model_dict |
|
7 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / config_adapter / _search_runner.py |
_build_hp_yaml_dict |
|
44 |
1 |
0 |
|
98% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / config_adapter / _search_runner.py |
_write_temp_hp_yaml |
|
7 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / config_adapter / _search_runner.py |
_build_machine |
|
9 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / config_adapter / _search_runner.py |
_resolve_search_dimensions |
|
11 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / config_adapter / _search_runner.py |
_post_filter |
|
29 |
2 |
0 |
|
93% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / config_adapter / _search_runner.py |
_format_result |
|
18 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / config_adapter / _search_runner.py |
search_strategies |
|
19 |
3 |
0 |
|
84% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / config_adapter / _search_runner.py |
(no function) |
|
20 |
0 |
4 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / config_adapter / _strategy_output.py |
_resolve_pp_degree |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / config_adapter / _strategy_output.py |
_write_json |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / config_adapter / _strategy_output.py |
_validate_strategy_and_yaml |
|
4 |
1 |
0 |
|
75% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / config_adapter / _strategy_output.py |
_load_yaml_to_inject |
|
16 |
6 |
0 |
|
62% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / config_adapter / _strategy_output.py |
_check_batch_derivation |
|
10 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / config_adapter / _strategy_output.py |
_inject_resolved_strategy |
|
23 |
1 |
0 |
|
96% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / config_adapter / _strategy_output.py |
_write_output_yaml |
|
7 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / config_adapter / _strategy_output.py |
write_strategy_config |
|
11 |
1 |
0 |
|
91% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / config_adapter / _strategy_output.py |
write_resolved_strategy |
|
13 |
1 |
0 |
|
92% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / config_adapter / _strategy_output.py |
write_ppb_config |
|
14 |
2 |
0 |
|
86% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / config_adapter / _strategy_output.py |
write_resolved_yaml |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / config_adapter / _strategy_output.py |
normalized_to_summary |
|
7 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / config_adapter / _strategy_output.py |
(no function) |
|
22 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / hyper_offload / __init__.py |
(no function) |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / hyper_offload / _patch_accelerator.py |
_patch |
|
51 |
49 |
0 |
|
4% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / hyper_offload / _patch_accelerator.py |
_patch.is_available |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / hyper_offload / _patch_accelerator.py |
_patch.current_accelerator |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / hyper_offload / _patch_accelerator.py |
(no function) |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / hyper_offload / api / __init__.py |
(no function) |
|
0 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / hyper_offload / api / config.py |
(no function) |
|
8 |
0 |
2 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / hyper_offload / api / opaque.py |
skip_offload |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / hyper_offload / api / opaque.py |
skip_offload.wrapper |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / hyper_offload / api / opaque.py |
(no function) |
|
7 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / hyper_offload / api / session.py |
ActivationDispatchMode.__init__ |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / hyper_offload / api / session.py |
ActivationDispatchMode.__torch_dispatch__ |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / hyper_offload / api / session.py |
OffloadSession.__init__ |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / hyper_offload / api / session.py |
OffloadSession.get_active |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / hyper_offload / api / session.py |
OffloadSession.__enter__ |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / hyper_offload / api / session.py |
OffloadSession.__exit__ |
|
17 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / hyper_offload / api / session.py |
(no function) |
|
20 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / hyper_offload / execution / __init__.py |
(no function) |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / hyper_offload / execution / base.py |
OpaqueRegionStart.forward |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / hyper_offload / execution / base.py |
OpaqueRegionStart.backward |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / hyper_offload / execution / base.py |
OpaqueRegionEnd.forward |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / hyper_offload / execution / base.py |
OpaqueRegionEnd.backward |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / hyper_offload / execution / base.py |
OpaqueRegionEnd.backward.bwd_dummy |
|
0 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / hyper_offload / execution / base.py |
BaseExecutor.__init__ |
|
8 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / hyper_offload / execution / base.py |
BaseExecutor.in_opaque_region |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / hyper_offload / execution / base.py |
BaseExecutor.enter_opaque_region |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / hyper_offload / execution / base.py |
BaseExecutor.exit_opaque_region |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / hyper_offload / execution / base.py |
BaseExecutor.execute_opaque_op |
|
20 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / hyper_offload / execution / base.py |
BaseExecutor.execute_opaque_op.fwd_dummy |
|
0 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / hyper_offload / execution / base.py |
BaseExecutor.dispatch |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / hyper_offload / execution / base.py |
BaseExecutor.on_op_begin |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / hyper_offload / execution / base.py |
BaseExecutor.on_op_end |
|
0 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / hyper_offload / execution / base.py |
BaseExecutor.apply_shadows |
|
8 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / hyper_offload / execution / base.py |
BaseExecutor.retained_sids |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / hyper_offload / execution / base.py |
BaseExecutor.make_shadow |
|
7 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / hyper_offload / execution / base.py |
BaseExecutor.reset |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / hyper_offload / execution / base.py |
(no function) |
|
37 |
0 |
2 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / hyper_offload / execution / replay / __init__.py |
(no function) |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / hyper_offload / execution / replay / executor.py |
ReplayExecutor.__init__ |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / hyper_offload / execution / replay / executor.py |
ReplayExecutor.on_op_begin |
|
7 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / hyper_offload / execution / replay / executor.py |
ReplayExecutor.on_op_end |
|
16 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / hyper_offload / execution / replay / executor.py |
(no function) |
|
13 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / hyper_offload / execution / tensor.py |
ShadowTensor.__new__ |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / hyper_offload / execution / tensor.py |
ShadowTensor.__init__ |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / hyper_offload / execution / tensor.py |
ShadowTensor.storage_id |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / hyper_offload / execution / tensor.py |
ShadowTensor.resolve |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / hyper_offload / execution / tensor.py |
ShadowTensor.__torch_dispatch__ |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / hyper_offload / execution / tensor.py |
ShadowTensor.__torch_dispatch__.unwrap |
|
9 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / hyper_offload / execution / tensor.py |
(no function) |
|
13 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / hyper_offload / execution / warmup / __init__.py |
(no function) |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / hyper_offload / execution / warmup / executor.py |
iter_tensors |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / hyper_offload / execution / warmup / executor.py |
WarmupExecutor.__init__ |
|
7 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / hyper_offload / execution / warmup / executor.py |
WarmupExecutor._enforce_budget |
|
17 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / hyper_offload / execution / warmup / executor.py |
WarmupExecutor.on_op_begin |
|
7 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / hyper_offload / execution / warmup / executor.py |
WarmupExecutor.on_op_end |
|
38 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / hyper_offload / execution / warmup / executor.py |
WarmupExecutor.finish |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / hyper_offload / execution / warmup / executor.py |
WarmupExecutor.reset |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / hyper_offload / execution / warmup / executor.py |
(no function) |
|
21 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / hyper_offload / execution / warmup / tracker.py |
ActivationTracker.__init__ |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / hyper_offload / execution / warmup / tracker.py |
ActivationTracker._ensure_id |
|
13 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / hyper_offload / execution / warmup / tracker.py |
ActivationTracker.get_activation_sid |
|
12 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / hyper_offload / execution / warmup / tracker.py |
ActivationTracker.register_op_activations |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / hyper_offload / execution / warmup / tracker.py |
ActivationTracker.storage_sizes |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / hyper_offload / execution / warmup / tracker.py |
ActivationTracker.clear_activations |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / hyper_offload / execution / warmup / tracker.py |
ActivationTracker.__repr__ |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / hyper_offload / execution / warmup / tracker.py |
(no function) |
|
15 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / hyper_offload / ir / __init__.py |
(no function) |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / hyper_offload / ir / replay.py |
(no function) |
|
7 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / hyper_offload / ir / schedule.py |
ResidencySchedule.add_pre |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / hyper_offload / ir / schedule.py |
ResidencySchedule.add_post |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / hyper_offload / ir / schedule.py |
ResidencySchedule.pre_actions |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / hyper_offload / ir / schedule.py |
ResidencySchedule.post_actions |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / hyper_offload / ir / schedule.py |
(no function) |
|
21 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / hyper_offload / ir / trace.py |
(no function) |
|
24 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / hyper_offload / planning / __init__.py |
(no function) |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / hyper_offload / planning / base.py |
ResidencyPlanner.build |
|
0 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / hyper_offload / planning / base.py |
(no function) |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / hyper_offload / planning / greedy.py |
_EvictionCandidate.distance |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / hyper_offload / planning / greedy.py |
_EvictionCandidate.start_interior |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / hyper_offload / planning / greedy.py |
_EvictionCandidate.end_interior |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / hyper_offload / planning / greedy.py |
GreedyResidencyPlanner.build |
|
13 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / hyper_offload / planning / greedy.py |
GreedyResidencyPlanner._group_accesses_by_storage |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / hyper_offload / planning / greedy.py |
GreedyResidencyPlanner._build_resident_footprint |
|
15 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / hyper_offload / planning / greedy.py |
GreedyResidencyPlanner._process_storage_accesses |
|
10 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / hyper_offload / planning / greedy.py |
GreedyResidencyPlanner._greedy_select |
|
11 |
1 |
0 |
|
91% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / hyper_offload / planning / greedy.py |
GreedyResidencyPlanner._emit_release_actions |
|
7 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / hyper_offload / planning / greedy.py |
GreedyResidencyPlanner._log_planning_summary |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / hyper_offload / planning / greedy.py |
GreedyResidencyPlanner._log_schedule_by_op |
|
10 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / hyper_offload / planning / greedy.py |
GreedyResidencyPlanner._earliest_safe_copy_start |
|
4 |
1 |
0 |
|
75% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / hyper_offload / planning / greedy.py |
GreedyResidencyPlanner._format_actions |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / hyper_offload / planning / greedy.py |
GreedyResidencyPlanner._add_eviction_actions |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / hyper_offload / planning / greedy.py |
(no function) |
|
49 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / hyper_offload / runtime / __init__.py |
(no function) |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / hyper_offload / runtime / bandwidth.py |
profile_transfer_bandwidth |
|
16 |
13 |
0 |
|
19% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / hyper_offload / runtime / bandwidth.py |
(no function) |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / hyper_offload / runtime / pinned_memory.py |
_align_to_bucket |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / hyper_offload / runtime / pinned_memory.py |
_bucket_for |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / hyper_offload / runtime / pinned_memory.py |
PinnedMemoryPool.__init__ |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / hyper_offload / runtime / pinned_memory.py |
PinnedMemoryPool.total_allocated |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / hyper_offload / runtime / pinned_memory.py |
PinnedMemoryPool.max_host_bytes |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / hyper_offload / runtime / pinned_memory.py |
PinnedMemoryPool._reclaim_locked |
|
8 |
1 |
0 |
|
88% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / hyper_offload / runtime / pinned_memory.py |
PinnedMemoryPool.acquire |
|
18 |
3 |
0 |
|
83% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / hyper_offload / runtime / pinned_memory.py |
PinnedMemoryPool.release |
|
10 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / hyper_offload / runtime / pinned_memory.py |
(no function) |
|
16 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / hyper_offload / runtime / residency.py |
PhysicalBuffer.device_storage |
|
16 |
4 |
0 |
|
75% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / hyper_offload / runtime / residency.py |
ResidencyManager.__init__ |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / hyper_offload / runtime / residency.py |
ResidencyManager._get_copy_stream |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / hyper_offload / runtime / residency.py |
ResidencyManager.resident_bytes |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / hyper_offload / runtime / residency.py |
ResidencyManager.device_resident_size |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / hyper_offload / runtime / residency.py |
ResidencyManager.wait_for_transfers |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / hyper_offload / runtime / residency.py |
ResidencyManager.sync_all_transfers |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / hyper_offload / runtime / residency.py |
ResidencyManager.bind |
|
9 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / hyper_offload / runtime / residency.py |
ResidencyManager.copy_d2h |
|
27 |
7 |
0 |
|
74% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / hyper_offload / runtime / residency.py |
ResidencyManager.copy_h2d |
|
29 |
10 |
0 |
|
66% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / hyper_offload / runtime / residency.py |
ResidencyManager.release_device |
|
9 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / hyper_offload / runtime / residency.py |
ResidencyManager.release_host |
|
9 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / hyper_offload / runtime / residency.py |
ResidencyManager.clear_runtime |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / hyper_offload / runtime / residency.py |
(no function) |
|
33 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / hyper_offload / runtime / timer.py |
DeviceTimer.__init__ |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / hyper_offload / runtime / timer.py |
DeviceTimer.start |
|
5 |
3 |
0 |
|
40% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / hyper_offload / runtime / timer.py |
DeviceTimer.stop |
|
10 |
6 |
0 |
|
40% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / hyper_offload / runtime / timer.py |
(no function) |
|
6 |
0 |
1 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / __init__.py |
(no function) |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / __init__.py |
(no function) |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / _backbone.py |
_Backbone.__init__ |
|
23 |
3 |
0 |
|
87% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / _backbone.py |
_Backbone.ccfg |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / _backbone.py |
_Backbone.ctx |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / _backbone.py |
_Backbone._load_eval_yaml_hook_cls |
|
25 |
25 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / _backbone.py |
_Backbone.update_config |
|
9 |
1 |
0 |
|
89% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / _backbone.py |
_Backbone._inner_static_mem |
|
16 |
1 |
0 |
|
94% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / _backbone.py |
_Backbone._inner_dynamic_mem |
|
13 |
2 |
0 |
|
85% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / _backbone.py |
_Backbone._inner_comm_mem |
|
16 |
1 |
0 |
|
94% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / _backbone.py |
_Backbone.__subplot_stages |
|
27 |
27 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / _backbone.py |
_Backbone.__plot_stages |
|
25 |
25 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / _backbone.py |
_Backbone.__update_stage_logs |
|
8 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / _backbone.py |
_Backbone.__preprocess_layer_custom_config_list |
|
11 |
4 |
0 |
|
64% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / _backbone.py |
_Backbone._estimate_backbone |
|
40 |
8 |
0 |
|
80% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / _backbone.py |
_Backbone.__estimate_stages_backbone |
|
26 |
8 |
0 |
|
69% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / _backbone.py |
_Backbone.__update_evaluator |
|
7 |
1 |
0 |
|
86% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / _backbone.py |
_Backbone.__update_next_layer_custom_function |
|
16 |
3 |
0 |
|
81% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / _backbone.py |
_Backbone.__chunk_stage_lay_loops |
|
34 |
5 |
0 |
|
85% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / _backbone.py |
_Backbone.__postprocess_stages |
|
38 |
8 |
0 |
|
79% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / _backbone.py |
_Backbone.__verbose_insights |
|
10 |
10 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / _backbone.py |
_Backbone.apply_hook |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / _backbone.py |
_Backbone.set_layer_custom |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / _backbone.py |
_Backbone.is_regular_layer |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / _backbone.py |
_Backbone.import_eval_yaml |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / _backbone.py |
_Backbone.get_num_layers |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / _backbone.py |
_Backbone.get_strategy |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / _backbone.py |
_Backbone.get_max_device_memory |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / _backbone.py |
_Backbone.print_stages |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / _backbone.py |
_Backbone.print_ctx |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / _backbone.py |
(no function) |
|
55 |
0 |
2 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / _bwd_overhead.py |
_BackwardOverhead.__init__ |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / _bwd_overhead.py |
_BackwardOverhead._fetch_node_and_switch_env |
|
14 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / _bwd_overhead.py |
_BackwardOverhead.estimate |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / _bwd_overhead.py |
_BackwardOverhead.vpp_1f1b_steady_overhead |
|
31 |
6 |
0 |
|
81% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / _bwd_overhead.py |
_BackwardOverhead.vpp_1f1b_steady_overhead.dyn |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / _bwd_overhead.py |
_BackwardOverhead.__stage_bwd_overhead_1f1b |
|
19 |
2 |
0 |
|
89% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / _bwd_overhead.py |
_BackwardOverhead.__stage_bwd_overhead_zbv |
|
31 |
5 |
0 |
|
84% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / _bwd_overhead.py |
(no function) |
|
8 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / _context.py |
NodeStatEval.__repr__ |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / _context.py |
_qname |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / _context.py |
NodeCommEval.__repr__ |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / _context.py |
NodeComputeEval.__repr__ |
|
10 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / _context.py |
NodeDynEval.__repr__ |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / _context.py |
NodeEval.__repr__ |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / _context.py |
Context.__init__ |
|
18 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / _context.py |
Context.__str__ |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / _context.py |
Context.print_node_eval |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / _context.py |
Context.eval |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / _context.py |
Context.init_tmp_buff |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / _context.py |
Context.copy_tmp_buff |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / _context.py |
Context.save2log |
|
20 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / _context.py |
(no function) |
|
59 |
0 |
2 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / _func_tracer.py |
_FuncTracer.__init__ |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / _func_tracer.py |
_FuncTracer.add2trace |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / _func_tracer.py |
_FuncTracer.is_constant_expr |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / _func_tracer.py |
_FuncTracer.scrap_term_symbols |
|
19 |
1 |
0 |
|
95% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / _func_tracer.py |
_FuncTracer.substitute |
|
17 |
2 |
0 |
|
88% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / _func_tracer.py |
_FuncTracer.fetch_node_from_lineno |
|
4 |
1 |
0 |
|
75% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / _func_tracer.py |
_FuncTracer.line_tracer |
|
56 |
56 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / _func_tracer.py |
_FuncTracer.extract_ast |
|
7 |
7 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / _func_tracer.py |
_FuncTracer.wrap |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / _func_tracer.py |
_FuncTracer.wrap.tracked_fun |
|
8 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / _func_tracer.py |
(no function) |
|
16 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / _hook_manager.py |
_HookManager.__init__ |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / _hook_manager.py |
_HookManager.fetch_hook_if_unimodal |
|
6 |
2 |
0 |
|
67% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / _hook_manager.py |
_HookManager.__is_valid_eval_func |
|
10 |
6 |
0 |
|
40% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / _hook_manager.py |
_HookManager.set_passes |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / _hook_manager.py |
_HookManager.__set_node_eval_fun |
|
22 |
7 |
0 |
|
68% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / _hook_manager.py |
_HookManager.__set_node_eval_comm_fun |
|
26 |
5 |
0 |
|
81% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / _hook_manager.py |
_HookManager.__resolve_compute_fun |
|
7 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / _hook_manager.py |
_HookManager.__resolve_compute_fun.zero |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / _hook_manager.py |
_HookManager.__set_node_eval_compute_fun |
|
19 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / _hook_manager.py |
_HookManager.set_head_eval_fun |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / _hook_manager.py |
_HookManager.set_tail_eval_fun |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / _hook_manager.py |
_HookManager.set_body_eval_fun |
|
9 |
2 |
0 |
|
78% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / _hook_manager.py |
_HookManager.set_attn_eval_fun |
|
8 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / _hook_manager.py |
_HookManager.set_ffn_eval_fun |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / _hook_manager.py |
_HookManager.set_expert_param_eval_fun |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / _hook_manager.py |
_HookManager.set_norm_eval_fun |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / _hook_manager.py |
_HookManager.set_pp_micro_factor_eval_fun |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / _hook_manager.py |
_HookManager.set_strategy |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / _hook_manager.py |
_HookManager.set_ccfg |
|
7 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / _hook_manager.py |
_HookManager.set_ccfg.custom_setter |
|
4 |
1 |
0 |
|
75% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / _hook_manager.py |
_HookManager.__wrap_mem_counter |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / _hook_manager.py |
_HookManager.__wrap_mem_counter.wrap |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / _hook_manager.py |
_HookManager.__custom_getattr |
|
21 |
6 |
0 |
|
71% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / _hook_manager.py |
_HookManager.__custom_getattr.constant |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / _hook_manager.py |
_HookManager.__custom_getattr.zero |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / _hook_manager.py |
_HookManager.import_eval_yaml |
|
31 |
1 |
0 |
|
97% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / _hook_manager.py |
_HookManager.is_regular_layer |
|
3 |
1 |
0 |
|
67% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / _hook_manager.py |
(no function) |
|
45 |
0 |
2 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / _ppb.py |
_PPB.__init__ |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / _ppb.py |
_PPB.add_to_ppb_list |
|
15 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / _ppb.py |
_PPB.lay_ppb |
|
31 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / _ppb.py |
_PPB.ppb_combine_bodies |
|
18 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / _ppb.py |
_PPB.lay_ppb_new |
|
72 |
7 |
0 |
|
90% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / _ppb.py |
_PPB.ppb_combine_bodies_new |
|
21 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / _ppb.py |
(no function) |
|
14 |
0 |
2 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / _utils.py |
_Utils.get_model_name |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / _utils.py |
_Utils.get_strategy |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / _utils.py |
_Utils.get_max_device_memory |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / _utils.py |
_Utils.get_num_layers |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / _utils.py |
_Utils.set_layer_custom |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / _utils.py |
_Utils.set_config |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / _utils.py |
_Utils.all_stage_micro_factors |
|
9 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / _utils.py |
_Utils.print_ccfg |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / _utils.py |
_Utils.print_ctx |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / _utils.py |
_Utils.print_node_eval |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / _utils.py |
_Utils.print_stages |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / _utils.py |
(no function) |
|
16 |
0 |
2 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / demo.py |
my_attn_num_param |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / demo.py |
custom |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / demo.py |
(no function) |
|
32 |
32 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / estimate_v2.py |
EvaluatorV2.__init__ |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / estimate_v2.py |
EvaluatorV2.reset_config |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / estimate_v2.py |
EvaluatorV2.load_hook_cls |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / estimate_v2.py |
EvaluatorV2.estimate_peak |
|
11 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / estimate_v2.py |
EvaluatorV2.estimate_peak_insight |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / estimate_v2.py |
EvaluatorV2.estimate_layer_memory |
|
11 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / estimate_v2.py |
EvaluatorV2.static_mem_stage |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / estimate_v2.py |
EvaluatorV2.dynamic_mem_stage |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / estimate_v2.py |
EvaluatorV2.logs_mem_stage |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / estimate_v2.py |
EvaluatorV2.static_mem_layer |
|
9 |
1 |
0 |
|
89% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / estimate_v2.py |
EvaluatorV2.dynamic_mem_layer |
|
9 |
1 |
0 |
|
89% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / estimate_v2.py |
EvaluatorV2.mem_fit |
|
14 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / estimate_v2.py |
estimate_memory |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / estimate_v2.py |
main |
|
35 |
35 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / estimate_v2.py |
(no function) |
|
29 |
1 |
3 |
|
97% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / evaluators / __init__.py |
(no function) |
|
0 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / evaluators / body.py |
EvalBody.num_params_layer |
|
10 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / evaluators / body.py |
EvalBody.stat_p_layer |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / evaluators / body.py |
EvalBody.stat_os_layer |
|
7 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / evaluators / body.py |
EvalBody.stat_grad_layer |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / evaluators / body.py |
EvalBody.layer_activ |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / evaluators / body.py |
EvalBody.fullrec_layer_activ |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / evaluators / body.py |
EvalBody.fullrec_layer_activ_gradclip |
|
10 |
2 |
0 |
|
80% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / evaluators / body.py |
EvalBody.fullrec_layer_comm_gradclip |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / evaluators / body.py |
EvalBody.act_cp_layer |
|
38 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / evaluators / body.py |
(no function) |
|
26 |
0 |
4 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / evaluators / comm.py |
EvalLayerComm.dp_comm_non_exp |
|
8 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / evaluators / comm.py |
EvalLayerComm.dp_comm_exp |
|
11 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / evaluators / comm.py |
EvalLayerComm.dp_comm_layer |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / evaluators / comm.py |
EvalLayerComm.tp_comm_non_exp |
|
7 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / evaluators / comm.py |
EvalLayerComm.tp_comm_exp |
|
9 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / evaluators / comm.py |
EvalLayerComm.tp_comm_layer |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / evaluators / comm.py |
EvalLayerComm.cp_comm_non_exp |
|
7 |
1 |
0 |
|
86% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / evaluators / comm.py |
EvalLayerComm.cp_comm_exp |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / evaluators / comm.py |
EvalLayerComm.cp_comm_layer |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / evaluators / comm.py |
EvalLayerComm.ep_comm_layer_balanced |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / evaluators / comm.py |
EvalLayerComm.ep_comm_layer_imbalanced |
|
16 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / evaluators / comm.py |
EvalLayerComm.ep_comm_layer |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / evaluators / comm.py |
EvalLayerComm.cp_comm_buffer |
|
22 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / evaluators / comm.py |
(no function) |
|
34 |
0 |
3 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / evaluators / compute.py |
EvalExpertCompute.router_compute_cost |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / evaluators / compute.py |
EvalExpertCompute.expert_compute_cost_balanced |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / evaluators / compute.py |
EvalExpertCompute.expert_compute_cost_imbalanced |
|
17 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / evaluators / compute.py |
EvalExpertCompute.shared_expert_compute_cost |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / evaluators / compute.py |
EvalExpertCompute.expert_layer_compute |
|
8 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / evaluators / compute.py |
(no function) |
|
13 |
0 |
3 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / evaluators / head.py |
EvalHead.num_params_embed |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / evaluators / head.py |
EvalHead.stat_embed_p |
|
7 |
1 |
0 |
|
86% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / evaluators / head.py |
EvalHead.stat_embed_os |
|
7 |
1 |
0 |
|
86% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / evaluators / head.py |
EvalHead.stat_embed_grad |
|
7 |
1 |
0 |
|
86% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / evaluators / head.py |
EvalHead.dp_comm_embed |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / evaluators / head.py |
EvalHead.tp_comm_embed |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / evaluators / head.py |
EvalHead.activ_embed |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / evaluators / head.py |
(no function) |
|
17 |
0 |
3 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / evaluators / layer_block.py |
EvalAttn.num_params_mla |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / evaluators / layer_block.py |
EvalAttn.num_params_attn |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / evaluators / layer_block.py |
EvalAttn.attn_qkv_activations |
|
11 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / evaluators / layer_block.py |
EvalAttn.attn_score_activations |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / evaluators / layer_block.py |
EvalAttn.attn_proj_activations |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / evaluators / layer_block.py |
EvalFFn.num_params_ffn |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / evaluators / layer_block.py |
EvalFFn.num_params_routed_expert |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / evaluators / layer_block.py |
EvalFFn.num_params_shared_expert |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / evaluators / layer_block.py |
EvalFFn.ffn_activations |
|
13 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / evaluators / layer_block.py |
EvalFFn.ffn_router_and_concat_activations |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / evaluators / layer_block.py |
EvalFFn.shared_exp_activations |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / evaluators / layer_block.py |
EvalFFn.routed_exp_activations |
|
9 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / evaluators / layer_block.py |
EvalFFn.ffn_moe_activations |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / evaluators / layer_block.py |
EvalNorm.num_params_norm |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / evaluators / layer_block.py |
EvalNorm.norm_activations |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / evaluators / layer_block.py |
(no function) |
|
38 |
0 |
3 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / evaluators / tail.py |
EvalMTP.num_params_mtp |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / evaluators / tail.py |
EvalMTP.stat_mtp_p |
|
12 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / evaluators / tail.py |
EvalMTP.stat_mtp_os |
|
12 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / evaluators / tail.py |
EvalMTP.stat_mtp_grad |
|
12 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / evaluators / tail.py |
EvalMTP.activ_mtp |
|
10 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / evaluators / tail.py |
EvalMTP.comm_mtp |
|
14 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / evaluators / tail.py |
EvalTailSingle.stat_output_single_p |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / evaluators / tail.py |
EvalTailSingle.stat_output_single_os |
|
6 |
1 |
0 |
|
83% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / evaluators / tail.py |
EvalTailSingle.stat_output_single_grad |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / evaluators / tail.py |
EvalTailSingle.activ_out_single |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / evaluators / tail.py |
EvalTailSingle.comm_out_single |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / evaluators / tail.py |
EvalTail.num_params_output |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / evaluators / tail.py |
EvalTail.stat_output_p |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / evaluators / tail.py |
EvalTail.stat_output_os |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / evaluators / tail.py |
EvalTail.stat_output_grad |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / evaluators / tail.py |
EvalTail.activ_output |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / evaluators / tail.py |
EvalTail.comm_output |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / evaluators / tail.py |
(no function) |
|
40 |
0 |
3 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / evaluators / utils.py |
EvalUtils.mb |
|
5 |
1 |
0 |
|
80% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / evaluators / utils.py |
EvalUtils.rec_coeff |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / evaluators / utils.py |
EvalUtils.eval_expr_insight |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / evaluators / utils.py |
EvalUtils.__log_ast_mem |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / evaluators / utils.py |
EvalUtils.__eval_ast_name |
|
8 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / evaluators / utils.py |
EvalUtils.__eval_ast_unary |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / evaluators / utils.py |
EvalUtils.__eval_ast_binop |
|
10 |
1 |
0 |
|
90% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / evaluators / utils.py |
EvalUtils.__eval_ast_call |
|
11 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / evaluators / utils.py |
EvalUtils.__eval_ast_mem |
|
11 |
3 |
0 |
|
73% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / evaluators / utils.py |
EvalUtils.pp_1f1b_micro_factor |
|
23 |
6 |
0 |
|
74% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / evaluators / utils.py |
EvalUtils.pp_seq1f1b_micro_factor |
|
26 |
10 |
0 |
|
62% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / evaluators / utils.py |
EvalUtils.pp_dualpipe_v_micro_factor |
|
7 |
3 |
0 |
|
57% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / evaluators / utils.py |
EvalUtils.pp_gpipe_micro_factor |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / evaluators / utils.py |
(no function) |
|
33 |
0 |
4 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / hook_base.py |
HookMetaclass.__init__ |
|
16 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / hook_base.py |
hook_runner |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / hook_base.py |
hook_runner.wrapper |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / hook_base.py |
MemEvalHook.get_hooks |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / hook_base.py |
MemEvalHook.run_hooks |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / hook_base.py |
(no function) |
|
14 |
0 |
2 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / hooks / __init__.py |
(no function) |
|
0 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / hooks / template.py |
Template.f |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / hooks / template.py |
Template.custom_ccfg |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / hooks / template.py |
Template.run_hooks |
|
10 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / hooks / template.py |
(no function) |
|
9 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / logger.py |
setup_logger |
|
8 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / logger.py |
(no function) |
|
9 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / score.py |
mape |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / score.py |
r2 |
|
9 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / score.py |
(no function) |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / size.py |
Unit.from_string |
|
11 |
2 |
0 |
|
82% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / size.py |
Unit.__str__ |
|
7 |
3 |
0 |
|
57% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / size.py |
Unit.__lt__ |
|
3 |
1 |
0 |
|
67% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / size.py |
Memory.__init__ |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / size.py |
Memory.from_string |
|
15 |
1 |
0 |
|
93% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / size.py |
Memory.from_b |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / size.py |
Memory.from_kb |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / size.py |
Memory.from_mb |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / size.py |
Memory.from_gb |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / size.py |
Memory.zero |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / size.py |
Memory.__str__ |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / size.py |
Memory.to |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / size.py |
Memory.to_gb |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / size.py |
Memory.to_mb |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / size.py |
Memory.to_kb |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / size.py |
Memory.to_b |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / size.py |
Memory.increase |
|
5 |
1 |
0 |
|
80% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / size.py |
Memory.__add__ |
|
8 |
2 |
0 |
|
75% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / size.py |
Memory.decrease |
|
5 |
1 |
0 |
|
80% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / size.py |
Memory.__sub__ |
|
8 |
1 |
0 |
|
88% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / size.py |
Memory.__lt__ |
|
4 |
1 |
0 |
|
75% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / size.py |
Memory.__le__ |
|
4 |
1 |
0 |
|
75% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / size.py |
Memory.__abs__ |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / size.py |
(no function) |
|
40 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / validators / __init__.py |
(no function) |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / validators / ep_constraints.py |
ConstraintResult.__bool__ |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / validators / ep_constraints.py |
EpConstraints.check_ep_divisibility |
|
7 |
1 |
0 |
|
86% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / validators / ep_constraints.py |
EpConstraints.check_expert_hidden_divisibility |
|
7 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / validators / ep_constraints.py |
EpConstraints.check_device_limit |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / validators / ep_constraints.py |
EpConstraints.check_ep_pp_stage_feasibility |
|
18 |
1 |
0 |
|
94% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / validators / ep_constraints.py |
EpConstraints.validate_all |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / memory_estimation / validators / ep_constraints.py |
(no function) |
|
19 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / __init__.py |
(no function) |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / balancing_adapter.py |
Pipeline.__init__ |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / balancing_adapter.py |
Pipeline.__str__ |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / balancing_adapter.py |
Pipeline.chunk_stage |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / balancing_adapter.py |
infer_pp_and_vpp |
|
7 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / balancing_adapter.py |
BalancingAdapter.__init__ |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / balancing_adapter.py |
BalancingAdapter.treat_pp_list |
|
27 |
2 |
0 |
|
93% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / balancing_adapter.py |
BalancingAdapter.treat_vpp_list |
|
30 |
7 |
0 |
|
77% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / balancing_adapter.py |
BalancingAdapter.treat_recompute_list |
|
10 |
6 |
0 |
|
40% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / balancing_adapter.py |
BalancingAdapter.treat_recompute |
|
6 |
1 |
0 |
|
83% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / balancing_adapter.py |
BalancingAdapter.treat_offset |
|
22 |
12 |
0 |
|
45% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / balancing_adapter.py |
BalancingAdapter.check_offset |
|
7 |
2 |
0 |
|
71% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / balancing_adapter.py |
BalancingAdapter.offset_checker |
|
5 |
2 |
0 |
|
60% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / balancing_adapter.py |
BalancingAdapter.make_valid |
|
29 |
6 |
0 |
|
79% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / balancing_adapter.py |
BalancingAdapter.default_offset |
|
13 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / balancing_adapter.py |
BalancingAdapter.default_recompute |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / balancing_adapter.py |
copy_offset |
|
3 |
1 |
0 |
|
67% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / balancing_adapter.py |
is_zero_d |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / balancing_adapter.py |
is_one_d |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / balancing_adapter.py |
is_two_d |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / balancing_adapter.py |
make_one_d |
|
7 |
1 |
0 |
|
86% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / balancing_adapter.py |
make_two_d |
|
7 |
2 |
0 |
|
71% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / balancing_adapter.py |
(no function) |
|
31 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / __init__.py |
(no function) |
|
0 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / _cost_model_variables.py |
_CostModVar.__init__ |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / _cost_model_variables.py |
_CostModVar._load_parser_cls |
|
26 |
4 |
0 |
|
85% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / _cost_model_variables.py |
_CostModVar.get_framework_parser_naive |
|
12 |
4 |
0 |
|
67% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / _cost_model_variables.py |
_CostModVar.get_framework_parser |
|
5 |
1 |
0 |
|
80% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / _cost_model_variables.py |
_CostModVar.update_config |
|
21 |
7 |
0 |
|
67% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / _cost_model_variables.py |
(no function) |
|
113 |
0 |
2 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / arch_hooks.py |
CWrap.__init__ |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / arch_hooks.py |
CWrap.set_ccfg |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / arch_hooks.py |
CWrap.get_model_name |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / arch_hooks.py |
CWrap.reset |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / arch_hooks.py |
CWrap.__getattr__ |
|
3 |
1 |
0 |
|
67% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / arch_hooks.py |
CWrap.set_strategy |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / arch_hooks.py |
CWrap.get_strategy |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / arch_hooks.py |
custom_default_transformer |
|
14 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / arch_hooks.py |
custom_llama2 |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / arch_hooks.py |
custom_mixtral |
|
15 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / arch_hooks.py |
custom_t5 |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / arch_hooks.py |
custom_t5.encode |
|
14 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / arch_hooks.py |
custom_t5.decode |
|
14 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / arch_hooks.py |
custom_t5.hook_encode |
|
3 |
1 |
0 |
|
67% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / arch_hooks.py |
custom_t5.hook_decode |
|
3 |
1 |
0 |
|
67% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / arch_hooks.py |
custom_pangualpha |
|
14 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / arch_hooks.py |
custom_deepseek3 |
|
22 |
5 |
0 |
|
77% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / arch_hooks.py |
custom_deepseek3.dense |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / arch_hooks.py |
custom_deepseek3.moe |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / arch_hooks.py |
custom_deepseek3.hook_dense |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / arch_hooks.py |
custom_deepseek3.hook_moe |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / arch_hooks.py |
custom_qwen |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / arch_hooks.py |
custom_cm |
|
10 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / arch_hooks.py |
custom_cm.custom_shard |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / arch_hooks.py |
custom_cm.wrap_hook |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / arch_hooks.py |
custom_cm.num_params_norm_cm |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / arch_hooks.py |
check_and_apply_custom_hook |
|
9 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / arch_hooks.py |
(no function) |
|
21 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / config.py |
YamlObject.from_dict |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / config.py |
YamlObject.__init__ |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / config.py |
YamlObject.__getattr__ |
|
4 |
1 |
0 |
|
75% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / config.py |
YamlObject.__copy__ |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / config.py |
YamlObject.__deepcopy__ |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / config.py |
YamlObject.__getstate__ |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / config.py |
YamlObject.__setstate__ |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / config.py |
YamlObject.to_dict |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / config.py |
YamlObject.dump |
|
5 |
1 |
0 |
|
80% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / config.py |
Config.__init__ |
|
18 |
4 |
0 |
|
78% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / config.py |
Config.__str__ |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / config.py |
Config.set_par |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / config.py |
(no function) |
|
20 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / cost_model_config.py |
(no function) |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / cost_model_preprocess.py |
detect_attention_type |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / cost_model_preprocess.py |
compute_kv_dim |
|
8 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / cost_model_preprocess.py |
CostModelConfig.__init__ |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / cost_model_preprocess.py |
CostModelConfig.__str__ |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / cost_model_preprocess.py |
CostModelConfig.__getattr__ |
|
5 |
1 |
0 |
|
80% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / cost_model_preprocess.py |
CostModelConfig.__copy__ |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / cost_model_preprocess.py |
CostModelConfig.__deepcopy__ |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / cost_model_preprocess.py |
CostModelConfig.__getstate__ |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / cost_model_preprocess.py |
CostModelConfig.__setstate__ |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / cost_model_preprocess.py |
CostModelConfig.fp_bytes |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / cost_model_preprocess.py |
CostModelConfig.print_stages_i |
|
13 |
4 |
0 |
|
69% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / cost_model_preprocess.py |
CostModelConfig.print_stages |
|
6 |
1 |
0 |
|
83% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / cost_model_preprocess.py |
CostModelConfig.count_layers |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / cost_model_preprocess.py |
CostModelConfig.print_parallelism |
|
9 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / cost_model_preprocess.py |
CostModelConfig.strategy_num_devices |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / cost_model_preprocess.py |
CostModelConfig.is_consistent_pp_config |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / cost_model_preprocess.py |
CostModelConfig.is_consistent_pp_config.is_valid_cfg |
|
9 |
1 |
0 |
|
89% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / cost_model_preprocess.py |
CostModelConfig.__maybe_set_int |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / cost_model_preprocess.py |
CostModelConfig.__strategy_target |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / cost_model_preprocess.py |
CostModelConfig.set_strategy |
|
43 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / cost_model_preprocess.py |
CostModelConfig.get_strategy |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / cost_model_preprocess.py |
CostModelConfig.get_strategy.strategy |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / cost_model_preprocess.py |
CostModelConfig.layer_custom_config_callback |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / cost_model_preprocess.py |
CostModelConfig.layer_custom_config_callback.wrap |
|
7 |
1 |
0 |
|
86% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / cost_model_preprocess.py |
(no function) |
|
36 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / cp_types.py |
_resolve_cp_algo |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / cp_types.py |
(no function) |
|
73 |
0 |
2 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / framework_parsers / __init__.py |
(no function) |
|
0 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / framework_parsers / _cost_model_parser.py |
_CostModelParser.__init__ |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / framework_parsers / _cost_model_parser.py |
_CostModelParser.parse |
|
0 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / framework_parsers / _cost_model_parser.py |
_CostModelParser.config_optimizer_shard |
|
7 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / framework_parsers / _cost_model_parser.py |
_CostModelParser.init_hff |
|
9 |
9 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / framework_parsers / _cost_model_parser.py |
_CostModelParser.config_comm_flag |
|
7 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / framework_parsers / _cost_model_parser.py |
_CostModelParser.config_dp_tp_exp |
|
13 |
3 |
0 |
|
77% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / framework_parsers / _cost_model_parser.py |
(no function) |
|
13 |
0 |
2 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / framework_parsers / cost_model_parser_hyper.py |
custom_vision_tower |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / framework_parsers / cost_model_parser_hyper.py |
custom_vision_tower_hook |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / framework_parsers / cost_model_parser_hyper.py |
CostModelParserHyperV2.parse |
|
24 |
1 |
0 |
|
96% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / framework_parsers / cost_model_parser_hyper.py |
CostModelParserHyperV2._resolve_model_config_pipeline |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / framework_parsers / cost_model_parser_hyper.py |
CostModelParserHyperV2._model_section |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / framework_parsers / cost_model_parser_hyper.py |
CostModelParserHyperV2._visual_seq_len_override |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / framework_parsers / cost_model_parser_hyper.py |
CostModelParserHyperV2._spec_int |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / framework_parsers / cost_model_parser_hyper.py |
CostModelParserHyperV2._apply_spec |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / framework_parsers / cost_model_parser_hyper.py |
CostModelParserHyperV2._apply_core_spec |
|
13 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / framework_parsers / cost_model_parser_hyper.py |
CostModelParserHyperV2._apply_moe_spec |
|
23 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / framework_parsers / cost_model_parser_hyper.py |
CostModelParserHyperV2._apply_scaling_spec |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / framework_parsers / cost_model_parser_hyper.py |
CostModelParserHyperV2._resolve_multimodal |
|
12 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / framework_parsers / cost_model_parser_hyper.py |
CostModelParserHyperV2._clone_submodule |
|
14 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / framework_parsers / cost_model_parser_hyper.py |
CostModelParserHyperV2._build_vision_submodule |
|
10 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / framework_parsers / cost_model_parser_hyper.py |
CostModelParserHyperV2._even_offset |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / framework_parsers / cost_model_parser_hyper.py |
CostModelParserHyperV2._front_loaded_offset |
|
9 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / framework_parsers / cost_model_parser_hyper.py |
CostModelParserHyperV2._resolve_sequence_length |
|
7 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / framework_parsers / cost_model_parser_hyper.py |
CostModelParserHyperV2._resolve_device_capacity |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / framework_parsers / cost_model_parser_hyper.py |
CostModelParserHyperV2._get_cfg_attr |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / framework_parsers / cost_model_parser_hyper.py |
CostModelParserHyperV2._config_to_flat_dict |
|
7 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / framework_parsers / cost_model_parser_hyper.py |
CostModelParserHyperV2._bytes_from_dtype |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / framework_parsers / cost_model_parser_hyper.py |
CostModelParserHyperV2._parse_parallelism |
|
7 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / framework_parsers / cost_model_parser_hyper.py |
CostModelParserHyperV2._parse_parallel_dimensions |
|
17 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / framework_parsers / cost_model_parser_hyper.py |
CostModelParserHyperV2._resolve_data_parallel |
|
10 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / framework_parsers / cost_model_parser_hyper.py |
CostModelParserHyperV2._parse_sequence_parallelism |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / framework_parsers / cost_model_parser_hyper.py |
CostModelParserHyperV2._parse_optimizer_parallelism |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / framework_parsers / cost_model_parser_hyper.py |
CostModelParserHyperV2._parse_batch |
|
13 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / framework_parsers / cost_model_parser_hyper.py |
CostModelParserHyperV2._parse_feature_flags |
|
19 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / framework_parsers / cost_model_parser_hyper.py |
CostModelParserHyperV2._parse_recompute |
|
17 |
3 |
0 |
|
82% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / framework_parsers / cost_model_parser_hyper.py |
CostModelParserHyperV2._init_bytes |
|
10 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / framework_parsers / cost_model_parser_hyper.py |
CostModelParserHyperV2._init_moe_strategy |
|
10 |
6 |
0 |
|
40% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / framework_parsers / cost_model_parser_hyper.py |
CostModelParserHyperV2._init_offset |
|
12 |
4 |
0 |
|
67% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / framework_parsers / cost_model_parser_hyper.py |
CostModelParserHyperV2.config_shard_emb |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / framework_parsers / cost_model_parser_hyper.py |
CostModelParserHyperV2.config_shard_recompute |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / framework_parsers / cost_model_parser_hyper.py |
CostModelParserHyperV2._init_shard |
|
9 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / framework_parsers / cost_model_parser_hyper.py |
(no function) |
|
48 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / framework_parsers / cost_model_parser_hyperparallel.py |
CostModelParserHyperparallel.parse |
|
29 |
25 |
0 |
|
14% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / framework_parsers / cost_model_parser_hyperparallel.py |
CostModelParserHyperparallel.__obj_to_config |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / framework_parsers / cost_model_parser_hyperparallel.py |
CostModelParserHyperparallel.__parse_init_code |
|
17 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / framework_parsers / cost_model_parser_hyperparallel.py |
CostModelParserHyperparallel.__parse_init_code.parse_args |
|
5 |
2 |
0 |
|
60% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / framework_parsers / cost_model_parser_hyperparallel.py |
CostModelParserHyperparallel.__parse_toml |
|
18 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / framework_parsers / cost_model_parser_hyperparallel.py |
CostModelParserHyperparallel.__parse_strat |
|
27 |
3 |
0 |
|
89% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / framework_parsers / cost_model_parser_hyperparallel.py |
CostModelParserHyperparallel.__parse_hyperparam |
|
24 |
4 |
0 |
|
83% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / framework_parsers / cost_model_parser_hyperparallel.py |
CostModelParserHyperparallel.__parse_moe |
|
12 |
4 |
0 |
|
67% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / framework_parsers / cost_model_parser_hyperparallel.py |
CostModelParserHyperparallel.__parse_feature_flag |
|
9 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / framework_parsers / cost_model_parser_hyperparallel.py |
CostModelParserHyperparallel.__parse_batch |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / framework_parsers / cost_model_parser_hyperparallel.py |
CostModelParserHyperparallel.__init_shard |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / framework_parsers / cost_model_parser_hyperparallel.py |
CostModelParserHyperparallel.__init_bytes |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / framework_parsers / cost_model_parser_hyperparallel.py |
(no function) |
|
22 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / framework_parsers / cost_model_parser_mindformers.py |
CostModelParserMindformers.parse |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / framework_parsers / cost_model_parser_mindformers.py |
CostModelParserMindformers.__config_parse_yaml_parallelism |
|
23 |
5 |
0 |
|
78% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / framework_parsers / cost_model_parser_mindformers.py |
CostModelParserMindformers.__config_parse_yaml_hyperparameters |
|
23 |
5 |
0 |
|
78% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / framework_parsers / cost_model_parser_mindformers.py |
CostModelParserMindformers.__config_parse_yaml_moe |
|
23 |
9 |
0 |
|
61% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / framework_parsers / cost_model_parser_mindformers.py |
CostModelParserMindformers.__config_parse_yaml_op_recompute |
|
8 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / framework_parsers / cost_model_parser_mindformers.py |
CostModelParserMindformers.config_shard_emb |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / framework_parsers / cost_model_parser_mindformers.py |
CostModelParserMindformers.__config_parse_yaml |
|
49 |
5 |
0 |
|
90% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / framework_parsers / cost_model_parser_mindformers.py |
(no function) |
|
12 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / framework_parsers / cost_model_parser_mindspeed.py |
CostModelParserMindspeed.parse |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / framework_parsers / cost_model_parser_mindspeed.py |
CostModelParserMindspeed.__config_parse_json_multimodals |
|
32 |
15 |
0 |
|
53% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / framework_parsers / cost_model_parser_mindspeed.py |
CostModelParserMindspeed.__complete_unimodal_pp_plan |
|
26 |
26 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / framework_parsers / cost_model_parser_mindspeed.py |
CostModelParserMindspeed.__search_and_parse_mods_ccfg |
|
8 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / framework_parsers / cost_model_parser_mindspeed.py |
CostModelParserMindspeed.__config_parse_json_parallelism |
|
12 |
1 |
0 |
|
92% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / framework_parsers / cost_model_parser_mindspeed.py |
CostModelParserMindspeed.__config_parse_json_hyperparameters |
|
11 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / framework_parsers / cost_model_parser_mindspeed.py |
CostModelParserMindspeed.__config_parse_json_moe |
|
8 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / framework_parsers / cost_model_parser_mindspeed.py |
CostModelParserMindspeed.__config_parse_json_op_recompute |
|
8 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / framework_parsers / cost_model_parser_mindspeed.py |
CostModelParserMindspeed.__config_parse_json |
|
50 |
1 |
0 |
|
98% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / framework_parsers / cost_model_parser_mindspeed.py |
(no function) |
|
14 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / generate_partitions.py |
PartitionGenerator.combine_partition_multimodal |
|
10 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / generate_partitions.py |
PartitionGenerator.generate_partitions_vpp |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / generate_partitions.py |
PartitionGenerator.__process_select_recompute |
|
15 |
6 |
0 |
|
60% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / generate_partitions.py |
PartitionGenerator.__process_full_recompute |
|
11 |
1 |
0 |
|
91% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / generate_partitions.py |
PartitionGenerator.__process_offset |
|
9 |
1 |
0 |
|
89% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / generate_partitions.py |
PartitionGenerator.generate_partitions_vpp_unimodal |
|
22 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / generate_partitions.py |
PartitionGenerator.insert_emb_out_partitions |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / generate_partitions.py |
PartitionGenerator.insert_emb_out_partitions_zbv |
|
10 |
4 |
0 |
|
60% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / generate_partitions.py |
PartitionGenerator.insert_emb_out_partitions_1f1b |
|
10 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / generate_partitions.py |
PartitionGenerator.first_and_last_non_empty_stage |
|
14 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / generate_partitions.py |
(no function) |
|
13 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / hardware.py |
Type.__init__ |
|
6 |
1 |
0 |
|
83% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / hardware.py |
Type.__str__ |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / hardware.py |
Type.__repr__ |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / hardware.py |
Type.devices_below_level |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / hardware.py |
Type.intra_node_num |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / hardware.py |
Type.levels_used |
|
8 |
1 |
0 |
|
88% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / hardware.py |
Type.level_assign |
|
36 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / hardware.py |
Machine.__init__ |
|
12 |
1 |
0 |
|
92% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / hardware.py |
Machine.update_num_if_none |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / hardware.py |
Machine.pipeline_bound |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / hardware.py |
prime_factors |
|
8 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / hardware.py |
all_factors_combinations |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / hardware.py |
all_factors_combinations.rec_factors |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / hardware.py |
all_divisors |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / hardware.py |
from_prime_factors |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / hardware.py |
split_node |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / hardware.py |
unique_factors |
|
7 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / hardware.py |
highest_power_of_2_divisor |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / hardware.py |
get_cp_topology |
|
9 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / hardware.py |
get_cp_bandwidth |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / hardware.py |
recommend_cp_max_by_attention |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / hardware.py |
(no function) |
|
34 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / common / layer_type.py |
(no function) |
|
7 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / debug.py |
PerfParts.__str__ |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / debug.py |
PerfParts.short_name |
|
22 |
1 |
0 |
|
95% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / debug.py |
RealParts.__str__ |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / debug.py |
MemParts.__str__ |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / debug.py |
Debug.__init__ |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / debug.py |
Debug.is_enabled |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / debug.py |
Debug.column_titles |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / debug.py |
Debug.values |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / debug.py |
Debug.write |
|
8 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / debug.py |
pastel |
|
16 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / debug.py |
near_white |
|
8 |
1 |
0 |
|
88% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / debug.py |
dim_color |
|
8 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / debug.py |
gen_colors |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / debug.py |
set_twin_handles |
|
16 |
16 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / debug.py |
Plot.__init__ |
|
7 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / debug.py |
Plot.make_table |
|
22 |
22 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / debug.py |
Plot.close |
|
9 |
9 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / debug.py |
Plot.parse_data |
|
23 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / debug.py |
plot_nd |
|
8 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / debug.py |
plot_vs_real |
|
7 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / debug.py |
plot_vs_real_comm_classified |
|
14 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / debug.py |
correlation_topk |
|
20 |
6 |
0 |
|
70% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / debug.py |
get_real_data |
|
18 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / debug.py |
get_diff_dims |
|
11 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / debug.py |
get_comm_classified_data |
|
29 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / debug.py |
estimation_in_real_parts |
|
8 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / debug.py |
real_in_parts |
|
14 |
2 |
0 |
|
86% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / debug.py |
correlation_with_classified_comms |
|
27 |
1 |
0 |
|
96% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / debug.py |
color_diff |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / debug.py |
color_correl |
|
9 |
5 |
0 |
|
44% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / debug.py |
print_diff |
|
11 |
11 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / debug.py |
get_distance_i |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / debug.py |
get_correl_i |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / debug.py |
print_part_x_file |
|
17 |
2 |
0 |
|
88% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / debug.py |
print_correlations_classified |
|
12 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / debug.py |
is_constant |
|
4 |
1 |
0 |
|
75% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / debug.py |
pearson_wait |
|
8 |
2 |
0 |
|
75% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / debug.py |
(no function) |
|
83 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / dimensions.py |
Dimension.__init__ |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / dimensions.py |
Dimension.__str__ |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / dimensions.py |
Dimension.__repr__ |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / dimensions.py |
Dimension.lname |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / dimensions.py |
Dimension.from_config |
|
6 |
3 |
0 |
|
50% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / dimensions.py |
Dimension.reset_bound |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / dimensions.py |
Dimension.set_bound |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / dimensions.py |
Dimension.get_bound |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / dimensions.py |
Dimension.is_valid |
|
9 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / dimensions.py |
Dimensions.__init__ |
|
10 |
2 |
0 |
|
80% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / dimensions.py |
Dimensions._reset_all_dims |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / dimensions.py |
Dimensions.__str__ |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / dimensions.py |
Dimensions.__repr__ |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / dimensions.py |
Dimensions.keys |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / dimensions.py |
Dimensions.global_batch_size |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / dimensions.py |
Dimensions.values |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / dimensions.py |
Dimensions.unique_name |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / dimensions.py |
Dimensions.has_dim |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / dimensions.py |
Dimensions._check_mbn_pp |
|
7 |
1 |
0 |
|
86% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / dimensions.py |
Dimensions._check_power_of_two |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / dimensions.py |
Dimensions.is_valid |
|
15 |
3 |
0 |
|
80% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / dimensions.py |
Dimensions.val |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / dimensions.py |
Dimensions.set |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / dimensions.py |
Dimensions.steal |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / dimensions.py |
get_dim |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / dimensions.py |
get_dims |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / dimensions.py |
_cp_params_from_args |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / dimensions.py |
_cp_ok_result |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / dimensions.py |
_cp_check_ulysses_heads |
|
11 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / dimensions.py |
_cp_check_device_sufficiency |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / dimensions.py |
_cp_collect_warnings |
|
19 |
1 |
0 |
|
95% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / dimensions.py |
_cp_check_memory |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / dimensions.py |
validate_cp_constraints |
|
18 |
1 |
0 |
|
94% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / dimensions.py |
(no function) |
|
53 |
0 |
2 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / global_config.py |
GlobalConfig.__init__ |
|
11 |
2 |
0 |
|
82% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / global_config.py |
GlobalConfig.dim_val |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / global_config.py |
GlobalConfig.global_batch_size |
|
9 |
2 |
0 |
|
78% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / global_config.py |
GlobalConfig.layer_num_for_offset |
|
6 |
1 |
0 |
|
83% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / global_config.py |
GlobalConfig.total_layer_num |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / global_config.py |
GlobalConfig.adapt_config_balancing |
|
9 |
1 |
0 |
|
89% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / global_config.py |
GlobalConfig.adapt_config |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / global_config.py |
GlobalConfig.write |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / global_config.py |
GlobalConfig.moe_valid |
|
8 |
1 |
0 |
|
88% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / global_config.py |
GlobalConfig.ep_constraints_valid |
|
17 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / global_config.py |
GlobalConfig.make_parallel_config_args |
|
11 |
2 |
0 |
|
82% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / global_config.py |
GlobalConfig.make_parallel_config |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / global_config.py |
GlobalConfig.set_parallel_config |
|
18 |
3 |
0 |
|
83% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / global_config.py |
GlobalConfig.space |
|
8 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / global_config.py |
GlobalConfig.range_space |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / global_config.py |
GlobalConfig.bool_space |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / global_config.py |
GlobalConfig.max_op |
|
11 |
2 |
0 |
|
82% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / global_config.py |
(no function) |
|
25 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / logger.py |
MyLogger.output |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / logger.py |
setup_logger |
|
11 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / logger.py |
set_verbose_level |
|
23 |
11 |
0 |
|
52% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / logger.py |
(no function) |
|
14 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / parallelize.py |
ParallelizeLayer.__init__ |
|
28 |
2 |
0 |
|
93% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / parallelize.py |
ParallelizeLayer.bound_space |
|
10 |
1 |
0 |
|
90% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / parallelize.py |
ParallelizeLayer.filtered_out |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / parallelize.py |
ParallelizeLayer.is_valid |
|
35 |
6 |
0 |
|
83% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / parallelize.py |
ParallelizeLayer.memory_estim |
|
9 |
1 |
0 |
|
89% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / parallelize.py |
ParallelizeLayer.generate_search_space |
|
28 |
15 |
0 |
|
46% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / parallelize.py |
ParallelizeLayer.device_loops |
|
9 |
1 |
0 |
|
89% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / parallelize.py |
ParallelizeLayer.batch_loops |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / parallelize.py |
ParallelizeLayer.parallel_loops |
|
8 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / parallelize.py |
ParallelizeLayer.inside_loop_nest |
|
17 |
6 |
0 |
|
65% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / parallelize.py |
ParallelizeLayer.order_search_space |
|
33 |
14 |
0 |
|
58% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / parallelize.py |
ParallelizeLayer.order_space_test_comm_classified |
|
15 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / parallelize.py |
ParallelizeLayer.order_space_test |
|
17 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / parallelize.py |
ParallelizeLayer.plot_title |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / parallelize.py |
ParallelizeLayer.run_generation_to_ordering |
|
16 |
4 |
0 |
|
75% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / parallelize.py |
ParallelizeLayer.to_ppb |
|
13 |
13 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / parallelize.py |
ParallelizeLayer.test_from_csv |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / parallelize.py |
ParallelizeLayer.test_from_csv_comm_classified |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / parallelize.py |
ParallelizeMultiModal.__init__ |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / parallelize.py |
Parallelize.__init__ |
|
15 |
2 |
0 |
|
87% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / parallelize.py |
Parallelize.__getattr__ |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / parallelize.py |
space_to_string |
|
28 |
4 |
0 |
|
86% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / parallelize.py |
pool_estimate_memory |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / parallelize.py |
pool_estimate_performance |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / parallelize.py |
(no function) |
|
44 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / run_nd.py |
_run_hyper_v2_search |
|
30 |
4 |
0 |
|
87% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / nd / run_nd.py |
(no function) |
|
56 |
2 |
0 |
|
96% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / perf_estimation / __init__.py |
(no function) |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / perf_estimation / comm_time.py |
_cp_resolve_topology |
|
9 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / perf_estimation / comm_time.py |
_cp_comm_zero |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / perf_estimation / comm_time.py |
_cp_comm_cost_common |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / perf_estimation / comm_time.py |
cp_comm_layer_detailed |
|
26 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / perf_estimation / comm_time.py |
fill_dp_table |
|
13 |
2 |
0 |
|
85% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / perf_estimation / comm_time.py |
fill_tp_table |
|
9 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / perf_estimation / comm_time.py |
fill_ep_table |
|
7 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / perf_estimation / comm_time.py |
dp_ratio |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / perf_estimation / comm_time.py |
comm_embed_ouput |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / perf_estimation / comm_time.py |
estimate_op_bulk_comm |
|
44 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / perf_estimation / comm_time.py |
estimate_op_bulk_comm_layer |
|
22 |
22 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / perf_estimation / comm_time.py |
prepare_context |
|
9 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / perf_estimation / comm_time.py |
estimate_from_mem_comm |
|
59 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / perf_estimation / comm_time.py |
estimate_comm |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / perf_estimation / comm_time.py |
level_efficiency |
|
5 |
1 |
0 |
|
80% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / perf_estimation / comm_time.py |
level_bandwidth |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / perf_estimation / comm_time.py |
level_latency |
|
5 |
1 |
0 |
|
80% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / perf_estimation / comm_time.py |
comm_throughput |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / perf_estimation / comm_time.py |
estimate_comm_size_time |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / perf_estimation / comm_time.py |
estimate_comm_score |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / perf_estimation / comm_time.py |
(no function) |
|
37 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / perf_estimation / estimate.py |
op_table |
|
16 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / perf_estimation / estimate.py |
estimate_op_bulk_comp |
|
27 |
1 |
0 |
|
96% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / perf_estimation / estimate.py |
estimate_comp |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / perf_estimation / estimate.py |
efficiency |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / perf_estimation / estimate.py |
throughput |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / perf_estimation / estimate.py |
estimate_comp_flop_time |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / perf_estimation / estimate.py |
get_dynamic_ratio |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / perf_estimation / estimate.py |
estimate_stage |
|
29 |
4 |
0 |
|
86% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / perf_estimation / estimate.py |
estimate_pipeline |
|
42 |
6 |
0 |
|
86% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / perf_estimation / estimate.py |
estimate_p2p_comm |
|
8 |
1 |
0 |
|
88% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / perf_estimation / estimate.py |
estimate_perf |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / perf_estimation / estimate.py |
estimate_p2p |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / perf_estimation / estimate.py |
estimate_layer_perf |
|
36 |
36 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / perf_estimation / estimate.py |
apply_regression_coefficients |
|
20 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / perf_estimation / estimate.py |
_resolve_estimate_args |
|
9 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / perf_estimation / estimate.py |
_finalize_perf |
|
13 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / perf_estimation / estimate.py |
estimate_performance |
|
28 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / perf_estimation / estimate.py |
(no function) |
|
32 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / perf_estimation / getters.py |
get_layer_custom_configs |
|
8 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / perf_estimation / getters.py |
get_recomp_factor |
|
8 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / perf_estimation / getters.py |
get_table_quantity |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / perf_estimation / getters.py |
(no function) |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / perf_estimation / utils_classes.py |
CustomConfig.__init__ |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / perf_estimation / utils_classes.py |
CustomConfig.__repr__ |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / perf_estimation / utils_classes.py |
CustomConfig.__str__ |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_nd / perf_estimation / utils_classes.py |
(no function) |
|
24 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / __init__.py |
__getattr__ |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / __init__.py |
(no function) |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / pp_config_builder / __init__.py |
(no function) |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / pp_config_builder / layer_loader.py |
_get_pipeline_layer_class |
|
3 |
1 |
0 |
|
67% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / pp_config_builder / layer_loader.py |
_apply_recompute_considered |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / pp_config_builder / layer_loader.py |
_head_tail_recompute_considered |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / pp_config_builder / layer_loader.py |
LayerBuilder.__init__ |
|
20 |
1 |
0 |
|
95% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / pp_config_builder / layer_loader.py |
LayerBuilder.memory_limit |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / pp_config_builder / layer_loader.py |
LayerBuilder.constant_memory |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / pp_config_builder / layer_loader.py |
LayerBuilder.use_backward_time |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / pp_config_builder / layer_loader.py |
LayerBuilder._post_process_layers |
|
6 |
1 |
0 |
|
83% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / pp_config_builder / layer_loader.py |
LayerBuilder._validate_recompute_consistency |
|
17 |
1 |
0 |
|
94% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / pp_config_builder / layer_loader.py |
LayerBuilder._validate_group_names |
|
9 |
1 |
0 |
|
89% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / pp_config_builder / layer_loader.py |
LayerBuilder._validate_num_layer_consistency |
|
8 |
3 |
0 |
|
62% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / pp_config_builder / layer_loader.py |
(no function) |
|
33 |
5 |
0 |
|
85% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / pp_config_builder / yaml_parser.py |
_validate_int |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / pp_config_builder / yaml_parser.py |
_to_bool |
|
23 |
2 |
0 |
|
91% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / pp_config_builder / yaml_parser.py |
YamlOptimizationConfig._validate_field_types |
|
13 |
3 |
0 |
|
77% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / pp_config_builder / yaml_parser.py |
YamlOptimizationConfig.validate |
|
17 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / pp_config_builder / yaml_parser.py |
_extract_required_fields |
|
19 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / pp_config_builder / yaml_parser.py |
_extract_optional_fields |
|
9 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / pp_config_builder / yaml_parser.py |
parse_yaml_for_optimization |
|
12 |
1 |
0 |
|
92% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / pp_config_builder / yaml_parser.py |
(no function) |
|
25 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / pp_modeling / __init__.py |
__getattr__ |
|
6 |
4 |
0 |
|
33% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / pp_modeling / __init__.py |
(no function) |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / pp_modeling / pp_balancer.py |
PPBalancer.__init__ |
|
5 |
1 |
0 |
|
80% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / pp_modeling / pp_balancer.py |
PPBalancer.yaml_config |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / pp_modeling / pp_balancer.py |
PPBalancer.pipeline |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / pp_modeling / pp_balancer.py |
PPBalancer._make_infeasible_output |
|
6 |
1 |
0 |
|
83% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / pp_modeling / pp_balancer.py |
PPBalancer._build_feasible_output |
|
3 |
1 |
0 |
|
67% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / pp_modeling / pp_balancer.py |
PPBalancer._pulp_has_feasible_solution |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / pp_modeling / pp_balancer.py |
PPBalancer._check_ilp_solve_status |
|
20 |
3 |
0 |
|
85% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / pp_modeling / pp_balancer.py |
PPBalancer.balance_with_ilp |
|
12 |
2 |
0 |
|
83% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / pp_modeling / pp_balancer.py |
PPBalancer._build_strategy_result |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / pp_modeling / pp_balancer.py |
PPBalancer._build_ilp_result |
|
14 |
4 |
0 |
|
71% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / pp_modeling / pp_balancer.py |
PPBalancer._extract_stage_partition |
|
15 |
2 |
0 |
|
87% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / pp_modeling / pp_balancer.py |
PPBalancer._assign_layers_no_vpp |
|
15 |
2 |
0 |
|
87% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / pp_modeling / pp_balancer.py |
PPBalancer._assign_layers_with_vpp |
|
17 |
17 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / pp_modeling / pp_balancer.py |
PPBalancer._extract_chunk_stage_recompute |
|
16 |
16 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / pp_modeling / pp_balancer.py |
PPBalancer._get_body_group_names |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / pp_modeling / pp_balancer.py |
PPBalancer._total_body_layers |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / pp_modeling / pp_balancer.py |
PPBalancer._get_body_layer_by_name |
|
5 |
1 |
0 |
|
80% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / pp_modeling / pp_balancer.py |
PPBalancer._extract_group_stage_recompute |
|
17 |
2 |
0 |
|
88% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / pp_modeling / pp_balancer.py |
PPBalancer._extract_layer_offset_from_ilp |
|
17 |
2 |
0 |
|
88% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / pp_modeling / pp_balancer.py |
(no function) |
|
35 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / pp_modeling / pp_structs.py |
(no function) |
|
23 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / pp_optimizer.py |
PPOptimizer.optimize |
|
21 |
7 |
0 |
|
67% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / pp_optimizer.py |
PPOptimizer._run_simulation |
|
20 |
4 |
0 |
|
80% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / pp_optimizer.py |
(no function) |
|
13 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / pp_sim_adapter.py |
PPSimulator.__init__ |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / pp_sim_adapter.py |
PPSimulator.simulate_from_ilp |
|
20 |
13 |
0 |
|
35% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / pp_sim_adapter.py |
(no function) |
|
10 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / run_pipeline_balance.py |
_str2bool |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / run_pipeline_balance.py |
build_arg_parser |
|
21 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / run_pipeline_balance.py |
_resolve_path |
|
3 |
1 |
0 |
|
67% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / run_pipeline_balance.py |
run |
|
37 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / run_pipeline_balance.py |
main |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / run_pipeline_balance.py |
(no function) |
|
16 |
1 |
0 |
|
94% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / sapp / __init__.py |
(no function) |
|
0 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / sapp / sapp_pipeline.py |
SappPipeline.__init__ |
|
16 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / sapp / sapp_pipeline.py |
SappPipeline.simulator |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / sapp / sapp_pipeline.py |
SappPipeline.has_some_memory_info |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / sapp / sapp_pipeline.py |
SappPipeline.construct_problem |
|
7 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / sapp / sapp_pipeline.py |
SappPipeline.solve_problem |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / sapp / sapp_pipeline.py |
SappPipeline.get_result |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / sapp / sapp_pipeline.py |
SappPipeline.get_memory_activation |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / sapp / sapp_pipeline.py |
SappPipeline.get_memory_parameter |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / sapp / sapp_pipeline.py |
SappPipeline.get_fw_time |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / sapp / sapp_pipeline.py |
SappPipeline.get_recompute_time |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / sapp / sapp_pipeline.py |
SappPipeline.get_time |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / sapp / sapp_pipeline.py |
SappPipeline.naive_layer_per_stage |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / sapp / sapp_pipeline.py |
SappPipeline.print_yaml_results |
|
8 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / sapp / sapp_pipeline.py |
SappPipeline.get_manual_memory_activation |
|
10 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / sapp / sapp_pipeline.py |
SappPipeline._get_layer_memory_activation |
|
9 |
1 |
0 |
|
89% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / sapp / sapp_pipeline.py |
SappPipeline.get_manual_memory_parameter |
|
21 |
2 |
0 |
|
90% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / sapp / sapp_pipeline.py |
SappPipeline.get_manual_time |
|
14 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / sapp / sapp_pipeline.py |
SappPipeline.get_manual_fw_time |
|
14 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / sapp / sapp_pipeline.py |
SappPipeline.get_manual_backward_time |
|
14 |
14 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / sapp / sapp_pipeline.py |
SappPipeline.get_manual_recompute_time |
|
12 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / sapp / sapp_pipeline.py |
SappPipeline._add_manual_recompute_time |
|
11 |
1 |
0 |
|
91% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / sapp / sapp_pipeline.py |
SappPipeline.simulate |
|
9 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / sapp / sapp_pipeline.py |
SappPipeline.simulate_naive |
|
16 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / sapp / sapp_pipeline.py |
SappPipeline.simulate_comparison |
|
32 |
4 |
0 |
|
88% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / sapp / sapp_pipeline.py |
SappPipeline.simulate_only_manual |
|
22 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / sapp / sapp_pipeline.py |
SappPipeline.simulate_yaml |
|
7 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / sapp / sapp_pipeline.py |
SappPipeline._calculate_activation_memory |
|
10 |
10 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / sapp / sapp_pipeline.py |
SappPipeline._compute_parameter_memory_manually_solver |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / sapp / sapp_pipeline.py |
SappPipeline._calculate_layer_parameter_memory |
|
7 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / sapp / sapp_pipeline.py |
SappPipeline._calculate_activation_memory_solver |
|
8 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / sapp / sapp_pipeline.py |
SappPipeline.debug_print_manual_theoretical_memory |
|
30 |
5 |
0 |
|
83% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / sapp / sapp_pipeline.py |
SappPipeline.split_layer_per_recompute |
|
13 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / sapp / sapp_pipeline.py |
SappPipeline.fuse_layer_per_recompute |
|
7 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / sapp / sapp_pipeline.py |
SappPipeline.simulate_manual |
|
20 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / sapp / sapp_pipeline.py |
SappPipeline.simulation |
|
14 |
2 |
0 |
|
86% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / sapp / sapp_pipeline.py |
SappPipeline._construct_problem_pulp_ |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / sapp / sapp_pipeline.py |
SappPipeline._recompute_considered |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / sapp / sapp_pipeline.py |
choose_interleave |
|
15 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / sapp / sapp_pipeline.py |
flatten |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / sapp / sapp_pipeline.py |
(no function) |
|
53 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / sapp / sapp_solver.py |
SappSolver.__init__ |
|
18 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / sapp / sapp_solver.py |
SappSolver._initialize_seq_pipe_layers |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / sapp / sapp_solver.py |
SappSolver._update_seq_pipe_memory |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / sapp / sapp_solver.py |
SappSolver._update_body_seq_memory |
|
16 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / sapp / sapp_solver.py |
SappSolver._update_head_seq_memory |
|
5 |
1 |
0 |
|
80% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / sapp / sapp_solver.py |
SappSolver._update_tail_seq_memory |
|
5 |
1 |
0 |
|
80% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / sapp / sapp_solver.py |
SappSolver._update_seq_pipe_time |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / sapp / sapp_solver.py |
SappSolver._update_layer_seq_time |
|
11 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / sapp / sapp_solver.py |
SappSolver.compute_forward_in_backward |
|
9 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / sapp / sapp_solver.py |
SappSolver.compute_lm_forward_in_backward |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / sapp / sapp_solver.py |
SappSolver.compute_activation_nums |
|
18 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / sapp / sapp_solver.py |
SappSolver.compute_activation_nums_dual |
|
13 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / sapp / sapp_solver.py |
SappSolver.compute_less_activation_nums |
|
13 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / sapp / sapp_solver.py |
SappSolver._compute_activation_seq_interleave |
|
26 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / sapp / sapp_solver.py |
SappSolver.compute_activation_seq_nums |
|
10 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / sapp / sapp_solver.py |
SappSolver.compute_seq_mem_activation |
|
13 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / sapp / sapp_solver.py |
SappSolver.compute_seq_mem_parameter |
|
9 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / sapp / sapp_solver.py |
SappSolver.compute_seq_mem_head_cost |
|
16 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / sapp / sapp_solver.py |
SappSolver.compute_seq_mem_tail_cost |
|
8 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / sapp / sapp_solver.py |
SappSolver.add_total_nb_layer_constraint |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / sapp / sapp_solver.py |
SappSolver.add_stage_nb_layer_constraint |
|
15 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / sapp / sapp_solver.py |
SappSolver._reserved_stage_positions |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / sapp / sapp_solver.py |
SappSolver.add_multimodal_sequence_constraint |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / sapp / sapp_solver.py |
SappSolver._add_frontier_lower_bound |
|
5 |
1 |
0 |
|
80% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / sapp / sapp_solver.py |
SappSolver._frontier_layer_sum |
|
3 |
1 |
0 |
|
67% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / sapp / sapp_solver.py |
SappSolver._dual_frontier_layer_sum |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / sapp / sapp_solver.py |
SappSolver._current_layer_sum |
|
7 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / sapp / sapp_solver.py |
SappSolver._previous_layer_sum |
|
7 |
2 |
0 |
|
71% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / sapp / sapp_solver.py |
SappSolver._add_frontier_upper_bounds |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / sapp / sapp_solver.py |
SappSolver._add_frontier_upper_bound |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / sapp / sapp_solver.py |
SappSolver.add_multimodal_recompute_constraint |
|
21 |
1 |
0 |
|
95% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / sapp / sapp_solver.py |
SappSolver.find_recompute_considered |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / sapp / sapp_solver.py |
SappSolver.max_stage_micro_eq_stage |
|
17 |
4 |
0 |
|
76% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / sapp / sapp_solver.py |
SappSolver.add_performance_constraint |
|
15 |
8 |
0 |
|
47% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / sapp / sapp_solver.py |
SappSolver.add_max_stage_constraint |
|
7 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / sapp / sapp_solver.py |
SappSolver._accumulate_body_param |
|
7 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / sapp / sapp_solver.py |
SappSolver.stage_param_memory |
|
11 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / sapp / sapp_solver.py |
SappSolver.stage_active_memory_per_micro |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / sapp / sapp_solver.py |
SappSolver.stage_active_memory |
|
7 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / sapp / sapp_solver.py |
SappSolver.init_overhead_variables |
|
26 |
19 |
0 |
|
27% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / sapp / sapp_solver.py |
SappSolver.stage_overhead_memory |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / sapp / sapp_solver.py |
SappSolver.add_pipeline_memory_constraint |
|
23 |
3 |
0 |
|
87% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / sapp / sapp_solver.py |
SappSolver._stage_activation_memory |
|
8 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / sapp / sapp_solver.py |
SappSolver.get_simulator_memory_activation |
|
8 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / sapp / sapp_solver.py |
SappSolver.get_simulator_memory_parameter |
|
13 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / sapp / sapp_solver.py |
SappSolver._get_stage_parameter_memory |
|
9 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / sapp / sapp_solver.py |
SappSolver.get_simulator_time |
|
14 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / sapp / sapp_solver.py |
SappSolver.get_simulator_forward_time |
|
14 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / sapp / sapp_solver.py |
SappSolver.get_simulator_backward_time |
|
14 |
14 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / sapp / sapp_solver.py |
SappSolver.get_simulator_recompute_time |
|
14 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / sapp / sapp_solver.py |
SappSolver.has_some_memory_info |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / sapp / sapp_solver.py |
SappSolver.add_optional_recompute_constraint |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / sapp / sapp_solver.py |
SappSolver.dump_problem |
|
10 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / sapp / sapp_solver.py |
SappSolver._print_body_layer_assignments |
|
17 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / sapp / sapp_solver.py |
SappSolver._print_debug_variables |
|
14 |
2 |
0 |
|
86% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / sapp / sapp_solver.py |
SappSolver.print_results |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / sapp / sapp_solver.py |
SappSolver.debug_print_solver_theoretical_memory |
|
19 |
4 |
0 |
|
79% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / sapp / sapp_solver.py |
SappSolver.solve |
|
8 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / sapp / sapp_solver.py |
SappSolver.result |
|
13 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / sapp / sapp_solver.py |
SappSolver._create_problem_ |
|
46 |
15 |
0 |
|
67% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / sapp / sapp_solver.py |
SappSolver._create_variables_to_solve_ |
|
29 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / sapp / sapp_solver.py |
SappSolver._max_stage_bound_i_fp |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / sapp / sapp_solver.py |
SappSolver._max_stage_bound_i_bp |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / sapp / sapp_solver.py |
SappSolver._max_stage_bound_head_tail |
|
16 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / sapp / sapp_solver.py |
SappSolver._total_sum |
|
8 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / sapp / sapp_solver.py |
SappSolver.body_layer_time |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / sapp / sapp_solver.py |
SappSolver._append_head_tail_time |
|
11 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / sapp / sapp_solver.py |
SappSolver.micro_batch_time |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / sapp / sapp_solver.py |
SappSolver._chunks_sum |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / sapp / sapp_solver.py |
SappSolver._prev_diff_sum |
|
13 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / sapp / sapp_solver.py |
SappSolver._next_diff_sum |
|
11 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / sapp / sapp_solver.py |
(no function) |
|
116 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / simulator / __init__.py |
(no function) |
|
0 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / simulator / causal_error.py |
CausalError.__init__ |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / simulator / causal_error.py |
CausalError.__str__ |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / simulator / causal_error.py |
CausalCommError.__init__ |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / simulator / causal_error.py |
CausalCommError.__str__ |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / simulator / causal_error.py |
(no function) |
|
11 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / simulator / pipeline_builder.py |
PipelineBuilder._inter_merge |
|
25 |
1 |
0 |
|
96% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / simulator / pipeline_builder.py |
PipelineBuilder._build_chain |
|
9 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / simulator / pipeline_builder.py |
PipelineBuilder.build_1f1b |
|
8 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / simulator / pipeline_builder.py |
PipelineBuilder.build_virtualpipeline |
|
13 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / simulator / pipeline_builder.py |
PipelineBuilder.build_virtualpipeline2 |
|
13 |
13 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / simulator / pipeline_builder.py |
PipelineBuilder.get_builder |
|
7 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / simulator / pipeline_builder.py |
(no function) |
|
17 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / simulator / plot_manager.py |
PlotMgr.__init__ |
|
13 |
2 |
0 |
|
85% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / simulator / plot_manager.py |
PlotMgr._set_block_ax |
|
8 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / simulator / plot_manager.py |
PlotMgr._get_block_indices |
|
21 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / simulator / plot_manager.py |
PlotMgr.draw_block |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / simulator / plot_manager.py |
PlotMgr.draw_comm |
|
10 |
4 |
0 |
|
60% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / simulator / plot_manager.py |
PlotMgr.draw_connect |
|
11 |
3 |
0 |
|
73% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / simulator / plot_manager.py |
PlotMgr.draw |
|
18 |
3 |
0 |
|
83% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / simulator / plot_manager.py |
PlotMgr.draw_loop |
|
16 |
9 |
0 |
|
44% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / simulator / plot_manager.py |
PlotMgr.draw_comm_loop |
|
16 |
9 |
0 |
|
44% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / simulator / plot_manager.py |
PlotMgr.draw_mem |
|
9 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / simulator / plot_manager.py |
PlotMgr.draw_info |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / simulator / plot_manager.py |
PlotMgr.save |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / simulator / plot_manager.py |
PlotMgr.show |
|
4 |
1 |
0 |
|
75% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / simulator / plot_manager.py |
(no function) |
|
21 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / simulator / pp_simulator.py |
PipelineSimulator.__init__ |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / simulator / pp_simulator.py |
PipelineSimulator.init |
|
23 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / simulator / pp_simulator.py |
PipelineSimulator.run |
|
16 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / simulator / pp_simulator.py |
PipelineSimulator.draw |
|
9 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / simulator / pp_simulator.py |
PipelineSimulator.show |
|
5 |
1 |
0 |
|
80% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / simulator / pp_simulator.py |
PipelineSimulator.save |
|
5 |
1 |
0 |
|
80% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / simulator / pp_simulator.py |
PipelineSimulator.print_info |
|
9 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / simulator / pp_simulator.py |
PipelineSimulator._base_init |
|
12 |
3 |
0 |
|
75% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / simulator / pp_simulator.py |
PipelineSimulator._input_format |
|
14 |
1 |
0 |
|
93% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / simulator / pp_simulator.py |
PipelineSimulator._statistic_init |
|
24 |
5 |
0 |
|
79% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / simulator / pp_simulator.py |
PipelineSimulator._update_block_mem |
|
12 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / simulator / pp_simulator.py |
PipelineSimulator._statistic_info |
|
16 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / simulator / pp_simulator.py |
PipelineSimulator._get_pre_label |
|
19 |
1 |
0 |
|
95% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / simulator / pp_simulator.py |
PipelineSimulator._build_block |
|
10 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / simulator / pp_simulator.py |
PipelineSimulator._build_comm_block |
|
16 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / simulator / pp_simulator.py |
PipelineSimulator._check_loop |
|
6 |
1 |
0 |
|
83% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / simulator / pp_simulator.py |
PipelineSimulator._check_comm_loop |
|
6 |
1 |
0 |
|
83% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / simulator / pp_simulator.py |
PipelineSimulator._create_lines |
|
18 |
1 |
0 |
|
94% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / simulator / pp_simulator.py |
PipelineSimulator._get_block_phase |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / simulator / pp_simulator.py |
PipelineSimulator._send_block_delay |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / simulator / pp_simulator.py |
PipelineSimulator._process_swap_gap3 |
|
13 |
2 |
0 |
|
85% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / simulator / pp_simulator.py |
PipelineSimulator._process_swap_gap4 |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / simulator / pp_simulator.py |
PipelineSimulator._process_swap |
|
5 |
1 |
0 |
|
80% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / simulator / pp_simulator.py |
PipelineSimulator.swap_send_rec |
|
13 |
3 |
0 |
|
77% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / simulator / pp_simulator.py |
PipelineSimulator.vpp_send_delay |
|
7 |
7 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / simulator / pp_simulator.py |
PipelineSimulator.residue_delay |
|
16 |
16 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / simulator / pp_simulator.py |
(no function) |
|
40 |
1 |
0 |
|
98% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / simulator / sim_block.py |
BlockSim.label |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / simulator / sim_block.py |
BlockSim.color_label |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / simulator / sim_block.py |
BlockSim.build_without_comm |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / simulator / sim_block.py |
BlockSim.build_with_comm |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / simulator / sim_block.py |
BlockSim.reset_time |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / simulator / sim_block.py |
BlockSim.loc_size |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / simulator / sim_block.py |
BlockSim.loop |
|
20 |
2 |
0 |
|
90% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / simulator / sim_block.py |
BlockSim.comm_loop |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / simulator / sim_block.py |
HeadBlockSim.label |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / simulator / sim_block.py |
HeadBlockSim.repr |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / simulator / sim_block.py |
HeadBlockSim.draw |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / simulator / sim_block.py |
HeadBlockSim.build_without_comm |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / simulator / sim_block.py |
HeadBlockSim.build_with_comm |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / simulator / sim_block.py |
HeadBlockSim.reset_time_recursive |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / simulator / sim_block.py |
MicroBlockSim.__post_init__ |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / simulator / sim_block.py |
MicroBlockSim.draw |
|
13 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / simulator / sim_block.py |
MicroBlockSim.reset_time_recursive |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / simulator / sim_block.py |
CommBlockSim.get_triangle |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / simulator / sim_block.py |
CommBlockSim.draw |
|
8 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / simulator / sim_block.py |
SendBlockSim.loc_size |
|
8 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / simulator / sim_block.py |
SendBlockSim.get_triangle |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / simulator / sim_block.py |
SendBlockSim.draw_comm |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / simulator / sim_block.py |
SendBlockSim.build_with_comm |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / simulator / sim_block.py |
SendBlockSim.loop |
|
3 |
1 |
0 |
|
67% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / simulator / sim_block.py |
SendBlockSim.comm_loop |
|
20 |
6 |
0 |
|
70% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / simulator / sim_block.py |
RecBlockSim.loc_size |
|
8 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / simulator / sim_block.py |
RecBlockSim.get_triangle |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / simulator / sim_block.py |
RecBlockSim.build_with_comm |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / simulator / sim_block.py |
(no function) |
|
94 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / simulator / utils.py |
format_2d_inputs |
|
9 |
1 |
0 |
|
89% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / simulator / utils.py |
apply_color |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / simulator / utils.py |
apply_format |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / simulator / utils.py |
color_mix |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / simulator / utils.py |
dfs_builder |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / simulator / utils.py |
dfs_builder.decorator |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / simulator / utils.py |
dfs_builder.decorator.wrapper |
|
13 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / simulator / utils.py |
timer |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / simulator / utils.py |
timer.wrapper |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / simulator / utils.py |
(no function) |
|
13 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / utils / __init__.py |
(no function) |
|
0 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / utils / check_rules.py |
_get_yaml_ast_depth |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / utils / check_rules.py |
check_yaml_depth_before_loading |
|
9 |
4 |
0 |
|
56% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / utils / check_rules.py |
(no function) |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / utils / computation_analyzer.py |
ComputationAnalyzer.__init__ |
|
18 |
1 |
0 |
|
94% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / utils / computation_analyzer.py |
ComputationAnalyzer._get_layer_list |
|
8 |
2 |
0 |
|
75% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / utils / computation_analyzer.py |
ComputationAnalyzer._get_timeline_data |
|
12 |
4 |
0 |
|
67% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / utils / computation_analyzer.py |
ComputationAnalyzer._parse_step_duration |
|
18 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / utils / computation_analyzer.py |
ComputationAnalyzer._load_json_data |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / utils / computation_analyzer.py |
ComputationAnalyzer._initialize_step_duration |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / utils / computation_analyzer.py |
ComputationAnalyzer._add_layer_object |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / utils / computation_analyzer.py |
ComputationAnalyzer._is_counted |
|
10 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / utils / computation_analyzer.py |
ComputationAnalyzer._forward_parser |
|
42 |
2 |
0 |
|
95% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / utils / computation_analyzer.py |
ComputationAnalyzer._process_timelines |
|
15 |
2 |
0 |
|
87% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / utils / computation_analyzer.py |
ComputationAnalyzer._parse_layer_objects |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / utils / computation_analyzer.py |
ComputationAnalyzer.parse_auto_partition_layer_name_list |
|
12 |
1 |
0 |
|
92% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / utils / computation_analyzer.py |
ComputationAnalyzer.parse_layer_with_computation_time_list |
|
14 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / utils / computation_analyzer.py |
ComputationAnalyzer.transform_layer_with_cost_list |
|
23 |
1 |
0 |
|
96% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / utils / computation_analyzer.py |
(no function) |
|
31 |
5 |
0 |
|
84% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / utils / compute_memory.py |
ComputeMemory.__init__ |
|
9 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / utils / compute_memory.py |
ComputeMemory.set_stages_a |
|
10 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / utils / compute_memory.py |
ComputeMemory.set_stages_b |
|
15 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / utils / compute_memory.py |
ComputeMemory.find_recompute_considered |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / utils / compute_memory.py |
ComputeMemory._compute_memory_parameter_local_ |
|
12 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / utils / compute_memory.py |
ComputeMemory._compute_memory_parameter_ |
|
22 |
19 |
0 |
|
14% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / utils / compute_memory.py |
ComputeMemory._compute_memory_activation_ |
|
7 |
4 |
0 |
|
43% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / utils / compute_memory.py |
ComputeMemory.zero_offset |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / utils / compute_memory.py |
ComputeMemory._compute_memories_layers_ |
|
13 |
2 |
0 |
|
85% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / utils / compute_memory.py |
ComputeMemory._compute_memories_layer_bodies_local_ |
|
13 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / utils / compute_memory.py |
ComputeMemory._compute_memories_layer_bodies_local_with_fix_ |
|
22 |
1 |
0 |
|
95% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / utils / compute_memory.py |
ComputeMemory._compute_memories_layer_bodies_ |
|
19 |
2 |
0 |
|
89% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / utils / compute_memory.py |
ComputeMemory._average_if_needed_fix |
|
15 |
6 |
0 |
|
60% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / utils / compute_memory.py |
ComputeMemory._average_if_needed |
|
12 |
5 |
0 |
|
58% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / utils / compute_memory.py |
ComputeMemory._compute_memory_head_ |
|
12 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / utils / compute_memory.py |
ComputeMemory._compute_memory_tail_ |
|
11 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / utils / compute_memory.py |
ComputeMemory.get_memory_const |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / utils / compute_memory.py |
ComputeMemory.get_memory_parameter |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / utils / compute_memory.py |
ComputeMemory.get_memory_activation |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / utils / compute_memory.py |
ComputeMemory.get_memory_head |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / utils / compute_memory.py |
ComputeMemory.get_memory_tail |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / utils / compute_memory.py |
compute_memories |
|
14 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / utils / compute_memory.py |
(no function) |
|
37 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / utils / config.py |
LayersDescription.__init__ |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / utils / config.py |
ModelInfo.__init__ |
|
7 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / utils / config.py |
ModelInfo.get_layer_by_type |
|
4 |
1 |
0 |
|
75% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / utils / config.py |
ModelInfo.set_stage_const_mem |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / utils / config.py |
ModelInfo.layer_memory_update |
|
12 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / utils / config.py |
ModelInfo.dump_json |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / utils / config.py |
_as_list |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / utils / config.py |
_accumulate_layer_times |
|
8 |
8 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / utils / config.py |
time_parser |
|
37 |
16 |
0 |
|
57% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / utils / config.py |
process_offset |
|
16 |
1 |
0 |
|
94% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / utils / config.py |
process_rec_config |
|
12 |
2 |
0 |
|
83% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / utils / config.py |
instantiate_stage |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / utils / config.py |
memory_parser |
|
49 |
2 |
0 |
|
96% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / utils / config.py |
initialize_layer_json |
|
21 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / utils / config.py |
get_stage_const_mem |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / utils / config.py |
_generate_offset_config |
|
13 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / utils / config.py |
_generate_offset_array |
|
25 |
4 |
0 |
|
84% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / utils / config.py |
_get_coef_matrix |
|
10 |
1 |
0 |
|
90% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / utils / config.py |
print_dryrun_config |
|
16 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / utils / config.py |
generate_solvable_config |
|
23 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / utils / config.py |
parse_training_config |
|
14 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / utils / config.py |
(no function) |
|
50 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / utils / error.py |
assert_sapp |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / utils / error.py |
check_in_bounds |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / utils / error.py |
(no function) |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / utils / interactive.py |
default_v |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / utils / interactive.py |
is_yes |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / utils / interactive.py |
is_empty |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / utils / interactive.py |
global_arguments |
|
21 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / utils / interactive.py |
make_layer |
|
26 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / utils / interactive.py |
dryrun_guide |
|
27 |
3 |
0 |
|
89% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / utils / interactive.py |
main |
|
24 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / utils / interactive.py |
(no function) |
|
18 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / utils / layer.py |
Layer.__init__ |
|
18 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / utils / layer.py |
Layer.__str__ |
|
18 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / utils / layer.py |
Layer.dump |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / utils / layer.py |
Layer.to_json |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / utils / layer.py |
Layer.find_recompute_considered |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / utils / layer.py |
Layer.compute_internal_time |
|
9 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / utils / layer.py |
Layer.update_internal_time_for_seqpp |
|
8 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / utils / layer.py |
Layer.compute_timer |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / utils / layer.py |
Layer.compute_memory |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / utils / layer.py |
generate_layers_list |
|
11 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / utils / layer.py |
filter_layer_type |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / utils / layer.py |
aggregate |
|
11 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / utils / layer.py |
aggregate.add_none |
|
5 |
1 |
0 |
|
80% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / utils / layer.py |
aggregate.add_rec_dict |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / utils / layer.py |
(no function) |
|
33 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / utils / logger.py |
setup_logger |
|
12 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / utils / logger.py |
setup_logger.output |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / utils / logger.py |
(no function) |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / utils / recompute.py |
sums |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / utils / recompute.py |
zero_if_none_var |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / utils / recompute.py |
zero_if_none |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / utils / recompute.py |
yaml_from_internal |
|
23 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / utils / recompute.py |
internal_from_yaml |
|
25 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / utils / recompute.py |
to_list |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / utils / recompute.py |
right_extend |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / utils / recompute.py |
make_all_indexes_local |
|
7 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / utils / recompute.py |
make_all_indexes |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / utils / recompute.py |
recomputes_from_indexes |
|
9 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / utils / recompute.py |
average |
|
14 |
2 |
0 |
|
86% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / utils / recompute.py |
assign_used |
|
7 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / utils / recompute.py |
get_used_list |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / utils / recompute.py |
get_unused_list |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / utils / recompute.py |
least_recomputed |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / utils / recompute.py |
most_recomputed |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / utils / recompute.py |
(no function) |
|
27 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / utils / stage.py |
Stage.__init__ |
|
7 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / utils / stage.py |
Stage.complete_nb_layer_rec_ |
|
9 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / utils / stage.py |
Stage.same_config |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / utils / stage.py |
Stage.same_global_config |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / utils / stage.py |
Stage.get_index_memory_var |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / utils / stage.py |
filter_stage_id |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / auto_parallel / sapp_ppb / utils / stage.py |
(no function) |
|
14 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / collectives / __init__.py |
(no function) |
|
0 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / collectives / cc.py |
_group_key |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / collectives / cc.py |
init_process_group |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / collectives / cc.py |
destroy_process_group |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / collectives / cc.py |
get_process_group_ranks |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / collectives / cc.py |
get_backend |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / collectives / cc.py |
split_group |
|
14 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / collectives / cc.py |
get_group_local_rank |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / collectives / cc.py |
mark_created_groups |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / collectives / cc.py |
(no function) |
|
13 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / __init__.py |
(no function) |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / examples / __init__.py |
(no function) |
|
0 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / examples / automodel_tp / __init__.py |
(no function) |
|
0 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / examples / automodel_tp / mock_modules.py |
RotaryEmbedding.__init__ |
|
8 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / examples / automodel_tp / mock_modules.py |
RotaryEmbedding.forward |
|
13 |
6 |
0 |
|
54% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / examples / automodel_tp / mock_modules.py |
rotate_half |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / examples / automodel_tp / mock_modules.py |
apply_rotary_pos_emb |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / examples / automodel_tp / mock_modules.py |
RMSNorm.__init__ |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / examples / automodel_tp / mock_modules.py |
RMSNorm.forward |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / examples / automodel_tp / mock_modules.py |
GroupQueryAttention.__init__ |
|
11 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / examples / automodel_tp / mock_modules.py |
GroupQueryAttention._expand_kv_heads |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / examples / automodel_tp / mock_modules.py |
GroupQueryAttention.forward |
|
19 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / examples / automodel_tp / mock_modules.py |
SwiGLUMLP.__init__ |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / examples / automodel_tp / mock_modules.py |
SwiGLUMLP.forward |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / examples / automodel_tp / mock_modules.py |
(no function) |
|
19 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / examples / automodel_tp / model.py |
AutoModelAdapterDecoderLayer.__init__ |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / examples / automodel_tp / model.py |
AutoModelAdapterDecoderLayer.forward |
|
9 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / examples / automodel_tp / model.py |
AutoModelAdapterModel.__init__ |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / examples / automodel_tp / model.py |
AutoModelAdapterModel.forward |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / examples / automodel_tp / model.py |
AutoModelAdapterForCausalLM.__init__ |
|
6 |
1 |
0 |
|
83% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / examples / automodel_tp / model.py |
AutoModelAdapterForCausalLM.forward |
|
11 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / examples / automodel_tp / model.py |
shard_for_sp |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / examples / automodel_tp / model.py |
build_model |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / examples / automodel_tp / model.py |
DataSampler.__init__ |
|
8 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / examples / automodel_tp / model.py |
DataSampler.sample |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / examples / automodel_tp / model.py |
DataSampler.__iter__ |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / examples / automodel_tp / model.py |
(no function) |
|
20 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / examples / automodel_tp / train.py |
parse_args |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / examples / automodel_tp / train.py |
load_config |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / examples / automodel_tp / train.py |
setup_distributed |
|
12 |
12 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / examples / automodel_tp / train.py |
build_model |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / examples / automodel_tp / train.py |
build_mesh_context |
|
12 |
12 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / examples / automodel_tp / train.py |
train_fn |
|
13 |
13 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / examples / automodel_tp / train.py |
inspect_graph |
|
21 |
21 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / examples / automodel_tp / train.py |
main |
|
44 |
44 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / examples / automodel_tp / train.py |
(no function) |
|
22 |
22 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / parallel_config.py |
PassConfig.__post_init__ |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / parallel_config.py |
PassConfig.validate |
|
8 |
2 |
0 |
|
75% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / parallel_config.py |
(no function) |
|
16 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / passes / __init__.py |
(no function) |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / passes / base.py |
GraphPass.name |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / passes / base.py |
GraphPass.run |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / passes / base.py |
GraphPass.__repr__ |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / passes / base.py |
(no function) |
|
11 |
0 |
3 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / passes / overlap / __init__.py |
(no function) |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / passes / overlap / schedule.py |
AutoOverlapPass.run |
|
8 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / passes / overlap / schedule.py |
AutoOverlapPass._find_wait_nodes |
|
7 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / passes / overlap / schedule.py |
AutoOverlapPass._move_wait_later |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / passes / overlap / schedule.py |
(no function) |
|
10 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / passes / parallel / __init__.py |
(no function) |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / passes / parallel / fsdp_pass.py |
FSDPPass.__init__ |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / passes / parallel / fsdp_pass.py |
FSDPPass.run |
|
26 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / passes / parallel / fsdp_pass.py |
FSDPPass._identify_params_in_fsdp_modules |
|
23 |
2 |
0 |
|
91% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / passes / parallel / fsdp_pass.py |
FSDPPass._shard_live_model_params |
|
17 |
3 |
0 |
|
82% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / passes / parallel / fsdp_pass.py |
FSDPPass._param_belongs_to_fsdp_module |
|
8 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / passes / parallel / fsdp_pass.py |
FSDPPass._get_parent_module_fqn |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / passes / parallel / fsdp_pass.py |
FSDPPass._insert_all_gather_for_params |
|
22 |
1 |
0 |
|
95% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / passes / parallel / fsdp_pass.py |
FSDPPass._insert_reduce_scatter_for_grads |
|
29 |
4 |
0 |
|
86% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / passes / parallel / fsdp_pass.py |
FSDPPass._build_trainable_state_indices |
|
11 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / passes / parallel / fsdp_pass.py |
FSDPPass._insert_reshard_logic |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / passes / parallel / fsdp_pass.py |
(no function) |
|
23 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / passes / parallel / pp_pass.py |
_sanitize |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / passes / parallel / pp_pass.py |
_iter_node_args |
|
9 |
1 |
0 |
|
89% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / passes / parallel / pp_pass.py |
_tensor_arg_stages |
|
10 |
1 |
0 |
|
90% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / passes / parallel / pp_pass.py |
_innermost_fqns |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / passes / parallel / pp_pass.py |
_even_split |
|
8 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / passes / parallel / pp_pass.py |
_auto_stage_split |
|
22 |
1 |
0 |
|
95% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / passes / parallel / pp_pass.py |
PpPass.__init__ |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / passes / parallel / pp_pass.py |
PpPass.run |
|
7 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / passes / parallel / pp_pass.py |
PpPass._resolve_run_context |
|
14 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / passes / parallel / pp_pass.py |
PpPass._classify_stage |
|
10 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / passes / parallel / pp_pass.py |
PpPass._build_and_install_stage |
|
9 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / passes / parallel / pp_pass.py |
PpPass._build_stage_graphs |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / passes / parallel / pp_pass.py |
PpPass._require_state_list |
|
4 |
1 |
0 |
|
75% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / passes / parallel / pp_pass.py |
PpPass._require_is_param |
|
5 |
2 |
0 |
|
60% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / passes / parallel / pp_pass.py |
PpPass._loss_node |
|
9 |
2 |
0 |
|
78% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / passes / parallel / pp_pass.py |
PpPass._resolve_group_and_stage |
|
10 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / passes / parallel / pp_pass.py |
PpPass._resolve_stage_plan |
|
19 |
2 |
0 |
|
89% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / passes / parallel / pp_pass.py |
PpPass._warn_unassigned_modules |
|
14 |
1 |
0 |
|
93% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / passes / parallel / pp_pass.py |
PpPass._classify_graph |
|
12 |
1 |
0 |
|
92% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / passes / parallel / pp_pass.py |
PpPass._classify_placeholders |
|
21 |
1 |
0 |
|
95% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / passes / parallel / pp_pass.py |
PpPass._classify_forward_nodes |
|
27 |
2 |
0 |
|
93% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / passes / parallel / pp_pass.py |
PpPass._classify_backward_nodes |
|
14 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / passes / parallel / pp_pass.py |
PpPass._assign_deferred_state |
|
10 |
9 |
0 |
|
10% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / passes / parallel / pp_pass.py |
PpPass._state_stage |
|
6 |
1 |
0 |
|
83% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / passes / parallel / pp_pass.py |
PpPass._stage_from_stack |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / passes / parallel / pp_pass.py |
PpPass._stage_from_state_arg |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / passes / parallel / pp_pass.py |
PpPass._propagate_anchor_stages |
|
11 |
1 |
0 |
|
91% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / passes / parallel / pp_pass.py |
PpPass._validate_adjacent_dataflow |
|
13 |
1 |
0 |
|
92% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / passes / parallel / pp_pass.py |
PpPass._anchor_grad_outputs |
|
14 |
3 |
0 |
|
79% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / passes / parallel / pp_pass.py |
PpPass._find_boundaries |
|
29 |
3 |
0 |
|
90% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / passes / parallel / pp_pass.py |
PpPass._find_boundaries.crossing |
|
9 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / passes / parallel / pp_pass.py |
PpPass._unique |
|
7 |
2 |
0 |
|
71% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / passes / parallel / pp_pass.py |
PpPass._copy_node |
|
9 |
1 |
0 |
|
89% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / passes / parallel / pp_pass.py |
PpPass._copy_node.map_fn |
|
7 |
2 |
0 |
|
71% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / passes / parallel / pp_pass.py |
PpPass._build_fwd_graph |
|
19 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / passes / parallel / pp_pass.py |
PpPass._create_fwd_placeholders |
|
16 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / passes / parallel / pp_pass.py |
PpPass._resolve_fwd_outputs |
|
13 |
2 |
0 |
|
85% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / passes / parallel / pp_pass.py |
PpPass._build_bwd_graph |
|
22 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / passes / parallel / pp_pass.py |
PpPass._create_bwd_placeholders |
|
17 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / passes / parallel / pp_pass.py |
PpPass._stage_grad_nodes |
|
9 |
1 |
0 |
|
89% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / passes / parallel / pp_pass.py |
PpPass._prune_live_model |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / passes / parallel / pp_pass.py |
PpPass._prune_live_model.is_kept |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / passes / parallel / pp_pass.py |
PpPass._prune_live_model.prune_container_child |
|
13 |
2 |
0 |
|
85% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / passes / parallel / pp_pass.py |
PpPass._prune_live_model.prune_plain_child |
|
8 |
3 |
0 |
|
62% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / passes / parallel / pp_pass.py |
PpPass._prune_live_model.prune |
|
8 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / passes / parallel / pp_pass.py |
PpPass._trainable_state_indices |
|
9 |
1 |
0 |
|
89% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / passes / parallel / pp_pass.py |
PpPass._value_spec |
|
12 |
4 |
0 |
|
67% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / passes / parallel / pp_pass.py |
PpPass._build_schedule |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / passes / parallel / pp_pass.py |
PpPass._install_stub |
|
13 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / passes / parallel / pp_pass.py |
(no function) |
|
87 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / passes / parallel / pp_schedule.py |
ScheduleGPipe.__init__ |
|
16 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / passes / parallel / pp_schedule.py |
ScheduleGPipe.forward |
|
17 |
1 |
0 |
|
94% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / passes / parallel / pp_schedule.py |
ScheduleGPipe._forward_sweep |
|
14 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / passes / parallel / pp_schedule.py |
ScheduleGPipe._backward_sweep |
|
14 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / passes / parallel / pp_schedule.py |
ScheduleGPipe._send_values |
|
8 |
1 |
0 |
|
88% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / passes / parallel / pp_schedule.py |
ScheduleGPipe._recv_values |
|
12 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / passes / parallel / pp_schedule.py |
ScheduleGPipe._global_rank |
|
3 |
1 |
0 |
|
67% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / passes / parallel / pp_schedule.py |
(no function) |
|
13 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / passes / pipeline.py |
PassPipeline.__init__ |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / passes / pipeline.py |
PassPipeline.build |
|
9 |
1 |
0 |
|
89% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / passes / pipeline.py |
PassPipeline.run |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / passes / pipeline.py |
PassPipeline.from_config |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / passes / pipeline.py |
DeadCodeEliminationPass.run |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / passes / pipeline.py |
CanonicalizeGraphPass.run |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / passes / pipeline.py |
(no function) |
|
19 |
0 |
3 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / sharding_config.py |
PassPlan.merge |
|
8 |
2 |
0 |
|
75% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / sharding_config.py |
PassPlan.fsdp_wrap |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / sharding_config.py |
PassPlan.fsdp_wrap_pattern |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / sharding_config.py |
PassPlan.is_fsdp_module |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / sharding_config.py |
PassPlan.get_fsdp_config |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / sharding_config.py |
PassPlan.pp_stage |
|
11 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / sharding_config.py |
create_sharding_plan_from_yaml |
|
24 |
2 |
0 |
|
92% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / sharding_config.py |
_yaml_section |
|
4 |
1 |
0 |
|
75% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / sharding_config.py |
_maybe_process_fsdp |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / sharding_config.py |
_process_fsdp |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / sharding_config.py |
_process_pp |
|
7 |
2 |
0 |
|
71% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / sharding_config.py |
create_simple_sharding_plan |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / sharding_config.py |
(no function) |
|
27 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / tracer / __init__.py |
(no function) |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / tracer / graph_tracer.py |
_skip_nested_compile |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / tracer / graph_tracer.py |
_patch_engine_backward |
|
6 |
2 |
0 |
|
67% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / tracer / graph_tracer.py |
_non_strict_tracing_context |
|
5 |
2 |
0 |
|
60% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / tracer / graph_tracer.py |
extract_module_state |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / tracer / graph_tracer.py |
_reparametrize_train_state |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / tracer / graph_tracer.py |
_iter_node_args |
|
7 |
1 |
0 |
|
86% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / tracer / graph_tracer.py |
_output_args |
|
12 |
2 |
0 |
|
83% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / tracer / graph_tracer.py |
_ancestors_of |
|
10 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / tracer / graph_tracer.py |
_has_module_stack |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / tracer / graph_tracer.py |
_annotate_autograd_backward |
|
26 |
1 |
0 |
|
96% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / tracer / graph_tracer.py |
_copy_fwd_metadata_to_bw_nodes |
|
8 |
1 |
0 |
|
88% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / tracer / graph_tracer.py |
_iter_seq_nr_nodes |
|
7 |
1 |
0 |
|
86% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / tracer / graph_tracer.py |
_copy_fwd_meta |
|
9 |
2 |
0 |
|
78% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / tracer / graph_tracer.py |
_fakeify_input |
|
3 |
1 |
0 |
|
67% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / tracer / graph_tracer.py |
_input_meta |
|
9 |
7 |
0 |
|
22% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / tracer / graph_tracer.py |
trace_model_graph |
|
33 |
3 |
0 |
|
91% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / tracer / graph_tracer.py |
trace_model_graph._fwd_bwd_fn |
|
19 |
1 |
0 |
|
95% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / tracer / graph_tracer.py |
run_traced_graph |
|
11 |
1 |
0 |
|
91% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / tracer / graph_tracer.py |
(no function) |
|
54 |
2 |
0 |
|
96% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / trainer.py |
GraphTrainer.__init__ |
|
11 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / trainer.py |
GraphTrainer.compile |
|
10 |
1 |
0 |
|
90% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / trainer.py |
GraphTrainer._init_device_mesh |
|
15 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / trainer.py |
GraphTrainer._build_pass_kwargs |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / trainer.py |
GraphTrainer.train_step |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / trainer.py |
GraphTrainer.optimizer_step |
|
6 |
1 |
0 |
|
83% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / trainer.py |
GraphTrainer.to |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / trainer.py |
GraphTrainer.set_pytree_pre_hook |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / trainer.py |
GraphTrainer._place_on_device |
|
4 |
1 |
0 |
|
75% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / trainer.py |
GraphTrainer.train |
|
18 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / trainer.py |
GraphTrainer._init_optimizer |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / trainer.py |
GraphTrainer._run_graph |
|
3 |
1 |
0 |
|
67% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / trainer.py |
GraphTrainer._accumulate_grads |
|
9 |
1 |
0 |
|
89% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / trainer.py |
GraphTrainer._trainable_params_in_state_order |
|
8 |
1 |
0 |
|
88% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / compile / trainer.py |
(no function) |
|
26 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / __init__.py |
(no function) |
|
0 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / checkpoint / __init__.py |
(no function) |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / checkpoint / base.py |
CheckpointerBase.save |
|
0 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / checkpoint / base.py |
CheckpointerBase.load |
|
0 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / checkpoint / base.py |
CheckpointerBase.maybe_wait_for_async_save |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / checkpoint / base.py |
CheckpointerBase.find_latest_checkpoint |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / checkpoint / base.py |
(no function) |
|
10 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / checkpoint / config.py |
(no function) |
|
20 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / checkpoint / conversion_ops.py |
_single_tensor |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / checkpoint / conversion_ops.py |
ConcatenateWithSections.__init__ |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / checkpoint / conversion_ops.py |
ConcatenateWithSections.convert |
|
8 |
8 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / checkpoint / conversion_ops.py |
ConcatenateWithSections.reverse_op |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / checkpoint / conversion_ops.py |
Split.__init__ |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / checkpoint / conversion_ops.py |
Split.convert |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / checkpoint / conversion_ops.py |
Split.reverse_op |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / checkpoint / conversion_ops.py |
AddScalar.__init__ |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / checkpoint / conversion_ops.py |
AddScalar.convert |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / checkpoint / conversion_ops.py |
AddScalar.reverse_op |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / checkpoint / conversion_ops.py |
InterleaveQKV.__init__ |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / checkpoint / conversion_ops.py |
InterleaveQKV._source_sections |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / checkpoint / conversion_ops.py |
InterleaveQKV.convert |
|
17 |
17 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / checkpoint / conversion_ops.py |
InterleaveQKV.reverse_op |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / checkpoint / conversion_ops.py |
DeinterleaveQKV.__init__ |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / checkpoint / conversion_ops.py |
DeinterleaveQKV._group_sections |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / checkpoint / conversion_ops.py |
DeinterleaveQKV.convert |
|
16 |
16 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / checkpoint / conversion_ops.py |
DeinterleaveQKV.reverse_op |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / checkpoint / conversion_ops.py |
InterleaveGateQKV.__init__ |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / checkpoint / conversion_ops.py |
InterleaveGateQKV.convert |
|
12 |
12 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / checkpoint / conversion_ops.py |
InterleaveGateQKV.reverse_op |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / checkpoint / conversion_ops.py |
DeinterleaveGateQKV.__init__ |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / checkpoint / conversion_ops.py |
DeinterleaveGateQKV._group_sections |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / checkpoint / conversion_ops.py |
DeinterleaveGateQKV.convert |
|
11 |
11 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / checkpoint / conversion_ops.py |
DeinterleaveGateQKV.reverse_op |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / checkpoint / conversion_ops.py |
(no function) |
|
52 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / checkpoint / dcp_checkpointer.py |
_empty_cache |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / checkpoint / dcp_checkpointer.py |
_get_rank |
|
3 |
1 |
0 |
|
67% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / checkpoint / dcp_checkpointer.py |
_barrier |
|
2 |
1 |
0 |
|
50% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / checkpoint / dcp_checkpointer.py |
_prepare_main_param_state_for_dcp_load |
|
22 |
3 |
0 |
|
86% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / checkpoint / dcp_checkpointer.py |
_ModelStrictLoadPlanner.__init__ |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / checkpoint / dcp_checkpointer.py |
_ModelStrictLoadPlanner.configure_planner |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / checkpoint / dcp_checkpointer.py |
_ModelStrictLoadPlanner.build_local_plan |
|
14 |
6 |
0 |
|
57% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / checkpoint / dcp_checkpointer.py |
_optimizer_parameters |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / checkpoint / dcp_checkpointer.py |
_prime_optimizer_state |
|
9 |
9 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / checkpoint / dcp_checkpointer.py |
_materialize_hyper_optimizer_state |
|
18 |
3 |
0 |
|
83% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / checkpoint / dcp_checkpointer.py |
_restore_optimizer_hyperparams |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / checkpoint / dcp_checkpointer.py |
_leaf_optimizers |
|
4 |
1 |
0 |
|
75% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / checkpoint / dcp_checkpointer.py |
initialize_optimizer_state |
|
17 |
8 |
0 |
|
53% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / checkpoint / dcp_checkpointer.py |
DistributedCheckpointer.__init__ |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / checkpoint / dcp_checkpointer.py |
DistributedCheckpointer._extra_state_path |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / checkpoint / dcp_checkpointer.py |
DistributedCheckpointer._is_complete |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / checkpoint / dcp_checkpointer.py |
DistributedCheckpointer._latest_pointer_path |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / checkpoint / dcp_checkpointer.py |
DistributedCheckpointer._write_latest_pointer |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / checkpoint / dcp_checkpointer.py |
DistributedCheckpointer._read_latest_pointer |
|
14 |
14 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / checkpoint / dcp_checkpointer.py |
DistributedCheckpointer._list_checkpoints |
|
15 |
15 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / checkpoint / dcp_checkpointer.py |
DistributedCheckpointer.find_latest_checkpoint |
|
7 |
7 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / checkpoint / dcp_checkpointer.py |
DistributedCheckpointer.maybe_wait_for_async_save |
|
14 |
12 |
0 |
|
14% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / checkpoint / dcp_checkpointer.py |
DistributedCheckpointer._finalize_checkpoint |
|
7 |
7 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / checkpoint / dcp_checkpointer.py |
DistributedCheckpointer.save |
|
26 |
26 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / checkpoint / dcp_checkpointer.py |
DistributedCheckpointer.load |
|
16 |
4 |
0 |
|
75% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / checkpoint / dcp_checkpointer.py |
DistributedCheckpointer._read_extra_state |
|
8 |
8 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / checkpoint / dcp_checkpointer.py |
(no function) |
|
55 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / checkpoint / registry.py |
Registry.__init__ |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / checkpoint / registry.py |
Registry.__getitem__ |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / checkpoint / registry.py |
Registry.__setitem__ |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / checkpoint / registry.py |
Registry.__delitem__ |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / checkpoint / registry.py |
Registry.__iter__ |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / checkpoint / registry.py |
Registry.__len__ |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / checkpoint / registry.py |
Registry.register |
|
5 |
2 |
0 |
|
60% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / checkpoint / registry.py |
Registry.register.decorator |
|
4 |
1 |
0 |
|
75% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / checkpoint / registry.py |
Registry.valid_keys |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / checkpoint / registry.py |
build_checkpointer |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / checkpoint / registry.py |
_dcp_checkpointer |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / checkpoint / registry.py |
(no function) |
|
17 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / checkpoint / weight_conversion.py |
_has_scoped_weight_transforms |
|
3 |
1 |
0 |
|
67% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / checkpoint / weight_conversion.py |
(no function) |
|
244 |
156 |
0 |
|
36% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / functional / __init__.py |
__getattr__ |
|
7 |
4 |
0 |
|
43% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / functional / __init__.py |
(no function) |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / functional / aggregate_hidden.py |
_AggregateHidden.forward |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / functional / aggregate_hidden.py |
_AggregateHidden.backward |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / functional / aggregate_hidden.py |
aggregate_hidden |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / functional / aggregate_hidden.py |
(no function) |
|
9 |
9 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / functional / attention_rescale.py |
_rescale_attention_outputs |
|
14 |
14 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / functional / attention_rescale.py |
_AttentionRescale.forward |
|
7 |
7 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / functional / attention_rescale.py |
_AttentionRescale.backward |
|
9 |
9 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / functional / attention_rescale.py |
attention_rescale |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / functional / attention_rescale.py |
(no function) |
|
11 |
11 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / functional / aux_loss.py |
_AuxLossAutoScaler.forward |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / functional / aux_loss.py |
_AuxLossAutoScaler.backward |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / functional / aux_loss.py |
_AuxLossAutoScaler.set_loss_scale |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / functional / aux_loss.py |
aux_loss_auto_scale |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / functional / aux_loss.py |
set_aux_loss_scale |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / functional / aux_loss.py |
(no function) |
|
12 |
12 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / functional / dsa_indexer.py |
dsa_indexer |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / functional / dsa_indexer.py |
(no function) |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / functional / dsa_kl_loss.py |
_DSAKLLoss.forward |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / functional / dsa_kl_loss.py |
_DSAKLLoss.backward |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / functional / dsa_kl_loss.py |
dsa_kl_loss |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / functional / dsa_kl_loss.py |
(no function) |
|
9 |
9 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / functional / dsa_sparse_attention.py |
dsa_sparse_attention |
|
9 |
9 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / functional / dsa_sparse_attention.py |
(no function) |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / functional / dsa_sparse_attention_rescale.py |
_SparseAttentionRescale.forward |
|
26 |
26 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / functional / dsa_sparse_attention_rescale.py |
_SparseAttentionRescale.backward |
|
20 |
20 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / functional / dsa_sparse_attention_rescale.py |
dsa_sparse_attention_rescale |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / functional / dsa_sparse_attention_rescale.py |
(no function) |
|
12 |
12 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / functional / grouped_matmul.py |
GMMFunction.forward |
|
12 |
12 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / functional / grouped_matmul.py |
GMMFunction.backward |
|
32 |
32 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / functional / grouped_matmul.py |
grouped_matmul |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / functional / grouped_matmul.py |
(no function) |
|
10 |
10 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / functional / mhc_post.py |
_MhcPost.forward |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / functional / mhc_post.py |
_MhcPost.backward |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / functional / mhc_post.py |
mhc_post |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / functional / mhc_post.py |
(no function) |
|
12 |
12 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / functional / mhc_pre.py |
_MhcPre.forward |
|
7 |
7 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / functional / mhc_pre.py |
_MhcPre.backward |
|
8 |
8 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / functional / mhc_pre.py |
mhc_pre |
|
8 |
8 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / functional / mhc_pre.py |
(no function) |
|
13 |
13 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / functional / moe_token_permute.py |
moe_token_permute |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / functional / moe_token_permute.py |
(no function) |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / functional / moe_token_unpermute.py |
moe_token_unpermute |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / functional / moe_token_unpermute.py |
(no function) |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / functional / npu_fusion_attention.py |
_npu_attention_mask |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / functional / npu_fusion_attention.py |
_causal_attention_mask |
|
9 |
9 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / functional / npu_fusion_attention.py |
_length_list |
|
10 |
10 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / functional / npu_fusion_attention.py |
_coalesce_lengths |
|
8 |
8 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / functional / npu_fusion_attention.py |
_packed_parameter_kwargs |
|
20 |
20 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / functional / npu_fusion_attention.py |
resolve_packed_sequence_lengths |
|
11 |
11 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / functional / npu_fusion_attention.py |
_attention_options |
|
8 |
8 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / functional / npu_fusion_attention.py |
_prepare_attention_inputs |
|
13 |
13 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / functional / npu_fusion_attention.py |
npu_fusion_attention_forward |
|
15 |
15 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / functional / npu_fusion_attention.py |
(no function) |
|
14 |
14 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / functional / npu_grouped_swiglu.py |
_get_npu_grouped_matmul |
|
11 |
11 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / functional / npu_grouped_swiglu.py |
_get_npu_grouped_matmul._NPUGroupedMatmul.forward |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / functional / npu_grouped_swiglu.py |
_get_npu_grouped_matmul._NPUGroupedMatmul.backward |
|
7 |
7 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / functional / npu_grouped_swiglu.py |
npu_grouped_swiglu |
|
13 |
13 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / functional / npu_grouped_swiglu.py |
(no function) |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / functional / rms_norm.py |
rms_norm |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / functional / rms_norm.py |
(no function) |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / functional / rotary_embedding.py |
apply_rotary_pos_emb |
|
14 |
14 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / functional / rotary_embedding.py |
apply_rotary_pos_emb_interleave |
|
33 |
33 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / functional / rotary_embedding.py |
(no function) |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / functional / sink_attention.py |
sink_attention |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / functional / sink_attention.py |
(no function) |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / functional / sinkhorn.py |
_Sinkhorn.forward |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / functional / sinkhorn.py |
_Sinkhorn.backward |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / functional / sinkhorn.py |
sinkhorn |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / functional / sinkhorn.py |
sinkhorn_knopps |
|
9 |
9 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / functional / sinkhorn.py |
(no function) |
|
13 |
13 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / functional / swiglu.py |
swiglu |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / functional / swiglu.py |
(no function) |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / losses / __init__.py |
(no function) |
|
10 |
2 |
0 |
|
80% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / losses / _vocab_parallel_cross_entropy.py |
_validate_target_type_base |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / losses / _vocab_parallel_cross_entropy.py |
_resolve_class_mesh_dim |
|
11 |
11 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / losses / _vocab_parallel_cross_entropy.py |
_is_floating_torch |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / losses / _vocab_parallel_cross_entropy.py |
_compute_vocab_start |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / losses / _vocab_parallel_cross_entropy.py |
distributed_log_softmax |
|
8 |
8 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / losses / _vocab_parallel_cross_entropy.py |
distributed_nll_loss_forward |
|
32 |
32 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / losses / _vocab_parallel_cross_entropy.py |
DistributedCrossEntropyFunction.forward |
|
46 |
46 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / losses / _vocab_parallel_cross_entropy.py |
DistributedCrossEntropyFunction.backward |
|
45 |
45 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / losses / _vocab_parallel_cross_entropy.py |
vocab_parallel_cross_entropy_local |
|
9 |
9 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / losses / _vocab_parallel_cross_entropy.py |
(no function) |
|
19 |
0 |
2 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / losses / chunked_cross_entropy.py |
_linear_cross_entropy_chunk |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / losses / chunked_cross_entropy.py |
_PrecomputedChunkLoss.forward |
|
15 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / losses / chunked_cross_entropy.py |
_PrecomputedChunkLoss.backward |
|
4 |
1 |
0 |
|
75% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / losses / chunked_cross_entropy.py |
_validate_chunk_options |
|
6 |
2 |
0 |
|
67% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / losses / chunked_cross_entropy.py |
_validate_chunk_shapes |
|
10 |
4 |
0 |
|
60% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / losses / chunked_cross_entropy.py |
_validate_chunk_dtypes_and_device |
|
8 |
3 |
0 |
|
62% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / losses / chunked_cross_entropy.py |
chunked_cross_entropy |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / losses / chunked_cross_entropy.py |
_get_output_value |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / losses / chunked_cross_entropy.py |
ChunkedCausalLMLoss.__init__ |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / losses / chunked_cross_entropy.py |
ChunkedCausalLMLoss._parallel_size |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / losses / chunked_cross_entropy.py |
ChunkedCausalLMLoss.bind_model |
|
20 |
20 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / losses / chunked_cross_entropy.py |
ChunkedCausalLMLoss.prepare_model_inputs |
|
15 |
15 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / losses / chunked_cross_entropy.py |
ChunkedCausalLMLoss.forward |
|
18 |
18 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / losses / chunked_cross_entropy.py |
(no function) |
|
40 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / losses / dispatcher.py |
_get_lm_head_weight |
|
9 |
9 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / losses / dispatcher.py |
calculate_loss |
|
21 |
21 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / losses / dispatcher.py |
(no function) |
|
10 |
2 |
0 |
|
80% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / losses / linear_ce.py |
FusedLinearCrossEntropy.__init__ |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / losses / linear_ce.py |
FusedLinearCrossEntropy.forward |
|
10 |
10 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / losses / linear_ce.py |
(no function) |
|
7 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / losses / masked_ce.py |
MaskedCrossEntropy.__call__ |
|
19 |
19 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / losses / masked_ce.py |
(no function) |
|
11 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / losses / model_output.py |
ModelOutputLoss.forward |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / losses / model_output.py |
(no function) |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / losses / mtp.py |
calculate_mtp_loss |
|
7 |
7 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / losses / mtp.py |
(no function) |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / modules / __init__.py |
__getattr__ |
|
7 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / modules / __init__.py |
(no function) |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / modules / dsa_attention.py |
apply_mome |
|
15 |
15 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / modules / dsa_attention.py |
_apply_attention_rope |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / modules / dsa_attention.py |
_apply_index_rope |
|
9 |
9 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / modules / dsa_attention.py |
_absorb_query_projection |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / modules / dsa_attention.py |
_restore_attention_projection |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / modules / dsa_attention.py |
DeepseekV32DSAAttention._validate_source |
|
8 |
8 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / modules / dsa_attention.py |
DeepseekV32DSAAttention._initialize_dimensions |
|
16 |
16 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / modules / dsa_attention.py |
DeepseekV32DSAAttention._initialize_options |
|
20 |
20 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / modules / dsa_attention.py |
DeepseekV32DSAAttention._initialize_source_layers |
|
13 |
13 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / modules / dsa_attention.py |
DeepseekV32DSAAttention.__init__ |
|
11 |
11 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / modules / dsa_attention.py |
DeepseekV32DSAAttention.make_transforms |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / modules / dsa_attention.py |
DeepseekV32DSAAttention._get_actual_seq_len |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / modules / dsa_attention.py |
DeepseekV32DSAAttention._validate_forward_inputs |
|
11 |
11 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / modules / dsa_attention.py |
DeepseekV32DSAAttention._project_attention_states |
|
11 |
11 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / modules / dsa_attention.py |
DeepseekV32DSAAttention._project_index_states |
|
7 |
7 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / modules / dsa_attention.py |
DeepseekV32DSAAttention._apply_auxiliary_loss |
|
7 |
7 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / modules / dsa_attention.py |
DeepseekV32DSAAttention.forward |
|
17 |
17 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / modules / dsa_attention.py |
DSAAttention.__init__ |
|
55 |
55 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / modules / dsa_attention.py |
DSAAttention._get_actual_seq_len |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / modules / dsa_attention.py |
DSAAttention._prepare_param_sink |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / modules / dsa_attention.py |
DSAAttention._validate_forward_inputs |
|
8 |
8 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / modules / dsa_attention.py |
DSAAttention._project_attention_states |
|
15 |
15 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / modules / dsa_attention.py |
DSAAttention._project_index_states |
|
9 |
9 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / modules / dsa_attention.py |
DSAAttention._run_sparse_attention |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / modules / dsa_attention.py |
DSAAttention._apply_auxiliary_loss |
|
7 |
7 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / modules / dsa_attention.py |
DSAAttention.forward |
|
20 |
20 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / modules / dsa_attention.py |
(no function) |
|
48 |
48 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / modules / gated_delta_net.py |
_l2norm |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / modules / gated_delta_net.py |
torch_chunk_gated_delta_rule |
|
47 |
3 |
0 |
|
94% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / modules / gated_delta_net.py |
_parse_version |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / modules / gated_delta_net.py |
_is_triton_gdn_input_supported |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / modules / gated_delta_net.py |
is_triton_gdn_available |
|
16 |
16 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / modules / gated_delta_net.py |
chunk_gated_delta_rule |
|
9 |
9 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / modules / gated_delta_net.py |
(no function) |
|
18 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / modules / gqa_attention.py |
rotate_half |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / modules / gqa_attention.py |
_projections_can_fuse |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / modules / gqa_attention.py |
GQAAttention._initialize_dimensions |
|
10 |
10 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / modules / gqa_attention.py |
GQAAttention._initialize_attention_options |
|
14 |
14 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / modules / gqa_attention.py |
GQAAttention._source_qkv |
|
21 |
21 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / modules / gqa_attention.py |
GQAAttention._initialize_projections |
|
12 |
12 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / modules / gqa_attention.py |
GQAAttention.__init__ |
|
12 |
12 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / modules / gqa_attention.py |
GQAAttention.make_transforms |
|
9 |
9 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / modules / gqa_attention.py |
GQAAttention.forward |
|
30 |
30 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / modules / gqa_attention.py |
GatedGQAAttention._initialize_dimensions |
|
18 |
18 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / modules / gqa_attention.py |
GatedGQAAttention._validate_source_projections |
|
15 |
15 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / modules / gqa_attention.py |
GatedGQAAttention._initialize_projection |
|
7 |
7 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / modules / gqa_attention.py |
GatedGQAAttention.__init__ |
|
14 |
14 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / modules / gqa_attention.py |
GatedGQAAttention.make_transforms |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / modules / gqa_attention.py |
GatedGQAAttention.forward |
|
32 |
32 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / modules / gqa_attention.py |
(no function) |
|
28 |
28 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / modules / grouped_experts.py |
GroupedExperts._source_parameters |
|
9 |
9 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / modules / grouped_experts.py |
GroupedExperts._source_dimensions |
|
15 |
15 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / modules / grouped_experts.py |
GroupedExperts._infer_ungated_hidden_size |
|
11 |
11 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / modules / grouped_experts.py |
GroupedExperts._activation |
|
20 |
20 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / modules / grouped_experts.py |
GroupedExperts._activation.glu |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / modules / grouped_experts.py |
GroupedExperts._source_layout |
|
15 |
15 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / modules / grouped_experts.py |
GroupedExperts._initialize_weights |
|
8 |
8 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / modules / grouped_experts.py |
GroupedExperts._initialize_biases |
|
16 |
16 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / modules / grouped_experts.py |
GroupedExperts.__init__ |
|
44 |
44 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / modules / grouped_experts.py |
GroupedExperts.reset_parameters |
|
8 |
8 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / modules / grouped_experts.py |
GroupedExperts.make_transforms |
|
9 |
9 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / modules / grouped_experts.py |
GroupedExperts._grouped_gemm_expert_forward |
|
24 |
24 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / modules / grouped_experts.py |
GroupedExperts.forward_expert_major |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / modules / grouped_experts.py |
GroupedExperts.forward |
|
9 |
9 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / modules / grouped_experts.py |
(no function) |
|
30 |
30 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / modules / kimi_delta_attention.py |
_is_triton_kda_input_supported |
|
14 |
14 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / modules / kimi_delta_attention.py |
is_triton_kda_available |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / modules / kimi_delta_attention.py |
_validate_kda_inputs |
|
17 |
6 |
0 |
|
65% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / modules / kimi_delta_attention.py |
_reshape_gate_bias |
|
3 |
1 |
0 |
|
67% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / modules / kimi_delta_attention.py |
torch_kda_gate |
|
13 |
4 |
0 |
|
69% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / modules / kimi_delta_attention.py |
_prepare_kda_inputs |
|
18 |
1 |
0 |
|
94% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / modules / kimi_delta_attention.py |
_initial_kda_state |
|
6 |
1 |
0 |
|
83% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / modules / kimi_delta_attention.py |
torch_recurrent_kda |
|
17 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / modules / kimi_delta_attention.py |
_inverse_unit_lower_triangular |
|
11 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / modules / kimi_delta_attention.py |
torch_chunk_kda |
|
47 |
1 |
0 |
|
98% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / modules / kimi_delta_attention.py |
torch_chunk_kda.chunk_view |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / modules / kimi_delta_attention.py |
chunk_kda |
|
16 |
12 |
0 |
|
25% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / modules / kimi_delta_attention.py |
_chunk_kda_summary_inputs |
|
10 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / modules / kimi_delta_attention.py |
_chunk_kda_summary_inputs.chunk_view |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / modules / kimi_delta_attention.py |
torch_kda_state_summary |
|
31 |
1 |
0 |
|
97% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / modules / kimi_delta_attention.py |
torch_apply_kda_state_summary |
|
3 |
1 |
0 |
|
67% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / modules / kimi_delta_attention.py |
torch_compose_kda_state_summaries |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / modules / kimi_delta_attention.py |
KimiRMSNormGated.__init__ |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / modules / kimi_delta_attention.py |
KimiRMSNormGated.forward |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / modules / kimi_delta_attention.py |
_make_short_convolution |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / modules / kimi_delta_attention.py |
KimiDeltaAttention.__init__ |
|
51 |
7 |
0 |
|
86% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / modules / kimi_delta_attention.py |
KimiDeltaAttention._run_short_convolution |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / modules / kimi_delta_attention.py |
KimiDeltaAttention._output_gate |
|
3 |
1 |
0 |
|
67% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / modules / kimi_delta_attention.py |
KimiDeltaAttention.forward |
|
18 |
4 |
0 |
|
78% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / modules / kimi_delta_attention.py |
(no function) |
|
34 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / modules / mhc.py |
_required_attribute |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / modules / mhc.py |
_required_config_value |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / modules / mhc.py |
_num_stream |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / modules / mhc.py |
_parameter |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / modules / mhc.py |
_validate_parameter_layout |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / modules / mhc.py |
MhcPreModule.__init__ |
|
38 |
38 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / modules / mhc.py |
MhcPreModule.forward |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / modules / mhc.py |
MhcPostModule.__init__ |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / modules / mhc.py |
MhcPostModule.forward |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / modules / mhc.py |
(no function) |
|
21 |
21 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / modules / mla_attention.py |
MLAAttention.__init__ |
|
52 |
52 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / modules / mla_attention.py |
MLAAttention.make_transforms |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / modules / mla_attention.py |
MLAAttention.forward |
|
26 |
26 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / modules / mla_attention.py |
(no function) |
|
15 |
15 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / modules / rms_norm.py |
RMSNorm.__init__ |
|
18 |
18 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / modules / rms_norm.py |
RMSNorm.forward |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / modules / rms_norm.py |
OffsetRMSNorm.__init__ |
|
8 |
8 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / modules / rms_norm.py |
OffsetRMSNorm.reset_parameters |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / modules / rms_norm.py |
OffsetRMSNorm.make_transforms |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / modules / rms_norm.py |
OffsetRMSNorm.forward |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / modules / rms_norm.py |
(no function) |
|
18 |
18 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / modules / shared_expert.py |
LinearWithMatmul.__init__ |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / modules / shared_expert.py |
LinearWithMatmul.forward |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / modules / shared_expert.py |
SharedExpert._separate_projections |
|
19 |
19 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / modules / shared_expert.py |
SharedExpert._initialize_fused_source |
|
11 |
11 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / modules / shared_expert.py |
SharedExpert._initialize_separate_linears |
|
10 |
10 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / modules / shared_expert.py |
SharedExpert._initialize_activation |
|
15 |
15 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / modules / shared_expert.py |
SharedExpert._initialize_activation.glu |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / modules / shared_expert.py |
SharedExpert.__init__ |
|
16 |
16 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / modules / shared_expert.py |
SharedExpert.make_transforms |
|
8 |
8 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / modules / shared_expert.py |
SharedExpert.forward |
|
10 |
10 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / modules / shared_expert.py |
(no function) |
|
24 |
24 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / modules / swiglu_mlp.py |
SwiGLUMLP._source_projections |
|
8 |
8 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / modules / swiglu_mlp.py |
SwiGLUMLP._validate_projection_contract |
|
18 |
18 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / modules / swiglu_mlp.py |
SwiGLUMLP._initialize_fc1 |
|
7 |
7 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / modules / swiglu_mlp.py |
SwiGLUMLP._initialize_fc2 |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / modules / swiglu_mlp.py |
SwiGLUMLP.__init__ |
|
14 |
14 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / modules / swiglu_mlp.py |
SwiGLUMLP.make_transforms |
|
7 |
7 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / modules / swiglu_mlp.py |
SwiGLUMLP.forward |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / modules / swiglu_mlp.py |
(no function) |
|
21 |
21 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / optim / __init__.py |
(no function) |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / optim / builders.py |
AdamW.__init__ |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / optim / builders.py |
AdamW.get_adamw_param_groups |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / optim / builders.py |
AdamW.get_optimizer |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / optim / builders.py |
Muon.__init__ |
|
11 |
1 |
0 |
|
91% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / optim / builders.py |
Muon.split_muon_adamw_params |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / optim / builders.py |
Muon.get_optimizer |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / optim / builders.py |
(no function) |
|
18 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / optim / lr_scheduler.py |
MultiLRScheduler.__init__ |
|
20 |
20 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / optim / lr_scheduler.py |
MultiLRScheduler.get_lr_scheduler |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / optim / lr_scheduler.py |
(no function) |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / optim / mixed_precision_optimizer.py |
_copy_tensor |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / optim / mixed_precision_optimizer.py |
_gradient_for_param |
|
5 |
1 |
0 |
|
80% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / optim / mixed_precision_optimizer.py |
MixedPrecisionOptimizer.__init__ |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / optim / mixed_precision_optimizer.py |
MixedPrecisionOptimizer.prepare_grads |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / optim / mixed_precision_optimizer.py |
MixedPrecisionOptimizer.step_with_ready_grads |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / optim / mixed_precision_optimizer.py |
MixedPrecisionOptimizer.reload_model_params |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / optim / mixed_precision_optimizer.py |
Float16OptimizerWithFloat16Params.__init__ |
|
11 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / optim / mixed_precision_optimizer.py |
Float16OptimizerWithFloat16Params._build_model_and_main_param_groups |
|
26 |
2 |
0 |
|
92% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / optim / mixed_precision_optimizer.py |
Float16OptimizerWithFloat16Params._copy_model_grads_to_main_grads |
|
12 |
1 |
0 |
|
92% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / optim / mixed_precision_optimizer.py |
Float16OptimizerWithFloat16Params._copy_main_params_to_model_params |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / optim / mixed_precision_optimizer.py |
Float16OptimizerWithFloat16Params._copy_model_params_to_main_params |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / optim / mixed_precision_optimizer.py |
Float16OptimizerWithFloat16Params.prepare_grads |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / optim / mixed_precision_optimizer.py |
Float16OptimizerWithFloat16Params.step_with_ready_grads |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / optim / mixed_precision_optimizer.py |
Float16OptimizerWithFloat16Params.step |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / optim / mixed_precision_optimizer.py |
Float16OptimizerWithFloat16Params.reload_model_params |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / optim / mixed_precision_optimizer.py |
Float16OptimizerWithFloat16Params.zero_grad |
|
12 |
2 |
0 |
|
83% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / optim / mixed_precision_optimizer.py |
Float16OptimizerWithFloat16Params.state_dict |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / optim / mixed_precision_optimizer.py |
Float16OptimizerWithFloat16Params.load_state_dict |
|
21 |
1 |
0 |
|
95% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / optim / mixed_precision_optimizer.py |
(no function) |
|
32 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / optim / parameter_groups.py |
get_parameter_names |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / optim / parameter_groups.py |
get_adamw_param_groups |
|
25 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / optim / parameter_groups.py |
split_muon_adamw_params |
|
15 |
3 |
0 |
|
80% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / optim / parameter_groups.py |
(no function) |
|
9 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / quantization / __init__.py |
(no function) |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / quantization / config.py |
LowPrecisionConfig.__post_init__ |
|
5 |
2 |
0 |
|
60% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / quantization / config.py |
(no function) |
|
8 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / quantization / functional / __init__.py |
(no function) |
|
7 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / quantization / functional / hifloat8_gmm_func.py |
_HiFloat8GroupedLinearFunction._validate_forward_inputs |
|
20 |
20 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / quantization / functional / hifloat8_gmm_func.py |
_HiFloat8GroupedLinearFunction._save_forward_context |
|
10 |
10 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / quantization / functional / hifloat8_gmm_func.py |
_HiFloat8GroupedLinearFunction.forward |
|
15 |
15 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / quantization / functional / hifloat8_gmm_func.py |
_HiFloat8GroupedLinearFunction.backward |
|
19 |
19 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / quantization / functional / hifloat8_gmm_func.py |
hifloat8_grouped_linear |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / quantization / functional / hifloat8_gmm_func.py |
(no function) |
|
15 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / quantization / functional / hifloat8_linear_func.py |
_as_matrix |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / quantization / functional / hifloat8_linear_func.py |
_HiFloat8LinearFunction.forward |
|
16 |
16 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / quantization / functional / hifloat8_linear_func.py |
_HiFloat8LinearFunction.backward |
|
16 |
16 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / quantization / functional / hifloat8_linear_func.py |
hifloat8_linear |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / quantization / functional / hifloat8_linear_func.py |
(no function) |
|
12 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / quantization / functional / mxfp8_gmm_func.py |
_MXFP8GroupedLinearFunction.forward |
|
35 |
35 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / quantization / functional / mxfp8_gmm_func.py |
_MXFP8GroupedLinearFunction.backward |
|
18 |
18 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / quantization / functional / mxfp8_gmm_func.py |
npu_quant_grouped_linear |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / quantization / functional / mxfp8_gmm_func.py |
(no function) |
|
10 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / quantization / functional / mxfp8_linear_func.py |
_as_matrix |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / quantization / functional / mxfp8_linear_func.py |
_MXFP8LinearFunction.forward |
|
16 |
16 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / quantization / functional / mxfp8_linear_func.py |
_MXFP8LinearFunction.backward |
|
17 |
17 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / quantization / functional / mxfp8_linear_func.py |
mxfp8_linear |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / quantization / functional / mxfp8_linear_func.py |
(no function) |
|
12 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / quantization / functional / npu_hifloat8.py |
HiFloat8NpuOps.__init__ |
|
12 |
12 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / quantization / functional / npu_hifloat8.py |
HiFloat8NpuOps.quant_dtype |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / quantization / functional / npu_hifloat8.py |
HiFloat8NpuOps.dynamic_quant |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / quantization / functional / npu_hifloat8.py |
HiFloat8NpuOps.quant_matmul |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / quantization / functional / npu_hifloat8.py |
HiFloat8NpuOps.validate_grouped_matmul |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / quantization / functional / npu_hifloat8.py |
HiFloat8NpuOps.quant_grouped_matmul |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / quantization / functional / npu_hifloat8.py |
_get_hifloat8_npu_ops |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / quantization / functional / npu_hifloat8.py |
validate_hifloat8_runtime |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / quantization / functional / npu_hifloat8.py |
validate_hifloat8_gmm_runtime |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / quantization / functional / npu_hifloat8.py |
_left_operand |
|
7 |
7 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / quantization / functional / npu_hifloat8.py |
_right_operand |
|
7 |
7 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / quantization / functional / npu_hifloat8.py |
hifloat8_matmul |
|
8 |
8 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / quantization / functional / npu_hifloat8.py |
hifloat8_grouped_matmul |
|
17 |
17 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / quantization / functional / npu_hifloat8.py |
(no function) |
|
23 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / quantization / functional / npu_mxfp8.py |
MXFP8NpuOps.__init__ |
|
10 |
10 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / quantization / functional / npu_mxfp8.py |
MXFP8NpuOps.validate_grouped_matmul |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / quantization / functional / npu_mxfp8.py |
MXFP8NpuOps.dynamic_mx_quant |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / quantization / functional / npu_mxfp8.py |
MXFP8NpuOps.dynamic_mx_quant_dual_axis |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / quantization / functional / npu_mxfp8.py |
MXFP8NpuOps.quant_matmul |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / quantization / functional / npu_mxfp8.py |
MXFP8NpuOps.grouped_dynamic_mx_quant |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / quantization / functional / npu_mxfp8.py |
MXFP8NpuOps.quant_grouped_matmul |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / quantization / functional / npu_mxfp8.py |
MXFP8NpuOps.is_e8m0_dtype |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / quantization / functional / npu_mxfp8.py |
_get_npu_ops |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / quantization / functional / npu_mxfp8.py |
validate_npu_runtime |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / quantization / functional / npu_mxfp8.py |
validate_npu_gmm_runtime |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / quantization / functional / npu_mxfp8.py |
_transpose_scale |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / quantization / functional / npu_mxfp8.py |
_left_operand |
|
7 |
7 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / quantization / functional / npu_mxfp8.py |
_right_operand |
|
7 |
7 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / quantization / functional / npu_mxfp8.py |
mxfp8_matmul |
|
7 |
7 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / quantization / functional / npu_mxfp8.py |
mxfp8_grouped_matmul |
|
14 |
14 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / quantization / functional / npu_mxfp8.py |
(no function) |
|
25 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / quantization / modules / __init__.py |
(no function) |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / quantization / modules / hifloat8_grouped_linear.py |
HiFloat8GroupedExperts.__init__ |
|
9 |
9 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / quantization / modules / hifloat8_grouped_linear.py |
HiFloat8GroupedExperts._initialize_quantizers |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / quantization / modules / hifloat8_grouped_linear.py |
HiFloat8GroupedExperts.from_module |
|
29 |
29 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / quantization / modules / hifloat8_grouped_linear.py |
HiFloat8GroupedExperts._grouped_forward |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / quantization / modules / hifloat8_grouped_linear.py |
HiFloat8GroupedExperts.forward |
|
19 |
19 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / quantization / modules / hifloat8_grouped_linear.py |
(no function) |
|
14 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / quantization / modules / hifloat8_linear.py |
HiFloat8Linear._initialize_format |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / quantization / modules / hifloat8_linear.py |
HiFloat8Linear._apply_low_precision |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / quantization / modules / hifloat8_linear.py |
replace_hifloat8_linear |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / quantization / modules / hifloat8_linear.py |
(no function) |
|
15 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / quantization / modules / linear.py |
QuantizedLinearBase.__init__ |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / quantization / modules / linear.py |
QuantizedLinearBase.from_linear |
|
9 |
9 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / quantization / modules / linear.py |
QuantizedLinearBase._initialize_format |
|
0 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / quantization / modules / linear.py |
QuantizedLinearBase._apply_low_precision |
|
0 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / quantization / modules / linear.py |
QuantizedLinearBase.forward |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / quantization / modules / linear.py |
(no function) |
|
16 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / quantization / modules / mxfp8_grouped_linear.py |
MXFP8GroupedExperts.__init__ |
|
9 |
9 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / quantization / modules / mxfp8_grouped_linear.py |
MXFP8GroupedExperts.from_module |
|
29 |
29 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / quantization / modules / mxfp8_grouped_linear.py |
MXFP8GroupedExperts._grouped_forward |
|
8 |
8 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / quantization / modules / mxfp8_grouped_linear.py |
MXFP8GroupedExperts.forward |
|
19 |
19 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / quantization / modules / mxfp8_grouped_linear.py |
(no function) |
|
13 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / quantization / modules / mxfp8_linear.py |
MXFP8Linear._initialize_format |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / quantization / modules / mxfp8_linear.py |
MXFP8Linear._apply_low_precision |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / quantization / modules / mxfp8_linear.py |
replace_mxfp8_linear |
|
8 |
8 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / quantization / modules / mxfp8_linear.py |
(no function) |
|
15 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / quantization / quantizers / __init__.py |
(no function) |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / quantization / quantizers / base.py |
Quantizer.quantize |
|
0 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / quantization / quantizers / base.py |
(no function) |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / quantization / quantizers / hifloat8.py |
HiFloat8Quantizer.__init__ |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / quantization / quantizers / hifloat8.py |
HiFloat8Quantizer.npu_ops |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / quantization / quantizers / hifloat8.py |
HiFloat8Quantizer.quant_dtype |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / quantization / quantizers / hifloat8.py |
HiFloat8Quantizer._validate_input |
|
17 |
17 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / quantization / quantizers / hifloat8.py |
HiFloat8Quantizer.quantize |
|
10 |
10 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / quantization / quantizers / hifloat8.py |
(no function) |
|
19 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / quantization / quantizers / mxfp8.py |
MXFP8Quantizer.__init__ |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / quantization / quantizers / mxfp8.py |
MXFP8Quantizer.npu_ops |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / quantization / quantizers / mxfp8.py |
MXFP8Quantizer.quantize |
|
28 |
28 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / quantization / quantizers / mxfp8.py |
(no function) |
|
10 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / quantization / tensor / __init__.py |
(no function) |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / quantization / tensor / base.py |
QuantizedTensorStorage.update_usage |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / quantization / tensor / base.py |
QuantizedTensorStorage.is_rowwise |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / quantization / tensor / base.py |
QuantizedTensorStorage.is_colwise |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / quantization / tensor / base.py |
QuantizedTensorStorage.get_metadata |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / quantization / tensor / base.py |
(no function) |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / quantization / tensor / hifloat8_tensor.py |
HiFloat8TensorStorage.__init__ |
|
9 |
9 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / quantization / tensor / hifloat8_tensor.py |
HiFloat8TensorStorage.update_usage |
|
7 |
7 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / quantization / tensor / hifloat8_tensor.py |
HiFloat8TensorStorage.is_rowwise |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / quantization / tensor / hifloat8_tensor.py |
HiFloat8TensorStorage.is_colwise |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / quantization / tensor / hifloat8_tensor.py |
HiFloat8TensorStorage.get_metadata |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / quantization / tensor / hifloat8_tensor.py |
HiFloat8Tensor.__new__ |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / quantization / tensor / hifloat8_tensor.py |
HiFloat8Tensor.__init__ |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / quantization / tensor / hifloat8_tensor.py |
HiFloat8Tensor._new_with_transformed_data |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / quantization / tensor / hifloat8_tensor.py |
HiFloat8Tensor._new_with_transformed_data._transform_optional |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / quantization / tensor / hifloat8_tensor.py |
HiFloat8Tensor.__repr__ |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / quantization / tensor / hifloat8_tensor.py |
(no function) |
|
23 |
0 |
2 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / quantization / tensor / mxfp8_tensor.py |
MXFP8TensorStorage.__init__ |
|
12 |
12 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / quantization / tensor / mxfp8_tensor.py |
MXFP8TensorStorage._validate_pair |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / quantization / tensor / mxfp8_tensor.py |
MXFP8TensorStorage.update_usage |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / quantization / tensor / mxfp8_tensor.py |
MXFP8TensorStorage.is_rowwise |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / quantization / tensor / mxfp8_tensor.py |
MXFP8TensorStorage.is_colwise |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / quantization / tensor / mxfp8_tensor.py |
MXFP8TensorStorage.get_metadata |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / quantization / tensor / mxfp8_tensor.py |
MXFP8Tensor.__new__ |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / quantization / tensor / mxfp8_tensor.py |
MXFP8Tensor.__init__ |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / quantization / tensor / mxfp8_tensor.py |
MXFP8Tensor._new_with_transformed_data |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / quantization / tensor / mxfp8_tensor.py |
MXFP8Tensor._new_with_transformed_data._transform_optional |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / quantization / tensor / mxfp8_tensor.py |
MXFP8Tensor.__repr__ |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / quantization / tensor / mxfp8_tensor.py |
(no function) |
|
17 |
0 |
2 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / quantization / tensor / quantized_tensor.py |
_contiguous_strides |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / quantization / tensor / quantized_tensor.py |
QuantizedTensor.__new__ |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / quantization / tensor / quantized_tensor.py |
QuantizedTensor._new_with_transformed_data |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / quantization / tensor / quantized_tensor.py |
QuantizedTensor.__torch_dispatch__ |
|
16 |
16 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / components / quantization / tensor / quantized_tensor.py |
(no function) |
|
9 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / __init__.py |
(no function) |
|
0 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / activation_checkpoint / __init__.py |
(no function) |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / activation_checkpoint / activation_checkpoint.py |
_StackedCtx.__init__ |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / activation_checkpoint / activation_checkpoint.py |
_StackedCtx.__enter__ |
|
8 |
3 |
0 |
|
62% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / activation_checkpoint / activation_checkpoint.py |
_StackedCtx.__exit__ |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / activation_checkpoint / activation_checkpoint.py |
_compose_context_fns |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / activation_checkpoint / activation_checkpoint.py |
_compose_context_fns.factory |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / activation_checkpoint / activation_checkpoint.py |
checkpoint |
|
41 |
16 |
0 |
|
61% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / activation_checkpoint / activation_checkpoint.py |
swap |
|
8 |
2 |
0 |
|
75% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / activation_checkpoint / activation_checkpoint.py |
checkpoint_exclude_wrapper |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / activation_checkpoint / activation_checkpoint.py |
(no function) |
|
24 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / activation_checkpoint / pinned_memory_pool.py |
_valid_positive_int |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / activation_checkpoint / pinned_memory_pool.py |
_storage_key |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / activation_checkpoint / pinned_memory_pool.py |
PinnedMemoryPool.__init__ |
|
9 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / activation_checkpoint / pinned_memory_pool.py |
PinnedMemoryPool.total_allocated |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / activation_checkpoint / pinned_memory_pool.py |
PinnedMemoryPool.max_host_bytes |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / activation_checkpoint / pinned_memory_pool.py |
PinnedMemoryPool.align_limit |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / activation_checkpoint / pinned_memory_pool.py |
PinnedMemoryPool._aligned_size |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / activation_checkpoint / pinned_memory_pool.py |
PinnedMemoryPool._reclaim_completed_locked |
|
21 |
1 |
0 |
|
95% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / activation_checkpoint / pinned_memory_pool.py |
PinnedMemoryPool._register_view_locked |
|
10 |
1 |
0 |
|
90% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / activation_checkpoint / pinned_memory_pool.py |
PinnedMemoryPool._checkout_locked |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / activation_checkpoint / pinned_memory_pool.py |
PinnedMemoryPool._find_available_locked |
|
8 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / activation_checkpoint / pinned_memory_pool.py |
PinnedMemoryPool._find_pending_locked |
|
8 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / activation_checkpoint / pinned_memory_pool.py |
PinnedMemoryPool.acquire |
|
53 |
19 |
0 |
|
64% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / activation_checkpoint / pinned_memory_pool.py |
PinnedMemoryPool.release |
|
13 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / activation_checkpoint / pinned_memory_pool.py |
(no function) |
|
35 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / activation_checkpoint / recompute_state.py |
_RecomputeInvocation.__init__ |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / activation_checkpoint / recompute_state.py |
_RecomputeInvocation.get_resource |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / activation_checkpoint / recompute_state.py |
_RecomputeInvocation.clear |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / activation_checkpoint / recompute_state.py |
RecomputeState.__init__ |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / activation_checkpoint / recompute_state.py |
RecomputeState.invocation_id |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / activation_checkpoint / recompute_state.py |
RecomputeState.get_resource |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / activation_checkpoint / recompute_state.py |
RecomputeState._clear_resources |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / activation_checkpoint / recompute_state.py |
_RecomputeContext.__init__ |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / activation_checkpoint / recompute_state.py |
_RecomputeContext.__enter__ |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / activation_checkpoint / recompute_state.py |
_RecomputeContext.__exit__ |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / activation_checkpoint / recompute_state.py |
get_recompute_state |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / activation_checkpoint / recompute_state.py |
is_recomputing |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / activation_checkpoint / recompute_state.py |
create_recompute_contexts |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / activation_checkpoint / recompute_state.py |
(no function) |
|
20 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / activation_checkpoint / swap.py |
_get_cpu_pinned_buf |
|
11 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / activation_checkpoint / swap.py |
_return_cpu_pinned_buf |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / activation_checkpoint / swap.py |
_collect_device_storage_ptrs |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / activation_checkpoint / swap.py |
_collect_device_storage_ptrs._collect |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / activation_checkpoint / swap.py |
SwapTensor.__init__ |
|
20 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / activation_checkpoint / swap.py |
SwapTensor.dedup_key |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / activation_checkpoint / swap.py |
SwapTensor.mark_duplicate_swap |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / activation_checkpoint / swap.py |
SwapTensor.protect_if_aliases |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / activation_checkpoint / swap.py |
SwapTensor.get_val |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / activation_checkpoint / swap.py |
SwapTensor.resize_device_storage |
|
8 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / activation_checkpoint / swap.py |
SwapTensor.async_load |
|
14 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / activation_checkpoint / swap.py |
SwapTensor.async_group_load |
|
12 |
5 |
0 |
|
58% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / activation_checkpoint / swap.py |
SwapTensor.release_cpu_buffer |
|
6 |
4 |
0 |
|
33% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / activation_checkpoint / swap.py |
SwapTensor.wait_load |
|
11 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / activation_checkpoint / swap.py |
SwapTensor.async_offload |
|
33 |
15 |
0 |
|
55% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / activation_checkpoint / swap.py |
SwapTensor.wait_offload |
|
11 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / activation_checkpoint / swap.py |
SwapTensor.state |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / activation_checkpoint / swap.py |
SwapTensor.__repr__ |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / activation_checkpoint / swap.py |
Storage.__init__ |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / activation_checkpoint / swap.py |
Storage.__getitem__ |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / activation_checkpoint / swap.py |
Storage.values |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / activation_checkpoint / swap.py |
Storage.clear |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / activation_checkpoint / swap.py |
Storage.iter_swap_tensors |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / activation_checkpoint / swap.py |
Storage.iter_swap_tensors._collect |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / activation_checkpoint / swap.py |
Storage.mark_duplicate_swaps |
|
11 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / activation_checkpoint / swap.py |
Storage.protect_alias_storage_ptrs |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / activation_checkpoint / swap.py |
Storage.protect_alias_storage_ptrs._protect_tensor |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / activation_checkpoint / swap.py |
Storage.launch_load |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / activation_checkpoint / swap.py |
Storage.launch_load._async_load |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / activation_checkpoint / swap.py |
Storage.resize_device_storage |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / activation_checkpoint / swap.py |
Storage.resize_device_storage._resize |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / activation_checkpoint / swap.py |
Storage.wait_load |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / activation_checkpoint / swap.py |
Storage.wait_load._wait_load |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / activation_checkpoint / swap.py |
Storage.release_cpu_buffers |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / activation_checkpoint / swap.py |
Storage.release_cpu_buffers._release |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / activation_checkpoint / swap.py |
Storage.wait_offload |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / activation_checkpoint / swap.py |
Storage.wait_offload._wait_offload |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / activation_checkpoint / swap.py |
Storage.launch_offload |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / activation_checkpoint / swap.py |
Storage.launch_offload._async_offload |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / activation_checkpoint / swap.py |
SwapGroup.__init__ |
|
11 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / activation_checkpoint / swap.py |
SwapGroup.add |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / activation_checkpoint / swap.py |
SwapGroup.protect_alias_tensors |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / activation_checkpoint / swap.py |
SwapGroup._collect_packable_tensors |
|
29 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / activation_checkpoint / swap.py |
SwapGroup._collect_packable_tensors._try_pack |
|
18 |
1 |
0 |
|
94% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / activation_checkpoint / swap.py |
SwapGroup.launch_offload |
|
43 |
15 |
0 |
|
65% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / activation_checkpoint / swap.py |
SwapGroup.wait_offload |
|
10 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / activation_checkpoint / swap.py |
SwapGroup.launch_load |
|
28 |
15 |
0 |
|
46% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / activation_checkpoint / swap.py |
SwapGroup.release_cpu_buffers |
|
8 |
6 |
0 |
|
25% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / activation_checkpoint / swap.py |
SwapGroup.wait_load |
|
17 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / activation_checkpoint / swap.py |
SwapManager.__init__ |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / activation_checkpoint / swap.py |
SwapManager.__new__ |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / activation_checkpoint / swap.py |
SwapManager.add_storage |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / activation_checkpoint / swap.py |
SwapManager.ensure_group |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / activation_checkpoint / swap.py |
SwapManager.launch_offload |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / activation_checkpoint / swap.py |
SwapManager.protect_alias_tensors |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / activation_checkpoint / swap.py |
SwapManager.wait_offload |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / activation_checkpoint / swap.py |
SwapManager.launch_load |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / activation_checkpoint / swap.py |
SwapManager.wait_load |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / activation_checkpoint / swap.py |
SwapManager.release_group_storage |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / activation_checkpoint / swap.py |
SwapManager.abort_group |
|
10 |
1 |
0 |
|
90% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / activation_checkpoint / swap.py |
SwapManager.get_current_group_name |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / activation_checkpoint / swap.py |
SwapManager.set_current_group_name |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / activation_checkpoint / swap.py |
SwapManager.active_group_count |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / activation_checkpoint / swap.py |
SwapManager.group_context |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / activation_checkpoint / swap.py |
SwapManager.is_last_group |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / activation_checkpoint / swap.py |
SwapManager.unregister_forward_prefetch_hooks |
|
10 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / activation_checkpoint / swap.py |
SwapManager.set_forward_prefetch_layer |
|
22 |
1 |
0 |
|
95% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / activation_checkpoint / swap.py |
SwapManager.set_forward_prefetch_layer._ensure_group_name |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / activation_checkpoint / swap.py |
SwapManager.set_forward_prefetch_layer._forward_pre_hook |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / activation_checkpoint / swap.py |
SwapManager.set_forward_prefetch_layer._forward_hook |
|
9 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / activation_checkpoint / swap.py |
SwapManager.set_forward_prefetch_layer._backward_pre_hook |
|
8 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / activation_checkpoint / swap.py |
SwapManager.set_forward_prefetch_layer._backward_hook |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / activation_checkpoint / swap.py |
SwapManager.set_forward_prefetch_layer._register_hooks_once |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / activation_checkpoint / swap.py |
SwapManager._get_copy_stream |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / activation_checkpoint / swap.py |
(no function) |
|
84 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / __init__.py |
(no function) |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / async_context_parallel.py |
_detach_if_available |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / async_context_parallel.py |
_launch_async_a2a_seq_to_head |
|
9 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / async_context_parallel.py |
_a2a_reconstruct |
|
7 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / async_context_parallel.py |
_normalize_dim |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / async_context_parallel.py |
_move_dim_to_front |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / async_context_parallel.py |
_move_dim_from_front |
|
8 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / async_context_parallel.py |
_launch_async_allgather_seq |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / async_context_parallel.py |
_allgather_reconstruct |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / async_context_parallel.py |
_to_local |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / async_context_parallel.py |
_to_cp_local |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / async_context_parallel.py |
_cp_execution_layout_from_input |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / async_context_parallel.py |
_make_async_cp_slot |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / async_context_parallel.py |
_slot_comm |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / async_context_parallel.py |
_slot_layout |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / async_context_parallel.py |
_wrap_async_cp_result |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / async_context_parallel.py |
AsyncContextParallel.__init__ |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / async_context_parallel.py |
AsyncContextParallel.apply |
|
12 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / async_context_parallel.py |
AsyncContextParallel._apply_colossal_async |
|
8 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / async_context_parallel.py |
AsyncContextParallel._apply_a2a_async |
|
23 |
1 |
0 |
|
96% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / async_context_parallel.py |
AsyncContextParallel._register_proj_hooks |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / async_context_parallel.py |
AsyncContextParallel._proj_post_hook |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / async_context_parallel.py |
AsyncContextParallel._register_ag_proj_hooks |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / async_context_parallel.py |
AsyncContextParallel._proj_ag_post_hook |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / async_context_parallel.py |
AsyncContextParallel._get_qkv_value |
|
8 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / async_context_parallel.py |
AsyncContextParallel._set_qkv_value |
|
8 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / async_context_parallel.py |
AsyncContextParallel._wait_a2a |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / async_context_parallel.py |
AsyncContextParallel._wait_a2a_dtensor |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / async_context_parallel.py |
AsyncContextParallel._wait_allgather |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / async_context_parallel.py |
AsyncContextParallel._apply_qkv_transforms |
|
9 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / async_context_parallel.py |
AsyncContextParallel._attn_pre_hook_ulysses |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / async_context_parallel.py |
AsyncContextParallel._attn_pre_hook_ulysses.make_a2a |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / async_context_parallel.py |
AsyncContextParallel._attn_pre_hook_ulysses.make_a2a.transform |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / async_context_parallel.py |
AsyncContextParallel._ulysses_cp_layout_from_input |
|
6 |
1 |
0 |
|
83% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / async_context_parallel.py |
AsyncContextParallel._attn_pre_hook_colossal |
|
24 |
7 |
0 |
|
71% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / async_context_parallel.py |
AsyncContextParallel._attn_pre_hook_hybrid |
|
22 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / async_context_parallel.py |
AsyncContextParallel._attn_post_hook_ata |
|
5 |
1 |
0 |
|
80% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / async_context_parallel.py |
AsyncContextParallel._attn_post_hook_ata._process |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / async_context_parallel.py |
AsyncContextParallel._proj_bwd_pre_hook |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / async_context_parallel.py |
AsyncContextParallel._proj_ag_bwd_pre_hook |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / async_context_parallel.py |
(no function) |
|
47 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / async_dsa_context_parallel.py |
_AsyncSequenceReplicateSlot.__init__ |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / async_dsa_context_parallel.py |
_AsyncSequenceReplicateSlot._extract_tensor_output |
|
5 |
3 |
0 |
|
40% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / async_dsa_context_parallel.py |
_AsyncSequenceReplicateSlot._local_tensor |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / async_dsa_context_parallel.py |
_AsyncSequenceReplicateSlot._make_slot |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / async_dsa_context_parallel.py |
_AsyncSequenceReplicateSlot._wrap_gathered |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / async_dsa_context_parallel.py |
_AsyncSequenceReplicateSlot.register_launch_hook |
|
7 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / async_dsa_context_parallel.py |
_AsyncSequenceReplicateSlot.register_launch_hook._post_hook |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / async_dsa_context_parallel.py |
_AsyncSequenceReplicateSlot.register_launch_hook._backward_pre_hook |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / async_dsa_context_parallel.py |
_AsyncSequenceReplicateSlot._producer_bwd_pre_hook |
|
8 |
2 |
0 |
|
75% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / async_dsa_context_parallel.py |
_AsyncSequenceReplicateSlot.launch |
|
10 |
4 |
0 |
|
60% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / async_dsa_context_parallel.py |
_AsyncSequenceReplicateSlot.wait |
|
17 |
3 |
0 |
|
82% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / async_dsa_context_parallel.py |
AsyncDSAIndexerContextParallel.apply |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / async_dsa_context_parallel.py |
AsyncDSASparseAttentionContextParallel.apply |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / async_dsa_context_parallel.py |
AsyncDSAIndexerLossContextParallel.apply |
|
7 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / async_dsa_context_parallel.py |
(no function) |
|
27 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / context_parallel.py |
_same_mesh_rank_list |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / context_parallel.py |
_same_mesh |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / context_parallel.py |
_is_foreign_dtensor |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / context_parallel.py |
_is_cp_composed_dtensor |
|
12 |
3 |
0 |
|
75% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / context_parallel.py |
_compose_cp_non_cp_mesh |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / context_parallel.py |
_compose_cp_non_cp_placements |
|
16 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / context_parallel.py |
_compose_cp_mesh |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / context_parallel.py |
_composed_placements |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / context_parallel.py |
_non_cp_placements_from_composed |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / context_parallel.py |
_cp_mesh_from_composed |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / context_parallel.py |
_split_cp_composed_layout |
|
11 |
2 |
0 |
|
82% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / context_parallel.py |
_non_cp_dtensor_layout |
|
10 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / context_parallel.py |
_localize_foreign_dtensor |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / context_parallel.py |
_cp_boundary_dim_size |
|
3 |
1 |
0 |
|
67% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / context_parallel.py |
_to_cp_dtensor |
|
16 |
2 |
0 |
|
88% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / context_parallel.py |
_output_layout_from_q |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / context_parallel.py |
_push_output_layout |
|
7 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / context_parallel.py |
_pop_output_layout |
|
9 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / context_parallel.py |
_drop_cp_from_output |
|
12 |
2 |
0 |
|
83% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / context_parallel.py |
_wrap_cp_output_dtensor |
|
9 |
2 |
0 |
|
78% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / context_parallel.py |
_finalize_output |
|
8 |
1 |
0 |
|
88% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / context_parallel.py |
_finalize_colossal_output |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / context_parallel.py |
_finalize_ata_output |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / context_parallel.py |
_is_tensor_or_dtensor |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / context_parallel.py |
_ensure_1d |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / context_parallel.py |
_build_2d_mesh |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / context_parallel.py |
_build_hybrid_cp_mesh |
|
9 |
3 |
0 |
|
67% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / context_parallel.py |
_scatter_seq_to_head |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / context_parallel.py |
_gather_head_to_seq |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / context_parallel.py |
_gather_seq |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / context_parallel.py |
ContextParallel.__init__ |
|
9 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / context_parallel.py |
ContextParallel.apply |
|
23 |
1 |
0 |
|
96% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / context_parallel.py |
ContextParallel._record_q_output_tp_layout |
|
9 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / context_parallel.py |
ContextParallel._pre_hook_colossal |
|
17 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / context_parallel.py |
ContextParallel._pre_hook_ulysses |
|
10 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / context_parallel.py |
ContextParallel._hybrid_cp_layout_from_input |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / context_parallel.py |
ContextParallel._ata_scatter_to_hybrid |
|
8 |
1 |
0 |
|
88% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / context_parallel.py |
ContextParallel._hybrid_kv_gather_placements |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / context_parallel.py |
ContextParallel._pre_hook_hybrid |
|
18 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / context_parallel.py |
ContextParallel._needs_hybrid_ata |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / context_parallel.py |
ContextParallel._post_hook_ata |
|
5 |
1 |
0 |
|
80% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / context_parallel.py |
ContextParallel._post_hook_ata._process |
|
6 |
1 |
0 |
|
83% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / context_parallel.py |
ContextParallel._post_hook_hybrid |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / context_parallel.py |
ContextParallel._post_hook_hybrid._process |
|
14 |
1 |
0 |
|
93% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / context_parallel.py |
ContextParallel._post_hook_colossal |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / context_parallel.py |
ContextParallel._post_hook_colossal._process |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / context_parallel.py |
ContextParallel._apply_lb_colossal |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / context_parallel.py |
ContextParallel._lb_colossal_forward |
|
23 |
23 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / context_parallel.py |
ContextParallel._lb_colossal_forward._fa |
|
7 |
7 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / context_parallel.py |
(no function) |
|
61 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / dsa_context_parallel.py |
_move_dim_to_front |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / dsa_context_parallel.py |
_DSASequenceReplicateGradientBridge.forward |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / dsa_context_parallel.py |
_DSASequenceReplicateGradientBridge.backward |
|
12 |
10 |
0 |
|
17% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / dsa_context_parallel.py |
DSASequenceReplicateCache.__init__ |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / dsa_context_parallel.py |
DSASequenceReplicateCache._signature |
|
6 |
2 |
0 |
|
67% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / dsa_context_parallel.py |
DSASequenceReplicateCache.begin |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / dsa_context_parallel.py |
DSASequenceReplicateCache.replicate |
|
19 |
4 |
0 |
|
79% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / dsa_context_parallel.py |
DSASequenceReplicateCache.clear |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / dsa_context_parallel.py |
_is_tensor_or_dtensor |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / dsa_context_parallel.py |
_to_sequence_shard |
|
3 |
1 |
0 |
|
67% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / dsa_context_parallel.py |
_to_sequence_replicate |
|
3 |
1 |
0 |
|
67% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / dsa_context_parallel.py |
_to_query_stats_shard |
|
3 |
1 |
0 |
|
67% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / dsa_context_parallel.py |
_maybe_replace_arg |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / dsa_context_parallel.py |
_maybe_replace_kwarg |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / dsa_context_parallel.py |
_apply_param_specs |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / dsa_context_parallel.py |
_validate_layout_and_mode |
|
6 |
1 |
0 |
|
83% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / dsa_context_parallel.py |
_validate_loss_variant |
|
4 |
1 |
0 |
|
75% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / dsa_context_parallel.py |
_query_stats_seq_dim |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / dsa_context_parallel.py |
_default_arg_index |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / dsa_context_parallel.py |
_finalize_output |
|
9 |
4 |
0 |
|
56% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / dsa_context_parallel.py |
_finalize_output.finalize_one |
|
5 |
1 |
0 |
|
80% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / dsa_context_parallel.py |
_dtensor_has_partial |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / dsa_context_parallel.py |
_dtensor_to_local_reducing_partial |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / dsa_context_parallel.py |
_register_boundary_hooks |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / dsa_context_parallel.py |
_register_boundary_hooks._finalize_output_hook |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / dsa_context_parallel.py |
_record_query_output_layout |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / dsa_context_parallel.py |
_read_value |
|
5 |
1 |
0 |
|
80% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / dsa_context_parallel.py |
_configure_sparse_attention_boundary |
|
19 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / dsa_context_parallel.py |
_apply_sparse_attention_boundary |
|
9 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / dsa_context_parallel.py |
_apply_sparse_attention_boundary._shard |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / dsa_context_parallel.py |
_apply_sparse_attention_boundary._replicate |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / dsa_context_parallel.py |
_apply_sparse_attention_boundary._pre_hook |
|
7 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / dsa_context_parallel.py |
DSAIndexerContextParallel.__init__ |
|
12 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / dsa_context_parallel.py |
DSAIndexerContextParallel.__repr__ |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / dsa_context_parallel.py |
DSAIndexerContextParallel._shard_query_side |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / dsa_context_parallel.py |
DSAIndexerContextParallel._replicate_key_side |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / dsa_context_parallel.py |
DSAIndexerContextParallel._build_specs |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / dsa_context_parallel.py |
DSAIndexerContextParallel._build_specs.shard |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / dsa_context_parallel.py |
DSAIndexerContextParallel._apply_with_specs |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / dsa_context_parallel.py |
DSAIndexerContextParallel._apply_with_specs._pre_hook |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / dsa_context_parallel.py |
DSAIndexerContextParallel.apply |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / dsa_context_parallel.py |
DSASparseAttentionContextParallel.__init__ |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / dsa_context_parallel.py |
DSASparseAttentionContextParallel.__repr__ |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / dsa_context_parallel.py |
DSASparseAttentionContextParallel.apply |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / dsa_context_parallel.py |
DSAIndexerLossContextParallel.__init__ |
|
35 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / dsa_context_parallel.py |
DSAIndexerLossContextParallel.__repr__ |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / dsa_context_parallel.py |
DSAIndexerLossContextParallel._shard_query_side |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / dsa_context_parallel.py |
DSAIndexerLossContextParallel._replicate_key_side |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / dsa_context_parallel.py |
DSAIndexerLossContextParallel._local_shape |
|
5 |
2 |
0 |
|
60% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / dsa_context_parallel.py |
DSAIndexerLossContextParallel._slice_local_key_grad |
|
15 |
15 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / dsa_context_parallel.py |
DSAIndexerLossContextParallel._process_outputs |
|
11 |
8 |
0 |
|
27% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / dsa_context_parallel.py |
DSAIndexerLossContextParallel._build_loss_specs |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / dsa_context_parallel.py |
DSAIndexerLossContextParallel._build_loss_specs.shard |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / dsa_context_parallel.py |
DSAIndexerLossContextParallel._build_loss_specs.stats_shard |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / dsa_context_parallel.py |
DSAIndexerLossContextParallel._read_key_indexer_shape |
|
5 |
1 |
0 |
|
80% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / dsa_context_parallel.py |
DSAIndexerLossContextParallel._get_local_idx |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / dsa_context_parallel.py |
DSAIndexerLossContextParallel._apply_with_loss_specs |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / dsa_context_parallel.py |
DSAIndexerLossContextParallel._apply_with_loss_specs._pre_hook |
|
11 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / dsa_context_parallel.py |
DSAIndexerLossContextParallel.apply |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / dsa_context_parallel.py |
DSAIndexerLossContextParallel.apply.replicate |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / dsa_context_parallel.py |
DSAIndexerLossContextParallel.apply.shared_replicate |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / dsa_context_parallel.py |
(no function) |
|
77 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / __init__.py |
(no function) |
|
12 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / api.py |
_save_impl |
|
29 |
3 |
0 |
|
90% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / api.py |
_save_impl.generate_final_plan_and_metadata |
|
9 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / api.py |
save |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / api.py |
_create_persist_process |
|
19 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / api.py |
async_save |
|
10 |
10 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / api.py |
async_save.async_callback |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / api.py |
_open_checkpoint |
|
8 |
2 |
0 |
|
75% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / api.py |
load |
|
24 |
1 |
0 |
|
96% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / api.py |
(no function) |
|
25 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / async_persist.py |
DataCopier.register |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / async_persist.py |
DataCopier.register.decorator |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / async_persist.py |
DataCopier.dispatch |
|
9 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / async_persist.py |
DataCopier.copy |
|
5 |
2 |
0 |
|
60% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / async_persist.py |
DataCopier._copy_others |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / async_persist.py |
_copy_tensor_to_cpu |
|
4 |
1 |
0 |
|
75% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / async_persist.py |
_copy_const |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / async_persist.py |
_copy_bytes |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / async_persist.py |
_copy_ndarray |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / async_persist.py |
_copy_tensor |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / async_persist.py |
_copy_dtensor |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / async_persist.py |
_copy_namespace |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / async_persist.py |
_copy_list |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / async_persist.py |
_copy_tuple |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / async_persist.py |
_copy_dict |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / async_persist.py |
AsyncSaveResponse.get_result |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / async_persist.py |
build_staged_state_dict |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / async_persist.py |
cleanup_and_reinit_process_group |
|
25 |
25 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / async_persist.py |
execute_async_persist_with_gloo |
|
10 |
10 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / async_persist.py |
execute_async_persist |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / async_persist.py |
resolve_async_persist_result |
|
29 |
1 |
0 |
|
97% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / async_persist.py |
batch_read_worker |
|
18 |
1 |
0 |
|
94% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / async_persist.py |
parallel_read_files |
|
7 |
7 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / async_persist.py |
construct_file_batches |
|
8 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / async_persist.py |
gather_all_results_from_storage |
|
10 |
10 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / async_persist.py |
assemble_file_path |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / async_persist.py |
_safe_remove_file |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / async_persist.py |
load_file |
|
18 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / async_persist.py |
write_file |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / async_persist.py |
(no function) |
|
81 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / filesystem_storage.py |
_MetadataUnpickler.find_class |
|
3 |
1 |
0 |
|
67% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / filesystem_storage.py |
FileSystemWriter.__init__ |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / filesystem_storage.py |
FileSystemWriter.initialize_writer |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / filesystem_storage.py |
FileSystemWriter.configure_writer |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / filesystem_storage.py |
FileSystemWriter.optimize_local_plan |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / filesystem_storage.py |
FileSystemWriter.optimize_global_plan |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / filesystem_storage.py |
FileSystemWriter._serialize_bytes_item |
|
4 |
1 |
0 |
|
75% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / filesystem_storage.py |
FileSystemWriter._write_bytes_items |
|
15 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / filesystem_storage.py |
FileSystemWriter._collect_tensors |
|
23 |
1 |
0 |
|
96% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / filesystem_storage.py |
FileSystemWriter._write_tensors |
|
11 |
1 |
0 |
|
91% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / filesystem_storage.py |
FileSystemWriter.execute_write |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / filesystem_storage.py |
FileSystemWriter.finalize_checkpoint |
|
13 |
1 |
0 |
|
92% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / filesystem_storage.py |
_copy_tensor_to_target |
|
4 |
1 |
0 |
|
75% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / filesystem_storage.py |
_fetch_bytes_file |
|
7 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / filesystem_storage.py |
_get_tensor_size |
|
3 |
1 |
0 |
|
67% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / filesystem_storage.py |
_get_storage_info |
|
4 |
1 |
0 |
|
75% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / filesystem_storage.py |
_validate_and_copy_tensor |
|
8 |
1 |
0 |
|
88% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / filesystem_storage.py |
_fetch_torch_tensor_file |
|
10 |
1 |
0 |
|
90% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / filesystem_storage.py |
_fetch_ms_tensor_file |
|
8 |
8 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / filesystem_storage.py |
_fetch_tensor_file |
|
3 |
1 |
0 |
|
67% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / filesystem_storage.py |
_apply_fetched |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / filesystem_storage.py |
_OpenFiles.__init__ |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / filesystem_storage.py |
_OpenFiles.reader |
|
7 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / filesystem_storage.py |
_OpenFiles.close |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / filesystem_storage.py |
_open_checkpoint_files |
|
3 |
1 |
0 |
|
67% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / filesystem_storage.py |
_broadcast_batch_bytes |
|
3 |
1 |
0 |
|
67% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / filesystem_storage.py |
FileSystemReader.__init__ |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / filesystem_storage.py |
FileSystemReader.initialize_reader |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / filesystem_storage.py |
FileSystemReader.load_metadata |
|
9 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / filesystem_storage.py |
FileSystemReader.configure_reader |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / filesystem_storage.py |
FileSystemReader.optimize_local_plan |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / filesystem_storage.py |
FileSystemReader.optimize_global_plan |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / filesystem_storage.py |
FileSystemReader._get_storage_path |
|
6 |
2 |
0 |
|
67% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / filesystem_storage.py |
FileSystemReader._group_items_by_file |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / filesystem_storage.py |
FileSystemReader._private_batches |
|
7 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / filesystem_storage.py |
FileSystemReader._fetch_from_storage |
|
7 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / filesystem_storage.py |
FileSystemReader.execute_read |
|
24 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / filesystem_storage.py |
(no function) |
|
59 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / layout.py |
get_current_layout |
|
23 |
1 |
0 |
|
96% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / layout.py |
save_layout |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / layout.py |
load_layout |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / layout.py |
combine_layout |
|
9 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / layout.py |
get_global_layout |
|
10 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / layout.py |
(no function) |
|
13 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / loader.py |
load_checkpoint |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / loader.py |
(no function) |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / metadata.py |
dtype_element_size |
|
14 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / metadata.py |
(no function) |
|
39 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / offline_transform.py |
_validate_state_dict_for_active_platform |
|
16 |
9 |
0 |
|
44% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / offline_transform.py |
_StateDictSplit.__post_init__ |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / offline_transform.py |
_parse_size_to_int |
|
10 |
10 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / offline_transform.py |
_build_shard_list_for_safetensors |
|
17 |
17 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / offline_transform.py |
_total_bytes_unique_keys_in_shards |
|
11 |
11 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / offline_transform.py |
_split_state_dict_into_shards |
|
21 |
21 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / offline_transform.py |
save_state_dict_as_huggingface_format |
|
24 |
24 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / offline_transform.py |
parse_checkpoint_from_huggingface |
|
29 |
29 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / offline_transform.py |
_mindspore_full_checkpoint_format_for_path |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / offline_transform.py |
_torch_full_checkpoint_format_for_path |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / offline_transform.py |
full_state_dict_to_dcp_format |
|
12 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / offline_transform.py |
dcp_to_full_state_dict |
|
13 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / offline_transform.py |
convert_full_checkpoint_to_dcp |
|
19 |
19 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / offline_transform.py |
(no function) |
|
35 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / planner.py |
WriteItem.tensor_storage_size |
|
13 |
2 |
0 |
|
85% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / planner.py |
LoadPlan.identity |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / planner.py |
SavePlanner.configure_planner |
|
0 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / planner.py |
SavePlanner.build_local_plan |
|
0 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / planner.py |
SavePlanner.build_global_plan |
|
0 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / planner.py |
SavePlanner.finalize_plan |
|
0 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / planner.py |
SavePlanner.get_data |
|
0 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / planner.py |
LoadPlanner.configure_planner |
|
0 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / planner.py |
LoadPlanner.build_local_plan |
|
0 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / planner.py |
LoadPlanner.build_global_plan |
|
0 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / planner.py |
LoadPlanner.finalize_plan |
|
0 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / planner.py |
LoadPlanner.acquire_tensor |
|
0 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / planner.py |
LoadPlanner.apply_tensor |
|
0 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / planner.py |
LoadPlanner.apply_bytes |
|
0 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / planner.py |
(no function) |
|
71 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / ragged_utils.py |
_decompose_flat_interval |
|
11 |
2 |
0 |
|
82% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / ragged_utils.py |
_decompose_flat_interval._decompose_axis |
|
18 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / ragged_utils.py |
compute_ragged_boxes |
|
15 |
2 |
0 |
|
87% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / ragged_utils.py |
create_ragged_write_items |
|
8 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / ragged_utils.py |
get_ragged_box_tensor |
|
6 |
1 |
0 |
|
83% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / ragged_utils.py |
(no function) |
|
15 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / reshard.py |
check_layout |
|
14 |
1 |
0 |
|
93% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / reshard.py |
check_layout.check_type_is_sequence |
|
2 |
1 |
0 |
|
50% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / reshard.py |
rank_id_to_dev_id_list |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / reshard.py |
infer_intersection |
|
13 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / reshard.py |
infer_intersection.is_valid_axis_list |
|
5 |
2 |
0 |
|
60% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / reshard.py |
ReshardHandler.__init__ |
|
34 |
9 |
0 |
|
74% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / reshard.py |
ReshardHandler._infer_inner_deredundancy_rank_list_by_from_layout |
|
21 |
21 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / reshard.py |
ReshardHandler.infer_all_tensor_offset |
|
11 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / reshard.py |
ReshardHandler.get_real_tensor |
|
17 |
2 |
0 |
|
88% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / reshard.py |
(no function) |
|
13 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / saver.py |
save_checkpoint |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / saver.py |
(no function) |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / standard_planner.py |
_own_plan_index |
|
4 |
1 |
0 |
|
75% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / standard_planner.py |
StandardSavePlanner.__init__ |
|
8 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / standard_planner.py |
StandardSavePlanner.configure_planner |
|
15 |
1 |
0 |
|
93% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / standard_planner.py |
StandardSavePlanner._build_cache_key |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / standard_planner.py |
StandardSavePlanner.build_local_plan |
|
36 |
16 |
0 |
|
56% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / standard_planner.py |
StandardSavePlanner.build_local_plan.compute_global_offsets |
|
14 |
14 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / standard_planner.py |
StandardSavePlanner.build_global_plan |
|
30 |
1 |
0 |
|
97% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / standard_planner.py |
StandardSavePlanner._global_chunks_for |
|
9 |
1 |
0 |
|
89% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / standard_planner.py |
StandardSavePlanner.finalize_plan |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / standard_planner.py |
StandardSavePlanner.get_cached |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / standard_planner.py |
StandardSavePlanner.cache_result |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / standard_planner.py |
StandardSavePlanner.get_data |
|
17 |
5 |
0 |
|
71% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / standard_planner.py |
create_read_items_for_chunk_list |
|
16 |
1 |
0 |
|
94% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / standard_planner.py |
_is_on_host |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / standard_planner.py |
StandardLoadPlanner.__init__ |
|
7 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / standard_planner.py |
StandardLoadPlanner.configure_planner |
|
10 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / standard_planner.py |
StandardLoadPlanner._rank_owns_dtensor_shard |
|
11 |
3 |
0 |
|
73% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / standard_planner.py |
StandardLoadPlanner.build_local_plan |
|
22 |
8 |
0 |
|
64% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / standard_planner.py |
StandardLoadPlanner.build_global_plan |
|
26 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / standard_planner.py |
StandardLoadPlanner.finalize_plan |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / standard_planner.py |
StandardLoadPlanner.acquire_tensor |
|
12 |
2 |
0 |
|
83% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / standard_planner.py |
StandardLoadPlanner.apply_tensor |
|
0 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / standard_planner.py |
StandardLoadPlanner.apply_bytes |
|
7 |
1 |
0 |
|
86% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / standard_planner.py |
_DcpMergeLoadPlanner.__init__ |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / standard_planner.py |
_DcpMergeLoadPlanner.configure_planner |
|
12 |
2 |
0 |
|
83% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / standard_planner.py |
(no function) |
|
51 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / state_dict.py |
get_optim_state_dict |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / state_dict.py |
set_optim_state_dict |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / state_dict.py |
(no function) |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / storage.py |
StorageWriter.initialize_writer |
|
0 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / storage.py |
StorageWriter.configure_writer |
|
0 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / storage.py |
StorageWriter.optimize_local_plan |
|
0 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / storage.py |
StorageWriter.optimize_global_plan |
|
0 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / storage.py |
StorageWriter.execute_write |
|
0 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / storage.py |
StorageWriter.finalize_checkpoint |
|
0 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / storage.py |
StorageReader.initialize_reader |
|
0 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / storage.py |
StorageReader.load_metadata |
|
0 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / storage.py |
StorageReader.configure_reader |
|
0 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / storage.py |
StorageReader.optimize_local_plan |
|
0 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / storage.py |
StorageReader.optimize_global_plan |
|
0 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / storage.py |
StorageReader.execute_read |
|
0 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / storage.py |
(no function) |
|
43 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / util.py |
dcp_timer_decorator |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / util.py |
dcp_timer_decorator.wrapper |
|
11 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / util.py |
check_path |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / util.py |
has_valid_filename |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / util.py |
narrow_tensor_by_index |
|
4 |
1 |
0 |
|
75% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / util.py |
chunk_to_area |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / util.py |
create_chunk_list_for_tensor |
|
31 |
5 |
0 |
|
84% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / util.py |
plan_ownership_masks |
|
29 |
3 |
0 |
|
90% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / util.py |
_least_loaded_owner_slot |
|
7 |
1 |
0 |
|
86% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / util.py |
traverse_state_dict |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / util.py |
traverse_state_dict._is_terminal |
|
11 |
1 |
0 |
|
91% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / util.py |
traverse_state_dict._traverse_obj |
|
8 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / util.py |
flatten_state_dict |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / util.py |
flatten_state_dict.flat_copy |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / util.py |
set_element |
|
18 |
1 |
0 |
|
94% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / util.py |
set_element.extend_list |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / util.py |
all_gather_object |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / util.py |
_existing_group |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / util.py |
_build_broadcast_groups |
|
19 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / util.py |
ensure_broadcast_groups |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / util.py |
_destroy_broadcast_groups |
|
8 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / util.py |
broadcast_groups_for_load |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / util.py |
_start_broadcast |
|
8 |
3 |
0 |
|
62% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / util.py |
_finish_broadcast |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / util.py |
broadcast_shard |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / util.py |
wait_broadcasts |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / util.py |
BroadcastBatcher.__init__ |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / util.py |
BroadcastBatcher.add |
|
11 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / util.py |
BroadcastBatcher.flush |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / util.py |
BroadcastBatcher._send |
|
21 |
1 |
0 |
|
95% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / util.py |
_shard_buffer |
|
5 |
1 |
0 |
|
80% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / util.py |
(no function) |
|
48 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / __init__.py |
(no function) |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / _collective_utils.py |
_ensure_mesh_process_groups |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / _collective_utils.py |
mesh_scatter |
|
8 |
8 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / _collective_utils.py |
mesh_scatter_ragged |
|
23 |
23 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / _collective_utils.py |
mesh_broadcast |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / _collective_utils.py |
(no function) |
|
10 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / _from_local_utils.py |
_ensure_mesh_process_groups |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / _from_local_utils.py |
mesh_broadcast |
|
8 |
8 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / _from_local_utils.py |
_tensor_meta |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / _from_local_utils.py |
check_tensor_meta |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / _from_local_utils.py |
_mesh_check_group |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / _from_local_utils.py |
run_from_local_checks |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / _from_local_utils.py |
(no function) |
|
13 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / _mesh_layout.py |
_is_int |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / _mesh_layout.py |
_as_tuple |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / _mesh_layout.py |
_flatten_inttuple |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / _mesh_layout.py |
_match_structure |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / _mesh_layout.py |
_numel |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / _mesh_layout.py |
_contiguous_strides |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / _mesh_layout.py |
_scale_inttuple |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / _mesh_layout.py |
_enumerate_offsets |
|
7 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / _mesh_layout.py |
_canonicalize_axis |
|
22 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / _mesh_layout.py |
_nested_from_flat |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / _mesh_layout.py |
_FlatLayout.__init__ |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / _mesh_layout.py |
_FlatLayout.numel |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / _mesh_layout.py |
_FlatLayout.cosize |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / _mesh_layout.py |
_FlatLayout.check_sorted |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / _mesh_layout.py |
_FlatLayout.check_orthogonal |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / _mesh_layout.py |
_FlatLayout.all_ranks_from_zero |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / _mesh_layout.py |
_MeshLayout.__init__ |
|
9 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / _mesh_layout.py |
_MeshLayout.from_sizes_strides |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / _mesh_layout.py |
_MeshLayout.sizes |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / _mesh_layout.py |
_MeshLayout.strides |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / _mesh_layout.py |
_MeshLayout.axes |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / _mesh_layout.py |
_MeshLayout.__len__ |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / _mesh_layout.py |
_MeshLayout.__iter__ |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / _mesh_layout.py |
_MeshLayout.__getitem__ |
|
7 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / _mesh_layout.py |
_MeshLayout.__eq__ |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / _mesh_layout.py |
_MeshLayout.__repr__ |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / _mesh_layout.py |
_MeshLayout.numel |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / _mesh_layout.py |
_MeshLayout.top_level_sizes |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / _mesh_layout.py |
_MeshLayout.all_ranks_from_zero |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / _mesh_layout.py |
_MeshLayout.check_non_overlap |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / _mesh_layout.py |
_MeshLayout.coalesce |
|
20 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / _mesh_layout.py |
_MeshLayout.composition |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / _mesh_layout.py |
_MeshLayout.nest |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / _mesh_layout.py |
_MeshLayout.splice |
|
7 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / _mesh_layout.py |
_MeshLayout.collapse |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / _mesh_layout.py |
_MeshLayout.remap_to_numpy |
|
22 |
2 |
0 |
|
91% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / _mesh_layout.py |
(no function) |
|
51 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / _ragged_utils.py |
_layout_has_ragged_shard |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / _ragged_utils.py |
_RaggedSlice.local_numel |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / _ragged_utils.py |
_normalize_global_shape |
|
4 |
1 |
0 |
|
75% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / _ragged_utils.py |
_compute_ragged_slice |
|
31 |
5 |
0 |
|
84% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / _ragged_utils.py |
_compute_ragged_splits |
|
4 |
1 |
0 |
|
75% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / _ragged_utils.py |
_interval_overlap_size |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / _ragged_utils.py |
_compute_ragged_all_to_all_splits |
|
16 |
3 |
0 |
|
81% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / _ragged_utils.py |
_slice_ragged_tensor |
|
5 |
1 |
0 |
|
80% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / _ragged_utils.py |
_scatter_ragged_tensor |
|
13 |
13 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / _ragged_utils.py |
(no function) |
|
20 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / _utils.py |
_a2a_reconstruct |
|
7 |
7 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / _utils.py |
_normalize_dim |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / _utils.py |
_move_dim_to_front |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / _utils.py |
_move_dim_from_front |
|
8 |
8 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / _utils.py |
_ensure_contiguous |
|
3 |
1 |
0 |
|
67% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / _utils.py |
get_op_name |
|
14 |
12 |
0 |
|
14% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / _utils.py |
tensor_type_cast |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / _utils.py |
get_device_handle |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / _utils.py |
device_count |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / _utils.py |
device_type |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / _utils.py |
device |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / _utils.py |
manual_seed |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / _utils.py |
get_rng_state |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / _utils.py |
set_rng_state |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / _utils.py |
get_created_group |
|
4 |
1 |
0 |
|
75% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / _utils.py |
create_group |
|
10 |
10 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / _utils.py |
split_group |
|
12 |
1 |
0 |
|
92% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / _utils.py |
init_process_group |
|
2 |
1 |
0 |
|
50% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / _utils.py |
_validate_intra_step |
|
7 |
7 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / _utils.py |
_compute_group_starts |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / _utils.py |
_build_groups_for_blocks |
|
8 |
8 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / _utils.py |
generate_groups_from_template |
|
33 |
33 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / _utils.py |
create_sub_groups |
|
31 |
31 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / _utils.py |
_TorchContiguousGrad.forward |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / _utils.py |
_TorchContiguousGrad.backward |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / _utils.py |
_TorchAsyncA2AFunction.forward |
|
8 |
8 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / _utils.py |
_TorchAsyncA2AFunction.backward |
|
11 |
11 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / _utils.py |
_TorchAsyncAllGatherFunction.forward |
|
7 |
7 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / _utils.py |
_TorchAsyncAllGatherFunction.backward |
|
12 |
12 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / _utils.py |
_AsyncA2ALazyBwd.forward |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / _utils.py |
_AsyncA2ALazyBwd.backward |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / _utils.py |
_TorchSyncHookFunction._role_enum |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / _utils.py |
_TorchSyncHookFunction.forward |
|
14 |
14 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / _utils.py |
_TorchSyncHookFunction.backward |
|
11 |
11 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / _utils.py |
_TorchP2PExchangeFunction.forward |
|
8 |
8 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / _utils.py |
_TorchP2PExchangeFunction.backward |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / _utils.py |
_TorchDifferentiableVariableAllGather.forward |
|
30 |
30 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / _utils.py |
_TorchDifferentiableVariableAllGather.backward |
|
13 |
13 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / _utils.py |
differentiable_all_gather_concat |
|
8 |
8 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / _utils.py |
chunk |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / _utils.py |
differentiable_all_to_all |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / _utils.py |
differentiable_all_reduce |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / _utils.py |
differentiable_reduce_scatter |
|
8 |
8 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / _utils.py |
differentiable_all_to_all_single |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / _utils.py |
differentiable_all_to_all_single_async |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / _utils.py |
differentiable_variable_all_gather |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / _utils.py |
wait_async_tensor |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / _utils.py |
differentiable_async_allgather_wait |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / _utils.py |
differentiable_async_a2a_wait |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / _utils.py |
differentiable_sync_hook |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / _utils.py |
p2p_exchange |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / _utils.py |
get_tensor_transform |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / _utils.py |
construct_strided_slice |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / _utils.py |
init_on_device |
|
12 |
12 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / _utils.py |
init_on_device._register_parameter |
|
7 |
7 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / _utils.py |
init_on_device._register_buffer |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / _utils.py |
(no function) |
|
94 |
1 |
0 |
|
99% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / debug / __init__.py |
(no function) |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / debug / _call_records.py |
DebugCall._render_self |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / debug / _call_records.py |
DebugCall.render |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / debug / _call_records.py |
OpCall._render_self |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / debug / _call_records.py |
CollectiveCall._render_self |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / debug / _call_records.py |
RedistributeCall._render_self |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / debug / _call_records.py |
AnnotateCall._render_self |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / debug / _call_records.py |
(no function) |
|
44 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / debug / _collective_tracer.py |
CollectiveTracer.__init__ |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / debug / _collective_tracer.py |
CollectiveTracer.install |
|
11 |
1 |
0 |
|
91% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / debug / _collective_tracer.py |
CollectiveTracer.install._make_wrapper |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / debug / _collective_tracer.py |
CollectiveTracer.install._make_wrapper.wrapper |
|
6 |
2 |
0 |
|
67% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / debug / _collective_tracer.py |
CollectiveTracer.uninstall |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / debug / _collective_tracer.py |
(no function) |
|
9 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / debug / _comm_debug_mode.py |
CommDebugMode.__init__ |
|
9 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / debug / _comm_debug_mode.py |
CommDebugMode.__enter__ |
|
14 |
3 |
0 |
|
79% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / debug / _comm_debug_mode.py |
CommDebugMode.__exit__ |
|
10 |
2 |
0 |
|
80% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / debug / _comm_debug_mode.py |
CommDebugMode.__repr__ |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / debug / _comm_debug_mode.py |
CommDebugMode._on_op_dispatch_enter |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / debug / _comm_debug_mode.py |
CommDebugMode._on_op_dispatch_exit |
|
5 |
1 |
0 |
|
80% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / debug / _comm_debug_mode.py |
CommDebugMode._on_collective_call |
|
26 |
2 |
0 |
|
92% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / debug / _comm_debug_mode.py |
CommDebugMode._on_module_event |
|
9 |
9 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / debug / _comm_debug_mode.py |
CommDebugMode._collect_module_info |
|
13 |
13 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / debug / _comm_debug_mode.py |
CommDebugMode._extract_tensor_infos |
|
14 |
5 |
0 |
|
64% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / debug / _comm_debug_mode.py |
CommDebugMode.get_comm_counts |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / debug / _comm_debug_mode.py |
CommDebugMode.get_total_counts |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / debug / _comm_debug_mode.py |
CommDebugMode.get_parameter_info |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / debug / _comm_debug_mode.py |
CommDebugMode.get_sharding_info |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / debug / _comm_debug_mode.py |
CommDebugMode.generate_comm_debug_tracing_table |
|
13 |
1 |
0 |
|
92% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / debug / _comm_debug_mode.py |
CommDebugMode.log_comm_debug_tracing_table_to_file |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / debug / _comm_debug_mode.py |
CommDebugMode.generate_json_dump |
|
12 |
12 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / debug / _comm_debug_mode.py |
CommDebugMode.generate_json_dump._record_to_dict |
|
31 |
31 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / debug / _comm_debug_mode.py |
CommDebugMode._collect_table_lines |
|
9 |
1 |
0 |
|
89% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / debug / _comm_debug_mode.py |
(no function) |
|
34 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / debug / _dispatch_logger.py |
log_dispatch_enter |
|
8 |
8 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / debug / _dispatch_logger.py |
log_dispatch_exit |
|
10 |
10 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / debug / _dispatch_logger.py |
(no function) |
|
7 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / debug / _module_tracker.py |
ModuleTracker.__init__ |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / debug / _module_tracker.py |
ModuleTracker.install |
|
9 |
9 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / debug / _module_tracker.py |
ModuleTracker.install._make_pre_hook |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / debug / _module_tracker.py |
ModuleTracker.install._make_pre_hook.hook |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / debug / _module_tracker.py |
ModuleTracker.install._make_post_hook |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / debug / _module_tracker.py |
ModuleTracker.install._make_post_hook.hook |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / debug / _module_tracker.py |
ModuleTracker.uninstall |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / debug / _module_tracker.py |
(no function) |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / device_mesh.py |
_host_tensor_from_numpy |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / device_mesh.py |
_MeshEnv.__init__ |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / device_mesh.py |
_MeshEnv.get_current_mesh |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / device_mesh.py |
_get_sub_rank_list |
|
13 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / device_mesh.py |
_normalize_backend_value |
|
9 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / device_mesh.py |
_normalize_backend_override |
|
15 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / device_mesh.py |
_should_defer_group_init |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / device_mesh.py |
DeviceMesh.__init__ |
|
11 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / device_mesh.py |
DeviceMesh._validate_device_type |
|
2 |
1 |
0 |
|
50% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / device_mesh.py |
DeviceMesh._resolve_layout_and_rank_map |
|
16 |
6 |
0 |
|
62% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / device_mesh.py |
DeviceMesh._refresh_mesh_view |
|
15 |
1 |
0 |
|
93% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / device_mesh.py |
DeviceMesh._set_mesh_dim_names |
|
12 |
3 |
0 |
|
75% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / device_mesh.py |
DeviceMesh._initialize_runtime_state |
|
11 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / device_mesh.py |
DeviceMesh._build_layout_from_mesh |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / device_mesh.py |
DeviceMesh._build_rank_map_from_mesh |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / device_mesh.py |
DeviceMesh._convert_rank_map_to_tensor |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / device_mesh.py |
DeviceMesh._get_mesh_tensor_from_full_mesh |
|
8 |
8 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / device_mesh.py |
DeviceMesh._compute_coordinate_on_dim |
|
9 |
2 |
0 |
|
78% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / device_mesh.py |
DeviceMesh._compute_coordinates_from_mesh |
|
7 |
7 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / device_mesh.py |
DeviceMesh.size |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / device_mesh.py |
DeviceMesh.get_coordinate |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / device_mesh.py |
DeviceMesh.__enter__ |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / device_mesh.py |
DeviceMesh.__exit__ |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / device_mesh.py |
DeviceMesh._convert_mesh_to_tensor |
|
8 |
1 |
0 |
|
88% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / device_mesh.py |
DeviceMesh._init_one_process_group |
|
13 |
13 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / device_mesh.py |
DeviceMesh._build_dim_split_ranks |
|
17 |
1 |
0 |
|
94% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / device_mesh.py |
DeviceMesh._cache_group_if_needed |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / device_mesh.py |
DeviceMesh._init_process_groups_for_layout |
|
16 |
1 |
0 |
|
94% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / device_mesh.py |
DeviceMesh._init_process_groups |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / device_mesh.py |
DeviceMesh.rank |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / device_mesh.py |
DeviceMesh.mesh_shape |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / device_mesh.py |
DeviceMesh.rank_list |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / device_mesh.py |
DeviceMesh.ndim |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / device_mesh.py |
DeviceMesh.shape |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / device_mesh.py |
DeviceMesh.root_mesh |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / device_mesh.py |
DeviceMesh.root_mesh |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / device_mesh.py |
DeviceMesh.sub_mesh |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / device_mesh.py |
DeviceMesh.get_flatten_mapping |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / device_mesh.py |
DeviceMesh.add_flatten_mapping |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / device_mesh.py |
DeviceMesh.__getitem__ |
|
13 |
2 |
0 |
|
85% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / device_mesh.py |
DeviceMesh._normalize_sub_mesh_dim_names |
|
7 |
2 |
0 |
|
71% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / device_mesh.py |
DeviceMesh._try_get_from_flatten_mapping |
|
3 |
1 |
0 |
|
67% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / device_mesh.py |
DeviceMesh._get_mesh_dim_by_name |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / device_mesh.py |
DeviceMesh._get_slice_mesh_layout |
|
31 |
5 |
0 |
|
84% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / device_mesh.py |
DeviceMesh._create_and_cache_sub_mesh |
|
32 |
9 |
0 |
|
72% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / device_mesh.py |
DeviceMesh.get_group |
|
9 |
4 |
0 |
|
56% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / device_mesh.py |
DeviceMesh.get_all_groups |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / device_mesh.py |
DeviceMesh.from_group |
|
29 |
29 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / device_mesh.py |
DeviceMesh.get_local_rank |
|
20 |
4 |
0 |
|
80% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / device_mesh.py |
DeviceMesh.flatten |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / device_mesh.py |
DeviceMesh._get_root_mesh |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / device_mesh.py |
DeviceMesh._validate_concatenate_inputs |
|
20 |
3 |
0 |
|
85% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / device_mesh.py |
DeviceMesh._collect_concatenate_anchor_meshes |
|
16 |
1 |
0 |
|
94% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / device_mesh.py |
DeviceMesh._collect_concatenate_anchor_meshes.add_anchor |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / device_mesh.py |
DeviceMesh._recover_concatenate_mesh_from_anchors |
|
10 |
3 |
0 |
|
70% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / device_mesh.py |
DeviceMesh._validate_concatenate_root_order |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / device_mesh.py |
DeviceMesh._collect_concatenate_metadata |
|
20 |
2 |
0 |
|
90% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / device_mesh.py |
DeviceMesh._build_concatenate_layout |
|
3 |
1 |
0 |
|
67% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / device_mesh.py |
DeviceMesh._set_concatenated_group_state |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / device_mesh.py |
DeviceMesh.concatenate |
|
12 |
2 |
0 |
|
83% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / device_mesh.py |
DeviceMesh._create_flatten_mesh |
|
26 |
7 |
0 |
|
73% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / device_mesh.py |
DeviceMesh._create_unflatten_mesh |
|
19 |
1 |
0 |
|
95% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / device_mesh.py |
DeviceMesh._flatten |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / device_mesh.py |
DeviceMesh._unflatten |
|
11 |
6 |
0 |
|
45% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / device_mesh.py |
DeviceMesh.assert_axis |
|
4 |
2 |
0 |
|
50% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / device_mesh.py |
DeviceMesh.axis_id |
|
4 |
1 |
0 |
|
75% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / device_mesh.py |
DeviceMesh.axis_index |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / device_mesh.py |
DeviceMesh.get_device_num_along_axis |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / device_mesh.py |
DeviceMesh.get_rank_list_along_axis |
|
29 |
2 |
0 |
|
93% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / device_mesh.py |
DeviceMesh.get_global_shape |
|
27 |
2 |
0 |
|
93% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / device_mesh.py |
DeviceMesh._materialize_dim_group |
|
16 |
5 |
0 |
|
69% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / device_mesh.py |
DeviceMesh.get_comm_group_by_axis |
|
18 |
8 |
0 |
|
56% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / device_mesh.py |
DeviceMesh.get_devices_for_axis |
|
31 |
31 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / device_mesh.py |
DeviceMesh.to_hash |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / device_mesh.py |
DeviceMesh.__repr__ |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / device_mesh.py |
DeviceMesh.__str__ |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / device_mesh.py |
DeviceMesh.__deepcopy__ |
|
8 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / device_mesh.py |
_device_mesh_map_key |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / device_mesh.py |
_register_device_mesh |
|
11 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / device_mesh.py |
_create_device_mesh |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / device_mesh.py |
init_device_mesh |
|
28 |
8 |
0 |
|
71% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / device_mesh.py |
(no function) |
|
133 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / dtensor.py |
_device_meshes_are_compatible |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / dtensor.py |
SkipDTensorDispatch.__init__ |
|
10 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / dtensor.py |
SkipDTensorDispatch.__enter__ |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / dtensor.py |
SkipDTensorDispatch.__exit__ |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / dtensor.py |
_is_alias_placements |
|
9 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / dtensor.py |
_build_layout |
|
14 |
1 |
0 |
|
93% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / dtensor.py |
_is_broadcastable |
|
9 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / dtensor.py |
_device_spec |
|
9 |
1 |
0 |
|
89% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / dtensor.py |
DTensor.__init_data__ |
|
21 |
3 |
0 |
|
86% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / dtensor.py |
DTensor.device_mesh |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / dtensor.py |
DTensor.placements |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / dtensor.py |
DTensor.layout |
|
3 |
1 |
0 |
|
67% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / dtensor.py |
DTensor.from_local |
|
18 |
2 |
0 |
|
89% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / dtensor.py |
DTensor.from_local_with_layout |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / dtensor.py |
DTensor._alias_placements |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / dtensor.py |
DTensor._from_converted_local |
|
10 |
2 |
0 |
|
80% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / dtensor.py |
DTensor.to |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / dtensor.py |
DTensor.float |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / dtensor.py |
DTensor.type_as |
|
13 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / dtensor.py |
DTensor._validate_factory_device |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / dtensor.py |
DTensor._new_const_tensor_op |
|
18 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / dtensor.py |
DTensor.new_zeros |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / dtensor.py |
DTensor.new_ones |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / dtensor.py |
DTensor.to_local |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / dtensor.py |
DTensor.tolist |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / dtensor.py |
DTensor.copy_ |
|
14 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / dtensor.py |
DTensor.zero_ |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / dtensor.py |
DTensor.fill_ |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / dtensor.py |
DTensor.shape |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / dtensor.py |
DTensor.size |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / dtensor.py |
DTensor.numel |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / dtensor.py |
DTensor.ndim |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / dtensor.py |
DTensor.dim |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / dtensor.py |
DTensor.local_shape |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / dtensor.py |
DTensor.redistribute |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / dtensor.py |
DTensor.reduce_partial |
|
7 |
7 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / dtensor.py |
DTensor.full_tensor |
|
10 |
1 |
0 |
|
90% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / dtensor.py |
_normalize_shard_dim |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / dtensor.py |
_distribute_tensor_with_communication |
|
23 |
6 |
0 |
|
74% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / dtensor.py |
distribute_tensor |
|
9 |
1 |
0 |
|
89% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / dtensor.py |
_distribute_module_param_source |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / dtensor.py |
_distribute_module_new_parameter |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / dtensor.py |
_distribute_module_set_param |
|
11 |
6 |
0 |
|
45% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / dtensor.py |
_distribute_module_iter_params |
|
5 |
3 |
0 |
|
40% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / dtensor.py |
_distribute_module_iter_buffers |
|
3 |
1 |
0 |
|
67% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / dtensor.py |
_distribute_module_named_modules |
|
5 |
3 |
0 |
|
40% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / dtensor.py |
_distribute_module_named_parameters |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / dtensor.py |
_replicate_submodule_params_buffers |
|
14 |
1 |
0 |
|
93% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / dtensor.py |
_distribute_module_run_partition_and_replicate |
|
7 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / dtensor.py |
_distribute_module_register_input_fn |
|
7 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / dtensor.py |
_distribute_module_register_output_fn |
|
7 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / dtensor.py |
distribute_module |
|
10 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / dtensor.py |
_dtensor_init_helper |
|
20 |
4 |
0 |
|
80% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / dtensor.py |
ones |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / dtensor.py |
empty |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / dtensor.py |
full |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / dtensor.py |
zeros |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / dtensor.py |
rand |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / dtensor.py |
randn |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / dtensor.py |
(no function) |
|
90 |
0 |
1 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / init_weights.py |
init_empty_weights |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / init_weights.py |
init_on_device |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / init_weights.py |
(no function) |
|
7 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / layout.py |
_extract_ragged_shard |
|
8 |
1 |
0 |
|
88% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / layout.py |
_replace_ragged_with_replicate |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / layout.py |
infer_balanced_chunk_range |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / layout.py |
infer_ceil_chunk_range |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / layout.py |
_infer_slice_area_by_rank |
|
25 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / layout.py |
infer_slice_area_by_rank |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / layout.py |
infer_slice_area_by_layout |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / layout.py |
_get_slice_tensor_by_layout |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / layout.py |
_get_slice_tensor_by_layout.get_slice_data |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / layout.py |
_infer_slice_shape_by_layout |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / layout.py |
Layout.__init__ |
|
16 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / layout.py |
Layout.from_device_mesh |
|
15 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / layout.py |
Layout.__call__ |
|
8 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / layout.py |
Layout.__deepcopy__ |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / layout.py |
Layout._process_placement_layout |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / layout.py |
Layout._process_alias_layout |
|
31 |
6 |
0 |
|
81% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / layout.py |
Layout.to_dict |
|
6 |
1 |
0 |
|
83% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / layout.py |
Layout.placement_to_tensor_map |
|
10 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / layout.py |
Layout._handle_zero_dim_placement |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / layout.py |
Layout._build_dim_map_from_placements |
|
17 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / layout.py |
Layout._placement_split_factor |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / layout.py |
Layout._build_order_positions |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / layout.py |
Layout._compute_expected_split_factors |
|
11 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / layout.py |
Layout._get_effective_shard_axes |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / layout.py |
Layout._reorder_dim_map_for_strided_shard |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / layout.py |
Layout._validate_strided_shard_split_factor |
|
11 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / layout.py |
Layout._extract_reduce_op |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / layout.py |
Layout._convert_dim_map_to_tensor_map |
|
9 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / layout.py |
Layout._build_readable_tensor_map |
|
10 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / layout.py |
Layout._build_readable_tensor_map._map_dim |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / layout.py |
Layout.tensor_map_to_placement |
|
25 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / layout.py |
Layout.__setstate__ |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / layout.py |
Layout.mesh |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / layout.py |
Layout.update_mesh |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / layout.py |
Layout.rank_list |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / layout.py |
Layout.rank_list |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / layout.py |
Layout.mesh_shape |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / layout.py |
Layout.alias_name |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / layout.py |
Layout.alias_tensor_map |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / layout.py |
Layout.alias_placements |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / layout.py |
Layout.set_alias_tensor_map |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / layout.py |
Layout.placements |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / layout.py |
Layout.set_placements |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / layout.py |
Layout.tensor_shape |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / layout.py |
Layout.tensor_stride |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / layout.py |
Layout.tensor_dtype |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / layout.py |
Layout.uneven_shard_mesh_dims |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / layout.py |
Layout.has_uneven_shard |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / layout.py |
Layout.set_tensor_meta |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / layout.py |
Layout.normal_placements |
|
3 |
1 |
0 |
|
67% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / layout.py |
Layout.ragged_shard |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / layout.py |
Layout.tensor_map |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / layout.py |
Layout.set_tensor_map |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / layout.py |
Layout.partial |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / layout.py |
Layout.set_partial_by_dev_axis |
|
7 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / layout.py |
Layout.get_partial_by_dev_id |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / layout.py |
Layout.is_dev_axis_apply_shard |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / layout.py |
Layout.is_dev_axis_apply_shard.flatten |
|
6 |
1 |
0 |
|
83% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / layout.py |
Layout.get_dev_axis_apply_shard_axis |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / layout.py |
Layout.reset_partial |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / layout.py |
Layout.is_partial |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / layout.py |
Layout.get_dim_split_num |
|
15 |
1 |
0 |
|
93% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / layout.py |
Layout.get_split_id |
|
22 |
2 |
0 |
|
91% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / layout.py |
Layout.get_global_shape |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / layout.py |
Layout.get_devices_for_axis |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / layout.py |
Layout.get_comm_group_by_axis |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / layout.py |
Layout.repeat_num |
|
13 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / layout.py |
Layout._to_compact_string |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / layout.py |
Layout.compact_str |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / layout.py |
Layout.update_compact_str |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / layout.py |
Layout.to_string |
|
16 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / layout.py |
Layout.__str__ |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / layout.py |
Layout.__repr__ |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / layout.py |
Layout.__eq__ |
|
8 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / layout.py |
(no function) |
|
106 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / parameter_init.py |
init_parameters |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / parameter_init.py |
(no function) |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / placement_types.py |
Placement.__repr__ |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / placement_types.py |
Placement.is_shard |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / placement_types.py |
Placement.is_replicate |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / placement_types.py |
Placement.is_partial |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / placement_types.py |
Placement.is_ragged_shard |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / placement_types.py |
Shard.__init__ |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / placement_types.py |
Shard.dim |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / placement_types.py |
Shard.uneven_shard |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / placement_types.py |
Shard.is_shard |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / placement_types.py |
Shard.__eq__ |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / placement_types.py |
Shard.__hash__ |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / placement_types.py |
Shard.__repr__ |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / placement_types.py |
Shard.__str__ |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / placement_types.py |
Shard.__setstate__ |
|
3 |
1 |
0 |
|
67% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / placement_types.py |
StridedShard.__init__ |
|
4 |
1 |
0 |
|
75% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / placement_types.py |
StridedShard.split_factor |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / placement_types.py |
StridedShard.__eq__ |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / placement_types.py |
StridedShard.__hash__ |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / placement_types.py |
StridedShard.__repr__ |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / placement_types.py |
StridedShard.__str__ |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / placement_types.py |
RaggedShard.__init__ |
|
17 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / placement_types.py |
RaggedShard.dims |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / placement_types.py |
RaggedShard.local_units |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / placement_types.py |
RaggedShard.is_ragged_shard |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / placement_types.py |
RaggedShard.__eq__ |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / placement_types.py |
RaggedShard.__hash__ |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / placement_types.py |
RaggedShard.__repr__ |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / placement_types.py |
RaggedShard.__str__ |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / placement_types.py |
Replicate.is_replicate |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / placement_types.py |
Replicate.__eq__ |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / placement_types.py |
Replicate.__hash__ |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / placement_types.py |
Replicate.__str__ |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / placement_types.py |
Partial.__init__ |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / placement_types.py |
Partial.reduce_op |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / placement_types.py |
Partial.is_partial |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / placement_types.py |
Partial.__eq__ |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / placement_types.py |
Partial.__hash__ |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / placement_types.py |
Partial.__repr__ |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / placement_types.py |
Partial.__str__ |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / placement_types.py |
(no function) |
|
52 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / random.py |
is_rng_supported_mesh |
|
9 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / random.py |
_PhiloxState.__init__ |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / random.py |
_PhiloxState.state |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / random.py |
_PhiloxState.offset |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / random.py |
_PhiloxState.offset |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / random.py |
_PhiloxState.seed |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / random.py |
_PhiloxState.seed |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / random.py |
_RNGStateTracker.__init__ |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / random.py |
_RNGStateTracker.distribute_region_enabled |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / random.py |
_RNGStateTracker.distribute_region_enabled |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / random.py |
_RNGStateTracker._distribute_region |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / random.py |
_RNGStateTracker._manual_seed |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / random.py |
OffsetBasedRNGTracker.__init__ |
|
8 |
8 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / random.py |
OffsetBasedRNGTracker._manual_seed |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / random.py |
OffsetBasedRNGTracker._get_device_state |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / random.py |
OffsetBasedRNGTracker._set_device_state |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / random.py |
OffsetBasedRNGTracker._distribute_region |
|
15 |
15 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / random.py |
OffsetBasedRNGTracker.compute_offset_incr |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / random.py |
OffsetBasedRNGTracker._set_pre_op_offset |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / random.py |
OffsetBasedRNGTracker._set_post_op_offset |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / random.py |
OffsetBasedRNGTracker._calc_shard_linear_idx |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / random.py |
_calc_first_shard_size |
|
8 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / random.py |
_calc_shard_info |
|
30 |
2 |
0 |
|
93% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / random.py |
_calc_shard_linear_idx |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / random.py |
_resolve_device |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / random.py |
_resolve_device.get_device |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / random.py |
manual_seed |
|
9 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / random.py |
local_shard_size_and_offset |
|
10 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / random.py |
fork_rng |
|
19 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / random.py |
(no function) |
|
56 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / redistribute_infer.py |
TensorMap.__init__ |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / redistribute_infer.py |
TensorMap.get_dim_by_idx |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / redistribute_infer.py |
TensorMap.get_index_by_value |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / redistribute_infer.py |
TensorMap.get_index_contain_value |
|
8 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / redistribute_infer.py |
DevMat.__init__ |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / redistribute_infer.py |
DevMat.get_dim_by_reverse_idx |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / redistribute_infer.py |
DevMat._get_combined_size |
|
7 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / redistribute_infer.py |
DevMat._get_devices_along_dim |
|
20 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / redistribute_infer.py |
DevMat.get_devices_along_dim |
|
15 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / redistribute_infer.py |
RedistributionOperatorInfer.__init__ |
|
8 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / redistribute_infer.py |
RedistributionOperatorInfer.insert_operator |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / redistribute_infer.py |
RedistributionOperatorInfer.infer_redistribution_operator |
|
25 |
4 |
0 |
|
84% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / redistribute_infer.py |
RedistributionOperatorInfer._handle_simple_split_case |
|
11 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / redistribute_infer.py |
RedistributionOperatorInfer._handle_tuple_split_case |
|
13 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / redistribute_infer.py |
RedistributionOperatorInfer.infer_split_by_axis |
|
16 |
1 |
0 |
|
94% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / redistribute_infer.py |
RedistributionOperatorInfer._handle_none_dim_permute_case |
|
24 |
4 |
0 |
|
83% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / redistribute_infer.py |
RedistributionOperatorInfer._handle_none_dim_tuple_permute_case |
|
29 |
23 |
0 |
|
21% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / redistribute_infer.py |
RedistributionOperatorInfer._handle_tuple_dim_permute_case |
|
29 |
11 |
0 |
|
62% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / redistribute_infer.py |
RedistributionOperatorInfer.infer_permute_by_axis |
|
16 |
2 |
0 |
|
88% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / redistribute_infer.py |
RedistributionOperatorInfer._handle_tuple_concat_case |
|
25 |
5 |
0 |
|
80% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / redistribute_infer.py |
RedistributionOperatorInfer._handle_simple_concat_case |
|
12 |
3 |
0 |
|
75% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / redistribute_infer.py |
RedistributionOperatorInfer.infer_concat_by_axis |
|
11 |
1 |
0 |
|
91% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / redistribute_infer.py |
RedistributionOperatorInfer.infer_ops_list |
|
29 |
1 |
0 |
|
97% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / redistribute_infer.py |
(no function) |
|
34 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / tensor_redistribution.py |
_construct_layout_tuple_for_transform_operator_list |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / tensor_redistribution.py |
TensorRedistribution.__init__ |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / tensor_redistribution.py |
TensorRedistribution._construct_reshape |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / tensor_redistribution.py |
TensorRedistribution._construct_all_concat |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / tensor_redistribution.py |
TensorRedistribution._construct_strided_slice |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / tensor_redistribution.py |
TensorRedistribution._construct_all_concat_new |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / tensor_redistribution.py |
TensorRedistribution._construct_all_split |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / tensor_redistribution.py |
TensorRedistribution._construct_all_to_all |
|
52 |
13 |
0 |
|
75% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / tensor_redistribution.py |
TensorRedistribution._apply_eazy_redistribute |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / tensor_redistribution.py |
TensorRedistribution._redistribution_without_shape |
|
6 |
1 |
0 |
|
83% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / tensor_redistribution.py |
TensorRedistribution._to_normal_layout |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / tensor_redistribution.py |
TensorRedistribution._redistribute_ragged |
|
23 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / tensor_redistribution.py |
TensorRedistribution.redistribution |
|
14 |
4 |
0 |
|
71% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / tensor_redistribution.py |
TensorRedistribution._redistribution_normal |
|
29 |
9 |
0 |
|
69% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / tensor_redistribution.py |
TensorRedistribution.ragged_to_normal |
|
12 |
2 |
0 |
|
83% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / tensor_redistribution.py |
TensorRedistribution.normal_to_ragged |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / tensor_redistribution.py |
TensorRedistribution.ragged_to_ragged_via_replicate |
|
5 |
1 |
0 |
|
80% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / tensor_redistribution.py |
TensorRedistribution.ragged_to_ragged |
|
16 |
4 |
0 |
|
75% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / tensor_redistribution.py |
TensorRedistribution._infer_transform_operator_list |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / tensor_redistribution.py |
TensorRedistribution._allreduce_along_dev_dim |
|
17 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / tensor_redistribution.py |
TensorRedistribution._reduce_scatter_along_dev_dim_with_axis |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / tensor_redistribution.py |
TensorRedistribution.reduce_partial |
|
43 |
34 |
0 |
|
21% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / tensor_redistribution.py |
TensorRedistribution.reduce_partial._reduce_pair_sort_key |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / tensor_redistribution.py |
(no function) |
|
45 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / expert_parallel / __init__.py |
(no function) |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / expert_parallel / expert_parallel.py |
AsyncHandle.__init__ |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / expert_parallel / expert_parallel.py |
AsyncHandle.wait |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / expert_parallel / expert_parallel.py |
_AsyncA2ALazyBwd.forward |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / expert_parallel / expert_parallel.py |
_AsyncA2ALazyBwd.backward |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / expert_parallel / expert_parallel.py |
_TorchContiguousGrad.forward |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / expert_parallel / expert_parallel.py |
_TorchContiguousGrad.backward |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / expert_parallel / expert_parallel.py |
_ensure_contiguous |
|
5 |
1 |
0 |
|
80% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / expert_parallel / expert_parallel.py |
differentiable_all_to_all_single |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / expert_parallel / expert_parallel.py |
differentiable_all_to_all_single_async |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / expert_parallel / expert_parallel.py |
differentiable_all_gather_concat |
|
8 |
4 |
0 |
|
50% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / expert_parallel / expert_parallel.py |
differentiable_reduce_scatter |
|
8 |
1 |
0 |
|
88% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / expert_parallel / expert_parallel.py |
wait_async_tensor |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / expert_parallel / expert_parallel.py |
_generate_permute_indices |
|
15 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / expert_parallel / expert_parallel.py |
_permute |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / expert_parallel / expert_parallel.py |
_unpermute |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / expert_parallel / expert_parallel.py |
_get_deredundency_mesh_info |
|
13 |
2 |
0 |
|
85% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / expert_parallel / expert_parallel.py |
_generate_deredundency_dispatch_indices |
|
20 |
1 |
0 |
|
95% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / expert_parallel / expert_parallel.py |
_scale_by_router_coeff |
|
8 |
1 |
0 |
|
88% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / expert_parallel / expert_parallel.py |
_scatter_add_first_dim |
|
9 |
4 |
0 |
|
56% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / expert_parallel / expert_parallel.py |
_DeredundencyCombineHandle.__init__ |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / expert_parallel / expert_parallel.py |
_DeredundencyCombineHandle.wait |
|
8 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / expert_parallel / expert_parallel.py |
BaseExpertParallel.apply |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / expert_parallel / expert_parallel.py |
BaseExpertParallel._partition_fn |
|
0 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / expert_parallel / expert_parallel.py |
BaseExpertParallel._token_dispatch |
|
0 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / expert_parallel / expert_parallel.py |
BaseExpertParallel._token_combine |
|
0 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / expert_parallel / expert_parallel.py |
AllToAllTokenDispatcher.dispatch |
|
19 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / expert_parallel / expert_parallel.py |
AllToAllTokenDispatcher._exchange_and_reorder_probs |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / expert_parallel / expert_parallel.py |
AllToAllTokenDispatcher.combine |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / expert_parallel / expert_parallel.py |
AllToAllTokenDispatcher.combine_start |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / expert_parallel / expert_parallel.py |
AllToAllTokenDispatcher.combine_wait |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / expert_parallel / expert_parallel.py |
DeredundencyTokenDispatcher._oep_gather_for_dispatch |
|
16 |
2 |
0 |
|
88% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / expert_parallel / expert_parallel.py |
DeredundencyTokenDispatcher.dispatch |
|
22 |
1 |
0 |
|
95% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / expert_parallel / expert_parallel.py |
DeredundencyTokenDispatcher._iep_exchange_counts |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / expert_parallel / expert_parallel.py |
DeredundencyTokenDispatcher.combine |
|
10 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / expert_parallel / expert_parallel.py |
DeredundencyTokenDispatcher._validate_combine_mesh |
|
2 |
1 |
0 |
|
50% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / expert_parallel / expert_parallel.py |
DeredundencyTokenDispatcher.combine_start |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / expert_parallel / expert_parallel.py |
DeredundencyTokenDispatcher.combine_wait |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / expert_parallel / expert_parallel.py |
_resolve_token_dispatcher |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / expert_parallel / expert_parallel.py |
_get_flattened_ep_mesh |
|
8 |
8 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / expert_parallel / expert_parallel.py |
ExpertParallel.__init__ |
|
7 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / expert_parallel / expert_parallel.py |
ExpertParallel._token_dispatch |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / expert_parallel / expert_parallel.py |
ExpertParallel._token_combine |
|
8 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / expert_parallel / expert_parallel.py |
ExpertParallel._partition_mesh |
|
3 |
1 |
0 |
|
67% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / expert_parallel / expert_parallel.py |
ExpertParallel._partition_fn |
|
10 |
1 |
0 |
|
90% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / expert_parallel / expert_parallel.py |
TensorParallel.apply |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / expert_parallel / expert_parallel.py |
TensorParallel._partition_fn |
|
10 |
1 |
0 |
|
90% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / expert_parallel / expert_parallel.py |
ExpertTensorParallel.__init__ |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / expert_parallel / expert_parallel.py |
ExpertTensorParallel._dispatch_mesh |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / expert_parallel / expert_parallel.py |
ExpertTensorParallel._token_dispatch |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / expert_parallel / expert_parallel.py |
ExpertTensorParallel._token_combine |
|
9 |
4 |
0 |
|
56% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / expert_parallel / expert_parallel.py |
ExpertTensorParallel._partition_fn |
|
10 |
1 |
0 |
|
90% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / expert_parallel / expert_parallel.py |
(no function) |
|
120 |
1 |
0 |
|
99% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / __init__.py |
(no function) |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / api.py |
_resolve_comm_fusion_zero_copy_default |
|
9 |
1 |
0 |
|
89% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / api.py |
_check_strict_keys |
|
10 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / api.py |
_resolve_local_tensor |
|
11 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / api.py |
_UnshardHandle.__init__ |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / api.py |
_UnshardHandle.wait |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / api.py |
HSDPModule.__init__ |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / api.py |
HSDPModule.hsdp_init |
|
8 |
8 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / api.py |
HSDPModule.set_requires_gradient_sync |
|
7 |
2 |
0 |
|
71% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / api.py |
HSDPModule.zero_grad |
|
7 |
2 |
0 |
|
71% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / api.py |
HSDPModule.set_modules_to_forward_prefetch |
|
8 |
2 |
0 |
|
75% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / api.py |
HSDPModule.set_modules_to_backward_prefetch |
|
8 |
3 |
0 |
|
62% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / api.py |
HSDPModule.reshard |
|
5 |
1 |
0 |
|
80% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / api.py |
HSDPModule.unshard |
|
10 |
2 |
0 |
|
80% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / api.py |
HSDPModule.load_state_dict |
|
24 |
24 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / api.py |
HSDPModule.set_is_last_backward |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / api.py |
HSDPModule.reset_iter_state |
|
7 |
1 |
0 |
|
86% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / api.py |
HSDPModule.set_requires_all_reduce |
|
8 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / api.py |
HSDPModule.set_reshard_after_forward |
|
8 |
1 |
0 |
|
88% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / api.py |
HSDPModule.set_reshard_after_backward |
|
8 |
1 |
0 |
|
88% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / api.py |
HSDPModule.set_reduce_op_type |
|
9 |
1 |
0 |
|
89% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / api.py |
HSDPModule.set_gradient_scaling_factor |
|
11 |
11 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / api.py |
_extend_module_with_hsdp_interface |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / api.py |
_get_root_modules |
|
12 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / api.py |
_get_root_modules._get_submodules |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / api.py |
_check_module_valid |
|
7 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / api.py |
_validate_module_for_fully_shard |
|
9 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / api.py |
_validate_hsdp_shard_size |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / api.py |
_validate_hsdp_threshold |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / api.py |
_validate_hsdp_optimizer_level |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / api.py |
_validate_hsdp_reduce_dtype |
|
8 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / api.py |
_check_hsdp_input_valid |
|
15 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / api.py |
_get_device_from_mesh |
|
13 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / api.py |
_normalize_replicate_params |
|
7 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / api.py |
_get_modules_parameters |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / api.py |
_validate_managed_params_source_shard_infos |
|
19 |
1 |
0 |
|
95% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / api.py |
fully_shard |
|
33 |
1 |
0 |
|
97% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / api.py |
get_model_state_dict |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / api.py |
set_model_state_dict |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / api.py |
hsdp_sync_stream |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / api.py |
(no function) |
|
54 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / hsdp_param.py |
_build_group_info_from_rank_list |
|
10 |
0 |
2 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / hsdp_param.py |
_build_group_info_from_process_group |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / hsdp_param.py |
HSDPParamV2.__repr__ |
|
7 |
2 |
0 |
|
71% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / hsdp_param.py |
HSDPParamV2.__init__ |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / hsdp_param.py |
HSDPParamV2._init_sharded_param |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / hsdp_param.py |
HSDPParamV2.init_dtype_attrs |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / hsdp_param.py |
HSDPParamV2.init_unsharded_param_buffers |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / hsdp_param.py |
HSDPParamV2.init_unsharded_param |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / hsdp_param.py |
HSDPParamV2.to_sharded |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / hsdp_param.py |
HSDPParamV2.to_unsharded |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / hsdp_param.py |
HSDPParamV2.to_sharded_dtensor |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / hsdp_param.py |
HSDPParamV2.to_accumulated_grad_if_needed |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / hsdp_param.py |
HSDPParamV2.accumulate_unsharded_grad_if_needed |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / hsdp_param.py |
HSDPParamV2.alloc_unsharded_param_buffers |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / hsdp_param.py |
HSDPParamV2.free_unsharded_param |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / hsdp_param.py |
HSDPParamV2.all_gather_inputs |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / hsdp_param.py |
HSDPParamV2.unsharded_param |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / hsdp_param.py |
HSDPParamV2.unsharded_grad_data |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / hsdp_param.py |
HSDPParamV2.unsharded_accumulated_grad_data |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / hsdp_param.py |
HSDPParamV2._sharded_local_tensor |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / hsdp_param.py |
HSDPParamV2._get_unsharded_param_data |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / hsdp_param.py |
HSDPParamV2.unshard |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / hsdp_param.py |
HSDPParamV2.wait_for_unshard |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / hsdp_param.py |
HSDPParamV2.shard |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / hsdp_param.py |
HSDPParamV2.reduce_scatter_grad |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / hsdp_param.py |
HSDPParamV2.all_reduce_grad |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / hsdp_param.py |
HSDPParamV2._resolve_process_group_name |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / hsdp_param.py |
HSDPParamV2._get_base_spmd_placements |
|
9 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / hsdp_param.py |
HSDPParamV2._get_data_parallel_shard_placement |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / hsdp_param.py |
HSDPParamV2._apply_data_parallel_placements |
|
9 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / hsdp_param.py |
HSDPParamV2._init_group_infos |
|
8 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / hsdp_param.py |
HSDPParamV2._build_layout_driven_group_info |
|
29 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / hsdp_param.py |
HSDPParamV2._normalize_unsharded_grad_to_local |
|
11 |
1 |
0 |
|
91% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / hsdp_param.py |
(no function) |
|
47 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / hsdp_scheduler.py |
ParamGroupCommCtx.__init__ |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / hsdp_scheduler.py |
HSDPSchedulerContext.__init__ |
|
12 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / hsdp_scheduler.py |
HSDPSchedulerV2.__init__ |
|
22 |
22 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / hsdp_scheduler.py |
HSDPSchedulerV2._init_platform |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / hsdp_scheduler.py |
HSDPSchedulerV2._new_cell_state |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / hsdp_scheduler.py |
HSDPSchedulerV2._register_hooks |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / hsdp_scheduler.py |
HSDPSchedulerV2._register_forward_backward_hooks |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / hsdp_scheduler.py |
HSDPSchedulerV2._get_managed_params |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / hsdp_scheduler.py |
HSDPSchedulerV2.set_reshard_after_forward |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / hsdp_scheduler.py |
HSDPSchedulerV2.set_reshard_after_backward |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / hsdp_scheduler.py |
HSDPSchedulerV2.set_requires_all_reduce |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / hsdp_scheduler.py |
HSDPSchedulerV2.reset_iter_state |
|
13 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / hsdp_scheduler.py |
HSDPSchedulerV2.set_requires_grad_sync |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / hsdp_scheduler.py |
HSDPSchedulerV2._hsdp_forward_pre_hook |
|
45 |
1 |
0 |
|
98% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / hsdp_scheduler.py |
HSDPSchedulerV2._lazy_init_all_states |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / hsdp_scheduler.py |
HSDPSchedulerV2._init_params_fqn |
|
22 |
3 |
0 |
|
86% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / hsdp_scheduler.py |
HSDPSchedulerV2._hsdp_forward_hook |
|
12 |
2 |
0 |
|
83% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / hsdp_scheduler.py |
HSDPSchedulerV2._hsdp_backward_pre_hook |
|
11 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / hsdp_scheduler.py |
HSDPSchedulerV2._hsdp_backward_hook |
|
7 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / hsdp_scheduler.py |
HSDPSchedulerV2._grouped_forward_pre_hook_skip |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / hsdp_scheduler.py |
HSDPSchedulerV2._grouped_forward_post_hook_skip |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / hsdp_scheduler.py |
HSDPSchedulerV2._grouped_forward_pre_hook |
|
7 |
1 |
0 |
|
86% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / hsdp_scheduler.py |
HSDPSchedulerV2._make_grouped_forward_post_hook |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / hsdp_scheduler.py |
HSDPSchedulerV2._make_grouped_forward_post_hook.grouped_post_hook |
|
7 |
1 |
0 |
|
86% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / hsdp_scheduler.py |
HSDPSchedulerV2.set_forward_prefetch_cells |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / hsdp_scheduler.py |
HSDPSchedulerV2.set_backward_prefetch_cells |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / hsdp_scheduler.py |
HSDPSchedulerV2._disable_forward_prefetch_for_recompute |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / hsdp_scheduler.py |
HSDPSchedulerV2._restore_forward_prefetch_after_recompute |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / hsdp_scheduler.py |
(no function) |
|
42 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / hsdp_state.py |
HSDPState.__init__ |
|
23 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / hsdp_state.py |
HSDPState.__repr__ |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / hsdp_state.py |
HSDPState._init_hsdp_params |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / hsdp_state.py |
HSDPState._move_states_to_device |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / hsdp_state.py |
HSDPState.set_reduce_op_type |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / hsdp_state.py |
HSDPState.shard |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / hsdp_state.py |
HSDPState.unshard |
|
9 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / hsdp_state.py |
HSDPState.prefetch |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / hsdp_state.py |
HSDPState.wait_for_unshard |
|
8 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / hsdp_state.py |
HSDPState.set_gradient_scaling_factor |
|
4 |
2 |
0 |
|
50% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / hsdp_state.py |
HSDPState.set_requires_all_reduce |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / hsdp_state.py |
(no function) |
|
21 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / hsdp_utils.py |
_named_parameters_with_duplicates |
|
10 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / hsdp_utils.py |
_named_parameters_with_duplicates.get_named_parameters |
|
3 |
1 |
0 |
|
67% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / hsdp_utils.py |
_get_param_module_infos |
|
14 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / hsdp_utils.py |
_get_param_module_infos.get_named_modules |
|
3 |
1 |
0 |
|
67% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / hsdp_utils.py |
get_managed_modules_parameters |
|
12 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / hsdp_utils.py |
infer_fully_shard_param_mode |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / hsdp_utils.py |
unwrap_dtensor_param |
|
8 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / hsdp_utils.py |
is_dtensor_managed_param |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / hsdp_utils.py |
get_dtensor_managed_mesh |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / hsdp_utils.py |
get_rank_list_for_axes |
|
19 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / hsdp_utils.py |
get_split_rank_lists_for_axes |
|
20 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / hsdp_utils.py |
get_hsdp_state |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / hsdp_utils.py |
apply_gradient_scaling_factor |
|
6 |
1 |
0 |
|
83% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / hsdp_utils.py |
(no function) |
|
43 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / utils.py |
DataParallelMeshInfo.__post_init__ |
|
2 |
1 |
0 |
|
50% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / utils.py |
FSDPMeshInfo.__post_init__ |
|
6 |
1 |
0 |
|
83% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / utils.py |
DDPMeshInfo.__post_init__ |
|
6 |
1 |
0 |
|
83% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / utils.py |
HSDPMeshInfo.__post_init__ |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / utils.py |
(no function) |
|
44 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / moe_utils.py |
sync_and_update_expert_bias |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / moe_utils.py |
_get_moe_layers |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / moe_utils.py |
MoEMonitorCallback.__init__ |
|
8 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / moe_utils.py |
MoEMonitorCallback.on_step_end |
|
10 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / moe_utils.py |
MoEMonitorCallback.register |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / moe_utils.py |
MoEMonitorCallback.remove |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / moe_utils.py |
(no function) |
|
10 |
0 |
2 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / __init__.py |
__getattr__ |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / __init__.py |
__dir__ |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / __init__.py |
_load_torch_optimizer_runtime |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / __init__.py |
_effective_optimizer_config |
|
7 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / __init__.py |
_filter_optimizer_config |
|
8 |
1 |
0 |
|
88% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / __init__.py |
_build_configured_optimizer |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / __init__.py |
get_hyper_optimizer |
|
11 |
1 |
0 |
|
91% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / __init__.py |
(no function) |
|
16 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / adamw.py |
adamw |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / adamw.py |
AdamW.__init__ |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / adamw.py |
AdamW.__setstate__ |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / adamw.py |
AdamW.__str__ |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / adamw.py |
AdamW.step |
|
34 |
6 |
0 |
|
82% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / adamw.py |
(no function) |
|
9 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / dtensor_compat.py |
_invalidate_lazy_cache |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / dtensor_compat.py |
detect_dtensor_backend |
|
15 |
6 |
0 |
|
60% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / dtensor_compat.py |
_extract_first_param |
|
7 |
2 |
0 |
|
71% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / dtensor_compat.py |
get_dtensor_cls |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / dtensor_compat.py |
get_device_mesh_cls |
|
5 |
2 |
0 |
|
60% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / dtensor_compat.py |
get_shard_cls |
|
5 |
2 |
0 |
|
60% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / dtensor_compat.py |
get_replicate_cls |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / dtensor_compat.py |
get_strided_shard_cls |
|
4 |
1 |
0 |
|
75% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / dtensor_compat.py |
_import_hyper_dtensor |
|
5 |
2 |
0 |
|
60% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / dtensor_compat.py |
_resolve_dtensor_union |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / dtensor_compat.py |
to_local_if_dtensor |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / dtensor_compat.py |
__getattr__ |
|
8 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / dtensor_compat.py |
__dir__ |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / dtensor_compat.py |
(no function) |
|
23 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / lr_scheduler.py |
get_constant_schedule_with_warmup |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / lr_scheduler.py |
get_constant_schedule_with_warmup._lr_lambda |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / lr_scheduler.py |
get_linear_schedule_with_warmup |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / lr_scheduler.py |
get_linear_schedule_with_warmup._lr_lambda |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / lr_scheduler.py |
get_cosine_schedule_with_warmup |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / lr_scheduler.py |
get_cosine_schedule_with_warmup.lr_lambda |
|
12 |
12 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / lr_scheduler.py |
LRSchedulersContainer.__init__ |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / lr_scheduler.py |
LRSchedulersContainer.__iter__ |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / lr_scheduler.py |
LRSchedulersContainer.__len__ |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / lr_scheduler.py |
LRSchedulersContainer.step |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / lr_scheduler.py |
LRSchedulersContainer.get_last_lr |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / lr_scheduler.py |
LRSchedulersContainer.state_dict |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / lr_scheduler.py |
LRSchedulersContainer.load_state_dict |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / lr_scheduler.py |
(no function) |
|
18 |
0 |
2 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / muon.py |
zeropower_via_newtonschulz5 |
|
32 |
10 |
0 |
|
69% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / muon.py |
compute_muon_slice_scale |
|
7 |
2 |
0 |
|
71% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / muon.py |
Muon.__init__ |
|
19 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / muon.py |
Muon.reset_optimizer_parameters |
|
11 |
2 |
0 |
|
82% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / muon.py |
Muon._validate_ns_coefficients |
|
17 |
2 |
0 |
|
88% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / muon.py |
Muon.__str__ |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / muon.py |
Muon.step |
|
51 |
21 |
0 |
|
59% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / muon.py |
Muon._run_post_update_fn |
|
7 |
7 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / muon.py |
Muon._classify_parameters_for_step |
|
25 |
8 |
0 |
|
68% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / muon.py |
Muon._update_muon_momentum |
|
40 |
12 |
0 |
|
70% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / muon.py |
Muon._process_unshard_params |
|
16 |
1 |
0 |
|
94% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / muon.py |
Muon._gather_and_compute_shard_updates |
|
35 |
35 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / muon.py |
Muon._process_shard_params |
|
17 |
17 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / muon.py |
Muon._group_by_shape |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / muon.py |
Muon._shape_to_core_shape |
|
7 |
5 |
0 |
|
29% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / muon.py |
Muon._reshape_ns_input |
|
14 |
11 |
0 |
|
21% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / muon.py |
Muon._prepare_ns_transform |
|
10 |
4 |
0 |
|
60% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / muon.py |
Muon._prepare_ns_transform.restore_view_updates |
|
4 |
1 |
0 |
|
75% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / muon.py |
Muon._validate_ns_transform |
|
15 |
15 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / muon.py |
Muon._compute_batched_ns_outputs_for_tensors |
|
47 |
14 |
0 |
|
70% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / muon.py |
Muon._split_into_memory_safe_batches |
|
16 |
3 |
0 |
|
81% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / muon.py |
Muon._compute_batched_ns_updates |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / muon.py |
Muon._compute_batched_ns_updates_fast |
|
33 |
3 |
0 |
|
91% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / muon.py |
Muon._compute_batched_ns_updates_with_transform |
|
47 |
4 |
0 |
|
91% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / muon.py |
Muon._fused_broadcast_and_apply |
|
51 |
51 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / muon.py |
Muon._build_param_shard_metadata |
|
9 |
6 |
0 |
|
33% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / muon.py |
Muon._get_shard_info |
|
7 |
7 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / muon.py |
Muon._batched_relay_broadcast |
|
15 |
15 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / muon.py |
(no function) |
|
61 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / muon_shard.py |
make_param_shard_meta |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / muon_shard.py |
_get_mesh_shape |
|
7 |
7 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / muon_shard.py |
_get_local_shape_kinds |
|
26 |
26 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / muon_shard.py |
_debug_param_shard_metadata |
|
12 |
12 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / muon_shard.py |
_allgather_shapes_for_group |
|
7 |
7 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / muon_shard.py |
build_param_shard_metadata_for_group |
|
31 |
31 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / muon_shard.py |
build_pad_ns_inputs |
|
20 |
20 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / muon_shard.py |
chunk_update_by_layout |
|
21 |
21 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / muon_shard.py |
_get_or_alloc_buffer |
|
11 |
11 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / muon_shard.py |
_early_return_tensors |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / muon_shard.py |
_prepare_gather_inputs |
|
23 |
23 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / muon_shard.py |
_pack_and_allgather |
|
10 |
10 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / muon_shard.py |
_unpack_gathered_results |
|
24 |
24 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / muon_shard.py |
fused_allgather_dtensor_params |
|
35 |
35 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / muon_shard.py |
(no function) |
|
31 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / optimizer.py |
ChainedOptimizer.__init__ |
|
9 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / optimizer.py |
ChainedOptimizer.__iter__ |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / optimizer.py |
ChainedOptimizer._rebind_param_attrs |
|
18 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / optimizer.py |
ChainedOptimizer.reset_optimizer_parameters |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / optimizer.py |
ChainedOptimizer._optimizer_parameter_view |
|
17 |
1 |
0 |
|
94% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / optimizer.py |
ChainedOptimizer.__str__ |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / optimizer.py |
ChainedOptimizer.step |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / optimizer.py |
ChainedOptimizer.zero_grad |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / optimizer.py |
ChainedOptimizer.optimizer |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / optimizer.py |
ChainedOptimizer.defaults |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / optimizer.py |
ChainedOptimizer.state_dict |
|
12 |
1 |
0 |
|
92% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / optimizer.py |
ChainedOptimizer._get_param_groups |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / optimizer.py |
ChainedOptimizer._set_param_groups |
|
9 |
9 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / optimizer.py |
ChainedOptimizer.load_state_dict |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / optimizer.py |
ChainedOptimizer._synchronize_steps |
|
15 |
1 |
0 |
|
93% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / optimizer.py |
BaseDistributedOptimizer.__init__ |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / optimizer.py |
BaseDistributedOptimizer._group_dtensor_by_mesh |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / optimizer.py |
BaseDistributedOptimizer.reset_optimizer_parameters |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / optimizer.py |
BaseDistributedOptimizer._auto_deduce_replica_count |
|
24 |
17 |
0 |
|
29% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / optimizer.py |
BaseDistributedOptimizer._split_replicate_groups |
|
24 |
22 |
0 |
|
8% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / optimizer.py |
BaseDistributedOptimizer._get_or_create_sub_pg |
|
37 |
37 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / optimizer.py |
BaseDistributedOptimizer._build_hsdp_batch |
|
28 |
16 |
0 |
|
43% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / optimizer.py |
BaseDistributedOptimizer._build_sub_batch_assignment |
|
20 |
20 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / optimizer.py |
BaseDistributedOptimizer._build_param_broadcast_info |
|
14 |
7 |
0 |
|
50% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / optimizer.py |
BaseDistributedOptimizer._broadcast_replicate_params_after_step |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / optimizer.py |
BaseDistributedOptimizer._broadcast_state_fused_for_ckpt |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / optimizer.py |
BaseDistributedOptimizer._collect_broadcast_tensors |
|
16 |
13 |
0 |
|
19% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / optimizer.py |
BaseDistributedOptimizer._compute_broadcast_batches |
|
15 |
15 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / optimizer.py |
BaseDistributedOptimizer._hierarchical_broadcast_buffer |
|
13 |
13 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / optimizer.py |
BaseDistributedOptimizer._hierarchical_broadcast_buffer_async |
|
21 |
4 |
0 |
|
81% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / optimizer.py |
BaseDistributedOptimizer._broadcast_op_fused |
|
29 |
25 |
0 |
|
14% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / optimizer.py |
BaseDistributedOptimizer.cleanup_synced_state |
|
13 |
13 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / optimizer.py |
AsyncReplicateBroadcaster.__init__ |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / optimizer.py |
AsyncReplicateBroadcaster.flush_group |
|
16 |
16 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / optimizer.py |
AsyncReplicateBroadcaster._flush_key |
|
38 |
38 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / optimizer.py |
AsyncReplicateBroadcaster._wait_and_release_oldest |
|
7 |
7 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / optimizer.py |
AsyncReplicateBroadcaster.wait_all |
|
2 |
1 |
0 |
|
50% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / optimizer.py |
(no function) |
|
59 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / sharding_category.py |
ParamLayoutSpec.shard_mesh_dims |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / sharding_category.py |
ParamLayoutSpec.shard_tensor_dims |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / sharding_category.py |
ParamLayoutSpec.is_last2d_sharded |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / sharding_category.py |
CommDomainKey.has_replicate_redundancy |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / sharding_category.py |
CommDomainKey.has_shard_group |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / sharding_category.py |
HSDPCommGroup.params |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / sharding_category.py |
HSDPCommGroup.add_param |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / sharding_category.py |
HSDPCommGroup.__len__ |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / sharding_category.py |
_normalize_tensor_dim |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / sharding_category.py |
extract_param_shard_spec |
|
24 |
24 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / sharding_category.py |
build_comm_domain_key |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / sharding_category.py |
group_parameters_for_hsdp |
|
16 |
9 |
0 |
|
44% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / sharding_category.py |
HSDPGroupAssignment.owned_params |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / sharding_category.py |
HSDPGroupAssignment.all_params |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / sharding_category.py |
HSDPGroupAssignment.is_replicated |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / sharding_category.py |
HSDPGroupAssignment.owner_rank_coord |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / sharding_category.py |
HSDPGroupAssignment.is_owned |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / sharding_category.py |
HSDPGroupAssignment._format_device_mesh |
|
8 |
8 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / sharding_category.py |
HSDPGroupAssignment.__str__ |
|
7 |
7 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / sharding_category.py |
get_multi_dim_logical_info |
|
8 |
8 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / sharding_category.py |
build_owner_by_size |
|
15 |
15 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / sharding_category.py |
select_owned_records |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / sharding_category.py |
(no function) |
|
101 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / swap_optimizer.py |
SwapOptimizerConfig.__post_init__ |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / swap_optimizer.py |
SwapOptimizer.__new__ |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / swap_optimizer.py |
swap_optimizer |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / swap_optimizer.py |
is_swap_optimizer |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / swap_optimizer.py |
(no function) |
|
17 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / swap_optimizer_base.py |
SwapSlot.bind_tensor |
|
7 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / swap_optimizer_base.py |
SwapSlot.checkpoint_tensor |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / swap_optimizer_base.py |
PipelineSwapRuntime.__init__ |
|
11 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / swap_optimizer_base.py |
PipelineSwapRuntime.packed_enabled |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / swap_optimizer_base.py |
PipelineSwapRuntime.partition |
|
26 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / swap_optimizer_base.py |
PipelineSwapRuntime.run_pipeline |
|
20 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / swap_optimizer_base.py |
PipelineSwapRuntime._run_packed_pipeline |
|
25 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / swap_optimizer_base.py |
PipelineSwapRuntime.release_packed_step_results |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / swap_optimizer_base.py |
PipelineSwapRuntime.refresh_swappable_slots |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / swap_optimizer_base.py |
PipelineSwapRuntime.synchronize_cpu_mirrors |
|
19 |
2 |
0 |
|
89% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / swap_optimizer_base.py |
PipelineSwapRuntime.prefetch |
|
18 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / swap_optimizer_base.py |
PipelineSwapRuntime.wait_prefetch |
|
10 |
1 |
0 |
|
90% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / swap_optimizer_base.py |
PipelineSwapRuntime.offload |
|
4 |
1 |
0 |
|
75% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / swap_optimizer_base.py |
PipelineSwapRuntime.offload_initial_slots |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / swap_optimizer_base.py |
PipelineSwapRuntime._enqueue_offload_slots |
|
10 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / swap_optimizer_base.py |
PipelineSwapRuntime.wait_offload |
|
4 |
1 |
0 |
|
75% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / swap_optimizer_base.py |
PipelineSwapRuntime._wait_offload_slots |
|
6 |
1 |
0 |
|
83% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / swap_optimizer_base.py |
PipelineSwapRuntime._unit_cost |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / swap_optimizer_base.py |
PipelineSwapRuntime._get_copy_stream |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / swap_optimizer_base.py |
PipelineSwapRuntime._record_current_stream_event |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / swap_optimizer_base.py |
PipelineSwapRuntime.populate_slot_metadata |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / swap_optimizer_base.py |
PipelineSwapRuntime.is_packable_template |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / swap_optimizer_base.py |
PipelineSwapRuntime.is_distributed_tensor |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / swap_optimizer_base.py |
PipelineSwapRuntime.validate_packed_devices |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / swap_optimizer_base.py |
PipelineSwapRuntime.prepare_packed_host |
|
34 |
1 |
0 |
|
97% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / swap_optimizer_base.py |
PipelineSwapRuntime.is_swappable_tensor |
|
21 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / swap_optimizer_base.py |
PipelineSwapRuntime.storage_nbytes |
|
7 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / swap_optimizer_base.py |
PipelineSwapRuntime.make_cpu_tensor |
|
10 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / swap_optimizer_base.py |
PipelineSwapRuntime.make_zero_cpu_tensor_like |
|
9 |
2 |
0 |
|
78% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / swap_optimizer_base.py |
PipelineSwapRuntime.make_device_tensor_like |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / swap_optimizer_base.py |
PipelineSwapRuntime.make_empty_device_tensor_like |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / swap_optimizer_base.py |
PipelineSwapRuntime.copy_to_device |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / swap_optimizer_base.py |
PipelineSwapRuntime.wait_prefetch_slot |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / swap_optimizer_base.py |
PipelineSwapRuntime.copy_to_cpu |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / swap_optimizer_base.py |
PipelineSwapRuntime.wait_offload_slot |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / swap_optimizer_base.py |
PipelineSwapRuntime.restore_device_storage |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / swap_optimizer_base.py |
PipelineSwapRuntime.release_device_storage |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / swap_optimizer_base.py |
PipelineSwapRuntime.device_handle |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / swap_optimizer_base.py |
PipelineSwapRuntime.current_stream |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / swap_optimizer_base.py |
PipelineSwapRuntime.new_stream |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / swap_optimizer_base.py |
PipelineSwapRuntime.stream_context |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / swap_optimizer_base.py |
PipelineSwapRuntime.record_event |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / swap_optimizer_base.py |
PipelineSwapRuntime.wait_event |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / swap_optimizer_base.py |
PipelineSwapRuntime.supports_packed_pipeline |
|
10 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / swap_optimizer_base.py |
PipelineSwapRuntime._storage_tensor |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / swap_optimizer_base.py |
PipelineSwapRuntime.begin_packed_step |
|
31 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / swap_optimizer_base.py |
PipelineSwapRuntime.enqueue_packed_prefetch |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / swap_optimizer_base.py |
PipelineSwapRuntime.wait_packed_prefetch |
|
5 |
1 |
0 |
|
80% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / swap_optimizer_base.py |
PipelineSwapRuntime.activate_packed_batch |
|
14 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / swap_optimizer_base.py |
PipelineSwapRuntime.enqueue_packed_offload_prefetch |
|
12 |
12 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / swap_optimizer_base.py |
PipelineSwapRuntime.wait_packed_offload |
|
4 |
1 |
0 |
|
75% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / swap_optimizer_base.py |
PipelineSwapRuntime.finish_packed_offload |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / swap_optimizer_base.py |
PipelineSwapRuntime.end_packed_step |
|
26 |
3 |
0 |
|
88% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / swap_optimizer_base.py |
PipelineSwapRuntime._build_packed_batch_plan |
|
19 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / swap_optimizer_base.py |
PipelineSwapRuntime._materialize_staging_arena |
|
16 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / swap_optimizer_base.py |
PipelineSwapRuntime._drop_packed_views |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / swap_optimizer_base.py |
PipelineSwapRuntime._copy_packed_to_device |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / swap_optimizer_base.py |
PipelineSwapRuntime._copy_packed_to_host |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / swap_optimizer_base.py |
PipelineSwapRuntime._require_staging_arena |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / swap_optimizer_base.py |
PipelineSwapRuntime._align_bytes |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / swap_optimizer_base.py |
OptimizerSwapAdapter.__init__ |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / swap_optimizer_base.py |
OptimizerSwapAdapter.matches |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / swap_optimizer_base.py |
OptimizerSwapAdapter.validate |
|
9 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / swap_optimizer_base.py |
OptimizerSwapAdapter.prepare_step |
|
19 |
1 |
0 |
|
95% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / swap_optimizer_base.py |
OptimizerSwapAdapter._prepare_packed_step |
|
28 |
3 |
0 |
|
89% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / swap_optimizer_base.py |
OptimizerSwapAdapter.iter_update_units |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / swap_optimizer_base.py |
OptimizerSwapAdapter.initial_slots |
|
7 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / swap_optimizer_base.py |
OptimizerSwapAdapter.step_batch |
|
39 |
39 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / swap_optimizer_base.py |
OptimizerSwapAdapter.finish_step |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / swap_optimizer_base.py |
OptimizerSwapAdapter.all_slots |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / swap_optimizer_base.py |
OptimizerSwapAdapter.checkpoint_state_dict |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / swap_optimizer_base.py |
OptimizerSwapAdapter.load_checkpoint_state_dict |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / swap_optimizer_base.py |
OptimizerSwapAdapter.publish_packed_state |
|
12 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / swap_optimizer_base.py |
OptimizerSwapAdapter.export_swappable_state |
|
24 |
2 |
0 |
|
92% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / swap_optimizer_base.py |
OptimizerSwapAdapter.strip_swappable_state |
|
10 |
1 |
0 |
|
90% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / swap_optimizer_base.py |
OptimizerSwapAdapter.load_swappable_state |
|
42 |
18 |
0 |
|
57% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / swap_optimizer_base.py |
OptimizerSwapAdapter._init_param_state |
|
26 |
2 |
0 |
|
92% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / swap_optimizer_base.py |
OptimizerSwapAdapter._register_present_slots |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / swap_optimizer_base.py |
OptimizerSwapAdapter._build_slots |
|
14 |
1 |
0 |
|
93% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / swap_optimizer_base.py |
OptimizerSwapAdapter._make_slot |
|
10 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / swap_optimizer_base.py |
OptimizerSwapAdapter._ordered_slots |
|
10 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / swap_optimizer_base.py |
OptimizerSwapAdapter._state_keys_for_param |
|
9 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / swap_optimizer_base.py |
OptimizerSwapAdapter._slot_tensor |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / swap_optimizer_base.py |
OptimizerSwapAdapter._configured_state_keys |
|
9 |
1 |
0 |
|
89% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / swap_optimizer_base.py |
OptimizerSwapAdapter._cast_state_tensor_like_torch |
|
10 |
6 |
0 |
|
40% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / swap_optimizer_base.py |
OptimizerSwapAdapter._cast_swappable_tensor_to_cpu |
|
5 |
2 |
0 |
|
60% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / swap_optimizer_base.py |
OptimizerSwapAdapter._default_state_keys |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / swap_optimizer_base.py |
TorchNativeAdamAdapter.matches |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / swap_optimizer_base.py |
TorchNativeAdamWAdapter.matches |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / swap_optimizer_base.py |
TorchNewAdamWAdapter.matches |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / swap_optimizer_base.py |
_iter_unique_slots |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / swap_optimizer_base.py |
_iter_unique_slot_objects |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / swap_optimizer_base.py |
_iter_unique_events |
|
10 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / swap_optimizer_base.py |
validate_state_keys |
|
7 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / swap_optimizer_base.py |
_new_adamw_cls |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / swap_optimizer_base.py |
_new_adamw_func |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / swap_optimizer_base.py |
SwapOptimizer.__init__ |
|
9 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / swap_optimizer_base.py |
SwapOptimizer.__getattr__ |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / swap_optimizer_base.py |
SwapOptimizer.param_groups |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / swap_optimizer_base.py |
SwapOptimizer.param_groups |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / swap_optimizer_base.py |
SwapOptimizer.state |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / swap_optimizer_base.py |
SwapOptimizer.defaults |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / swap_optimizer_base.py |
SwapOptimizer.add_param_group |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / swap_optimizer_base.py |
SwapOptimizer.zero_grad |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / swap_optimizer_base.py |
SwapOptimizer.step |
|
8 |
8 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / swap_optimizer_base.py |
SwapOptimizer.state_dict |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / swap_optimizer_base.py |
SwapOptimizer.load_state_dict |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / swap_optimizer_base.py |
SwapOptimizer._build_adapter |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / swap_optimizer_base.py |
SwapOptimizer._no_grad_context |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / swap_optimizer_base.py |
(no function) |
|
192 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / utils.py |
_get_rank |
|
6 |
2 |
0 |
|
67% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / utils.py |
info_rank0 |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / utils.py |
warning_rank0 |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / utils.py |
debug_rank0 |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / utils.py |
get_device_count |
|
6 |
2 |
0 |
|
67% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / utils.py |
get_current_device |
|
6 |
2 |
0 |
|
67% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / utils.py |
empty_accelerator_cache |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / utils.py |
_install_logger_methods |
|
6 |
1 |
0 |
|
83% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / optimizer / utils.py |
(no function) |
|
13 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / __init__.py |
(no function) |
|
8 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / comm_compute_overlap.py |
CommComputeOverlap.__init__ |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / comm_compute_overlap.py |
CommComputeOverlap.coordinator |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / comm_compute_overlap.py |
CommComputeOverlap.wrap_dispatch |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / comm_compute_overlap.py |
CommComputeOverlap.wrap_dispatch._wrapped |
|
7 |
7 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / comm_compute_overlap.py |
CommComputeOverlap.wrap_combine |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / comm_compute_overlap.py |
CommComputeOverlap.wrap_combine._wrapped |
|
7 |
7 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / comm_compute_overlap.py |
CommComputeOverlap._ensure_worker |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / comm_compute_overlap.py |
CommComputeOverlap._submit_bwd_task |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / comm_compute_overlap.py |
CommComputeOverlap._start_bwd_task |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / comm_compute_overlap.py |
CommComputeOverlap._worker_loop |
|
7 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / comm_compute_overlap.py |
CommComputeOverlap._run_bwd_task |
|
9 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / comm_compute_overlap.py |
CommComputeOverlap.close |
|
14 |
3 |
0 |
|
79% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / comm_compute_overlap.py |
CommComputeOverlap._run |
|
23 |
3 |
0 |
|
87% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / comm_compute_overlap.py |
CommComputeOverlap.run |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / comm_compute_overlap.py |
(no function) |
|
28 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / hook_coordinator.py |
HookCoordinator.__init__ |
|
8 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / hook_coordinator.py |
HookCoordinator.enable |
|
8 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / hook_coordinator.py |
HookCoordinator._on_barrier_trip |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / hook_coordinator.py |
HookCoordinator.disable |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / hook_coordinator.py |
HookCoordinator.depart |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / hook_coordinator.py |
HookCoordinator.is_enabled |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / hook_coordinator.py |
HookCoordinator.rendezvous |
|
16 |
1 |
0 |
|
94% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / hook_coordinator.py |
HookCoordinator.notify_dispatched |
|
9 |
2 |
0 |
|
78% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / hook_coordinator.py |
(no function) |
|
17 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / mpipe / __init__.py |
(no function) |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / mpipe / executor_base.py |
MPipeTransposeExecutorBase.__init__ |
|
17 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / mpipe / executor_base.py |
MPipeTransposeExecutorBase.reset |
|
9 |
1 |
0 |
|
89% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / mpipe / executor_base.py |
MPipeTransposeExecutorBase._send_meta |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / mpipe / executor_base.py |
MPipeTransposeExecutorBase._recv_meta |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / mpipe / executor_base.py |
MPipeTransposeExecutorBase._output_arity |
|
8 |
4 |
0 |
|
50% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / mpipe / executor_base.py |
MPipeTransposeExecutorBase._input_arity |
|
6 |
2 |
0 |
|
67% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / mpipe / executor_base.py |
MPipeTransposeExecutorBase._global_rank |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / mpipe / executor_base.py |
MPipeTransposeExecutorBase._as_tuple |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / mpipe / executor_base.py |
MPipeTransposeExecutorBase._kwargs_for |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / mpipe / executor_base.py |
MPipeTransposeExecutorBase.broadcast_params |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / mpipe / executor_base.py |
MPipeTransposeExecutorBase.transpose_forward |
|
24 |
1 |
0 |
|
96% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / mpipe / executor_base.py |
MPipeTransposeExecutorBase._save_grad_for_bwd |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / mpipe / executor_base.py |
MPipeTransposeExecutorBase._transfer_out_for_fwd |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / mpipe / executor_base.py |
MPipeTransposeExecutorBase.fwd_send |
|
7 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / mpipe / executor_base.py |
MPipeTransposeExecutorBase.fwd_recv |
|
6 |
3 |
0 |
|
50% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / mpipe / executor_base.py |
MPipeTransposeExecutorBase._recv_device |
|
7 |
1 |
0 |
|
86% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / mpipe / executor_base.py |
MPipeTransposeExecutorBase._recv_payload |
|
12 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / mpipe / executor_base.py |
MPipeTransposeExecutorBase._fwd_recv_one |
|
7 |
7 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / mpipe / executor_base.py |
MPipeTransposeExecutorBase._fwd_recv_all |
|
9 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / mpipe / executor_base.py |
MPipeTransposeExecutorBase.fwd_bwd_graph_input_send |
|
7 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / mpipe / executor_base.py |
MPipeTransposeExecutorBase.fwd_bwd_graph_input_recv |
|
10 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / mpipe / executor_base.py |
MPipeTransposeExecutorBase.grad_send |
|
10 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / mpipe / executor_base.py |
MPipeTransposeExecutorBase.grad_recv_with_backward |
|
15 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / mpipe / executor_base.py |
MPipeTransposeExecutorBase.reduce_tower_grads |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / mpipe / executor_base.py |
MPipeTransposeExecutorBase.transpose_backward |
|
9 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / mpipe / executor_base.py |
MPipeTransposeExecutorBase._contiguous |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / mpipe / executor_base.py |
MPipeTransposeExecutorBase._broadcast_tensors |
|
0 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / mpipe / executor_base.py |
MPipeTransposeExecutorBase._detached_forward |
|
0 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / mpipe / executor_base.py |
MPipeTransposeExecutorBase._connected_forward |
|
0 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / mpipe / executor_base.py |
MPipeTransposeExecutorBase._mark_requires_grad |
|
0 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / mpipe / executor_base.py |
MPipeTransposeExecutorBase._explicit_forward_before_backward |
|
0 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / mpipe / executor_base.py |
MPipeTransposeExecutorBase._detach_for_wire |
|
0 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / mpipe / executor_base.py |
MPipeTransposeExecutorBase._zeros_like |
|
0 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / mpipe / executor_base.py |
MPipeTransposeExecutorBase._owner_transpose_backward |
|
0 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / mpipe / executor_base.py |
MPipeTransposeExecutorBase._snapshot_tower_grads |
|
0 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / mpipe / executor_base.py |
MPipeTransposeExecutorBase._reduce_grads |
|
0 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / mpipe / executor_base.py |
(no function) |
|
57 |
0 |
3 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / mpipe / sampler.py |
PPRankOwnedSampler.__init__ |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / mpipe / sampler.py |
PPRankOwnedSampler.__iter__ |
|
7 |
7 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / mpipe / sampler.py |
PPRankOwnedSampler.__len__ |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / mpipe / sampler.py |
PPRankOwnedSampler.set_epoch |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / mpipe / sampler.py |
PPRankOwnedSampler.state_dict |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / mpipe / sampler.py |
PPRankOwnedSampler.load_state_dict |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / mpipe / sampler.py |
mpipe_owned_micros |
|
10 |
3 |
0 |
|
70% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / mpipe / sampler.py |
(no function) |
|
9 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / mpipe / schedule.py |
ScheduleMPipeTranspose.__init__ |
|
18 |
1 |
0 |
|
94% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / mpipe / schedule.py |
ScheduleMPipeTranspose._module_has_trainable_params |
|
7 |
4 |
0 |
|
43% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / mpipe / schedule.py |
ScheduleMPipeTranspose.preprocess_module |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / mpipe / schedule.py |
ScheduleMPipeTranspose.output_consumer |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / mpipe / schedule.py |
ScheduleMPipeTranspose.has_trainable_preprocess |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / mpipe / schedule.py |
ScheduleMPipeTranspose.owner_backward |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / mpipe / schedule.py |
ScheduleMPipeTranspose.num_transpose_layers |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / mpipe / schedule.py |
ScheduleMPipeTranspose.num_transpose_micro_batches |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / mpipe / schedule.py |
ScheduleMPipeTranspose.owned_micros |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / mpipe / schedule.py |
ScheduleMPipeTranspose.owner_of |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / mpipe / schedule.py |
ScheduleMPipeTranspose._setup_mpipe_execution |
|
10 |
3 |
0 |
|
70% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / mpipe / schedule.py |
ScheduleMPipeTranspose._apply_backward_retain_flag |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / mpipe / schedule.py |
ScheduleMPipeTranspose.run_microbatches |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / mpipe / schedule.py |
ScheduleMPipeTranspose.construct_exec_order |
|
25 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / mpipe / schedule.py |
ScheduleMPipeTranspose._build_transpose_prefix |
|
36 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / mpipe / schedule.py |
ScheduleMPipeTranspose._build_owner_backward_suffix |
|
7 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / mpipe / schedule.py |
ScheduleMPipeTranspose._build_owner_backward_suffix.owner_of |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / mpipe / schedule.py |
ScheduleMPipeTranspose._insert_rank0_preprocess_steps |
|
16 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / mpipe / schedule.py |
(no function) |
|
39 |
0 |
2 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / mpipe / step_types.py |
(no function) |
|
12 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / pipeline_swap.py |
unregister_layer_swap_hooks |
|
13 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / pipeline_swap.py |
PipelineSwapSession.__init__ |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / pipeline_swap.py |
PipelineSwapSession._key |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / pipeline_swap.py |
PipelineSwapSession.manages |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / pipeline_swap.py |
PipelineSwapSession.group_context |
|
7 |
1 |
0 |
|
86% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / pipeline_swap.py |
PipelineSwapSession.require_forward_context |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / pipeline_swap.py |
PipelineSwapSession.group_name |
|
4 |
1 |
0 |
|
75% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / pipeline_swap.py |
PipelineSwapSession.wait_load |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / pipeline_swap.py |
PipelineSwapSession.protect_aliases |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / pipeline_swap.py |
PipelineSwapSession.close |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / pipeline_swap.py |
_is_compute_step |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / pipeline_swap.py |
_is_comm_step |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / pipeline_swap.py |
_is_composite_compute_step |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / pipeline_swap.py |
_ComputeLeaf.__init__ |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / pipeline_swap.py |
_iter_compute_leaf_steps |
|
7 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / pipeline_swap.py |
_collect_compute_leaves |
|
12 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / pipeline_swap.py |
_append_after |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / pipeline_swap.py |
_append_before |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / pipeline_swap.py |
_iter_steps_by_priority |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / pipeline_swap.py |
_comm_block_anchor |
|
8 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / pipeline_swap.py |
_post_compute_anchor |
|
18 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / pipeline_swap.py |
_post_compute_launch_anchor |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / pipeline_swap.py |
_load_launch_anchor |
|
9 |
1 |
0 |
|
89% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / pipeline_swap.py |
inject_pipeline_swap_steps |
|
37 |
1 |
0 |
|
97% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / pipeline_swap.py |
_protect_pipeline_owned_tensors |
|
9 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / pipeline_swap.py |
swap_launch_offload |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / pipeline_swap.py |
swap_wait_offload |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / pipeline_swap.py |
swap_launch_load |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / pipeline_swap.py |
swap_wait_load |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / pipeline_swap.py |
(no function) |
|
50 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / scheduler.py |
MetaStep.__init__ |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / scheduler.py |
MetaStep.micro_index |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / scheduler.py |
MetaStep.stage_index |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / scheduler.py |
MetaStep.type |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / scheduler.py |
MetaStep.sub_steps |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / scheduler.py |
MetaStep.boundary_p2p |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / scheduler.py |
MetaStep.__eq__ |
|
3 |
1 |
0 |
|
67% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / scheduler.py |
MetaStep.__ne__ |
|
3 |
1 |
0 |
|
67% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / scheduler.py |
MetaStep.__hash__ |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / scheduler.py |
MetaStep.__str__ |
|
4 |
2 |
0 |
|
50% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / scheduler.py |
MetaStep.__repr__ |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / scheduler.py |
MetaStep.from_str |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / scheduler.py |
generate_stage_to_rank_mapping |
|
16 |
14 |
0 |
|
12% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / scheduler.py |
generate_rank_to_stage_mapping |
|
7 |
7 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / scheduler.py |
iter_leaf_meta_steps |
|
7 |
4 |
0 |
|
43% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / scheduler.py |
PipelineContext.__init__ |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / scheduler.py |
_exec_fsdp_unshard |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / scheduler.py |
_exec_fsdp_reshard |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / scheduler.py |
_exec_fsdp_reduce_grad |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / scheduler.py |
PipelineScheduleRuntime.__init__ |
|
38 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / scheduler.py |
PipelineScheduleRuntime.kwargs_batch_dim |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / scheduler.py |
PipelineScheduleRuntime.register_custom_function |
|
1 |
0 |
1 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / scheduler.py |
PipelineScheduleRuntime._stage_hsdp_modules |
|
10 |
2 |
0 |
|
80% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / scheduler.py |
PipelineScheduleRuntime._inject_local_fsdp_actions |
|
11 |
1 |
0 |
|
91% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / scheduler.py |
PipelineScheduleRuntime._prepare_fsdp_backward |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / scheduler.py |
PipelineScheduleRuntime._inject_local_pp_swap_actions |
|
7 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / scheduler.py |
PipelineScheduleRuntime._build_stage_to_rank_index |
|
0 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / scheduler.py |
PipelineScheduleRuntime.construct_exec_order |
|
0 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / scheduler.py |
PipelineScheduleRuntime.build_exec_order |
|
15 |
3 |
0 |
|
80% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / scheduler.py |
PipelineScheduleRuntime._init_p2p_multi_stream_groups |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / scheduler.py |
PipelineScheduleRuntime._p2p_op |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / scheduler.py |
PipelineScheduleRuntime.convert_stages_dict |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / scheduler.py |
PipelineScheduleRuntime.split_microbatches |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / scheduler.py |
PipelineScheduleRuntime._to_spec |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / scheduler.py |
PipelineScheduleRuntime._normalize_args_batch_dim |
|
7 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / scheduler.py |
PipelineScheduleRuntime._normalize_kwargs_batch_dim |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / scheduler.py |
PipelineScheduleRuntime._check_stages |
|
8 |
3 |
0 |
|
62% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / scheduler.py |
PipelineScheduleRuntime._init_stages |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / scheduler.py |
PipelineScheduleRuntime._on_forward_chunk_done |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / scheduler.py |
PipelineScheduleRuntime.run |
|
11 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / scheduler.py |
PipelineScheduleRuntime.run_with_dataiterator |
|
15 |
1 |
0 |
|
93% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / scheduler.py |
PipelineScheduleRuntime.sync_shared_parameters_grad |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / scheduler.py |
PipelineScheduleRuntime.update_losses |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / scheduler.py |
PipelineScheduleRuntime.enable_dxdw_split |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / scheduler.py |
PipelineScheduleRuntime._wait_p2p |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / scheduler.py |
PipelineScheduleRuntime._resolve_batch_p2p_group |
|
7 |
2 |
0 |
|
71% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / scheduler.py |
PipelineScheduleRuntime._ensure_batch_p2p_group_initialized |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / scheduler.py |
PipelineScheduleRuntime._ensure_p2p_multi_stream_groups_initialized |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / scheduler.py |
PipelineScheduleRuntime._drain_inflight_p2p |
|
6 |
1 |
0 |
|
83% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / scheduler.py |
PipelineScheduleRuntime._batched_issue |
|
5 |
1 |
0 |
|
80% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / scheduler.py |
PipelineScheduleRuntime.recv_fwd |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / scheduler.py |
PipelineScheduleRuntime.recv_bwd |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / scheduler.py |
PipelineScheduleRuntime.wait_fwd_recv |
|
3 |
1 |
0 |
|
67% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / scheduler.py |
PipelineScheduleRuntime.wait_bwd_recv |
|
3 |
1 |
0 |
|
67% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / scheduler.py |
PipelineScheduleRuntime.send_fwd |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / scheduler.py |
PipelineScheduleRuntime.send_bwd |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / scheduler.py |
PipelineScheduleRuntime._arm_boundary |
|
8 |
6 |
0 |
|
25% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / scheduler.py |
PipelineScheduleRuntime._finish_boundary |
|
3 |
1 |
0 |
|
67% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / scheduler.py |
PipelineScheduleRuntime.exec_boundary_p2p |
|
14 |
11 |
0 |
|
21% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / scheduler.py |
PipelineScheduleRuntime._exec_batch_send_recv |
|
25 |
25 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / scheduler.py |
PipelineScheduleRuntime._assert_in_unshard_if_needed |
|
6 |
4 |
0 |
|
33% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / scheduler.py |
PipelineScheduleRuntime._exec_step |
|
34 |
9 |
0 |
|
74% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / scheduler.py |
PipelineScheduleRuntime.execute_fwd_leaf |
|
11 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / scheduler.py |
PipelineScheduleRuntime._exec_pipeline_data_steps |
|
56 |
36 |
0 |
|
36% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / scheduler.py |
PipelineScheduleRuntime._exec_pipeline_swap_step |
|
10 |
4 |
0 |
|
60% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / scheduler.py |
PipelineScheduleRuntime.run_microbatches |
|
36 |
8 |
0 |
|
78% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / scheduler.py |
_OverlapPhantom.__init__ |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / scheduler.py |
_expand_overlap_slots |
|
11 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / scheduler.py |
_process_rank_items |
|
10 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / scheduler.py |
_insert_dataload_before_fwd |
|
8 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / scheduler.py |
_insert_dataload_before_fwd._loads_for |
|
7 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / scheduler.py |
_column_scan_insert_comms |
|
18 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / scheduler.py |
coalesce_p2p |
|
15 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / scheduler.py |
coalesce_p2p._flush |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / scheduler.py |
attach_fwd_boundary_p2p |
|
21 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / scheduler.py |
_p2p_run_after |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / scheduler.py |
_split_boundary_run |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / scheduler.py |
_split_boundary_run._is_own_fwd_send |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / scheduler.py |
split_overlap_dxdw |
|
22 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / scheduler.py |
add_send_recv |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / scheduler.py |
add_send_recv.stage_to_rank |
|
7 |
5 |
0 |
|
29% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / scheduler.py |
add_send_recv._fwd_peer |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / scheduler.py |
add_send_recv._bwd_peer |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / scheduler.py |
add_send_recv._insert_comms_for_step |
|
17 |
4 |
0 |
|
76% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / scheduler.py |
_step_dep_ready |
|
12 |
12 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / scheduler.py |
_simulate_aligned_schedule |
|
15 |
15 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / scheduler.py |
_simulate_aligned_schedule._emit_bubble |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / scheduler.py |
_simulate_aligned_schedule._emit_step |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / scheduler.py |
_simulate_aligned_schedule._step_rank_at |
|
16 |
16 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / scheduler.py |
auto_align_and_add_send_recv |
|
14 |
14 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / scheduler.py |
auto_align_and_add_send_recv.stage_to_rank |
|
7 |
7 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / scheduler.py |
auto_align_and_add_send_recv._fwd_peer |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / scheduler.py |
auto_align_and_add_send_recv._bwd_peer |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / scheduler.py |
auto_align_and_add_send_recv._find_1f1b_boundary |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / scheduler.py |
auto_align_and_add_send_recv._insert_comms_for_step |
|
17 |
17 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / scheduler.py |
ScheduleGPipe.__init__ |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / scheduler.py |
ScheduleGPipe._build_stage_to_rank_index |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / scheduler.py |
ScheduleGPipe.construct_exec_order |
|
18 |
18 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / scheduler.py |
Schedule1F1B.__init__ |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / scheduler.py |
Schedule1F1B._build_stage_to_rank_index |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / scheduler.py |
Schedule1F1B.construct_exec_order |
|
51 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / scheduler.py |
ScheduleInterleaved1F1B.__init__ |
|
8 |
1 |
0 |
|
88% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / scheduler.py |
ScheduleInterleaved1F1B._init_round_layout |
|
10 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / scheduler.py |
ScheduleInterleaved1F1B.construct_exec_order |
|
8 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / scheduler.py |
ScheduleInterleaved1F1B._build_stage_to_rank_index |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / scheduler.py |
ScheduleInterleaved1F1B.warmup_ops |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / scheduler.py |
ScheduleInterleaved1F1B.forward_stage_index |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / scheduler.py |
ScheduleInterleaved1F1B.backward_stage_index |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / scheduler.py |
ScheduleInterleaved1F1B._short_micro |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / scheduler.py |
ScheduleInterleaved1F1B._trailing_bubble |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / scheduler.py |
ScheduleInterleaved1F1B._emit_warmup_ops |
|
14 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / scheduler.py |
ScheduleInterleaved1F1B._emit_cooldown_ops |
|
13 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / scheduler.py |
ScheduleInterleaved1F1B._emit_1f1b_ops |
|
18 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / scheduler.py |
ScheduleInterleaved1F1B._collect_fwd_bwd_steps |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / scheduler.py |
ScheduleInterleaved1F1B._pair_into_overlap_b_f |
|
8 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / scheduler.py |
ScheduleInterleaved1F1B._emit_1f1b_overlap_ops |
|
9 |
2 |
0 |
|
78% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / scheduler.py |
ScheduleInterleaved1F1B._emit_1f1b_overlap_ops.emit_fwd |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / scheduler.py |
ScheduleInterleaved1F1B._emit_1f1b_overlap_ops.emit_bwd |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / scheduler.py |
ScheduleInterleaved1F1B.construct_stage_exec_order |
|
16 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / scheduler.py |
detect_cycle_in_graph |
|
39 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / scheduler.py |
output_cycle_results |
|
8 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / scheduler.py |
parse_and_validate |
|
23 |
5 |
0 |
|
78% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / scheduler.py |
parse_and_validate.parse_elements |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / scheduler.py |
generate_operations |
|
35 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / scheduler.py |
generate_operations.stage_to_rank |
|
7 |
5 |
0 |
|
29% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / scheduler.py |
generate_operations.find_send_target |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / scheduler.py |
generate_operations.find_recv_source |
|
8 |
1 |
0 |
|
88% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / scheduler.py |
validate_pipeline_execution |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / scheduler.py |
_next_active_stage_indices |
|
13 |
1 |
0 |
|
92% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / scheduler.py |
add_fsdp_unshard_reshard |
|
18 |
3 |
0 |
|
83% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / scheduler.py |
add_fsdp_reduce_grad |
|
18 |
2 |
0 |
|
89% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / scheduler.py |
(no function) |
|
181 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / stage.py |
SharedParameterInfo.__init__ |
|
9 |
2 |
0 |
|
78% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / stage.py |
SharedParameterInfo.parameter |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / stage.py |
SharedParameterInfo.shared_stage |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / stage.py |
SharedParameterInfo.__repr__ |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / stage.py |
SharedParameterInfo.__str__ |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / stage.py |
PipelineStage.__init__ |
|
20 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / stage.py |
PipelineStage.has_backward |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / stage.py |
PipelineStage.init |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / stage.py |
PipelineStage._init_pp_group |
|
11 |
2 |
0 |
|
82% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / stage.py |
PipelineStage.clear_states |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / stage.py |
PipelineStage._check_shared_parameters |
|
10 |
8 |
0 |
|
20% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / stage.py |
PipelineStage._sync_shared_parameters |
|
10 |
8 |
0 |
|
20% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / stage.py |
PipelineStage._global_rank |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / stage.py |
PipelineStage._init_shared_parameter_group |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / stage.py |
PipelineStage.sync_shared_parameters_grad |
|
26 |
2 |
0 |
|
92% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / stage.py |
PipelineStage._check_src_stage |
|
5 |
3 |
0 |
|
40% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / stage.py |
PipelineStage._check_dst_stage |
|
5 |
3 |
0 |
|
40% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / stage.py |
PipelineStage._update_layout |
|
10 |
10 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / stage.py |
PipelineStage._get_layout_rank_list |
|
18 |
18 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / stage.py |
PipelineStage.get_last_stage_sens |
|
14 |
9 |
0 |
|
36% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / stage.py |
PipelineStage._construct_forward_recv_info |
|
12 |
12 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / stage.py |
PipelineStage._communicate_meta |
|
14 |
14 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / stage.py |
PipelineStage.fwd_recv_specs |
|
13 |
13 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / stage.py |
PipelineStage.exec_fwd_recv_ops |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / stage.py |
PipelineStage._construct_backward_recv_info |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / stage.py |
PipelineStage._extract_meta_from_tensor |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / stage.py |
PipelineStage.exec_fwd_send_ops |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / stage.py |
PipelineStage.fwd_send_specs |
|
20 |
1 |
0 |
|
95% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / stage.py |
PipelineStage.bwd_recv_specs |
|
10 |
1 |
0 |
|
90% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / stage.py |
PipelineStage.exec_bwd_recv_ops |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / stage.py |
PipelineStage.exec_bwd_send_ops |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / stage.py |
PipelineStage.bwd_send_specs |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / stage.py |
PipelineStage.execute_reduce_grad |
|
24 |
24 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / stage.py |
PipelineStage._build_padded_sens |
|
12 |
12 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / stage.py |
PipelineStage._output_requires_grad_mask |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / stage.py |
(no function) |
|
49 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / utils.py |
BatchDimSpec.__init__ |
|
3 |
1 |
0 |
|
67% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / utils.py |
BatchDimSpec.__repr__ |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / utils.py |
BatchDimSpec.__str__ |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / utils.py |
BatchDimSpec.from_tuple |
|
3 |
1 |
0 |
|
67% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / utils.py |
BatchDimSpec.from_dict |
|
3 |
1 |
0 |
|
67% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / utils.py |
_RecvInfo.__init__ |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / utils.py |
_RecvInfo.global_rank |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / utils.py |
_RecvInfo.buffer |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / utils.py |
_RecvInfo.buffer |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / utils.py |
_RecvInfo.requires_grad |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / utils.py |
_RecvInfo.requires_grad |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / utils.py |
(no function) |
|
21 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / __init__.py |
(no function) |
|
0 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / _op_dispatch.py |
_apply_shard_offset_to_rng_args |
|
13 |
13 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / _op_dispatch.py |
get_no_skip_ops |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / _op_dispatch.py |
get_dtensor_dispatch |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / _op_dispatch.py |
LayoutCacheKey.__init__ |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / _op_dispatch.py |
LayoutCacheKey.from_cache_values |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / _op_dispatch.py |
LayoutCacheKey.__eq__ |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / _op_dispatch.py |
LayoutCacheKey.__hash__ |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / _op_dispatch.py |
LayoutCacheKey.__repr__ |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / _op_dispatch.py |
LayoutCacheManager.__init__ |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / _op_dispatch.py |
LayoutCacheManager.get_instance |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / _op_dispatch.py |
LayoutCacheManager.get_layout_cache |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / _op_dispatch.py |
LayoutCacheManager.distributed_op |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / _op_dispatch.py |
LayoutCacheManager.clear_cache |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / _op_dispatch.py |
OpDispatcher.__init__ |
|
9 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / _op_dispatch.py |
OpDispatcher._register_distributed_ops |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / _op_dispatch.py |
OpDispatcher._resolve_distributed_op_module |
|
14 |
3 |
0 |
|
79% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / _op_dispatch.py |
OpDispatcher._register_single_distributed_op |
|
12 |
1 |
0 |
|
92% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / _op_dispatch.py |
OpDispatcher._merge_default |
|
11 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / _op_dispatch.py |
OpDispatcher.safe_load_yaml_from_dir |
|
13 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / _op_dispatch.py |
OpDispatcher._dispatch_random_op |
|
19 |
8 |
0 |
|
58% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / _op_dispatch.py |
OpDispatcher._func_dropout_ext_inplace |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / _op_dispatch.py |
OpDispatcher._random_op_returns_self |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / _op_dispatch.py |
OpDispatcher._wrap_random_result |
|
9 |
2 |
0 |
|
78% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / _op_dispatch.py |
OpDispatcher._unwrap_value |
|
7 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / _op_dispatch.py |
OpDispatcher._unwrap_args |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / _op_dispatch.py |
OpDispatcher._unwrap_kwargs |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / _op_dispatch.py |
OpDispatcher._collect_dtensors |
|
7 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / _op_dispatch.py |
OpDispatcher._validate_ragged_dispatch |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / _op_dispatch.py |
OpDispatcher._dispatch_ragged_elementwise |
|
9 |
1 |
0 |
|
89% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / _op_dispatch.py |
OpDispatcher._gather_dtensors_to_full |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / _op_dispatch.py |
OpDispatcher._gather_dtensors_to_full.gather |
|
7 |
2 |
0 |
|
71% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / _op_dispatch.py |
OpDispatcher._should_bypass_dispatch |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / _op_dispatch.py |
OpDispatcher._validate_inplace_partial_inputs |
|
19 |
17 |
0 |
|
11% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / _op_dispatch.py |
OpDispatcher._should_dispatch_loss_parallel |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / _op_dispatch.py |
OpDispatcher._check_decomposed_ce_op_in_loss_parallel |
|
14 |
3 |
0 |
|
79% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / _op_dispatch.py |
OpDispatcher._dispatch_loss_parallel |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / _op_dispatch.py |
OpDispatcher._check_ce_op_without_loss_parallel_context |
|
7 |
1 |
0 |
|
86% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / _op_dispatch.py |
OpDispatcher._normalize_aclop_args |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / _op_dispatch.py |
OpDispatcher._is_aclop_packed |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / _op_dispatch.py |
OpDispatcher._call_op_impl |
|
3 |
1 |
0 |
|
67% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / _op_dispatch.py |
OpDispatcher._handle_unregistered_op |
|
17 |
4 |
0 |
|
76% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / _op_dispatch.py |
OpDispatcher._dispatch_layout_infer |
|
17 |
1 |
0 |
|
94% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / _op_dispatch.py |
OpDispatcher._restore_inplace_dtensor_result |
|
3 |
1 |
0 |
|
67% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / _op_dispatch.py |
OpDispatcher._lookup_or_infer_layout |
|
10 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / _op_dispatch.py |
OpDispatcher.dispatch |
|
36 |
5 |
0 |
|
86% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / _op_dispatch.py |
(no function) |
|
107 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / api.py |
_has_kwargs |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / api.py |
_get_param_name |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / api.py |
_convert_sharding_plan |
|
20 |
1 |
0 |
|
95% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / api.py |
_convert_sharding_plan._is_placement_tuple |
|
15 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / api.py |
_convert_sharding_plan._to_layout |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / api.py |
_convert_sharding_plan._convert_value |
|
15 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / api.py |
_convert_sharding_plan._convert_forward_plan |
|
16 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / api.py |
_parallel_in |
|
19 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / api.py |
_parallel_in._get_layout |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / api.py |
_parallel_out |
|
18 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / api.py |
_forward_pre_hook |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / api.py |
_forward_pre_with_kwargs_hook |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / api.py |
_forward_hook |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / api.py |
_forward_with_kwargs_hook |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / api.py |
_register_hook |
|
20 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / api.py |
_register_hook._register_cell_hook |
|
7 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / api.py |
_register_hook._set_layouts |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / api.py |
_register_local_tensor_hook |
|
9 |
1 |
0 |
|
89% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / api.py |
_register_local_tensor_hook.hook_func |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / api.py |
_register_local_tensor_hook.hook_func._recursive_to_local |
|
8 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / api.py |
_shard_callable |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / api.py |
_shard_callable._shard_wrapper |
|
8 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / api.py |
shard_module |
|
43 |
2 |
0 |
|
95% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / api.py |
parallelize_value_and_grad |
|
10 |
10 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / api.py |
parallelize_value_and_grad.CellWrapper.__init__ |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / api.py |
parallelize_value_and_grad.CellWrapper.construct |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / api.py |
parallelize_value_and_grad.CellWrapper.forward |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / api.py |
parallelize_value_and_grad.wrapper |
|
38 |
38 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / api.py |
(no function) |
|
26 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / custom_shard.py |
_process_custom_shard_inputs |
|
20 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / custom_shard.py |
_wrap_custom_shard_outputs |
|
16 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / custom_shard.py |
custom_shard |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / custom_shard.py |
custom_shard.wrapped |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / custom_shard.py |
(no function) |
|
11 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / dfunction.py |
_LocalCallable.__init__ |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / dfunction.py |
_LocalCallable.__call__ |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / dfunction.py |
DFunction.forward |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / dfunction.py |
DFunction.backward |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / dfunction.py |
DFunction.apply |
|
6 |
1 |
0 |
|
83% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / dfunction.py |
DFunction._get_local_callable |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / dfunction.py |
DFunction._get_local_callable._local_fn |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / dfunction.py |
(no function) |
|
18 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / __init__.py |
(no function) |
|
0 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_activation_with_axis.py |
_normalize_activation_with_axis_args |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_activation_with_axis.py |
ActivationWithAxisDistributedOp.preprocess |
|
7 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_activation_with_axis.py |
ActivationWithAxisDistributedOp.infer_layout |
|
14 |
1 |
0 |
|
93% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_activation_with_axis.py |
ActivationWithAxisDistributedOp.check_layout |
|
13 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_activation_with_axis.py |
(no function) |
|
8 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_all_gather_matmul.py |
_normalize_agm_args |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_all_gather_matmul.py |
AllGatherMatmulDistributedOp.preprocess |
|
9 |
9 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_all_gather_matmul.py |
AllGatherMatmulDistributedOp._set_partial_from_k |
|
12 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_all_gather_matmul.py |
AllGatherMatmulDistributedOp._validate_input_layouts |
|
9 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_all_gather_matmul.py |
AllGatherMatmulDistributedOp.infer_layout |
|
21 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_all_gather_matmul.py |
(no function) |
|
12 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_argmax_with_value_ops.py |
_normalize_argmax_with_value_args |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_argmax_with_value_ops.py |
ArgMaxWithValueDistributedOp.preprocess |
|
8 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_argmax_with_value_ops.py |
ArgMaxWithValueDistributedOp.infer_layout |
|
24 |
1 |
0 |
|
96% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_argmax_with_value_ops.py |
(no function) |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_argsort.py |
_normalize_argsort_args |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_argsort.py |
ArgsortDistributedOp.preprocess |
|
12 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_argsort.py |
ArgsortDistributedOp.infer_layout |
|
15 |
1 |
0 |
|
93% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_argsort.py |
(no function) |
|
8 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_atleast_1d.py |
_normalize_atleast_1d_args |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_atleast_1d.py |
Atleast1DDistributedOp.preprocess |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_atleast_1d.py |
Atleast1DDistributedOp.infer_layout |
|
26 |
4 |
0 |
|
85% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_atleast_1d.py |
Atleast1DDistributedOp.infer_layout.idx_to_alias |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_atleast_1d.py |
(no function) |
|
7 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_cell_backward_hook.py |
CellBackwardHookDistributedOp.wrap_output |
|
13 |
2 |
0 |
|
85% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_cell_backward_hook.py |
(no function) |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_chunk_view.py |
_normalize_chunk_view_args |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_chunk_view.py |
ChunkViewDistributedOp._calculate_output_count |
|
5 |
1 |
0 |
|
80% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_chunk_view.py |
ChunkViewDistributedOp.preprocess |
|
7 |
7 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_chunk_view.py |
ChunkViewDistributedOp.infer_layout |
|
30 |
1 |
0 |
|
97% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_chunk_view.py |
(no function) |
|
9 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_concat.py |
_normalize_concat_args |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_concat.py |
ConcatDistributedOp.preprocess |
|
9 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_concat.py |
ConcatDistributedOp.infer_layout |
|
20 |
1 |
0 |
|
95% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_concat.py |
(no function) |
|
7 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_conv3d.py |
_normalize_conv3d_args |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_conv3d.py |
Conv3dDistributedOp.__init__ |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_conv3d.py |
Conv3dDistributedOp._validate_row_parallelism |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_conv3d.py |
Conv3dDistributedOp._validate_column_parallelism |
|
15 |
1 |
0 |
|
93% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_conv3d.py |
Conv3dDistributedOp.preprocess |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_conv3d.py |
Conv3dDistributedOp.infer_layout |
|
25 |
2 |
0 |
|
92% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_conv3d.py |
Conv3dDistributedOp.get_expand_impl |
|
21 |
21 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_conv3d.py |
Conv3dDistributedOp.get_expand_impl.distributed_conv3d_impl |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_conv3d.py |
(no function) |
|
11 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_cumsum.py |
_normalize_cumsum_args |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_cumsum.py |
CumsumDistributedOp.preprocess |
|
14 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_cumsum.py |
CumsumDistributedOp.infer_layout |
|
16 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_cumsum.py |
(no function) |
|
8 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_elementwise.py |
_partial_signature |
|
6 |
2 |
0 |
|
67% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_elementwise.py |
_contributes_to_partial_output |
|
6 |
1 |
0 |
|
83% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_elementwise.py |
_zero_contribution |
|
8 |
8 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_elementwise.py |
_unwrap_local_value |
|
7 |
2 |
0 |
|
71% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_elementwise.py |
_collect_layout_and_shape |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_elementwise.py |
_build_elementwise_cache_values |
|
11 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_elementwise.py |
ElementWiseDistributedOp.preprocess |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_elementwise.py |
ElementWiseDistributedOp.infer_layout |
|
21 |
3 |
0 |
|
86% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_elementwise.py |
ElementWiseDistributedOp._handle_no_input_shapes |
|
6 |
1 |
0 |
|
83% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_elementwise.py |
ElementWiseDistributedOp._align_layouts_and_shapes |
|
8 |
1 |
0 |
|
88% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_elementwise.py |
ElementWiseDistributedOp._compute_output_shape |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_elementwise.py |
ElementWiseDistributedOp._merge_all_layouts |
|
8 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_elementwise.py |
ElementWiseDistributedOp._merge_partial_status |
|
15 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_elementwise.py |
ElementWiseDistributedOp._check_shard_partial_conflict |
|
17 |
5 |
0 |
|
71% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_elementwise.py |
ElementWiseDistributedOp._check_all_inputs_broadcasts_and_partial |
|
16 |
2 |
0 |
|
88% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_elementwise.py |
ElementWiseDistributedOp._broadcast_shapes |
|
14 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_elementwise.py |
ElementWiseDistributedOp._align_tensor_maps_for_broadcast |
|
7 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_elementwise.py |
ElementWiseDistributedOp._normalize_tensor_map_element |
|
7 |
3 |
0 |
|
57% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_elementwise.py |
ElementWiseDistributedOp._denormalize_tensor_map_element |
|
5 |
2 |
0 |
|
60% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_elementwise.py |
ElementWiseDistributedOp._merge_tensor_maps_for_broadcast |
|
30 |
2 |
0 |
|
93% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_elementwise.py |
ElementWiseDistributedOp._create_output_layout |
|
17 |
2 |
0 |
|
88% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_elementwise.py |
ElementWiseWithPartialDistributedOp.__init__ |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_elementwise.py |
AddDistributedOp._canonicalize_binary_operands |
|
19 |
9 |
0 |
|
53% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_elementwise.py |
AddDistributedOp.preprocess |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_elementwise.py |
AddDistributedOp.infer_layout |
|
14 |
6 |
0 |
|
57% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_elementwise.py |
AddDistributedOp.get_expand_impl |
|
13 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_elementwise.py |
AddDistributedOp.get_expand_impl._expand_impl |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_elementwise.py |
(no function) |
|
35 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_embedding.py |
_normalize_embedding_args |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_embedding.py |
EmbeddingDistributedOp.preprocess |
|
8 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_embedding.py |
EmbeddingDistributedOp.infer_layout |
|
26 |
3 |
0 |
|
88% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_embedding.py |
EmbeddingDistributedOp._parse_params |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_embedding.py |
EmbeddingDistributedOp._handle_rp_input |
|
20 |
2 |
0 |
|
90% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_embedding.py |
EmbeddingDistributedOp.get_expand_impl |
|
9 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_embedding.py |
EmbeddingDistributedOp.get_expand_impl.distributed_embedding_impl |
|
20 |
5 |
0 |
|
75% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_embedding.py |
(no function) |
|
11 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_expand.py |
_normalize_expand_args |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_expand.py |
_normalize_expand_as_args |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_expand.py |
ExpandDistributedOp.preprocess |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_expand.py |
ExpandDistributedOp._validate_input_layouts |
|
26 |
6 |
0 |
|
77% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_expand.py |
ExpandDistributedOp.infer_layout |
|
22 |
1 |
0 |
|
95% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_expand.py |
ExpandAsDistributedOp.preprocess |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_expand.py |
ExpandAsDistributedOp.infer_layout |
|
31 |
3 |
0 |
|
90% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_expand.py |
(no function) |
|
13 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_expand_dims.py |
_normalize_expand_dims_args |
|
3 |
1 |
0 |
|
67% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_expand_dims.py |
ExpandDimsDistributedOp.preprocess |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_expand_dims.py |
ExpandDimsDistributedOp.infer_layout |
|
28 |
8 |
0 |
|
71% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_expand_dims.py |
(no function) |
|
7 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_flatten.py |
_normalize_flatten_args |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_flatten.py |
FlattenDistributedOp.preprocess |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_flatten.py |
FlattenDistributedOp.infer_layout |
|
31 |
5 |
0 |
|
84% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_flatten.py |
(no function) |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_gather.py |
_normalize_index_select_args |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_gather.py |
_normalize_gatherd_args |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_gather.py |
_normalize_gathernd_args |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_gather.py |
IndexSelectDistributedOp.preprocess |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_gather.py |
IndexSelectDistributedOp.infer_layout |
|
22 |
3 |
0 |
|
86% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_gather.py |
IndexSelectDistributedOp.get_expand_impl |
|
9 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_gather.py |
IndexSelectDistributedOp.get_expand_impl.expand_impl |
|
20 |
20 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_gather.py |
GatherDDistributedOp.preprocess |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_gather.py |
GatherDDistributedOp.infer_layout |
|
35 |
6 |
0 |
|
83% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_gather.py |
GatherDDistributedOp.get_expand_impl |
|
12 |
3 |
0 |
|
75% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_gather.py |
GatherDDistributedOp.get_expand_impl.distributed_gatherd_impl |
|
18 |
18 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_gather.py |
GatherNdDistributedOp.preprocess |
|
8 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_gather.py |
GatherNdDistributedOp.infer_layout |
|
14 |
1 |
0 |
|
93% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_gather.py |
GatherNdDistributedOp._parse_input_layouts |
|
9 |
4 |
0 |
|
56% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_gather.py |
GatherNdDistributedOp._validate_tensor_maps |
|
15 |
2 |
0 |
|
87% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_gather.py |
GatherNdDistributedOp._get_input_shapes |
|
14 |
3 |
0 |
|
79% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_gather.py |
GatherNdDistributedOp._normalize_shape |
|
9 |
4 |
0 |
|
56% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_gather.py |
GatherNdDistributedOp._get_k_and_trailing_rank |
|
11 |
4 |
0 |
|
64% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_gather.py |
GatherNdDistributedOp._is_none_axis |
|
5 |
1 |
0 |
|
80% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_gather.py |
(no function) |
|
24 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_getitem.py |
_normalize___getitem___args |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_getitem.py |
_is_long_tensor |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_getitem.py |
_is_bool_tensor |
|
5 |
3 |
0 |
|
40% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_getitem.py |
_is_advanced_elem |
|
7 |
3 |
0 |
|
57% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_getitem.py |
_build_key_element_descriptor |
|
19 |
6 |
0 |
|
68% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_getitem.py |
_key_cache_descriptor |
|
10 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_getitem.py |
_desc_action_to_expanded |
|
14 |
1 |
0 |
|
93% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_getitem.py |
_descriptor_to_expanded_actions |
|
27 |
3 |
0 |
|
89% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_getitem.py |
_unwrap_key_for_local |
|
5 |
2 |
0 |
|
60% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_getitem.py |
_copy_partial_state |
|
3 |
1 |
0 |
|
67% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_getitem.py |
_is_full_slice_action |
|
6 |
1 |
0 |
|
83% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_getitem.py |
GetItemDistributedOp.preprocess |
|
10 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_getitem.py |
GetItemDistributedOp._reject_bool_mask |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_getitem.py |
GetItemDistributedOp._validate_int_action |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_getitem.py |
GetItemDistributedOp._validate_slice_action |
|
7 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_getitem.py |
GetItemDistributedOp._validate_advanced_action |
|
7 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_getitem.py |
GetItemDistributedOp._validate_input_layouts |
|
13 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_getitem.py |
GetItemDistributedOp._infer_shard_dim0_int |
|
21 |
1 |
0 |
|
95% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_getitem.py |
GetItemDistributedOp.infer_layout |
|
18 |
1 |
0 |
|
94% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_getitem.py |
GetItemDistributedOp.get_expand_impl |
|
8 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_getitem.py |
GetItemDistributedOp.get_expand_impl.ragged_getitem_impl |
|
8 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_getitem.py |
GetItemDistributedOp.wrap_output |
|
3 |
1 |
0 |
|
67% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_getitem.py |
GetItemDistributedOp._infer_basic_output_layout |
|
16 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_getitem.py |
GetItemDistributedOp._compute_advanced_index_info |
|
16 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_getitem.py |
GetItemDistributedOp._infer_advanced_output_layout |
|
26 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_getitem.py |
GetItemDistributedOp._infer_advanced_output_layout._append_non_advanced |
|
4 |
1 |
0 |
|
75% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_getitem.py |
_broadcast_shapes |
|
15 |
5 |
0 |
|
67% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_getitem.py |
(no function) |
|
47 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_histc_ext.py |
_normalize_histc_args |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_histc_ext.py |
HistcExtDistributedOp.__init__ |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_histc_ext.py |
HistcExtDistributedOp.preprocess |
|
10 |
10 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_histc_ext.py |
HistcExtDistributedOp.infer_layout |
|
26 |
2 |
0 |
|
92% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_histc_ext.py |
(no function) |
|
10 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_inplace_scatter_value.py |
_normalize_inplace_scatter_value_args |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_inplace_scatter_value.py |
InplaceScatterValueDistributedOp.preprocess |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_inplace_scatter_value.py |
InplaceScatterValueDistributedOp.infer_layout |
|
27 |
1 |
0 |
|
96% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_inplace_scatter_value.py |
(no function) |
|
7 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_isin.py |
_normalize_isin_args |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_isin.py |
IsinDistributedOp.preprocess |
|
8 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_isin.py |
IsinDistributedOp.infer_layout |
|
14 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_isin.py |
(no function) |
|
7 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_masked_scatter.py |
_normalize_masked_scatter_args |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_masked_scatter.py |
MaskedScatterDistributedOp.preprocess |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_masked_scatter.py |
MaskedScatterDistributedOp.infer_layout |
|
13 |
1 |
0 |
|
92% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_masked_scatter.py |
(no function) |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_matmul.py |
_propagate_partial_from_inputs |
|
12 |
1 |
0 |
|
92% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_matmul.py |
_normalize_matmul_ext_args |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_matmul.py |
MatMulExtDistributedOp.preprocess |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_matmul.py |
MatMulExtDistributedOp.infer_layout |
|
25 |
5 |
0 |
|
80% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_matmul.py |
_normalize_matmul_args |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_matmul.py |
MatMulDistributedOp.preprocess |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_matmul.py |
MatMulDistributedOp.infer_layout |
|
34 |
5 |
0 |
|
85% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_matmul.py |
BaseBatchMatMulDistributedOp._merge_batch_entry |
|
9 |
2 |
0 |
|
78% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_matmul.py |
BaseBatchMatMulDistributedOp._is_none_entry |
|
3 |
1 |
0 |
|
67% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_matmul.py |
BaseBatchMatMulDistributedOp._merge_batches |
|
9 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_matmul.py |
BaseBatchMatMulDistributedOp._build_output_layout |
|
10 |
2 |
0 |
|
80% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_matmul.py |
_normalize_batch_matmul_ext_args |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_matmul.py |
BatchMatMulExtDistributedOp.preprocess |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_matmul.py |
BatchMatMulExtDistributedOp.infer_layout |
|
16 |
3 |
0 |
|
81% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_matmul.py |
_normalize_batch_matmul_args |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_matmul.py |
BatchMatMulDistributedOp.preprocess |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_matmul.py |
BatchMatMulDistributedOp.infer_layout |
|
24 |
5 |
0 |
|
79% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_matmul.py |
_normalize_linear_args |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_matmul.py |
LinearDistributedOp.preprocess |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_matmul.py |
LinearDistributedOp.infer_layout |
|
34 |
6 |
0 |
|
82% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_matmul.py |
LinearDistributedOp.get_expand_impl |
|
14 |
2 |
0 |
|
86% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_matmul.py |
LinearDistributedOp.get_expand_impl.expand_impl |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_matmul.py |
(no function) |
|
30 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_matmul_reduce_scatter.py |
_normalize_mrs_args |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_matmul_reduce_scatter.py |
MatmulReduceScatterDistributedOp.preprocess |
|
8 |
8 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_matmul_reduce_scatter.py |
MatmulReduceScatterDistributedOp._validate_input_layouts |
|
9 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_matmul_reduce_scatter.py |
MatmulReduceScatterDistributedOp.infer_layout |
|
17 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_matmul_reduce_scatter.py |
(no function) |
|
10 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_ms_flash_attention_score.py |
_resolve_input_layout |
|
5 |
1 |
0 |
|
80% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_ms_flash_attention_score.py |
_normalize_ms_flash_attention_score_args |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_ms_flash_attention_score.py |
FlashAttentionScoreDistributedOp.__init__ |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_ms_flash_attention_score.py |
FlashAttentionScoreDistributedOp.preprocess |
|
9 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_ms_flash_attention_score.py |
FlashAttentionScoreDistributedOp._to_python_scalar |
|
3 |
1 |
0 |
|
67% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_ms_flash_attention_score.py |
FlashAttentionScoreDistributedOp._validate_key_value_layout_identity |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_ms_flash_attention_score.py |
FlashAttentionScoreDistributedOp._validate_input_layouts |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_ms_flash_attention_score.py |
FlashAttentionScoreDistributedOp._is_dynamic_shape |
|
7 |
7 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_ms_flash_attention_score.py |
FlashAttentionScoreDistributedOp._get_dynamic_shape_info |
|
11 |
11 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_ms_flash_attention_score.py |
FlashAttentionScoreDistributedOp._is_attn_mask_compressed |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_ms_flash_attention_score.py |
FlashAttentionScoreDistributedOp._compute_sparse_params |
|
23 |
23 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_ms_flash_attention_score.py |
FlashAttentionScoreDistributedOp._compute_sparse_params_dynamic |
|
26 |
26 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_ms_flash_attention_score.py |
FlashAttentionScoreDistributedOp._adjust_actual_seq_len_for_tnd_cp |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_ms_flash_attention_score.py |
FlashAttentionScoreDistributedOp._adjust_actual_seq_len |
|
23 |
23 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_ms_flash_attention_score.py |
FlashAttentionScoreDistributedOp._validate_attn_mask |
|
14 |
14 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_ms_flash_attention_score.py |
FlashAttentionScoreDistributedOp._validate_real_shift_configuration |
|
7 |
7 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_ms_flash_attention_score.py |
FlashAttentionScoreDistributedOp.infer_layout |
|
24 |
2 |
0 |
|
92% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_ms_flash_attention_score.py |
FlashAttentionScoreDistributedOp._infer_softmax_layout_conservatively |
|
10 |
1 |
0 |
|
90% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_ms_flash_attention_score.py |
FlashAttentionScoreDistributedOp._infer_softmax_layout_by_input_layout |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_ms_flash_attention_score.py |
FlashAttentionScoreDistributedOp._build_softmax_tensor_map |
|
24 |
1 |
0 |
|
96% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_ms_flash_attention_score.py |
FlashAttentionScoreDistributedOp._validate_sharding_consistency |
|
10 |
2 |
0 |
|
80% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_ms_flash_attention_score.py |
FlashAttentionScoreDistributedOp._check_batch_consistency |
|
9 |
2 |
0 |
|
78% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_ms_flash_attention_score.py |
FlashAttentionScoreDistributedOp._check_hidden_consistency |
|
9 |
2 |
0 |
|
78% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_ms_flash_attention_score.py |
FlashAttentionScoreDistributedOp._check_dim_consistency |
|
9 |
1 |
0 |
|
89% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_ms_flash_attention_score.py |
FlashAttentionScoreDistributedOp._check_seq_sharding_compatibility |
|
17 |
17 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_ms_flash_attention_score.py |
FlashAttentionScoreDistributedOp._truncate_result |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_ms_flash_attention_score.py |
FlashAttentionScoreDistributedOp._adjust_head_num |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_ms_flash_attention_score.py |
FlashAttentionScoreDistributedOp._compute_adjusted_sparse_params |
|
9 |
9 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_ms_flash_attention_score.py |
FlashAttentionScoreDistributedOp._adjust_tnd_layout_params |
|
15 |
15 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_ms_flash_attention_score.py |
FlashAttentionScoreDistributedOp.get_expand_impl |
|
7 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_ms_flash_attention_score.py |
FlashAttentionScoreDistributedOp.get_expand_impl._expanded_impl |
|
24 |
24 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_ms_flash_attention_score.py |
FlashAttentionScoreDistributedOp._apply_seq_split_adjustments |
|
26 |
26 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_ms_flash_attention_score.py |
FlashAttentionScoreDistributedOp._get_seq_dim_idx |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_ms_flash_attention_score.py |
FlashAttentionScoreDistributedOp._normalize_dim_map |
|
3 |
1 |
0 |
|
67% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_ms_flash_attention_score.py |
FlashAttentionScoreDistributedOp._get_split_info |
|
15 |
15 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_ms_flash_attention_score.py |
FlashAttentionScoreDistributedOp._calculate_tnd_split_params |
|
11 |
11 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_ms_flash_attention_score.py |
FlashAttentionScoreDistributedOp._get_dim_split_num |
|
17 |
17 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_ms_flash_attention_score.py |
FlashAttentionScoreDistributedOp._get_split_id |
|
27 |
27 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_ms_flash_attention_score.py |
FlashAttentionScoreDistributedOp._validate_tnd_cp_requirements |
|
12 |
12 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_ms_flash_attention_score.py |
FlashAttentionScoreDistributedOp._create_replicated_placeholder_layout |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_ms_flash_attention_score.py |
(no function) |
|
82 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_nonzero.py |
_normalize_nonzero_args |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_nonzero.py |
NonzeroDistributedOp.preprocess |
|
7 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_nonzero.py |
NonzeroDistributedOp.infer_layout |
|
21 |
1 |
0 |
|
95% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_nonzero.py |
NonzeroDistributedOp.infer_layout._create_replicated_layout |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_nonzero.py |
(no function) |
|
7 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_norm.py |
_normalize_rmsnorm_args |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_norm.py |
_normalize_layernorm_args |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_norm.py |
NormDistributedOp.preprocess |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_norm.py |
NormDistributedOp.infer_layout |
|
29 |
1 |
0 |
|
97% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_norm.py |
LayerNormDistributedOp.preprocess |
|
10 |
10 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_norm.py |
LayerNormDistributedOp.infer_layout |
|
26 |
1 |
0 |
|
96% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_norm.py |
(no function) |
|
11 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_npu_flash_attention_score.py |
_set_lb_override |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_npu_flash_attention_score.py |
_clear_lb_override |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_npu_flash_attention_score.py |
_get_lb_override |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_npu_flash_attention_score.py |
_normalize_npu_fusion_attention_args |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_npu_flash_attention_score.py |
NPUFlashAttentionScoreDistributedOp.__init__ |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_npu_flash_attention_score.py |
NPUFlashAttentionScoreDistributedOp.preprocess |
|
10 |
10 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_npu_flash_attention_score.py |
NPUFlashAttentionScoreDistributedOp._is_dynamic_shape |
|
7 |
7 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_npu_flash_attention_score.py |
NPUFlashAttentionScoreDistributedOp._get_dynamic_shape_info |
|
11 |
11 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_npu_flash_attention_score.py |
NPUFlashAttentionScoreDistributedOp._is_attn_mask_compressed |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_npu_flash_attention_score.py |
NPUFlashAttentionScoreDistributedOp._validate_atten_mask_for_lb |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_npu_flash_attention_score.py |
NPUFlashAttentionScoreDistributedOp._adjust_atten_mask_for_seq_split |
|
9 |
9 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_npu_flash_attention_score.py |
NPUFlashAttentionScoreDistributedOp._compute_sparse_params |
|
23 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_npu_flash_attention_score.py |
NPUFlashAttentionScoreDistributedOp._compute_sparse_params_dynamic |
|
26 |
26 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_npu_flash_attention_score.py |
NPUFlashAttentionScoreDistributedOp._adjust_actual_seq_len_for_tnd_cp |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_npu_flash_attention_score.py |
NPUFlashAttentionScoreDistributedOp._adjust_actual_seq_len |
|
23 |
23 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_npu_flash_attention_score.py |
NPUFlashAttentionScoreDistributedOp._validate_atten_mask |
|
14 |
1 |
0 |
|
93% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_npu_flash_attention_score.py |
NPUFlashAttentionScoreDistributedOp._validate_pse_configuration |
|
7 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_npu_flash_attention_score.py |
NPUFlashAttentionScoreDistributedOp._validate_input_layouts |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_npu_flash_attention_score.py |
NPUFlashAttentionScoreDistributedOp.infer_layout |
|
21 |
2 |
0 |
|
90% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_npu_flash_attention_score.py |
NPUFlashAttentionScoreDistributedOp._infer_softmax_layout_conservatively |
|
10 |
10 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_npu_flash_attention_score.py |
NPUFlashAttentionScoreDistributedOp._infer_softmax_layout_by_input_layout |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_npu_flash_attention_score.py |
NPUFlashAttentionScoreDistributedOp._build_softmax_tensor_map |
|
24 |
1 |
0 |
|
96% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_npu_flash_attention_score.py |
NPUFlashAttentionScoreDistributedOp._create_default_softmax_layout |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_npu_flash_attention_score.py |
NPUFlashAttentionScoreDistributedOp._validate_sharding_consistency |
|
10 |
1 |
0 |
|
90% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_npu_flash_attention_score.py |
NPUFlashAttentionScoreDistributedOp._check_batch_consistency |
|
9 |
1 |
0 |
|
89% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_npu_flash_attention_score.py |
NPUFlashAttentionScoreDistributedOp._check_hidden_consistency |
|
9 |
1 |
0 |
|
89% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_npu_flash_attention_score.py |
NPUFlashAttentionScoreDistributedOp._check_dim_consistency |
|
9 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_npu_flash_attention_score.py |
NPUFlashAttentionScoreDistributedOp._check_seq_sharding_compatibility |
|
17 |
9 |
0 |
|
47% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_npu_flash_attention_score.py |
NPUFlashAttentionScoreDistributedOp._truncate_result |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_npu_flash_attention_score.py |
NPUFlashAttentionScoreDistributedOp._adjust_head_num |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_npu_flash_attention_score.py |
NPUFlashAttentionScoreDistributedOp._compute_adjusted_sparse_params |
|
9 |
9 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_npu_flash_attention_score.py |
NPUFlashAttentionScoreDistributedOp._adjust_tnd_layout_params |
|
15 |
15 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_npu_flash_attention_score.py |
NPUFlashAttentionScoreDistributedOp.get_expand_impl |
|
6 |
1 |
0 |
|
83% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_npu_flash_attention_score.py |
NPUFlashAttentionScoreDistributedOp.get_expand_impl._expanded_impl |
|
14 |
14 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_npu_flash_attention_score.py |
NPUFlashAttentionScoreDistributedOp._apply_seq_split_adjustments |
|
30 |
30 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_npu_flash_attention_score.py |
NPUFlashAttentionScoreDistributedOp._get_seq_dim_idx |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_npu_flash_attention_score.py |
NPUFlashAttentionScoreDistributedOp._normalize_dim_map |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_npu_flash_attention_score.py |
NPUFlashAttentionScoreDistributedOp._get_split_info |
|
15 |
1 |
0 |
|
93% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_npu_flash_attention_score.py |
NPUFlashAttentionScoreDistributedOp._calculate_tnd_split_params |
|
11 |
11 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_npu_flash_attention_score.py |
NPUFlashAttentionScoreDistributedOp._get_dim_split_num |
|
17 |
10 |
0 |
|
41% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_npu_flash_attention_score.py |
NPUFlashAttentionScoreDistributedOp._get_split_id |
|
27 |
27 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_npu_flash_attention_score.py |
NPUFlashAttentionScoreDistributedOp._validate_tnd_cp_requirements |
|
12 |
12 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_npu_flash_attention_score.py |
NPUFlashAttentionScoreDistributedOp._create_replicated_placeholder_layout |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_npu_flash_attention_score.py |
(no function) |
|
82 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_one_hot_ext.py |
_normalize_one_hot_ext_args |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_one_hot_ext.py |
OneHotExtDistributedOp.preprocess |
|
11 |
11 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_one_hot_ext.py |
OneHotExtDistributedOp.infer_layout |
|
20 |
1 |
0 |
|
95% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_one_hot_ext.py |
OneHotExtDistributedOp.get_expand_impl |
|
12 |
1 |
0 |
|
92% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_one_hot_ext.py |
OneHotExtDistributedOp.get_expand_impl.expanded_one_hot |
|
22 |
22 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_one_hot_ext.py |
OneHotExtDistributedOp._validate_num_classes |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_one_hot_ext.py |
OneHotExtDistributedOp._validate_indices_dtype |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_one_hot_ext.py |
OneHotExtDistributedOp._get_sharded_axes |
|
12 |
4 |
0 |
|
67% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_one_hot_ext.py |
OneHotExtDistributedOp._validate_axis |
|
5 |
1 |
0 |
|
80% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_one_hot_ext.py |
OneHotExtDistributedOp._validate_multi_dim_restriction |
|
9 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_one_hot_ext.py |
OneHotExtDistributedOp._validate_inputs_layouts |
|
6 |
3 |
0 |
|
50% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_one_hot_ext.py |
OneHotExtDistributedOp._infer_output_tensor_map |
|
9 |
1 |
0 |
|
89% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_one_hot_ext.py |
OneHotExtDistributedOp._create_layout_from_tensor_map |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_one_hot_ext.py |
OneHotExtDistributedOp._tensor_map_to_alias_tensor_map |
|
12 |
3 |
0 |
|
75% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_one_hot_ext.py |
(no function) |
|
19 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_ops.py |
DistributedOp.__init__ |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_ops.py |
DistributedOp._check_partial_inputs |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_ops.py |
DistributedOp.preprocess |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_ops.py |
DistributedOp.infer_layout |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_ops.py |
DistributedOp.get_expand_impl |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_ops.py |
DistributedOp.wrap_output |
|
11 |
1 |
0 |
|
91% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_ops.py |
(no function) |
|
10 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_ops_register.py |
register_distributed_op |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_ops_register.py |
get_distributed_op |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_ops_register.py |
(no function) |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_outer.py |
_normalize_outer_args |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_outer.py |
_get_alias_shard_set |
|
3 |
1 |
0 |
|
67% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_outer.py |
OuterDistributedOp.preprocess |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_outer.py |
OuterDistributedOp.infer_layout |
|
19 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_outer.py |
(no function) |
|
8 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_pad.py |
_normalize_pad_args |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_pad.py |
PadDistributedOp.preprocess |
|
7 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_pad.py |
PadDistributedOp.infer_layout |
|
26 |
3 |
0 |
|
88% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_pad.py |
(no function) |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_reduce.py |
_normalize_reduce_args |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_reduce.py |
ReduceExtDistributedOpBase.__init__ |
|
5 |
1 |
0 |
|
80% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_reduce.py |
ReduceExtDistributedOpBase.preprocess |
|
13 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_reduce.py |
ReduceExtDistributedOpBase.infer_layout |
|
11 |
3 |
0 |
|
73% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_reduce.py |
ReduceExtDistributedOpBase._infer_output_layout |
|
11 |
2 |
0 |
|
82% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_reduce.py |
ReduceExtDistributedOpBase._handle_all_axis_reduce |
|
7 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_reduce.py |
ReduceExtDistributedOpBase.replace_axis_with_none |
|
15 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_reduce.py |
ReduceExtDistributedOpBase._flatten_aliases |
|
6 |
1 |
0 |
|
83% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_reduce.py |
ReduceExtDistributedOpBase._replace_keepdim |
|
10 |
2 |
0 |
|
80% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_reduce.py |
ReduceExtDistributedOpBase._replace_dropdim |
|
13 |
4 |
0 |
|
69% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_reduce.py |
ReduceExtDistributedOpBase._compact_tensor_map |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_reduce.py |
ReduceExtDistributedOpBase._compact_tensor_map._compress |
|
8 |
6 |
0 |
|
25% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_reduce.py |
ReduceExtDistributedOpBase._apply_partial |
|
7 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_reduce.py |
SumExtDistributedOp.__init__ |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_reduce.py |
MeanExtDistributedOp.__init__ |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_reduce.py |
ReduceMaxDistributedOp.__init__ |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_reduce.py |
ProdExtDistributedOp.__init__ |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_reduce.py |
AllExtDistributedOp.__init__ |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_reduce.py |
MaxDistributedOp.__init__ |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_reduce.py |
MaxDistributedOp.preprocess |
|
18 |
1 |
0 |
|
94% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_reduce.py |
MaxDistributedOp.infer_layout |
|
18 |
2 |
0 |
|
89% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_reduce.py |
MinDistributedOp.__init__ |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_reduce.py |
(no function) |
|
39 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_repeat.py |
_normalize_repeat_args |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_repeat.py |
RepeatDistributedOp.preprocess |
|
7 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_repeat.py |
RepeatDistributedOp.infer_layout |
|
39 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_repeat.py |
(no function) |
|
7 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_repeat_interleave.py |
_normalize_repeat_interleave_args |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_repeat_interleave.py |
RepeatInterleaveDistributedOp._validate_input_layouts |
|
12 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_repeat_interleave.py |
RepeatInterleaveDistributedOp.preprocess |
|
15 |
2 |
0 |
|
87% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_repeat_interleave.py |
RepeatInterleaveDistributedOp.infer_layout |
|
21 |
3 |
0 |
|
86% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_repeat_interleave.py |
(no function) |
|
10 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_reshape.py |
_normalize_reshape_args |
|
11 |
5 |
0 |
|
55% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_reshape.py |
_filter_none_split_tensor_map |
|
14 |
9 |
0 |
|
36% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_reshape.py |
ReshapeDistributedOp.__init__ |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_reshape.py |
ReshapeDistributedOp._get_dynamic_shape_info |
|
7 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_reshape.py |
ReshapeDistributedOp._handle_dynamic_shape |
|
22 |
3 |
0 |
|
86% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_reshape.py |
ReshapeDistributedOp._merge_unshared_axis |
|
13 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_reshape.py |
ReshapeDistributedOp._cal_output_layout_and_dst_shape |
|
19 |
8 |
0 |
|
58% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_reshape.py |
ReshapeDistributedOp._normalize_shape |
|
5 |
2 |
0 |
|
60% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_reshape.py |
ReshapeDistributedOp._compute_output_tensor_map |
|
16 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_reshape.py |
ReshapeDistributedOp._validate_reshape_shard |
|
10 |
3 |
0 |
|
70% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_reshape.py |
ReshapeDistributedOp._apply_partial_status |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_reshape.py |
ReshapeDistributedOp.preprocess |
|
7 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_reshape.py |
ReshapeDistributedOp._infer_reshape_layout |
|
12 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_reshape.py |
ReshapeDistributedOp.infer_layout |
|
7 |
1 |
0 |
|
86% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_reshape.py |
ReshapeDistributedOp.get_expand_impl |
|
8 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_reshape.py |
ReshapeDistributedOp.get_expand_impl.expand_impl |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_reshape.py |
(no function) |
|
22 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_rotary_position_embedding.py |
_normalize_rpe_args |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_rotary_position_embedding.py |
_normalize_npu_rotary_mul_args |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_rotary_position_embedding.py |
RotaryPositionEmbeddingDistributedOp._validate_input_layouts |
|
11 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_rotary_position_embedding.py |
RotaryPositionEmbeddingDistributedOp.preprocess |
|
13 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_rotary_position_embedding.py |
RotaryPositionEmbeddingDistributedOp.infer_layout |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_rotary_position_embedding.py |
(no function) |
|
10 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_scaled_dot_product_attention.py |
_normalize_sdpa_args |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_scaled_dot_product_attention.py |
ScaledDotProductAttentionDistributedOp.preprocess |
|
9 |
9 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_scaled_dot_product_attention.py |
ScaledDotProductAttentionDistributedOp._normalize_dim_map |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_scaled_dot_product_attention.py |
ScaledDotProductAttentionDistributedOp._get_dims |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_scaled_dot_product_attention.py |
ScaledDotProductAttentionDistributedOp._get_dim_split_num |
|
17 |
8 |
0 |
|
53% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_scaled_dot_product_attention.py |
ScaledDotProductAttentionDistributedOp._get_split_info |
|
8 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_scaled_dot_product_attention.py |
ScaledDotProductAttentionDistributedOp._get_split_id |
|
26 |
16 |
0 |
|
38% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_scaled_dot_product_attention.py |
ScaledDotProductAttentionDistributedOp._validate_sharding_consistency |
|
16 |
1 |
0 |
|
94% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_scaled_dot_product_attention.py |
ScaledDotProductAttentionDistributedOp._build_causal_mask_for_chunk |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_scaled_dot_product_attention.py |
ScaledDotProductAttentionDistributedOp._adjust_attn_mask_for_sp |
|
17 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_scaled_dot_product_attention.py |
ScaledDotProductAttentionDistributedOp._validate_input_layouts |
|
10 |
1 |
0 |
|
90% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_scaled_dot_product_attention.py |
ScaledDotProductAttentionDistributedOp._validate_tensor_strategy |
|
8 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_scaled_dot_product_attention.py |
ScaledDotProductAttentionDistributedOp._validate_mesh_kv_identity |
|
8 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_scaled_dot_product_attention.py |
ScaledDotProductAttentionDistributedOp._validate_sharding_forbidden |
|
7 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_scaled_dot_product_attention.py |
ScaledDotProductAttentionDistributedOp.infer_layout |
|
13 |
1 |
0 |
|
92% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_scaled_dot_product_attention.py |
ScaledDotProductAttentionDistributedOp.get_expand_impl |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_scaled_dot_product_attention.py |
ScaledDotProductAttentionDistributedOp.get_expand_impl._expanded_impl |
|
16 |
4 |
0 |
|
75% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_scaled_dot_product_attention.py |
(no function) |
|
30 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_scatter.py |
_normalize_scatter_args |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_scatter.py |
ScatterDistributedOp.preprocess |
|
9 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_scatter.py |
ScatterDistributedOp.infer_layout |
|
28 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_scatter.py |
(no function) |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_scatter_update.py |
_normalize_scatter_update_args |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_scatter_update.py |
ScatterUpdateDistributedOp.__init__ |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_scatter_update.py |
ScatterUpdateDistributedOp.preprocess |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_scatter_update.py |
ScatterUpdateDistributedOp.infer_layout |
|
17 |
2 |
0 |
|
88% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_scatter_update.py |
ScatterUpdateDistributedOp._validate_strategy |
|
23 |
2 |
0 |
|
91% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_scatter_update.py |
(no function) |
|
9 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_setitem.py |
_normalize___setitem___args |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_setitem.py |
_compute_lhs_shape |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_setitem.py |
_compute_basic_lhs_shape |
|
23 |
23 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_setitem.py |
_compute_slice_len |
|
15 |
15 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_setitem.py |
_append_non_advanced_shape |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_setitem.py |
_compute_advanced_lhs_shape |
|
27 |
27 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_setitem.py |
_compute_lhs_layout |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_setitem.py |
_build_broadcast_value_layout |
|
8 |
8 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_setitem.py |
SetItemDistributedOp.preprocess |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_setitem.py |
SetItemDistributedOp._preprocess_setitem_key |
|
11 |
11 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_setitem.py |
SetItemDistributedOp._preprocess_setitem_value |
|
20 |
20 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_setitem.py |
SetItemDistributedOp.infer_layout |
|
11 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_setitem.py |
SetItemDistributedOp._validate_input_layouts |
|
10 |
1 |
0 |
|
90% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_setitem.py |
SetItemDistributedOp._validate_dtensor_value |
|
5 |
1 |
0 |
|
80% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_setitem.py |
SetItemDistributedOp._validate_value_layout |
|
13 |
2 |
0 |
|
85% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_setitem.py |
SetItemDistributedOp.wrap_output |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_setitem.py |
_validate_value_broadcast |
|
6 |
1 |
0 |
|
83% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_setitem.py |
(no function) |
|
31 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_slice.py |
_normalize_slice_args |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_slice.py |
SliceDistributedOp.preprocess |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_slice.py |
SliceDistributedOp._is_shard_dim |
|
14 |
6 |
0 |
|
57% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_slice.py |
SliceDistributedOp._check_layout |
|
8 |
1 |
0 |
|
88% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_slice.py |
SliceDistributedOp.infer_layout |
|
10 |
2 |
0 |
|
80% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_slice.py |
SliceDistributedOp.get_expand_impl |
|
9 |
2 |
0 |
|
78% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_slice.py |
SliceDistributedOp.get_expand_impl.expand_impl |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_slice.py |
(no function) |
|
9 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_slice_ext.py |
_normalize_slice_ext_args |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_slice_ext.py |
SliceExtDistributedOp.preprocess |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_slice_ext.py |
SliceExtDistributedOp.infer_layout |
|
15 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_slice_ext.py |
(no function) |
|
7 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_sort.py |
_normalize_sort_args |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_sort.py |
SortDistributedOp.preprocess |
|
12 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_sort.py |
SortDistributedOp.infer_layout |
|
18 |
2 |
0 |
|
89% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_sort.py |
(no function) |
|
7 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_split.py |
_normalize_split_with_size_args |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_split.py |
_normalize_split_args |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_split.py |
_normalize_split_tensor_args |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_split.py |
_normalize_tensor_split_args |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_split.py |
SplitWithSizeDistributedOp.preprocess |
|
7 |
7 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_split.py |
SplitWithSizeDistributedOp.infer_layout |
|
14 |
2 |
0 |
|
86% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_split.py |
SplitWithSizeViewDistributedOp.preprocess |
|
7 |
7 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_split.py |
SplitWithSizeViewDistributedOp.infer_layout |
|
14 |
14 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_split.py |
SplitDistributedOp.preprocess |
|
9 |
9 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_split.py |
SplitDistributedOp.infer_layout |
|
14 |
1 |
0 |
|
93% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_split.py |
SplitTensorDistributedOp.preprocess |
|
7 |
7 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_split.py |
SplitTensorDistributedOp.infer_layout |
|
14 |
2 |
0 |
|
86% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_split.py |
SplitTensorViewDistributedOp.preprocess |
|
7 |
7 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_split.py |
SplitTensorViewDistributedOp.infer_layout |
|
14 |
14 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_split.py |
TensorSplitDistributedOp.preprocess |
|
16 |
10 |
0 |
|
38% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_split.py |
TensorSplitDistributedOp.infer_layout |
|
14 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_split.py |
(no function) |
|
26 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_squeeze.py |
_normalize_squeeze_args |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_squeeze.py |
SqueezeDistributedOp.preprocess |
|
8 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_squeeze.py |
SqueezeDistributedOp.infer_layout |
|
15 |
1 |
0 |
|
93% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_squeeze.py |
SqueezeDistributedOp._compute_squeeze_layout |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_squeeze.py |
SqueezeDistributedOp._handle_scalar_case |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_squeeze.py |
SqueezeDistributedOp._validate_input_shape |
|
4 |
1 |
0 |
|
75% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_squeeze.py |
SqueezeDistributedOp._get_dims_to_squeeze |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_squeeze.py |
SqueezeDistributedOp._get_all_squeezable_dims |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_squeeze.py |
SqueezeDistributedOp._get_specified_dims_to_squeeze |
|
13 |
3 |
0 |
|
77% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_squeeze.py |
SqueezeDistributedOp._validate_axis_range |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_squeeze.py |
SqueezeDistributedOp._validate_axis_for_squeeze |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_squeeze.py |
SqueezeDistributedOp._create_output_layout |
|
13 |
2 |
0 |
|
85% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_squeeze.py |
SqueezeDistributedOp._copy_partial_operations |
|
5 |
3 |
0 |
|
40% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_squeeze.py |
(no function) |
|
18 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_stack.py |
_normalize_stack_args |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_stack.py |
StackDistributedOp.preprocess |
|
9 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_stack.py |
StackDistributedOp.infer_layout |
|
28 |
1 |
0 |
|
96% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_stack.py |
StackDistributedOp.infer_layout.idx_to_alias |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_stack.py |
(no function) |
|
7 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_swiglu.py |
_normalize_swiglu_args |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_swiglu.py |
SwiGLUDistributedOp.preprocess |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_swiglu.py |
SwiGLUDistributedOp.infer_layout |
|
17 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_swiglu.py |
(no function) |
|
7 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_topk.py |
_normalize_topk_args |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_topk.py |
TopKDistributedOp.preprocess |
|
10 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_topk.py |
TopKDistributedOp.infer_layout |
|
22 |
1 |
0 |
|
95% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_topk.py |
(no function) |
|
7 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_transpose.py |
_normalize_transpose_args |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_transpose.py |
TransposeDistributedOp.preprocess |
|
14 |
1 |
0 |
|
93% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_transpose.py |
TransposeDistributedOp.infer_layout |
|
33 |
2 |
0 |
|
94% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_transpose.py |
(no function) |
|
7 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_tuple_elementwise.py |
_unwrap_local_value |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_tuple_elementwise.py |
TupleElementWiseDistributedOp.preprocess |
|
12 |
2 |
0 |
|
83% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_tuple_elementwise.py |
TupleElementWiseDistributedOp.infer_layout |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_tuple_elementwise.py |
(no function) |
|
7 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_unbind.py |
_normalize_unbind_args |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_unbind.py |
UnbindDistributedOp.preprocess |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_unbind.py |
UnbindDistributedOp.infer_layout |
|
20 |
1 |
0 |
|
95% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_unbind.py |
(no function) |
|
7 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_vstack.py |
_normalize_vstack_args |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_vstack.py |
_promote_tensor_map |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_vstack.py |
VstackDistributedOp.preprocess |
|
13 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_vstack.py |
VstackDistributedOp.infer_layout |
|
12 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_vstack.py |
(no function) |
|
9 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / sharding_plan.py |
(no function) |
|
8 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / tensor_parallel / __init__.py |
__getattr__ |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / tensor_parallel / __init__.py |
__dir__ |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / tensor_parallel / __init__.py |
(no function) |
|
8 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / tensor_parallel / _ce_op_registry.py |
get_loss_parallel_op_names |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / tensor_parallel / _ce_op_registry.py |
get_decomposed_ce_op_names |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / tensor_parallel / _ce_op_registry.py |
register_loss_parallel_op_names |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / tensor_parallel / _ce_op_registry.py |
unregister_loss_parallel_op_names |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / tensor_parallel / _ce_op_registry.py |
is_loss_parallel_op |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / tensor_parallel / _ce_op_registry.py |
is_decomposed_ce_op |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / tensor_parallel / _ce_op_registry.py |
clear_extended_op_names |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / tensor_parallel / _ce_op_registry.py |
(no function) |
|
14 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / tensor_parallel / api.py |
_named_children |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / tensor_parallel / api.py |
_tensor_parallel_mesh_context |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / tensor_parallel / api.py |
_validate_tp_mesh_dim |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / tensor_parallel / api.py |
parallelize_module |
|
21 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / tensor_parallel / api.py |
parallelize_module._apply_path |
|
9 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / tensor_parallel / api.py |
(no function) |
|
14 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / tensor_parallel / loss_parallel.py |
loss_parallel |
|
22 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / tensor_parallel / loss_parallel.py |
is_loss_parallel_active |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / tensor_parallel / loss_parallel.py |
get_loss_parallel_count |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / tensor_parallel / loss_parallel.py |
_get_loss_parallel_token |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / tensor_parallel / loss_parallel.py |
_get_loss_parallel_mesh |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / tensor_parallel / loss_parallel.py |
_get_loss_parallel_strict |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / tensor_parallel / loss_parallel.py |
(no function) |
|
20 |
0 |
2 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / tensor_parallel / loss_parallel_ops_common.py |
_is_dtensor |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / tensor_parallel / loss_parallel_ops_common.py |
_is_shard_on_last_dim |
|
9 |
2 |
0 |
|
78% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / tensor_parallel / loss_parallel_ops_common.py |
_is_replicate |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / tensor_parallel / loss_parallel_ops_common.py |
_get_mesh_and_dim |
|
7 |
7 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / tensor_parallel / loss_parallel_ops_common.py |
_get_local_tensor |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / tensor_parallel / loss_parallel_ops_common.py |
_get_full_tensor |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / tensor_parallel / loss_parallel_ops_common.py |
_validate_target_type_base |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / tensor_parallel / loss_parallel_ops_common.py |
_validate_mesh_and_shard |
|
9 |
9 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / tensor_parallel / loss_parallel_ops_common.py |
_validate_cross_entropy_params |
|
14 |
14 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / tensor_parallel / loss_parallel_ops_common.py |
_check_context_and_layout |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / tensor_parallel / loss_parallel_ops_common.py |
(no function) |
|
17 |
0 |
2 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / tensor_parallel / mc2.py |
get_hcomm_info |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / tensor_parallel / mc2.py |
_require_torch_npu |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / tensor_parallel / mc2.py |
_normalize_sequence_dim |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / tensor_parallel / mc2.py |
_move_dim_to_front |
|
4 |
1 |
0 |
|
75% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / tensor_parallel / mc2.py |
_move_front_to_dim |
|
4 |
1 |
0 |
|
75% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / tensor_parallel / mc2.py |
AllGatherMatmulFunction.forward |
|
11 |
11 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / tensor_parallel / mc2.py |
AllGatherMatmulFunction.backward |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / tensor_parallel / mc2.py |
MatmulReduceScatterFunction.forward |
|
11 |
11 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / tensor_parallel / mc2.py |
MatmulReduceScatterFunction.backward |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / tensor_parallel / mc2.py |
MC2Linear.configure_mc2 |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / tensor_parallel / mc2.py |
MC2Linear.from_linear |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / tensor_parallel / mc2.py |
MC2Linear._mc2_forward |
|
25 |
25 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / tensor_parallel / mc2.py |
MC2Linear.forward |
|
9 |
9 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / tensor_parallel / mc2.py |
(no function) |
|
28 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / tensor_parallel / mc2_style.py |
_replace_with_mc2_linear |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / tensor_parallel / mc2_style.py |
MC2ColwiseParallel.__init__ |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / tensor_parallel / mc2_style.py |
MC2ColwiseParallel.apply |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / tensor_parallel / mc2_style.py |
MC2RowwiseParallel.__init__ |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / tensor_parallel / mc2_style.py |
MC2RowwiseParallel.apply |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / tensor_parallel / mc2_style.py |
(no function) |
|
15 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / tensor_parallel / style.py |
_cast_fp_tensor |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / tensor_parallel / style.py |
_src_data_rank_for_tensor |
|
3 |
1 |
0 |
|
67% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / tensor_parallel / style.py |
ParallelStyle.apply |
|
0 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / tensor_parallel / style.py |
ColwiseParallel.__init__ |
|
8 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / tensor_parallel / style.py |
ColwiseParallel.__repr__ |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / tensor_parallel / style.py |
ColwiseParallel._prepare_input_fn |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / tensor_parallel / style.py |
ColwiseParallel._partition_linear_fn |
|
8 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / tensor_parallel / style.py |
ColwiseParallel._partition_embedding_fn |
|
8 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / tensor_parallel / style.py |
ColwiseParallel._prepare_output_fn |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / tensor_parallel / style.py |
ColwiseParallel.apply |
|
9 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / tensor_parallel / style.py |
ColwiseParallel.apply.partition_fn |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / tensor_parallel / style.py |
ColwiseParallel.apply.input_fn |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / tensor_parallel / style.py |
ColwiseParallel.apply.output_fn |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / tensor_parallel / style.py |
RowwiseParallel.__init__ |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / tensor_parallel / style.py |
RowwiseParallel.__repr__ |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / tensor_parallel / style.py |
RowwiseParallel._prepare_input_fn |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / tensor_parallel / style.py |
RowwiseParallel._partition_linear_fn |
|
9 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / tensor_parallel / style.py |
RowwiseParallel._partition_embedding_fn |
|
8 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / tensor_parallel / style.py |
RowwiseParallel._prepare_output_fn |
|
12 |
2 |
0 |
|
83% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / tensor_parallel / style.py |
RowwiseParallel.apply |
|
11 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / tensor_parallel / style.py |
RowwiseParallel.apply.partition_fn |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / tensor_parallel / style.py |
RowwiseParallel.apply.input_fn |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / tensor_parallel / style.py |
RowwiseParallel.apply.output_fn |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / tensor_parallel / style.py |
SequenceParallel.__init__ |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / tensor_parallel / style.py |
SequenceParallel.__repr__ |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / tensor_parallel / style.py |
SequenceParallel._prepare_input_fn |
|
8 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / tensor_parallel / style.py |
SequenceParallel._prepare_output_fn |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / tensor_parallel / style.py |
SequenceParallel.apply |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / tensor_parallel / style.py |
SequenceParallel.apply.partition_fn |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / tensor_parallel / style.py |
SequenceParallel.apply.input_fn |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / tensor_parallel / style.py |
SequenceParallel.apply.output_fn |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / tensor_parallel / style.py |
PrepareModuleInput.__init__ |
|
15 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / tensor_parallel / style.py |
PrepareModuleInput._prepare_input_arg |
|
10 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / tensor_parallel / style.py |
PrepareModuleInput._prepare_input_fn |
|
10 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / tensor_parallel / style.py |
PrepareModuleInput._prepare_input_kwarg_fn |
|
8 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / tensor_parallel / style.py |
PrepareModuleInput.apply |
|
7 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / tensor_parallel / style.py |
PrepareModuleInput.apply._pre_hook |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / tensor_parallel / style.py |
PrepareModuleInput.__repr__ |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / tensor_parallel / style.py |
PrepareModuleOutput.__init__ |
|
7 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / tensor_parallel / style.py |
PrepareModuleOutput._prepare_out_fn |
|
21 |
2 |
0 |
|
90% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / tensor_parallel / style.py |
PrepareModuleOutput.apply |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / tensor_parallel / style.py |
PrepareModuleOutput.apply._hook |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / tensor_parallel / style.py |
PrepareModuleOutput.__repr__ |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / tensor_parallel / style.py |
PrepareModuleInputOutput.__init__ |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / tensor_parallel / style.py |
PrepareModuleInputOutput.apply |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / tensor_parallel / style.py |
PrepareModuleInputOutput.__repr__ |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / tensor_parallel / style.py |
NoParallel.__init__ |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / tensor_parallel / style.py |
NoParallel.__repr__ |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / tensor_parallel / style.py |
NoParallel._prepare_input_fn |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / tensor_parallel / style.py |
NoParallel._prepare_output_fn |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / tensor_parallel / style.py |
NoParallel.apply |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / tensor_parallel / style.py |
NoParallel.apply.input_fn |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / tensor_parallel / style.py |
NoParallel.apply.output_fn |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / tensor_parallel / style.py |
(no function) |
|
66 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / utils / __init__.py |
(no function) |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / utils / clip_grad.py |
_is_dtensor |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / utils / clip_grad.py |
_str_to_reduce_op |
|
7 |
2 |
0 |
|
71% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / utils / clip_grad.py |
_normalize_parameters |
|
5 |
2 |
0 |
|
60% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / utils / clip_grad.py |
_param_device |
|
3 |
1 |
0 |
|
67% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / utils / clip_grad.py |
_get_grad_obj |
|
4 |
1 |
0 |
|
75% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / utils / clip_grad.py |
_get_local_grad |
|
8 |
3 |
0 |
|
62% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / utils / clip_grad.py |
_get_param_mesh_info |
|
7 |
1 |
0 |
|
86% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / utils / clip_grad.py |
_sum_p_norms |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / utils / clip_grad.py |
_foreach_p_norms |
|
13 |
13 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / utils / clip_grad.py |
_per_tensor_norms |
|
17 |
5 |
0 |
|
71% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / utils / clip_grad.py |
_compute_local_norm |
|
18 |
18 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / utils / clip_grad.py |
_get_total_norm |
|
8 |
3 |
0 |
|
62% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / utils / clip_grad.py |
_total_norm_inf |
|
14 |
14 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / utils / clip_grad.py |
_total_norm_sum |
|
9 |
9 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / utils / clip_grad.py |
_reduction_signature |
|
15 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / utils / clip_grad.py |
_total_norm_fsdp2_aligned |
|
15 |
1 |
0 |
|
93% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / utils / clip_grad.py |
_build_coalesce_buffer |
|
20 |
7 |
0 |
|
65% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / utils / clip_grad.py |
_coalesce_partial_reduce |
|
31 |
5 |
0 |
|
84% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / utils / clip_grad.py |
_build_grad_groups |
|
37 |
6 |
0 |
|
84% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / utils / clip_grad.py |
_clip_grads_with_norm_ |
|
17 |
6 |
0 |
|
65% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / utils / clip_grad.py |
clip_grad_norm_ |
|
15 |
3 |
0 |
|
80% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / utils / clip_grad.py |
(no function) |
|
44 |
4 |
0 |
|
91% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / utils / shape_utils.py |
compute_local_shape_and_global_offset |
|
20 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / utils / shape_utils.py |
compute_local_shape_and_global_offset_by_ceil_chunk |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / utils / shape_utils.py |
(no function) |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / __init__.py |
(no function) |
|
0 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / batching / __init__.py |
(no function) |
|
8 |
8 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / batching / attention_runtime.py |
build_dense_attention_masks |
|
24 |
24 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / batching / attention_runtime.py |
AttentionRuntimeAdapter.build_packed_seq_params |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / batching / attention_runtime.py |
(no function) |
|
9 |
9 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / batching / build_collate_fn.py |
_get_sequence_parallel_size |
|
8 |
8 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / batching / build_collate_fn.py |
DataCollator.__call__ |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / batching / build_collate_fn.py |
TextPackingCollator.__call__ |
|
23 |
23 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / batching / build_collate_fn.py |
MainCollator.__call__ |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / batching / build_collate_fn.py |
build_indexed_collate_fn |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / batching / build_collate_fn.py |
build_online_text_collate_fn |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / batching / build_collate_fn.py |
(no function) |
|
22 |
22 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / batching / build_dataloader.py |
calculate_num_micro_batches |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / batching / build_dataloader.py |
_is_iterable_dataset |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / batching / build_dataloader.py |
_supports_output_index_for_resume |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / batching / build_dataloader.py |
_normalize_source_samples |
|
11 |
11 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / batching / build_dataloader.py |
_restore_index_buffer |
|
21 |
21 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / batching / build_dataloader.py |
_OutputIndexDataset.__init__ |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / batching / build_dataloader.py |
_OutputIndexDataset.__len__ |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / batching / build_dataloader.py |
_OutputIndexDataset.__getitem__ |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / batching / build_dataloader.py |
_OutputIndexDataset.get_item |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / batching / build_dataloader.py |
build_dataloader |
|
35 |
35 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / batching / build_dataloader.py |
FixedBatchDataLoader.__init__ |
|
10 |
10 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / batching / build_dataloader.py |
FixedBatchDataLoader.set_epoch |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / batching / build_dataloader.py |
TextTokenBatcher.__post_init__ |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / batching / build_dataloader.py |
TextTokenBatcher.put_item |
|
7 |
7 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / batching / build_dataloader.py |
TextTokenBatcher.is_ready_for_micro_batch |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / batching / build_dataloader.py |
TextTokenBatcher.get_micro_batch |
|
17 |
17 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / batching / build_dataloader.py |
TextTokenBatcher.empty |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / batching / build_dataloader.py |
_IndexBufferDynamicBatchRuntime.__init__ |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / batching / build_dataloader.py |
_IndexBufferDynamicBatchRuntime.put_source_item |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / batching / build_dataloader.py |
_IndexBufferDynamicBatchRuntime.get_buffer_state |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / batching / build_dataloader.py |
_IndexBufferDynamicBatchRuntime.restore_buffer |
|
7 |
7 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / batching / build_dataloader.py |
_FullBufferDynamicBatchRuntime.__init__ |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / batching / build_dataloader.py |
_FullBufferDynamicBatchRuntime.put_source_item |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / batching / build_dataloader.py |
_FullBufferDynamicBatchRuntime.get_buffer_state |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / batching / build_dataloader.py |
_FullBufferDynamicBatchRuntime.restore_buffer |
|
7 |
7 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / batching / build_dataloader.py |
DynamicBatchDataLoader.__init__ |
|
34 |
34 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / batching / build_dataloader.py |
DynamicBatchDataLoader.__iter__ |
|
8 |
8 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / batching / build_dataloader.py |
DynamicBatchDataLoader._batch_data_generator |
|
15 |
15 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / batching / build_dataloader.py |
DynamicBatchDataLoader.state_dict |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / batching / build_dataloader.py |
DynamicBatchDataLoader.load_state_dict |
|
15 |
15 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / batching / build_dataloader.py |
DynamicBatchDataLoader.set_epoch |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / batching / build_dataloader.py |
(no function) |
|
57 |
57 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / batching / data_collator.py |
_validate_positive_int |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / batching / data_collator.py |
calculate_num_micro_batches |
|
7 |
7 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / batching / data_collator.py |
MakeMicroBatchCollator.__post_init__ |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / batching / data_collator.py |
MakeMicroBatchCollator._normalize_feature |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / batching / data_collator.py |
MakeMicroBatchCollator.__call__ |
|
13 |
13 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / batching / data_collator.py |
(no function) |
|
18 |
18 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / batching / dataloader.py |
DataLoader.__init__ |
|
7 |
7 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / batching / dataloader.py |
DataLoader.set_epoch |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / batching / dataloader.py |
(no function) |
|
9 |
9 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / batching / get_batch.py |
ParallelBatch.__init__ |
|
27 |
27 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / batching / get_batch.py |
ParallelBatch.__call__ |
|
21 |
21 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / batching / get_batch.py |
ParallelBatch._read_source_batch |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / batching / get_batch.py |
ParallelBatch._normalize_source_batch |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / batching / get_batch.py |
ParallelBatch._resolve_sequence_boundaries |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / batching / get_batch.py |
ParallelBatch._build_local_position_ids |
|
24 |
24 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / batching / get_batch.py |
ParallelBatch._build_loss_mask |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / batching / get_batch.py |
ParallelBatch._build_attention_data |
|
13 |
13 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / batching / get_batch.py |
ParallelBatch._split_model_and_loss_inputs |
|
11 |
11 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / batching / get_batch.py |
(no function) |
|
19 |
19 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / batching / sequence_boundaries.py |
OnlineBoundaryResolver.resolve |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / batching / sequence_boundaries.py |
IndexedBoundaryResolver.__init__ |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / batching / sequence_boundaries.py |
IndexedBoundaryResolver.resolve |
|
21 |
21 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / batching / sequence_boundaries.py |
(no function) |
|
9 |
9 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / constants.py |
(no function) |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / dataset_logging.py |
_get_rank |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / dataset_logging.py |
_DatasetDebugRankFilter.__init__ |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / dataset_logging.py |
_DatasetDebugRankFilter.filter |
|
8 |
8 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / dataset_logging.py |
_DatasetLogFormatter.formatTime |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / dataset_logging.py |
DatasetLogger.debug |
|
10 |
8 |
0 |
|
20% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / dataset_logging.py |
get_dataset_logger |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / dataset_logging.py |
enable_dataset_logging |
|
21 |
21 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / dataset_logging.py |
(no function) |
|
22 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / indexed / __init__.py |
_extend_native_path |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / indexed / __init__.py |
(no function) |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / indexed / indexed_blended_dataset.py |
BlendedDataset.__init__ |
|
18 |
18 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / indexed / indexed_blended_dataset.py |
BlendedDataset._collect_dataset_identifiers |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / indexed / indexed_blended_dataset.py |
BlendedDataset.__len__ |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / indexed / indexed_blended_dataset.py |
BlendedDataset.__getitem__ |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / indexed / indexed_blended_dataset.py |
BlendedDataset._build_indices |
|
23 |
23 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / indexed / indexed_blended_dataset.py |
BlendedDataset._build_indices_in_memory |
|
9 |
9 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / indexed / indexed_blended_dataset.py |
_normalize_weights |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / indexed / indexed_blended_dataset.py |
(no function) |
|
20 |
20 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / indexed / indexed_data_config.py |
BlendedMegatronDatasetConfig.__post_init__ |
|
33 |
33 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / indexed / indexed_data_config.py |
GPTDatasetConfig.__post_init__ |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / indexed / indexed_data_config.py |
_discover_indexed_paths |
|
26 |
26 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / indexed / indexed_data_config.py |
resolve_data_paths |
|
26 |
26 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / indexed / indexed_data_config.py |
build_gpt_dataset_config |
|
13 |
13 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / indexed / indexed_data_config.py |
_walk_bin_directories |
|
8 |
8 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / indexed / indexed_data_config.py |
(no function) |
|
43 |
43 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / indexed / indexed_data_reader.py |
_IndexReader.__init__ |
|
20 |
20 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / indexed / indexed_data_reader.py |
_IndexReader.__del__ |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / indexed / indexed_data_reader.py |
_IndexReader.__len__ |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / indexed / indexed_data_reader.py |
_IndexReader.__getitem__ |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / indexed / indexed_data_reader.py |
IndexedDataReader.__init__ |
|
7 |
7 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / indexed / indexed_data_reader.py |
IndexedDataReader.initialize |
|
20 |
20 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / indexed / indexed_data_reader.py |
IndexedDataReader.__getstate__ |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / indexed / indexed_data_reader.py |
IndexedDataReader.__setstate__ |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / indexed / indexed_data_reader.py |
IndexedDataReader.__del__ |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / indexed / indexed_data_reader.py |
IndexedDataReader.__len__ |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / indexed / indexed_data_reader.py |
IndexedDataReader._getitem_mmap |
|
17 |
17 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / indexed / indexed_data_reader.py |
IndexedDataReader._getitem_file |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / indexed / indexed_data_reader.py |
IndexedDataReader.__getitem__ |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / indexed / indexed_data_reader.py |
IndexedDataReader.get |
|
17 |
17 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / indexed / indexed_data_reader.py |
IndexedDataReader.sequence_lengths |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / indexed / indexed_data_reader.py |
IndexedDataReader.document_indices |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / indexed / indexed_data_reader.py |
IndexedDataReader.get_document_indices |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / indexed / indexed_data_reader.py |
IndexedDataReader.set_document_indices |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / indexed / indexed_data_reader.py |
IndexedDataReader._require_index |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / indexed / indexed_data_reader.py |
(no function) |
|
37 |
37 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / indexed / indexed_dataset.py |
IndexedPretrainDatasetBuilder.__init__ |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / indexed / indexed_dataset.py |
IndexedPretrainDatasetBuilder.build |
|
13 |
13 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / indexed / indexed_dataset.py |
IndexedPretrainDatasetBuilder._build_dataset_splits |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / indexed / indexed_dataset.py |
IndexedPretrainDatasetBuilder._build_pretrain_dataset_splits |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / indexed / indexed_dataset.py |
IndexedPretrainDatasetBuilder._build_instruction_dataset_splits |
|
10 |
10 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / indexed / indexed_dataset.py |
IndexedPretrainDatasetBuilder._resolve_data_paths |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / indexed / indexed_dataset.py |
IndexedPretrainDatasetBuilder._build_gpt_dataset_config |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / indexed / indexed_dataset.py |
IndexedPretrainDatasetBuilder._select_dataset_type |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / indexed / indexed_dataset.py |
IndexedPretrainDatasetBuilder._select_split_builder |
|
9 |
9 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / indexed / indexed_dataset.py |
build_indexed_dataset |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / indexed / indexed_dataset.py |
(no function) |
|
25 |
25 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / indexed / indexed_helpers.py |
build_sample_index |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / indexed / indexed_helpers.py |
build_blending_indices |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / indexed / indexed_helpers.py |
(no function) |
|
29 |
29 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / indexed / indexed_lazy_dataset.py |
LazyDatasetProxy.__init__ |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / indexed / indexed_lazy_dataset.py |
LazyDatasetProxy.__len__ |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / indexed / indexed_lazy_dataset.py |
LazyDatasetProxy.__getitem__ |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / indexed / indexed_lazy_dataset.py |
LazyDatasetProxy.unique_identifiers |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / indexed / indexed_lazy_dataset.py |
LazyDatasetProxy._get_dataset |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / indexed / indexed_lazy_dataset.py |
(no function) |
|
12 |
12 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / indexed / indexed_pretrain_dataset.py |
_build_document_index |
|
10 |
10 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / indexed / indexed_pretrain_dataset.py |
_build_shuffle_index |
|
9 |
9 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / indexed / indexed_pretrain_dataset.py |
_IndexedPretrainDataset.__init__ |
|
33 |
33 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / indexed / indexed_pretrain_dataset.py |
_IndexedPretrainDataset._build_ltor_fields |
|
12 |
12 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / indexed / indexed_pretrain_dataset.py |
_IndexedPretrainDataset._finalize |
|
0 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / indexed / indexed_pretrain_dataset.py |
_IndexedPretrainDataset.numel_low_level_dataset |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / indexed / indexed_pretrain_dataset.py |
_IndexedPretrainDataset.build_low_level_dataset |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / indexed / indexed_pretrain_dataset.py |
_IndexedPretrainDataset._key_config_attributes |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / indexed / indexed_pretrain_dataset.py |
_IndexedPretrainDataset.build_unique_identifiers |
|
8 |
8 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / indexed / indexed_pretrain_dataset.py |
_IndexedPretrainDataset.__len__ |
|
0 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / indexed / indexed_pretrain_dataset.py |
_IndexedPretrainDataset.__getitem__ |
|
0 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / indexed / indexed_pretrain_dataset.py |
_MockDataset.__init__ |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / indexed / indexed_pretrain_dataset.py |
_MockDataset.__len__ |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / indexed / indexed_pretrain_dataset.py |
MockGPTDataset.__getitem__ |
|
15 |
15 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / indexed / indexed_pretrain_dataset.py |
GPTDataset._finalize |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / indexed / indexed_pretrain_dataset.py |
GPTDataset._build_document_sample_shuffle_indices |
|
49 |
49 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / indexed / indexed_pretrain_dataset.py |
GPTDataset.numel_low_level_dataset |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / indexed / indexed_pretrain_dataset.py |
GPTDataset.build_low_level_dataset |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / indexed / indexed_pretrain_dataset.py |
GPTDataset.__len__ |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / indexed / indexed_pretrain_dataset.py |
GPTDataset.__getitem__ |
|
15 |
15 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / indexed / indexed_pretrain_dataset.py |
GPTDataset._query_document_sample_shuffle_indices |
|
21 |
21 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / indexed / indexed_pretrain_dataset.py |
GPTFromMRDataset._finalize |
|
10 |
10 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / indexed / indexed_pretrain_dataset.py |
GPTFromMRDataset.numel_low_level_dataset |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / indexed / indexed_pretrain_dataset.py |
GPTFromMRDataset.is_multimodal |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / indexed / indexed_pretrain_dataset.py |
GPTFromMRDataset.is_split_by_sequence |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / indexed / indexed_pretrain_dataset.py |
GPTFromMRDataset.build_low_level_dataset |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / indexed / indexed_pretrain_dataset.py |
GPTFromMRDataset.__len__ |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / indexed / indexed_pretrain_dataset.py |
GPTFromMRDataset.__getitem__ |
|
18 |
18 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / indexed / indexed_pretrain_dataset.py |
_get_ltor_masks_and_position_ids |
|
23 |
23 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / indexed / indexed_pretrain_dataset.py |
(no function) |
|
62 |
62 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / indexed / indexed_simple_blended_dataset.py |
SimpleBlendedDataset.__init__ |
|
8 |
8 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / indexed / indexed_simple_blended_dataset.py |
SimpleBlendedDataset.__len__ |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / indexed / indexed_simple_blended_dataset.py |
SimpleBlendedDataset.__getitem__ |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / indexed / indexed_simple_blended_dataset.py |
SimpleBlendedDataset._build_locations |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / indexed / indexed_simple_blended_dataset.py |
SimpleBlendedDataset._build_interleaved_locations |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / indexed / indexed_simple_blended_dataset.py |
(no function) |
|
10 |
10 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / indexed / indexed_split_builder.py |
_get_prefixes_weights_and_sizes_for_blend |
|
11 |
11 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / indexed / indexed_split_builder.py |
_build_low_level_dataset_and_split_indices |
|
13 |
13 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / indexed / indexed_split_builder.py |
IndexedDatasetSplitBuilder.__init__ |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / indexed / indexed_split_builder.py |
IndexedDatasetSplitBuilder.build |
|
20 |
20 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / indexed / indexed_split_builder.py |
IndexedDatasetSplitBuilder._build_mock_dataset_splits |
|
9 |
9 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / indexed / indexed_split_builder.py |
IndexedDatasetSplitBuilder._build_splits_from_blend |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / indexed / indexed_split_builder.py |
IndexedDatasetSplitBuilder._build_splits_from_independent_blends |
|
14 |
14 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / indexed / indexed_split_builder.py |
IndexedDatasetSplitBuilder._build_multiple_source_splits |
|
25 |
25 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / indexed / indexed_split_builder.py |
IndexedDatasetSplitBuilder._build_mid_level_dataset_splits |
|
21 |
21 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / indexed / indexed_split_builder.py |
IndexedDatasetSplitBuilder._build_lazy_mid_level_dataset_split |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / indexed / indexed_split_builder.py |
IndexedDatasetSplitBuilder._build_lazy_mid_level_dataset_split._dataset_factory |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / indexed / indexed_split_builder.py |
IndexedDatasetSplitBuilder._build_dataset_for_dataloader |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / indexed / indexed_split_builder.py |
(no function) |
|
34 |
34 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / indexed / io.py |
safe_import |
|
11 |
1 |
0 |
|
91% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / indexed / io.py |
_is_object_storage_path |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / indexed / io.py |
_parse_s3_path |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / indexed / io.py |
_get_index_cache_path |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / indexed / io.py |
_cache_index_file |
|
25 |
25 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / indexed / io.py |
DType.code_from_dtype |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / indexed / io.py |
DType.dtype_from_code |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / indexed / io.py |
DType.size |
|
5 |
2 |
0 |
|
60% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / indexed / io.py |
DType.optimal_dtype |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / indexed / io.py |
_IndexWriter.__init__ |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / indexed / io.py |
_IndexWriter.__enter__ |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / indexed / io.py |
_IndexWriter.__exit__ |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / indexed / io.py |
_IndexWriter.write |
|
10 |
1 |
0 |
|
90% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / indexed / io.py |
_IndexWriter._sequence_pointers |
|
7 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / indexed / io.py |
_IndexReader.__init__ |
|
28 |
2 |
0 |
|
93% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / indexed / io.py |
_IndexReader.close |
|
10 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / indexed / io.py |
_IndexReader.__del__ |
|
4 |
2 |
0 |
|
50% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / indexed / io.py |
_IndexReader.__len__ |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / indexed / io.py |
_IndexReader.__getitem__ |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / indexed / io.py |
_BinReader.read |
|
0 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / indexed / io.py |
_MMapBinReader.__init__ |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / indexed / io.py |
_MMapBinReader.read |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / indexed / io.py |
_MMapBinReader.close |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / indexed / io.py |
_MMapBinReader.__del__ |
|
4 |
2 |
0 |
|
50% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / indexed / io.py |
_FileBinReader.__init__ |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / indexed / io.py |
_FileBinReader.read |
|
8 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / indexed / io.py |
_S3BinReader.__init__ |
|
10 |
10 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / indexed / io.py |
_S3BinReader._extract_from_cache |
|
7 |
7 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / indexed / io.py |
_S3BinReader.read |
|
9 |
9 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / indexed / io.py |
_S3BinReader.__del__ |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / indexed / io.py |
_MultiStorageClientBinReader.__init__ |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / indexed / io.py |
_MultiStorageClientBinReader.read |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / indexed / io.py |
IndexedDataset.__init__ |
|
9 |
9 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / indexed / io.py |
IndexedDataset.initialize |
|
18 |
18 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / indexed / io.py |
IndexedDataset.__len__ |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / indexed / io.py |
IndexedDataset.__getitem__ |
|
16 |
7 |
0 |
|
56% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / indexed / io.py |
IndexedDataset.get |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / indexed / io.py |
IndexedDataset.sequence_lengths |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / indexed / io.py |
IndexedDataset.document_indices |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / indexed / io.py |
IndexedDataset.exists |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / indexed / io.py |
IndexedDatasetBuilder.__init__ |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / indexed / io.py |
IndexedDatasetBuilder.add_item |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / indexed / io.py |
IndexedDatasetBuilder.add_document |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / indexed / io.py |
IndexedDatasetBuilder.end_document |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / indexed / io.py |
IndexedDatasetBuilder.add_index |
|
12 |
12 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / indexed / io.py |
IndexedDatasetBuilder.finalize |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / indexed / io.py |
get_idx_path |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / indexed / io.py |
get_bin_path |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / indexed / io.py |
_normalize_prefix |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / indexed / io.py |
(no function) |
|
101 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / parallel / __init__.py |
(no function) |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / parallel / batch_parallel.py |
shard_batch_for_cp |
|
39 |
1 |
0 |
|
97% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / parallel / batch_parallel.py |
_shard_seq_lens_for_cp |
|
42 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / parallel / batch_parallel.py |
CPBatchSharder.__init__ |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / parallel / batch_parallel.py |
CPBatchSharder.shard |
|
16 |
16 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / parallel / batch_parallel.py |
TPBatchBroadcaster.__init__ |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / parallel / batch_parallel.py |
TPBatchBroadcaster.broadcast |
|
31 |
31 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / parallel / batch_parallel.py |
(no function) |
|
14 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / parallel / batch_sampler.py |
_validate_positive_integer |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / parallel / batch_sampler.py |
_DatasetBatchSampler.__init__ |
|
20 |
20 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / parallel / batch_sampler.py |
_DatasetBatchSampler._validate_consumed_samples |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / parallel / batch_sampler.py |
_DatasetBatchSampler.__iter__ |
|
14 |
14 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / parallel / batch_sampler.py |
_DatasetBatchSampler.__len__ |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / parallel / batch_sampler.py |
_DatasetBatchSampler._resolve_index |
|
7 |
7 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / parallel / batch_sampler.py |
_DatasetBatchSampler.state_dict |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / parallel / batch_sampler.py |
_DatasetBatchSampler.enable_source_batch_resume |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / parallel / batch_sampler.py |
_DatasetBatchSampler.load_state_dict |
|
15 |
15 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / parallel / batch_sampler.py |
_DatasetBatchSampler.set_epoch |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / parallel / batch_sampler.py |
_CyclicDatasetBatchSampler.__init__ |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / parallel / batch_sampler.py |
_CyclicDatasetBatchSampler.load_state_dict |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / parallel / batch_sampler.py |
_CyclicDatasetBatchSampler.__len__ |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / parallel / batch_sampler.py |
_CyclicDatasetBatchSampler.__iter__ |
|
22 |
22 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / parallel / batch_sampler.py |
_CyclicDatasetBatchSampler._active_samples |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / parallel / batch_sampler.py |
_resolve_index_mapping |
|
9 |
9 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / parallel / batch_sampler.py |
build_dataset_batch_sampler |
|
12 |
12 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / parallel / batch_sampler.py |
(no function) |
|
30 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / parallel / build_barrier.py |
OnlineDatasetBarrier.__init__ |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / parallel / build_barrier.py |
OnlineDatasetBarrier.__call__ |
|
14 |
14 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / parallel / build_barrier.py |
(no function) |
|
9 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / parallel / dataloader_parallel.py |
_always_build |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / parallel / dataloader_parallel.py |
_no_barrier |
|
0 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / parallel / dataloader_parallel.py |
_is_dataloader_rank |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / parallel / dataloader_parallel.py |
_is_global_cache_builder_rank |
|
8 |
8 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / parallel / dataloader_parallel.py |
create_dataloader_parallel_context |
|
18 |
18 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / parallel / dataloader_parallel.py |
split_iterable_dataset_by_dp |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / parallel / dataloader_parallel.py |
build_dataset_for_dataloader |
|
14 |
14 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / parallel / dataloader_parallel.py |
(no function) |
|
30 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / text / __init__.py |
__getattr__ |
|
6 |
3 |
0 |
|
50% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / text / __init__.py |
__dir__ |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / text / __init__.py |
(no function) |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / text / build_data_transform.py |
_get_record_value |
|
9 |
9 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / text / build_data_transform.py |
IdentityDataTransform.__init__ |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / text / build_data_transform.py |
IdentityDataTransform.__call__ |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / text / build_data_transform.py |
PlaintextTransform.__post_init__ |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / text / build_data_transform.py |
PlaintextTransform.__call__ |
|
11 |
11 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / text / build_data_transform.py |
TextConversationTransform.__post_init__ |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / text / build_data_transform.py |
TextConversationTransform.__call__ |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / text / build_data_transform.py |
build_llm_data_transform |
|
7 |
7 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / text / build_data_transform.py |
(no function) |
|
27 |
27 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / text / build_dataset.py |
_build_dataloader_context |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / text / build_dataset.py |
_get_indexed_split_sizes |
|
14 |
14 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / text / build_dataset.py |
build_online_text_dataset |
|
12 |
12 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / text / build_dataset.py |
build_indexed_text_dataset |
|
13 |
13 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / text / build_dataset.py |
(no function) |
|
15 |
15 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / text / build_tokenizer.py |
_DatasetTokenizer.__init__ |
|
24 |
24 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / text / build_tokenizer.py |
_DatasetTokenizer.__len__ |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / text / build_tokenizer.py |
_DatasetTokenizer.eos |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / text / build_tokenizer.py |
_DatasetTokenizer.pad |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / text / build_tokenizer.py |
_HFAutoTokenizer.__init__ |
|
12 |
12 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / text / build_tokenizer.py |
_HFAutoTokenizer.__getattr__ |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / text / build_tokenizer.py |
_HFAutoTokenizer.__setattr__ |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / text / build_tokenizer.py |
_HFAutoTokenizer.__len__ |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / text / build_tokenizer.py |
_HFAutoTokenizer.__call__ |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / text / build_tokenizer.py |
AutoTokenizer.from_pretrained |
|
17 |
17 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / text / build_tokenizer.py |
build_tokenizer |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / text / build_tokenizer.py |
(no function) |
|
23 |
23 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / text / chat_template.py |
Registry.__init__ |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / text / chat_template.py |
Registry.__getitem__ |
|
5 |
2 |
0 |
|
60% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / text / chat_template.py |
Registry.__setitem__ |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / text / chat_template.py |
Registry.__delitem__ |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / text / chat_template.py |
Registry.__iter__ |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / text / chat_template.py |
Registry.__len__ |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / text / chat_template.py |
Registry.register |
|
5 |
2 |
0 |
|
60% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / text / chat_template.py |
Registry.register.decorator |
|
4 |
1 |
0 |
|
75% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / text / chat_template.py |
Registry.valid_keys |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / text / chat_template.py |
_format_janus_message |
|
19 |
2 |
0 |
|
89% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / text / chat_template.py |
_janus_labels |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / text / chat_template.py |
build_chat_template |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / text / chat_template.py |
ChatTemplate.__init__ |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / text / chat_template.py |
ChatTemplate.save_pretrained |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / text / chat_template.py |
ChatTemplate.encode_messages |
|
0 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / text / chat_template.py |
ChatTemplate.get_jinja_template |
|
0 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / text / chat_template.py |
DefaultTemplate.encode_messages |
|
12 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / text / chat_template.py |
DefaultTemplate.get_jinja_template |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / text / chat_template.py |
TokenizerTemplate._update_prefix_labels |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / text / chat_template.py |
TokenizerTemplate.encode_messages |
|
18 |
1 |
0 |
|
94% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / text / chat_template.py |
TokenizerTemplate.get_jinja_template |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / text / chat_template.py |
GptOssTokenizerTemplate.__init__ |
|
5 |
1 |
0 |
|
80% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / text / chat_template.py |
GptOssTokenizerTemplate._update_prefix_labels |
|
9 |
1 |
0 |
|
89% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / text / chat_template.py |
Llama2Template.encode_messages |
|
20 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / text / chat_template.py |
Llama2Template.get_jinja_template |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / text / chat_template.py |
JanusTemplate.encode_messages |
|
22 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / text / chat_template.py |
JanusTemplate.get_jinja_template |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / text / chat_template.py |
ChatmlTemplate.encode_messages |
|
15 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / text / chat_template.py |
ChatmlTemplate.get_jinja_template |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / text / chat_template.py |
(no function) |
|
53 |
0 |
2 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / text / online / __init__.py |
(no function) |
|
0 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / text / online / online_dataset.py |
build_online_dataset |
|
10 |
10 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / text / online / online_dataset.py |
(no function) |
|
10 |
10 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / text / online / online_iterable_dataset.py |
build_online_iterable_dataset |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / text / online / online_iterable_dataset.py |
build_online_iterable_dataset.dataset_factory |
|
12 |
12 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / text / online / online_iterable_dataset.py |
(no function) |
|
8 |
8 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / text / online / online_mapping_dataset.py |
OnlineMappingDataset.__init__ |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / text / online / online_mapping_dataset.py |
OnlineMappingDataset.__len__ |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / text / online / online_mapping_dataset.py |
OnlineMappingDataset.__getitem__ |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / text / online / online_mapping_dataset.py |
build_online_mapping_dataset |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / text / online / online_mapping_dataset.py |
build_online_mapping_dataset.dataset_factory |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / text / online / online_mapping_dataset.py |
(no function) |
|
14 |
14 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / text / online / online_utils.py |
_is_nonempty_plaintext_sample |
|
11 |
11 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / text / online / online_utils.py |
_filter_online_plaintext_dataset |
|
10 |
10 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / text / online / online_utils.py |
resolve_online_data_files |
|
25 |
25 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / text / online / online_utils.py |
load_online_hf_dataset |
|
16 |
16 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / text / online / online_utils.py |
normalize_online_dataloader_context |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / text / online / online_utils.py |
(no function) |
|
15 |
15 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / text / transform_dataset.py |
_supports_output_index_for_resume |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / text / transform_dataset.py |
_has_trainable_labels |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / text / transform_dataset.py |
_normalize_transformed_samples |
|
11 |
11 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / text / transform_dataset.py |
_transform_samples |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / text / transform_dataset.py |
_LLMTransformDataset.__init__ |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / text / transform_dataset.py |
_LLMTransformDataset.__len__ |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / text / transform_dataset.py |
_LLMTransformDataset.get_item |
|
10 |
10 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / text / transform_dataset.py |
_LLMTransformDataset.__getitem__ |
|
7 |
7 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / text / transform_dataset.py |
_LLMIterableTransformDataset.__init__ |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / text / transform_dataset.py |
_LLMIterableTransformDataset.output_index_for_resume |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / text / transform_dataset.py |
_LLMIterableTransformDataset.output_index_for_resume |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / text / transform_dataset.py |
_LLMIterableTransformDataset.get_item |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / text / transform_dataset.py |
_LLMIterableTransformDataset._prepare_model_samples |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / text / transform_dataset.py |
_LLMIterableTransformDataset.__iter__ |
|
13 |
13 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / text / transform_dataset.py |
_LLMIterableTransformDataset.state_dict |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / text / transform_dataset.py |
_LLMIterableTransformDataset.load_state_dict |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / text / transform_dataset.py |
_LLMIterableTransformDataset.set_epoch |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / text / transform_dataset.py |
_wrap_llm_dataset |
|
9 |
9 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / text / transform_dataset.py |
apply_llm_data_transform |
|
7 |
7 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / text / transform_dataset.py |
(no function) |
|
32 |
32 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / tools / __init__.py |
(no function) |
|
0 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / tools / huggingface_offline.py |
_download_jsonl |
|
26 |
26 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / tools / huggingface_offline.py |
_parse_bool |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / tools / huggingface_offline.py |
_add_huggingface_arguments |
|
7 |
7 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / tools / huggingface_offline.py |
_add_conversion_arguments |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / tools / huggingface_offline.py |
_get_args |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / tools / huggingface_offline.py |
main |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / tools / huggingface_offline.py |
(no function) |
|
16 |
16 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / tools / offline_config.py |
OfflinePreparationConfig._slug |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / tools / offline_config.py |
OfflinePreparationConfig.dataset_label |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / tools / offline_config.py |
OfflinePreparationConfig.json_keys_list |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / tools / offline_config.py |
OfflinePreparationConfig.download_root |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / tools / offline_config.py |
OfflinePreparationConfig.resolved_json_path |
|
11 |
11 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / tools / offline_config.py |
OfflinePreparationConfig.resolved_output_prefix |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / tools / offline_config.py |
OfflinePreparationConfig.to_offline_args |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / tools / offline_config.py |
(no function) |
|
45 |
45 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / tools / offline_preparation.py |
build_tokenizer |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / tools / offline_preparation.py |
build_huggingface_tokenizer |
|
10 |
10 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / tools / offline_preparation.py |
append_eod |
|
8 |
8 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / tools / offline_preparation.py |
Encoder.__init__ |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / tools / offline_preparation.py |
Encoder.initializer |
|
11 |
11 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / tools / offline_preparation.py |
Encoder.split |
|
8 |
8 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / tools / offline_preparation.py |
Encoder.encode |
|
23 |
23 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / tools / offline_preparation.py |
Partition.__init__ |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / tools / offline_preparation.py |
Partition.print_processing_stats |
|
8 |
8 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / tools / offline_preparation.py |
Partition.split_sentences |
|
15 |
15 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / tools / offline_preparation.py |
Partition.process_json_file |
|
45 |
45 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / tools / offline_preparation.py |
_parse_bool |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / tools / offline_preparation.py |
_add_offline_source_arguments |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / tools / offline_preparation.py |
_add_offline_worker_arguments |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / tools / offline_preparation.py |
get_args |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / tools / offline_preparation.py |
get_file_name |
|
12 |
12 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / tools / offline_preparation.py |
check_files_exist |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / tools / offline_preparation.py |
find_optimal_num_workers |
|
12 |
12 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / tools / offline_preparation.py |
_process_json_file_worker |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / tools / offline_preparation.py |
_split_sentences_worker |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / tools / offline_preparation.py |
_wait_for_processes |
|
9 |
9 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / tools / offline_preparation.py |
_count_jsonl_lines |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / tools / offline_preparation.py |
_resolve_input_files |
|
11 |
11 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / tools / offline_preparation.py |
_merge_partition_outputs |
|
10 |
10 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / tools / offline_preparation.py |
_cleanup_intermediate_files |
|
16 |
16 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / tools / offline_preparation.py |
_validate_preparation_args |
|
26 |
26 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / tools / offline_preparation.py |
_partition_input_data |
|
23 |
23 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / tools / offline_preparation.py |
_write_input_partitions |
|
16 |
16 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / tools / offline_preparation.py |
_split_partitions |
|
10 |
10 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / tools / offline_preparation.py |
_encode_partitions |
|
9 |
9 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / tools / offline_preparation.py |
prepare_offline_dataset |
|
17 |
17 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / tools / offline_preparation.py |
main |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / tools / offline_preparation.py |
(no function) |
|
61 |
61 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / tools / read_indexed_dataset.py |
_path_prefix |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / tools / read_indexed_dataset.py |
_parse_args |
|
12 |
12 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / tools / read_indexed_dataset.py |
main |
|
28 |
28 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / tools / read_indexed_dataset.py |
(no function) |
|
13 |
13 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / vlm / __init__.py |
(no function) |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / vlm / build_data_transform.py |
VLMChatTransform.__init__ |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / vlm / build_data_transform.py |
VLMChatTransform._normalize_messages |
|
19 |
19 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / vlm / build_data_transform.py |
VLMChatTransform._encode |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / vlm / build_data_transform.py |
VLMChatTransform._modal_runs |
|
13 |
13 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / vlm / build_data_transform.py |
VLMChatTransform._drop_truncated_images |
|
28 |
28 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / vlm / build_data_transform.py |
VLMChatTransform._truncate_and_pad |
|
12 |
12 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / vlm / build_data_transform.py |
VLMChatTransform.__call__ |
|
9 |
9 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / vlm / build_data_transform.py |
build_vlm_data_transform |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / vlm / build_data_transform.py |
(no function) |
|
16 |
16 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / vlm / build_processor.py |
build_processor |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / vlm / build_processor.py |
(no function) |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / vlm / collator.py |
VLMCollator.__call__ |
|
8 |
8 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / vlm / collator.py |
build_vlm_collator |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / vlm / collator.py |
(no function) |
|
9 |
9 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / vlm / constants.py |
(no function) |
|
8 |
8 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / vlm / dataset.py |
VLMDataset.__init__ |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / vlm / dataset.py |
VLMDataset._resolve_record_paths |
|
13 |
13 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / vlm / dataset.py |
VLMDataset._resolve |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / vlm / dataset.py |
VLMDataset.__len__ |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / vlm / dataset.py |
VLMDataset.__getitem__ |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / vlm / dataset.py |
_TransformDataset.__init__ |
|
10 |
10 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / vlm / dataset.py |
_TransformDataset._has_trainable |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / vlm / dataset.py |
_TransformDataset.__len__ |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / vlm / dataset.py |
_TransformDataset.__getitem__ |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / vlm / dataset.py |
build_vlm_dataset |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / vlm / dataset.py |
(no function) |
|
22 |
22 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / vlm / get_batch.py |
VLMBatchProcessor.normalize_source_batch |
|
8 |
8 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / vlm / get_batch.py |
VLMBatchProcessor.prepare_batch |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / vlm / get_batch.py |
VLMGetBatch.__init__ |
|
8 |
8 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / vlm / get_batch.py |
VLMGetBatch.__call__ |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / vlm / get_batch.py |
build_vlm_get_batch |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / data / vlm / get_batch.py |
(no function) |
|
15 |
15 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / __init__.py |
(no function) |
|
9 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / __init__.py |
(no function) |
|
0 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / applier.py |
_validate_inner_wrapper_injections |
|
19 |
3 |
0 |
|
84% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / applier.py |
_validate_cp_compute_injections |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / applier.py |
_validate_ep_compute_injections |
|
10 |
1 |
0 |
|
90% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / applier.py |
_preflight_compute_injection |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / applier.py |
_log_injection_choice |
|
10 |
1 |
0 |
|
90% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / applier.py |
_apply_plan_special_handlers |
|
8 |
6 |
0 |
|
25% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / applier.py |
_get_active_mesh |
|
6 |
1 |
0 |
|
83% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / applier.py |
_get_tp_submesh |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / applier.py |
_get_cp_submesh |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / applier.py |
_get_ep_submesh |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / applier.py |
_expert_mesh_layout |
|
9 |
1 |
0 |
|
89% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / applier.py |
_build_expert_mesh |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / applier.py |
build_expert_mesh |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / applier.py |
_apply_phase_c |
|
29 |
5 |
0 |
|
83% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / applier.py |
(no function) |
|
27 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / default_templates.py |
_multi_dim |
|
8 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / default_templates.py |
_hid |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / default_templates.py |
_out |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / default_templates.py |
_infer_colwise_vs_rowwise |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / default_templates.py |
_moe_expert_tp_placement |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / default_templates.py |
_placement_for_role |
|
17 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / default_templates.py |
_build_spec_from_template |
|
34 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / default_templates.py |
_finalize_tp_local_attr_plans |
|
6 |
1 |
0 |
|
83% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / default_templates.py |
(no function) |
|
33 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / forward_rewriter.py |
_is_subsequence |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / forward_rewriter.py |
_compute_parameters |
|
7 |
3 |
0 |
|
57% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / forward_rewriter.py |
_validate_explicit_compute_parameters |
|
3 |
1 |
0 |
|
67% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / forward_rewriter.py |
_validate_compute_parameter_names |
|
9 |
1 |
0 |
|
89% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / forward_rewriter.py |
_validate_required_compute_parameters |
|
5 |
1 |
0 |
|
80% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / forward_rewriter.py |
validate_local_compute_signature |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / forward_rewriter.py |
validate_wrapped_forward |
|
16 |
4 |
0 |
|
75% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / forward_rewriter.py |
_install_bias_suppression |
|
7 |
7 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / forward_rewriter.py |
_install_bias_suppression.bias_free_forward |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / forward_rewriter.py |
_add_bias_to_primary_output |
|
17 |
5 |
0 |
|
71% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / forward_rewriter.py |
_maybe_add_deferred_biases |
|
10 |
8 |
0 |
|
20% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / forward_rewriter.py |
_keep_loss_parallel_output |
|
4 |
2 |
0 |
|
50% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / forward_rewriter.py |
_descendant_boundary_fqns |
|
4 |
1 |
0 |
|
75% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / forward_rewriter.py |
_bind_input_indices |
|
12 |
3 |
0 |
|
75% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / forward_rewriter.py |
_wrap_production_forward |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / forward_rewriter.py |
_wrap_production_forward.production_forward |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / forward_rewriter.py |
_wrap_validate_forward |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / forward_rewriter.py |
_wrap_validate_forward.validate_forward |
|
16 |
5 |
0 |
|
69% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / forward_rewriter.py |
_validate_out_src |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / forward_rewriter.py |
_validate_out_dst |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / forward_rewriter.py |
_normalize_placements_ndim |
|
6 |
1 |
0 |
|
83% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / forward_rewriter.py |
_validate_outputs |
|
20 |
5 |
0 |
|
75% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / forward_rewriter.py |
_rewrap_local_outputs |
|
29 |
7 |
0 |
|
76% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / forward_rewriter.py |
_wrap_local_region_forward |
|
13 |
1 |
0 |
|
92% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / forward_rewriter.py |
_wrap_local_region_forward.local_region_forward |
|
21 |
2 |
0 |
|
90% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / forward_rewriter.py |
_apply_custom_inner_wrapper |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / forward_rewriter.py |
_find_dtensor_argument |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / forward_rewriter.py |
_validate_inner_dispatch_output |
|
11 |
2 |
0 |
|
82% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / forward_rewriter.py |
_rewrap_inner_tensor |
|
3 |
1 |
0 |
|
67% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / forward_rewriter.py |
_rewrap_inner_outputs |
|
8 |
1 |
0 |
|
88% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / forward_rewriter.py |
_install_inner_adapter |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / forward_rewriter.py |
_install_inner_adapter.adapted |
|
23 |
3 |
0 |
|
87% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / forward_rewriter.py |
_wrap_inner_attention |
|
38 |
3 |
0 |
|
92% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / forward_rewriter.py |
_is_vocab_parallel_embed |
|
6 |
4 |
0 |
|
33% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / forward_rewriter.py |
_wrap_vocab_parallel_embedding |
|
7 |
7 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / forward_rewriter.py |
_wrap_vocab_parallel_embedding.masked_embedding_forward |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / forward_rewriter.py |
_ForwardRewriteRequest.__init__ |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / forward_rewriter.py |
_restore_attr |
|
3 |
1 |
0 |
|
67% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / forward_rewriter.py |
_commit_forward_rewrite |
|
10 |
4 |
0 |
|
60% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / forward_rewriter.py |
_classify_rewrite_result |
|
15 |
1 |
0 |
|
93% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / forward_rewriter.py |
_install_bound_forward |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / forward_rewriter.py |
install_local_compute_forward_adapters |
|
9 |
2 |
0 |
|
78% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / forward_rewriter.py |
_wrap_module_forward |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / forward_rewriter.py |
_wrap_module_forward.local_param_forward |
|
11 |
3 |
0 |
|
73% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / forward_rewriter.py |
(no function) |
|
57 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / fsdp_adapter.py |
_is_expert_source_mesh |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / fsdp_adapter.py |
_get_checkpoint_wrapped_module |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / fsdp_adapter.py |
FSDP2Manager.__init__ |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / fsdp_adapter.py |
FSDP2Manager._resolve_replicate_params |
|
10 |
10 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / fsdp_adapter.py |
FSDP2Manager._build_expert_fsdp_mesh |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / fsdp_adapter.py |
FSDP2Manager._build_dense_fsdp_mesh |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / fsdp_adapter.py |
FSDP2Manager._build_compatibility_fsdp_mesh |
|
13 |
13 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / fsdp_adapter.py |
FSDP2Manager._build_fsdp_actual_mesh |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / fsdp_adapter.py |
FSDP2Manager._build_mixed_precision_policy |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / fsdp_adapter.py |
FSDP2Manager._build_offload_policy |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / fsdp_adapter.py |
FSDP2Manager._build_fully_shard_kwargs |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / fsdp_adapter.py |
FSDP2Manager._find_transformer_block_modules |
|
21 |
21 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / fsdp_adapter.py |
FSDP2Manager._find_expert_wrap_modules |
|
13 |
13 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / fsdp_adapter.py |
FSDP2Manager._find_wrap_modules |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / fsdp_adapter.py |
FSDP2Manager._resolve_parameter_owners |
|
10 |
10 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / fsdp_adapter.py |
FSDP2Manager._uses_expert_mesh |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / fsdp_adapter.py |
FSDP2Manager._build_managed_replicate_params |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / fsdp_adapter.py |
FSDP2Manager._configure_source_layout_gradient_scaling |
|
9 |
2 |
0 |
|
78% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / fsdp_adapter.py |
FSDP2Manager._configure_prefetch |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / fsdp_adapter.py |
FSDP2Manager._order_wrap_modules |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / fsdp_adapter.py |
FSDP2Manager._parallelize_child_units |
|
13 |
13 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / fsdp_adapter.py |
FSDP2Manager._parallelize_root |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / fsdp_adapter.py |
FSDP2Manager._log_gradient_scaling |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / fsdp_adapter.py |
FSDP2Manager.parallelize |
|
12 |
12 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / fsdp_adapter.py |
_instantiate_fsdp2 |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / fsdp_adapter.py |
_apply_fsdp2 |
|
8 |
8 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / fsdp_adapter.py |
(no function) |
|
51 |
0 |
2 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / function_module.py |
FunctionModule.__init__ |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / function_module.py |
FunctionModule.forward |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / function_module.py |
FunctionModule.extra_repr |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / function_module.py |
(no function) |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / local_compute_context.py |
local_compute_context |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / local_compute_context.py |
(no function) |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / local_region.py |
_bind_arg_names |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / local_region.py |
_normalize_out_placements |
|
10 |
10 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / local_region.py |
_unwrap_region_inputs |
|
22 |
22 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / local_region.py |
_wrap_region_outputs |
|
13 |
13 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / local_region.py |
local_region |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / local_region.py |
local_region.decorator |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / local_region.py |
local_region.decorator.wrapped |
|
7 |
7 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / local_region.py |
(no function) |
|
13 |
13 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / parameter_sharding.py |
_get_attr_by_path |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / parameter_sharding.py |
_set_param_by_path |
|
7 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / parameter_sharding.py |
_resolve_module |
|
6 |
1 |
0 |
|
83% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / parameter_sharding.py |
_local_params_context |
|
6 |
2 |
0 |
|
67% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / parameter_sharding.py |
_temp_local_params |
|
16 |
8 |
0 |
|
50% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / parameter_sharding.py |
_stack_moe_experts |
|
23 |
1 |
0 |
|
96% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / parameter_sharding.py |
_resolve_parameter_source_meshes |
|
14 |
4 |
0 |
|
71% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / parameter_sharding.py |
_shard_planned_parameters |
|
13 |
5 |
0 |
|
62% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / parameter_sharding.py |
_build_runtime_source_shard_info |
|
10 |
3 |
0 |
|
70% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / parameter_sharding.py |
_shard_module_params |
|
13 |
12 |
0 |
|
8% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / parameter_sharding.py |
detect_tied_weights |
|
11 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / parameter_sharding.py |
_broadcast_tied_param |
|
12 |
2 |
0 |
|
83% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / parameter_sharding.py |
_replicate_tied_weights |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / parameter_sharding.py |
(no function) |
|
25 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / planner.py |
ShardingPlanner.__init__ |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / planner.py |
ShardingPlanner._derive_boundary_specs |
|
15 |
1 |
0 |
|
93% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / planner.py |
ShardingPlanner._finalize_boundary_specs |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / planner.py |
ShardingPlanner._log_explanation |
|
2 |
1 |
0 |
|
50% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / planner.py |
ShardingPlanner._classify_boundary_groups |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / planner.py |
ShardingPlanner.plan |
|
20 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / planner.py |
ShardingPlanner._get_architecture |
|
14 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / planner.py |
ShardingPlanner._validate_dtensor_axes |
|
18 |
5 |
0 |
|
72% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / planner.py |
ShardingPlanner._build_mesh_dim_names |
|
7 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / planner.py |
ShardingPlanner._check_overrides_no_dp |
|
7 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / planner.py |
ShardingPlanner._check_overrides_no_dp._declares_dp |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / planner.py |
ShardingPlanner._warn_uncovered_function_modules |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / planner.py |
ShardingPlanner._classify_all_params |
|
9 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / planner.py |
ShardingPlanner._group_by_boundary |
|
23 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / planner.py |
ShardingPlanner._explicit_boundary_type |
|
9 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / planner.py |
ShardingPlanner._moe_boundary_type |
|
7 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / planner.py |
ShardingPlanner._dense_boundary_type |
|
11 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / planner.py |
ShardingPlanner._infer_boundary_type |
|
12 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / planner.py |
ShardingPlanner._validate_ep_extend |
|
13 |
1 |
0 |
|
92% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / planner.py |
ShardingPlanner._detect_expert_layout |
|
19 |
5 |
0 |
|
74% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / planner.py |
ShardingPlanner._mark_old_style_experts |
|
9 |
9 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / planner.py |
ShardingPlanner._mark_extended_expert_params |
|
19 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / planner.py |
ShardingPlanner._set_extended_ep_contract |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / planner.py |
ShardingPlanner._mark_hf_native_moe |
|
14 |
2 |
0 |
|
86% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / planner.py |
ShardingPlanner._finalize_fused_expert_tp_guard |
|
19 |
6 |
0 |
|
68% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / planner.py |
ShardingPlanner._should_defer_rowwise_bias |
|
11 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / planner.py |
ShardingPlanner._validate_colwise_bias |
|
7 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / planner.py |
ShardingPlanner._collect_deferred_biases |
|
26 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / planner.py |
ShardingPlanner._finalize_deferred_biases |
|
13 |
1 |
0 |
|
92% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / planner.py |
ShardingPlanner._check_param_uniqueness |
|
8 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / planner.py |
ShardingPlanner._check_full_declaration |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / planner.py |
ShardingPlanner._check_shard_divisibility |
|
27 |
2 |
0 |
|
93% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / planner.py |
ShardingPlanner._check_all_trainable_params_covered |
|
19 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / planner.py |
ShardingPlanner._mark_terminal |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / planner.py |
ShardingPlanner._topological_sort_by_forward_order |
|
12 |
2 |
0 |
|
83% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / planner.py |
ShardingPlanner._detect_tied_pairs |
|
11 |
9 |
0 |
|
18% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / planner.py |
validate_model_compatibility |
|
21 |
1 |
0 |
|
95% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / planner.py |
(no function) |
|
76 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / precompiled_boundary.py |
_classify_collective |
|
15 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / precompiled_boundary.py |
_get_arg |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / precompiled_boundary.py |
_set_arg |
|
9 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / precompiled_boundary.py |
BoundaryExecutionOp.execute |
|
0 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / precompiled_boundary.py |
RedistOp.execute |
|
14 |
10 |
0 |
|
29% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / precompiled_boundary.py |
PrecompiledBoundary.__init__ |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / precompiled_boundary.py |
PrecompiledBoundary._lower_execution_op |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / precompiled_boundary.py |
PrecompiledBoundary._compile_input_plan |
|
9 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / precompiled_boundary.py |
PrecompiledBoundary._compile_output_plan |
|
13 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / precompiled_boundary.py |
PrecompiledBoundary.redistribute_inputs |
|
7 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / precompiled_boundary.py |
PrecompiledBoundary.redistribute_outputs |
|
16 |
10 |
0 |
|
38% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / precompiled_boundary.py |
(no function) |
|
31 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / rule_resolver.py |
_last_segment |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / rule_resolver.py |
_is_glob_key |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / rule_resolver.py |
_merge_plan_overrides |
|
25 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / rule_resolver.py |
_iter_named_placements |
|
8 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / rule_resolver.py |
_validate_override_axes |
|
7 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / rule_resolver.py |
_merge_contract_field |
|
11 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / rule_resolver.py |
_warn_dropped_params |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / rule_resolver.py |
_merge_into |
|
7 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / rule_resolver.py |
_normalize_contract_fields |
|
4 |
1 |
0 |
|
75% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / rule_resolver.py |
_suggest_insert_skeleton |
|
27 |
4 |
0 |
|
85% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / rule_resolver.py |
_insert_spec |
|
15 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / rule_resolver.py |
fill_context_kwargs |
|
9 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / rule_resolver.py |
_is_delayed_target |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / rule_resolver.py |
_check_target_config_keys |
|
12 |
2 |
0 |
|
83% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / rule_resolver.py |
_require_region_dispatch |
|
8 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / rule_resolver.py |
_build_local_compute_factory |
|
10 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / rule_resolver.py |
_resolve_local_compute_fn |
|
16 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / rule_resolver.py |
_resolve_inner_target |
|
9 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / rule_resolver.py |
_resolve_inner_wrapper |
|
32 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / rule_resolver.py |
_resolve_inner_wrapper._apply_target |
|
9 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / rule_resolver.py |
_resolve_inner_output_placements |
|
18 |
1 |
0 |
|
94% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / rule_resolver.py |
_inner_target_name |
|
6 |
1 |
0 |
|
83% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / rule_resolver.py |
(no function) |
|
34 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / source_shard.py |
_source_dim_names |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / source_shard.py |
_source_sub_mesh |
|
10 |
5 |
0 |
|
50% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / source_shard.py |
_check_fsdp_owned_axes |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / source_shard.py |
build_source_shard_info |
|
27 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / source_shard.py |
_build_dtensor_source_shard_info |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / source_shard.py |
_build_parameter_source_shard_info |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / source_shard.py |
_record_parameter_source_shard_info |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / source_shard.py |
_build_source_shard_info_by_param |
|
13 |
13 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / source_shard.py |
_get_default_source_shard_info |
|
8 |
8 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / source_shard.py |
_build_managed_source_shard_info |
|
16 |
16 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / source_shard.py |
_source_infos_for_fully_shard |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / source_shard.py |
(no function) |
|
25 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / special_handlers.py |
_shard_gated_delta |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / special_handlers.py |
_collect_special_handlers |
|
11 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / special_handlers.py |
(no function) |
|
9 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / tp_collective_lowering.py |
TPExecutionOp.execute |
|
10 |
10 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / tp_collective_lowering.py |
TPCollectiveLowerer.__call__ |
|
25 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / tp_collective_lowering.py |
create_tp_collective_lowerer |
|
12 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / _builder / tp_collective_lowering.py |
(no function) |
|
26 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / activation_checkpoint.py |
_resolve_torch_op |
|
11 |
1 |
0 |
|
91% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / activation_checkpoint.py |
_resolve_op_attr |
|
7 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / activation_checkpoint.py |
_existing_ops |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / activation_checkpoint.py |
_default_compute_intensive_ops |
|
5 |
2 |
0 |
|
60% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / activation_checkpoint.py |
_ffpa_forward_ops |
|
5 |
1 |
0 |
|
80% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / activation_checkpoint.py |
_build_selective_ac_must_save_ops |
|
7 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / activation_checkpoint.py |
_get_checkpoint_wrapped_module |
|
5 |
1 |
0 |
|
80% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / activation_checkpoint.py |
_find_transformer_block_modules |
|
22 |
3 |
0 |
|
86% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / activation_checkpoint.py |
ignore_sac_ops |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / activation_checkpoint.py |
ensure_profiler_ops_sac_ignored |
|
11 |
11 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / activation_checkpoint.py |
ensure_fsdp_ops_sac_ignored |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / activation_checkpoint.py |
_make_selective_checkpoint_policy_fn |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / activation_checkpoint.py |
_make_selective_checkpoint_policy_fn.selective_checkpointing_policy |
|
11 |
11 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / activation_checkpoint.py |
make_selective_checkpoint_context_fn |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / activation_checkpoint.py |
make_selective_checkpoint_context_fn.selective_checkpoint_context_fn |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / activation_checkpoint.py |
_find_transformer_layer_container_infos |
|
14 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / activation_checkpoint.py |
_flatten_layer_container_infos |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / activation_checkpoint.py |
_should_use_hf_native_gradient_checkpointing |
|
9 |
9 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / activation_checkpoint.py |
_wrap_layer_containers |
|
10 |
1 |
0 |
|
90% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / activation_checkpoint.py |
_is_checkpoint_wrapped |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / activation_checkpoint.py |
_find_checkpoint_wrappers |
|
7 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / activation_checkpoint.py |
_register_forward_prefetch_layers |
|
12 |
1 |
0 |
|
92% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / activation_checkpoint.py |
_wrap_first_existing_attr |
|
14 |
4 |
0 |
|
71% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / activation_checkpoint.py |
apply_submodule_checkpointing |
|
15 |
2 |
0 |
|
87% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / activation_checkpoint.py |
apply_submodule_checkpointing.submodule_checkpoint_wrapper |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / activation_checkpoint.py |
_detect_kv_sharing_and_maybe_disable_cache |
|
16 |
11 |
0 |
|
31% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / activation_checkpoint.py |
_apply_activation_checkpointing |
|
47 |
5 |
0 |
|
89% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / activation_checkpoint.py |
(no function) |
|
50 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / apply.py |
apply_sharding_plan |
|
25 |
1 |
0 |
|
96% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / apply.py |
(no function) |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / attention_swap.py |
attention_swap_policy |
|
7 |
7 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / attention_swap.py |
validate_attention_swap |
|
10 |
10 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / attention_swap.py |
_record_attention_target |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / attention_swap.py |
_find_attention_targets |
|
8 |
8 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / attention_swap.py |
_find_attention_targets.visit |
|
9 |
9 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / attention_swap.py |
apply_attention_swap |
|
15 |
15 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / attention_swap.py |
(no function) |
|
18 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / compile.py |
_get_attribute |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / compile.py |
_normalize_declared_layers |
|
14 |
14 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / compile.py |
_qualify_declared_layers |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / compile.py |
_layers_from_path |
|
7 |
7 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / compile.py |
get_compile_layers |
|
16 |
16 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / compile.py |
_install_dynamo_mapping_get_polyfill |
|
15 |
15 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / compile.py |
_install_dynamo_mapping_get_polyfill._mapping_get |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / compile.py |
resolve_compile_kwargs |
|
7 |
7 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / compile.py |
apply_compile |
|
13 |
13 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / compile.py |
_resolve_compile_config |
|
10 |
10 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / compile.py |
(no function) |
|
20 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / __init__.py |
(no function) |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / attention.py |
_mome_cp_halo_exchange |
|
7 |
1 |
0 |
|
86% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / attention.py |
_mome_cp_halo_exchange.apply_mome_with_halo |
|
15 |
15 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / attention.py |
_mla_cp_alltoall |
|
7 |
1 |
0 |
|
86% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / attention.py |
_mla_cp_alltoall.mla_with_sequence_head_exchange |
|
15 |
15 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / attention.py |
_dsa_cp_alltoall |
|
19 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / attention.py |
_dsa_cp_alltoall.index_with_gathered_sequence |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / attention.py |
_dsa_cp_alltoall.sparse_attention_with_gathered_kv |
|
21 |
21 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / attention.py |
_dsa_cp_alltoall.CPDSAKLLoss.apply |
|
7 |
7 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / attention.py |
head_tail_load_balance_attention |
|
19 |
19 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / attention.py |
head_tail_load_balance_attention.run_half |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / attention.py |
_cp_offset_causal_mask |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / attention.py |
(no function) |
|
20 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / collectives.py |
_SeqAllToAll.forward |
|
10 |
10 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / collectives.py |
_SeqAllToAll.backward |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / collectives.py |
_UlyssesContext.group |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / collectives.py |
_UlyssesContext.size |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / collectives.py |
_UlyssesContext.rank |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / collectives.py |
_sequence_to_head |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / collectives.py |
_head_to_sequence |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / collectives.py |
_gather_sequence |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / collectives.py |
_slice_sequence |
|
4 |
1 |
0 |
|
75% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / collectives.py |
_global_seq_len |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / collectives.py |
_slice_sink |
|
7 |
7 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / collectives.py |
_normalize_dim |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / collectives.py |
_reconstruct_all_to_all |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / collectives.py |
_move_dim_to_front |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / collectives.py |
_move_dim_from_front |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / collectives.py |
_prepare_ulysses_send |
|
9 |
9 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / collectives.py |
_AsyncAllGatherWait.forward |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / collectives.py |
_AsyncAllGatherWait.backward |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / collectives.py |
_AsyncUlyssesWait.forward |
|
7 |
7 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / collectives.py |
_AsyncUlyssesWait.backward |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / collectives.py |
AsyncCPCollective.wait |
|
10 |
10 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / collectives.py |
async_cp_allgather_launch |
|
10 |
10 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / collectives.py |
async_ulysses_seq_to_head_launch |
|
11 |
11 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / collectives.py |
_ulysses_all_to_all |
|
18 |
18 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / collectives.py |
ulysses_seq_to_head |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / collectives.py |
ulysses_head_to_seq |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / collectives.py |
_AllGatherAlongDim.forward |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / collectives.py |
_AllGatherAlongDim.backward |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / collectives.py |
flex_cp_allgather |
|
5 |
2 |
0 |
|
60% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / collectives.py |
_build_hybrid_cp_submeshes |
|
28 |
28 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / collectives.py |
hybrid_cp_attention |
|
9 |
9 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / collectives.py |
(no function) |
|
71 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / gated_delta_net.py |
_global_peer_rank |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / gated_delta_net.py |
_slice_local_cp |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / gated_delta_net.py |
_slice_qkv_local_cp |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / gated_delta_net.py |
_local_tensor_at_cp_boundary |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / gated_delta_net.py |
_kv_groups |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / gated_delta_net.py |
_apply_local_padding_mask |
|
7 |
7 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / gated_delta_net.py |
_validate_gdn_activation |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / gated_delta_net.py |
_all_to_all_previous_rank_halo |
|
19 |
19 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / gated_delta_net.py |
_causal_conv1d_with_cp_halo |
|
12 |
12 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / gated_delta_net.py |
_l2norm_torch |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / gated_delta_net.py |
_prepare_gdn_chunks_for_summary |
|
30 |
30 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / gated_delta_net.py |
_compute_gdn_state_summary_from_prepared |
|
18 |
18 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / gated_delta_net.py |
_checkpoint_gdn_state_summary |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / gated_delta_net.py |
_checkpoint_gdn_state_summary.recompute |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / gated_delta_net.py |
_run_prepared_gdn_chunks |
|
20 |
20 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / gated_delta_net.py |
_RecvInitialStateP2PFunction.forward |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / gated_delta_net.py |
_RecvInitialStateP2PFunction.backward |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / gated_delta_net.py |
_SendFinalStateP2PFunction.forward |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / gated_delta_net.py |
_SendFinalStateP2PFunction.backward |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / gated_delta_net.py |
_apply_gdn_state_summary |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / gated_delta_net.py |
_gdn_state_p2p_summary |
|
23 |
23 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / gated_delta_net.py |
_GDNStateP2PTritonFunction.forward |
|
35 |
35 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / gated_delta_net.py |
_GDNStateP2PTritonFunction.backward |
|
28 |
28 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / gated_delta_net.py |
_gdn_state_p2p_triton |
|
8 |
8 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / gated_delta_net.py |
_differentiable_all_to_all_shard |
|
43 |
43 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / gated_delta_net.py |
GatedDeltaNetUlyssesCP.__init__ |
|
17 |
17 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / gated_delta_net.py |
GatedDeltaNetUlyssesCP._validate_module |
|
13 |
13 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / gated_delta_net.py |
GatedDeltaNetUlyssesCP._seq_to_head |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / gated_delta_net.py |
GatedDeltaNetUlyssesCP._head_to_seq |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / gated_delta_net.py |
GatedDeltaNetUlyssesCP._seq_to_head_qkvba |
|
15 |
15 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / gated_delta_net.py |
GatedDeltaNetUlyssesCP._local_conv_weight |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / gated_delta_net.py |
GatedDeltaNetUlyssesCP._local_conv_bias |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / gated_delta_net.py |
GatedDeltaNetUlyssesCP.forward |
|
46 |
46 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / gated_delta_net.py |
GatedDeltaNetP2PCP.__init__ |
|
15 |
15 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / gated_delta_net.py |
GatedDeltaNetP2PCP._validate_module |
|
16 |
16 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / gated_delta_net.py |
GatedDeltaNetP2PCP._conv1d_with_halo |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / gated_delta_net.py |
GatedDeltaNetP2PCP.forward |
|
37 |
37 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / gated_delta_net.py |
(no function) |
|
73 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / kimi_delta_attention.py |
_run_local_kda |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / kimi_delta_attention.py |
_is_official_kimi_layer |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / kimi_delta_attention.py |
_num_value_heads |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / kimi_delta_attention.py |
_value_head_dim |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / kimi_delta_attention.py |
_key_projection_dim |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / kimi_delta_attention.py |
_value_projection_dim |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / kimi_delta_attention.py |
_gate_lower_bound |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / kimi_delta_attention.py |
_uses_safe_gate |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / kimi_delta_attention.py |
_uses_short_conv |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / kimi_delta_attention.py |
_forget_gate_projection |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / kimi_delta_attention.py |
_output_gate_projection |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / kimi_delta_attention.py |
_format_layer_output |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / kimi_delta_attention.py |
_global_peer_rank |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / kimi_delta_attention.py |
_local_tensor_at_cp_boundary |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / kimi_delta_attention.py |
_all_to_all_previous_rank_halo |
|
19 |
19 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / kimi_delta_attention.py |
_causal_short_convs_with_cp_halo |
|
22 |
22 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / kimi_delta_attention.py |
_slice_local_heads |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / kimi_delta_attention.py |
_differentiable_all_to_all_shard |
|
32 |
32 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / kimi_delta_attention.py |
_RecvKDAInitialState.forward |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / kimi_delta_attention.py |
_RecvKDAInitialState.backward |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / kimi_delta_attention.py |
_SendKDAFinalState.forward |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / kimi_delta_attention.py |
_SendKDAFinalState.backward |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / kimi_delta_attention.py |
_eager_kda_state_p2p |
|
18 |
18 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / kimi_delta_attention.py |
KimiDeltaAttentionUlyssesCP.__init__ |
|
17 |
17 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / kimi_delta_attention.py |
KimiDeltaAttentionUlyssesCP._validate_inputs |
|
25 |
25 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / kimi_delta_attention.py |
KimiDeltaAttentionUlyssesCP._seq_to_head |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / kimi_delta_attention.py |
KimiDeltaAttentionUlyssesCP._head_to_seq |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / kimi_delta_attention.py |
KimiDeltaAttentionUlyssesCP.forward |
|
12 |
12 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / kimi_delta_attention.py |
KimiDeltaAttentionP2PCP.__init__ |
|
16 |
16 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / kimi_delta_attention.py |
KimiDeltaAttentionP2PCP.forward |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / kimi_delta_attention.py |
KimiDeltaAttentionLayerUlyssesCP.__init__ |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / kimi_delta_attention.py |
KimiDeltaAttentionLayerUlyssesCP._validate_module |
|
28 |
28 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / kimi_delta_attention.py |
KimiDeltaAttentionLayerUlyssesCP._pack_projected_inputs |
|
9 |
9 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / kimi_delta_attention.py |
KimiDeltaAttentionLayerUlyssesCP._local_short_conv |
|
9 |
9 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / kimi_delta_attention.py |
KimiDeltaAttentionLayerUlyssesCP.forward |
|
41 |
41 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / kimi_delta_attention.py |
KimiDeltaAttentionLayerP2PCP.__init__ |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / kimi_delta_attention.py |
KimiDeltaAttentionLayerP2PCP._validate_module |
|
26 |
26 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / kimi_delta_attention.py |
KimiDeltaAttentionLayerP2PCP.forward |
|
30 |
30 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / kimi_delta_attention.py |
KimiDeltaAttentionLayerP2PCP._run_local_kda |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / kimi_delta_attention.py |
(no function) |
|
63 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / wrappers.py |
_attn_implementation |
|
5 |
1 |
0 |
|
80% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / wrappers.py |
is_sdpa_attention |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / wrappers.py |
is_flex_attention |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / wrappers.py |
is_hf_style_attention |
|
9 |
2 |
0 |
|
78% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / wrappers.py |
_prepare_cp_sdpa_kwargs |
|
22 |
22 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / wrappers.py |
_prepare_head_tail_sdpa_kwargs |
|
10 |
10 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / wrappers.py |
_prepare_hybrid_sdpa_kwargs |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / wrappers.py |
_cp_sdpa_call |
|
5 |
1 |
0 |
|
80% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / wrappers.py |
_ulysses_attention_call |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / wrappers.py |
_ulysses_qkv_forward |
|
15 |
15 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / wrappers.py |
_require_ulysses_cp_mesh |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / wrappers.py |
_normalize_hf_sdpa_gqa |
|
14 |
14 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / wrappers.py |
_bind_qkv_invocation |
|
26 |
26 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / wrappers.py |
_bind_qkv_invocation.attention_call |
|
15 |
15 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / wrappers.py |
sdpa_qkv_cp_wrapper |
|
7 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / wrappers.py |
sdpa_qkv_cp_wrapper.cp_forward |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / wrappers.py |
flex_qkv_cp_wrapper |
|
7 |
7 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / wrappers.py |
flex_qkv_cp_wrapper.cp_forward |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / wrappers.py |
sdpa_hf_cp_wrapper |
|
8 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / wrappers.py |
sdpa_hf_cp_wrapper.cp_forward |
|
9 |
1 |
0 |
|
89% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / wrappers.py |
sdpa_hf_cp_wrapper.cp_forward.cp_aware_sdpa |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / wrappers.py |
flex_hf_cp_wrapper |
|
9 |
9 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / wrappers.py |
flex_hf_cp_wrapper.cp_forward |
|
9 |
9 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / wrappers.py |
flex_hf_cp_wrapper.cp_forward.cp_aware_flex |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / wrappers.py |
sdpa_qkv_load_balance_cp_wrapper |
|
7 |
7 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / wrappers.py |
sdpa_qkv_load_balance_cp_wrapper.cp_forward |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / wrappers.py |
sdpa_hf_load_balance_cp_wrapper |
|
8 |
8 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / wrappers.py |
sdpa_hf_load_balance_cp_wrapper.cp_forward |
|
9 |
9 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / wrappers.py |
sdpa_hf_load_balance_cp_wrapper.cp_forward.load_balanced_sdpa |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / wrappers.py |
sdpa_qkv_ulysses_cp_wrapper |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / wrappers.py |
sdpa_qkv_ulysses_cp_wrapper.cp_forward |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / wrappers.py |
flex_qkv_ulysses_cp_wrapper |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / wrappers.py |
flex_qkv_ulysses_cp_wrapper.cp_forward |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / wrappers.py |
sdpa_hf_ulysses_cp_wrapper |
|
7 |
7 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / wrappers.py |
sdpa_hf_ulysses_cp_wrapper.cp_forward |
|
9 |
9 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / wrappers.py |
sdpa_hf_ulysses_cp_wrapper.cp_forward.ulysses_sdpa |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / wrappers.py |
flex_hf_ulysses_cp_wrapper |
|
8 |
8 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / wrappers.py |
flex_hf_ulysses_cp_wrapper.cp_forward |
|
9 |
9 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / wrappers.py |
flex_hf_ulysses_cp_wrapper.cp_forward.ulysses_flex |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / wrappers.py |
_validate_hybrid_config |
|
9 |
9 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / wrappers.py |
_apply_qkv_hybrid_wrapper |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / wrappers.py |
_apply_qkv_hybrid_wrapper.cp_forward |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / wrappers.py |
sdpa_qkv_hybrid_cp_wrapper |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / wrappers.py |
flex_qkv_hybrid_cp_wrapper |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / wrappers.py |
sdpa_hf_hybrid_cp_wrapper |
|
7 |
7 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / wrappers.py |
sdpa_hf_hybrid_cp_wrapper.cp_forward |
|
9 |
9 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / wrappers.py |
sdpa_hf_hybrid_cp_wrapper.cp_forward.hybrid_sdpa |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / wrappers.py |
flex_hf_hybrid_cp_wrapper |
|
8 |
8 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / wrappers.py |
flex_hf_hybrid_cp_wrapper.cp_forward |
|
9 |
9 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / wrappers.py |
flex_hf_hybrid_cp_wrapper.cp_forward.hybrid_flex |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / wrappers.py |
_input_cp_sharding |
|
6 |
1 |
0 |
|
83% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / wrappers.py |
_input_cp_sharding.forward_with_sequence_sharding |
|
20 |
5 |
0 |
|
75% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / wrappers.py |
_validate_ulysses_requirements |
|
19 |
3 |
0 |
|
84% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / wrappers.py |
mla_dsa_ulysses_cp_wrapper |
|
33 |
5 |
0 |
|
85% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / wrappers.py |
mla_dsa_ulysses_cp_wrapper.forward_with_ulysses_adapters |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / context_parallel / wrappers.py |
(no function) |
|
64 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / expert_parallel / __init__.py |
(no function) |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / expert_parallel / collectives.py |
_backend_supports_uneven_a2a |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / expert_parallel / collectives.py |
_EPAllToAllUneven.forward |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / expert_parallel / collectives.py |
_EPAllToAllUneven.backward |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / expert_parallel / collectives.py |
_EPAllToAllPadded._pad_and_exchange |
|
10 |
10 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / expert_parallel / collectives.py |
_EPAllToAllPadded._unpad |
|
7 |
7 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / expert_parallel / collectives.py |
_EPAllToAllPadded.forward |
|
9 |
9 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / expert_parallel / collectives.py |
_EPAllToAllPadded.backward |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / expert_parallel / collectives.py |
ep_all_to_all |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / expert_parallel / collectives.py |
(no function) |
|
20 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / expert_parallel / experts.py |
resolve_swiglu_weights |
|
15 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / expert_parallel / experts.py |
_local_swiglu_expert_forward |
|
31 |
12 |
0 |
|
61% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / expert_parallel / experts.py |
_get_global_expert_count |
|
9 |
9 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / expert_parallel / experts.py |
bind_local_expert_forward |
|
13 |
13 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / expert_parallel / experts.py |
_prepare_ep_dispatch |
|
14 |
14 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / expert_parallel / experts.py |
_run_ep_local_experts |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / expert_parallel / experts.py |
_aggregate_ep_outputs |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / expert_parallel / experts.py |
ep_routed_forward |
|
11 |
11 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / expert_parallel / experts.py |
require_attrs |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / expert_parallel / experts.py |
describe_moe_module |
|
15 |
15 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / expert_parallel / experts.py |
(no function) |
|
17 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / expert_parallel / recipes.py |
_require_ep_group |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / expert_parallel / recipes.py |
_require_moe_interface |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / expert_parallel / recipes.py |
build_ep_compute |
|
7 |
1 |
0 |
|
86% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / expert_parallel / recipes.py |
build_ep_compute.compute_fn |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / expert_parallel / recipes.py |
routed_only_ep_compute_fn |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / expert_parallel / recipes.py |
mixtral_ep_compute_fn |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / expert_parallel / recipes.py |
mixtral_ep_compute_fn.compute_fn |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / expert_parallel / recipes.py |
qwen2moe_ep_compute_fn |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / expert_parallel / recipes.py |
qwen2moe_ep_compute_fn.combine |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / expert_parallel / recipes.py |
deepseekv3_ep_compute_fn |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / expert_parallel / recipes.py |
deepseekv3_ep_compute_fn.combine |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / expert_parallel / recipes.py |
(no function) |
|
18 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / expert_parallel / routing.py |
_softmax_topk_router |
|
14 |
2 |
0 |
|
86% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / expert_parallel / routing.py |
_topk_router_module |
|
8 |
2 |
0 |
|
75% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / expert_parallel / routing.py |
_sigmoid_group_router |
|
29 |
8 |
0 |
|
72% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / expert_parallel / routing.py |
_sigmoid_group_router._attr |
|
4 |
1 |
0 |
|
75% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / expert_parallel / routing.py |
(no function) |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / mesh.py |
_init_topology_mesh |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / mesh.py |
MeshContext.pp_enabled |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / mesh.py |
MeshContext.dp_cp_mesh |
|
12 |
12 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / mesh.py |
MeshContext.cp_mesh |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / mesh.py |
MeshContext.build_meshs |
|
41 |
41 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / mesh.py |
_validate_dense_tp_rank_layout |
|
8 |
8 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / mesh.py |
_build_device_mesh_from_accelerator |
|
12 |
12 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / mesh.py |
(no function) |
|
51 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / plan.py |
ShardingPlan._format_named_placement |
|
4 |
1 |
0 |
|
75% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / plan.py |
ShardingPlan._format_injection_callable |
|
8 |
8 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / plan.py |
ShardingPlan._append_parameter_explanation |
|
14 |
5 |
0 |
|
64% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / plan.py |
ShardingPlan._append_communication_explanation |
|
13 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / plan.py |
ShardingPlan._append_injection_explanation |
|
13 |
6 |
0 |
|
54% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / plan.py |
ShardingPlan._append_handler_explanation |
|
3 |
1 |
0 |
|
67% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / plan.py |
ShardingPlan.explain |
|
12 |
2 |
0 |
|
83% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / plan.py |
(no function) |
|
19 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / recipe_spec.py |
PlacementMismatchError.__init__ |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / recipe_spec.py |
resolve_placements |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / recipe_spec.py |
parse_placement |
|
11 |
2 |
0 |
|
82% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / recipe_spec.py |
parse_named_placement |
|
8 |
2 |
0 |
|
75% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / recipe_spec.py |
_normalize_out_fields |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / recipe_spec.py |
_make_decorator |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / recipe_spec.py |
_make_decorator.decorator |
|
18 |
2 |
0 |
|
89% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / recipe_spec.py |
require_injection_meta |
|
7 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / recipe_spec.py |
(no function) |
|
71 |
0 |
2 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / tensor_parallel / __init__.py |
(no function) |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / tensor_parallel / head_count.py |
_is_head_sharded |
|
9 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / tensor_parallel / head_count.py |
_tp_degree |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / tensor_parallel / head_count.py |
update_module_head_counts |
|
21 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / tensor_parallel / head_count.py |
collect_auto_head_attrs |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / tensor_parallel / head_count.py |
normalize_tp_divide_attrs |
|
22 |
5 |
0 |
|
77% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / tensor_parallel / head_count.py |
build_tp_local_attr_plan |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / tensor_parallel / head_count.py |
_update_user_tp_attrs |
|
24 |
2 |
0 |
|
92% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / tensor_parallel / head_count.py |
maybe_update_head_counts |
|
10 |
1 |
0 |
|
90% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / tensor_parallel / head_count.py |
(no function) |
|
24 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / tensor_parallel / param_role.py |
_match_any |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / tensor_parallel / param_role.py |
_match_pattern |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / tensor_parallel / param_role.py |
_match_rule |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / tensor_parallel / param_role.py |
_build_default_rules |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / tensor_parallel / param_role.py |
ParameterClassifier.__init__ |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / tensor_parallel / param_role.py |
ParameterClassifier.classify |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / tensor_parallel / param_role.py |
ParameterClassifier.classify_param |
|
16 |
4 |
0 |
|
75% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / distributed / tensor_parallel / param_role.py |
(no function) |
|
29 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / infer / __init__.py |
(no function) |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / infer / generation.py |
_get_output |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / infer / generation.py |
_model_forward |
|
13 |
13 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / infer / generation.py |
_resolve_context_parallel_rank_world |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / infer / generation.py |
_init_cache |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / infer / generation.py |
_cache_shard_info |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / infer / generation.py |
_cache_seq_len |
|
8 |
8 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / infer / generation.py |
_cache_batch_size |
|
8 |
8 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / infer / generation.py |
_resolve_prefix_length |
|
18 |
18 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / infer / generation.py |
_prepare_prefix_attention_mask |
|
14 |
14 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / infer / generation.py |
_init_cache_with_prefix |
|
11 |
11 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / infer / generation.py |
_update_cache |
|
14 |
14 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / infer / generation.py |
_resolve_mask_dtype |
|
10 |
10 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / infer / generation.py |
_build_decode_key_mask |
|
8 |
8 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / infer / generation.py |
_combined_attention_mask |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / infer / generation.py |
_build_prefill_mask |
|
17 |
17 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / infer / generation.py |
_build_prefill_position_ids |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / infer / generation.py |
_prompt_lengths |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / infer / generation.py |
_finalize_sequences |
|
17 |
17 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / infer / generation.py |
_validate_generate_inputs |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / infer / generation.py |
_finalize_zero_new_tokens |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / infer / generation.py |
_prepare_generation_context |
|
10 |
10 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / infer / generation.py |
_prefill |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / infer / generation.py |
_required_logits |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / infer / generation.py |
_finalize_prefill_outputs |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / infer / generation.py |
_append_next_token |
|
8 |
8 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / infer / generation.py |
_should_finish_generation |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / infer / generation.py |
_decode |
|
7 |
7 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / infer / generation.py |
generate |
|
26 |
26 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / infer / generation.py |
(no function) |
|
36 |
36 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / infer / kv_cache.py |
_validate_pair_shapes |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / infer / kv_cache.py |
detach_and_validate_past_key_values |
|
8 |
8 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / infer / kv_cache.py |
SequenceShardInfo.local_seq_len |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / infer / kv_cache.py |
get_sequence_shard_info |
|
11 |
11 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / infer / kv_cache.py |
shard_past_key_values |
|
11 |
11 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / infer / kv_cache.py |
KVCache.__init__ |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / infer / kv_cache.py |
KVCache.is_empty |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / infer / kv_cache.py |
KVCache.clear |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / infer / kv_cache.py |
KVCache.update |
|
7 |
7 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / infer / kv_cache.py |
KVCache.merge |
|
23 |
23 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / infer / kv_cache.py |
KVCache._detach_and_validate |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / infer / kv_cache.py |
KVCache._validate_pair_shapes |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / infer / kv_cache.py |
KVCache._is_opaque_cache |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / infer / kv_cache.py |
ContextParallelKVCache.__init__ |
|
8 |
8 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / infer / kv_cache.py |
ContextParallelKVCache.update_full |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / infer / kv_cache.py |
ContextParallelKVCache.update_local |
|
7 |
7 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / infer / kv_cache.py |
ContextParallelKVCache.merge_local |
|
23 |
23 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / infer / kv_cache.py |
ContextParallelKVCache.clear |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / infer / kv_cache.py |
ContextParallelKVCache._validate_shard_info |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / infer / kv_cache.py |
ContextParallelKVCache._validate_local_seq_len |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / infer / kv_cache.py |
(no function) |
|
39 |
39 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / infer / mixin.py |
GenerateMixin.generate |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / infer / mixin.py |
(no function) |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / infer / sampler.py |
_filter_top_k |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / infer / sampler.py |
greedy_sample |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / infer / sampler.py |
top_k_sample |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / infer / sampler.py |
top_p_sample |
|
13 |
13 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / infer / sampler.py |
apply_repetition_penalty |
|
12 |
12 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / infer / sampler.py |
sample_next_token |
|
8 |
8 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / infer / sampler.py |
(no function) |
|
9 |
9 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / infer / utils.py |
GenerationConfig.__post_init__ |
|
9 |
9 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / infer / utils.py |
GenerationConfig._validate_sampling |
|
10 |
10 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / infer / utils.py |
GenerationConfig._validate_prefix |
|
13 |
13 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / infer / utils.py |
GenerationConfig._validate_context_parallel |
|
9 |
9 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / infer / utils.py |
GenerationConfig._validate_callables |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / infer / utils.py |
build_position_ids |
|
9 |
9 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / infer / utils.py |
gather_context_parallel_logits |
|
22 |
22 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / infer / utils.py |
gather_tensor_parallel_logits |
|
19 |
19 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / infer / utils.py |
prepare_logits_for_sampling |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / infer / utils.py |
apply_logits_processors |
|
8 |
8 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / infer / utils.py |
should_stop_generation |
|
11 |
11 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / infer / utils.py |
build_causal_mask |
|
14 |
14 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / infer / utils.py |
append_attention_mask |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / infer / utils.py |
(no function) |
|
44 |
44 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / __init__.py |
(no function) |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / __init__.py |
(no function) |
|
0 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / activation_checkpoint / __init__.py |
(no function) |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / activation_checkpoint / activation_swap.py |
_strip_ckpt_wrapped_module_prefix |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / activation_checkpoint / activation_swap.py |
FuncCell.__init__ |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / activation_checkpoint / activation_swap.py |
FuncCell.construct |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / activation_checkpoint / activation_swap.py |
_is_shared_function_callable |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / activation_checkpoint / activation_swap.py |
_iter_wrappable_callable_attrs |
|
7 |
1 |
0 |
|
86% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / activation_checkpoint / activation_swap.py |
_mark_wrapped |
|
4 |
2 |
0 |
|
50% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / activation_checkpoint / activation_swap.py |
_get_wrapped_callable |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / activation_checkpoint / activation_swap.py |
_raise_callable_already_wrapped |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / activation_checkpoint / activation_swap.py |
_check_callable_attr_not_wrapped |
|
3 |
1 |
0 |
|
67% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / activation_checkpoint / activation_swap.py |
_check_and_mark_callable |
|
5 |
3 |
0 |
|
40% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / activation_checkpoint / activation_swap.py |
_check_and_mark_wrapped |
|
19 |
3 |
0 |
|
84% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / activation_checkpoint / activation_swap.py |
ActivationWrapper.__init__ |
|
12 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / activation_checkpoint / activation_swap.py |
ActivationWrapper._wrapped_module |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / activation_checkpoint / activation_swap.py |
ActivationWrapper.construct |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / activation_checkpoint / activation_swap.py |
ActivationWrapper.__getattr__ |
|
9 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / activation_checkpoint / activation_swap.py |
ActivationWrapper.unwrap_cell |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / activation_checkpoint / activation_swap.py |
ActivationWrapper.__getitem__ |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / activation_checkpoint / activation_swap.py |
ActivationWrapper.cells_and_names |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / activation_checkpoint / activation_swap.py |
ActivationWrapper.parameters_and_names |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / activation_checkpoint / activation_swap.py |
ActivationWrapper.update_parameters_name |
|
13 |
3 |
0 |
|
77% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / activation_checkpoint / activation_swap.py |
base_check_fn |
|
7 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / activation_checkpoint / activation_swap.py |
_normalize_device |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / activation_checkpoint / activation_swap.py |
AsyncSaveOnCpu.__init__ |
|
10 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / activation_checkpoint / activation_swap.py |
AsyncSaveOnCpu.__init__.pack_to_cpu |
|
18 |
6 |
0 |
|
67% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / activation_checkpoint / activation_swap.py |
AsyncSaveOnCpu.__init__.unpack_from_cpu |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / activation_checkpoint / activation_swap.py |
SwapWrapper.__init__ |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / activation_checkpoint / activation_swap.py |
SwapWrapper.construct |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / activation_checkpoint / activation_swap.py |
swap_wrapper |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / activation_checkpoint / activation_swap.py |
swap_tensor_wrapper |
|
16 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / activation_checkpoint / activation_swap.py |
swap_tensor_wrapper._apply |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / activation_checkpoint / activation_swap.py |
swap_tensor_wrapper._map |
|
7 |
1 |
0 |
|
86% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / activation_checkpoint / activation_swap.py |
(no function) |
|
44 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / activation_checkpoint / checkpoint_exclude_wrapper.py |
_RecomputedInputHandle.__init__ |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / activation_checkpoint / checkpoint_exclude_wrapper.py |
_RecomputedInputHandle.mark_used |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / activation_checkpoint / checkpoint_exclude_wrapper.py |
_RecomputedInputHandle.materialize |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / activation_checkpoint / checkpoint_exclude_wrapper.py |
_RecomputedInputHandle.get_recomputed_tensor |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / activation_checkpoint / checkpoint_exclude_wrapper.py |
_ExcludeCache.__init__ |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / activation_checkpoint / checkpoint_exclude_wrapper.py |
_ExcludeCache.save |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / activation_checkpoint / checkpoint_exclude_wrapper.py |
_ExcludeCache.pop |
|
7 |
1 |
0 |
|
86% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / activation_checkpoint / checkpoint_exclude_wrapper.py |
_ExcludeCache.clear |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / activation_checkpoint / checkpoint_exclude_wrapper.py |
_pack_saved_tensor |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / activation_checkpoint / checkpoint_exclude_wrapper.py |
_unpack_saved_tensor |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / activation_checkpoint / checkpoint_exclude_wrapper.py |
_saved_tensors_context |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / activation_checkpoint / checkpoint_exclude_wrapper.py |
_append_tensor_inputs |
|
10 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / activation_checkpoint / checkpoint_exclude_wrapper.py |
_collect_tensor_inputs |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / activation_checkpoint / checkpoint_exclude_wrapper.py |
_mark_recompute_inputs |
|
17 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / activation_checkpoint / checkpoint_exclude_wrapper.py |
_restore_recompute_inputs |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / activation_checkpoint / checkpoint_exclude_wrapper.py |
_resolve_input |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / activation_checkpoint / checkpoint_exclude_wrapper.py |
_materialize_recompute_inputs |
|
7 |
2 |
0 |
|
71% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / activation_checkpoint / checkpoint_exclude_wrapper.py |
_has_used_input |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / activation_checkpoint / checkpoint_exclude_wrapper.py |
_get_replay_placeholder |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / activation_checkpoint / checkpoint_exclude_wrapper.py |
_make_replay_placeholder_output |
|
6 |
2 |
0 |
|
67% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / activation_checkpoint / checkpoint_exclude_wrapper.py |
_get_recompute_trigger |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / activation_checkpoint / checkpoint_exclude_wrapper.py |
_RecomputeBoundary.forward |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / activation_checkpoint / checkpoint_exclude_wrapper.py |
_RecomputeBoundary.backward |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / activation_checkpoint / checkpoint_exclude_wrapper.py |
_finalize_save_outputs |
|
18 |
5 |
0 |
|
72% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / activation_checkpoint / checkpoint_exclude_wrapper.py |
CheckpointExcludeWrapper.__init__ |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / activation_checkpoint / checkpoint_exclude_wrapper.py |
CheckpointExcludeWrapper.construct |
|
21 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / activation_checkpoint / checkpoint_exclude_wrapper.py |
checkpoint_exclude_wrapper |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / activation_checkpoint / checkpoint_exclude_wrapper.py |
(no function) |
|
57 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / activation_checkpoint / checkpoint_wrapper.py |
CheckpointWrapper.__init__ |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / activation_checkpoint / checkpoint_wrapper.py |
CheckpointWrapper._do_checkpoint |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / activation_checkpoint / checkpoint_wrapper.py |
CheckpointWrapper.construct |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / activation_checkpoint / checkpoint_wrapper.py |
ckpt_wrapper |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / activation_checkpoint / checkpoint_wrapper.py |
(no function) |
|
8 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / activation_checkpoint / sac.py |
_VersionWrapper.__init__ |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / activation_checkpoint / sac.py |
_VersionWrapper.get_val |
|
4 |
1 |
0 |
|
75% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / activation_checkpoint / sac.py |
_SwapCacheEntry.__init__ |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / activation_checkpoint / sac.py |
_maybe_detach |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / activation_checkpoint / sac.py |
SelectiveCheckpointContext.__init__ |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / activation_checkpoint / sac.py |
ignore_sac_ops |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / activation_checkpoint / sac.py |
_CachingMindSporeDispatchMode.__init__ |
|
7 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / activation_checkpoint / sac.py |
_CachingMindSporeDispatchMode.__ms_dispatch__ |
|
21 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / activation_checkpoint / sac.py |
_CachedMindSporeDispatchMode.__init__ |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / activation_checkpoint / sac.py |
_CachedMindSporeDispatchMode.__ms_dispatch__ |
|
16 |
1 |
0 |
|
94% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / activation_checkpoint / sac.py |
create_selective_checkpoint_contexts |
|
8 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / activation_checkpoint / sac.py |
create_selective_checkpoint_contexts.policy_fn |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / activation_checkpoint / sac.py |
(no function) |
|
25 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / autograd_compat.py |
requires_grad |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / autograd_compat.py |
requires_grad |
|
3 |
1 |
0 |
|
67% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / autograd_compat.py |
grad |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / autograd_compat.py |
grad |
|
14 |
4 |
0 |
|
71% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / autograd_compat.py |
is_leaf |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / autograd_compat.py |
retains_grad |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / autograd_compat.py |
grad_fn |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / autograd_compat.py |
output_nr |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / autograd_compat.py |
retain_grad |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / autograd_compat.py |
detach |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / autograd_compat.py |
_is_same_size |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / autograd_compat.py |
_calculate_shape |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / autograd_compat.py |
_tensor_or_tensors_to_tuple |
|
5 |
1 |
0 |
|
80% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / autograd_compat.py |
_make_grads |
|
17 |
5 |
0 |
|
71% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / autograd_compat.py |
backward |
|
16 |
1 |
0 |
|
94% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / autograd_compat.py |
enable_mindspore_backward_compat |
|
13 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / autograd_compat.py |
(no function) |
|
31 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / dtensor.py |
_MindSporeGetItem.__call__ |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / dtensor.py |
DTensorBase.__new__ |
|
25 |
3 |
0 |
|
88% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / dtensor.py |
DTensorBase.asnumpy |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / dtensor.py |
DTensorBase.__str__ |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / dtensor.py |
DTensorBase.__getitem__ |
|
10 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / dtensor.py |
DTensorBase.__copy__ |
|
23 |
3 |
0 |
|
87% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / dtensor.py |
DTensorBase.__fallback__ |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / dtensor.py |
DTensorBase._need_contiguous |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / dtensor.py |
DTensorBase.device |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / dtensor.py |
DTensorBase.data |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / dtensor.py |
DTensorBase.data |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / dtensor.py |
DTensorBase.set_data |
|
16 |
10 |
0 |
|
38% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / dtensor.py |
DTensorBase.has_init |
|
3 |
1 |
0 |
|
67% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / dtensor.py |
DTensorBase.init |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / dtensor.py |
DTensorBase.init |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / dtensor.py |
DTensorBase.local_param_info |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / dtensor.py |
DTensorBase.local_param_info |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / dtensor.py |
DTensorBase._alias_placements |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / dtensor.py |
DTensorBase.to |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / dtensor.py |
(no function) |
|
34 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / __init__.py |
(no function) |
|
0 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / _version_utils.py |
copy_without_bumping_version |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / _version_utils.py |
(no function) |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / hook_function.py |
PostBackwardFunction.forward |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / hook_function.py |
PostBackwardFunction.backward |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / hook_function.py |
PostBackwardFunction.apply |
|
26 |
26 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / hook_function.py |
(no function) |
|
8 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / param.py |
_to_dtype_if_needed |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / param.py |
_pad_dim0_for_communication |
|
8 |
1 |
0 |
|
88% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / param.py |
_split_sharded_param |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / param.py |
_pack_dim0_reduce_scatter_input |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / param.py |
_unpack_dim0_all_gather_output |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / param.py |
make_contiguous_strides_for |
|
18 |
18 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / param.py |
ParameterHookMigrator.__init__ |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / param.py |
ParameterHookMigrator._save_backward_hooks |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / param.py |
ParameterHookMigrator._migrate_backward_hooks |
|
9 |
9 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / param.py |
MindSporeHSDPParamV2.__init__ |
|
30 |
30 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / param.py |
MindSporeHSDPParamV2._get_base_spmd_placements |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / param.py |
MindSporeHSDPParamV2._apply_data_parallel_placements |
|
13 |
13 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / param.py |
MindSporeHSDPParamV2._build_sharding_spec |
|
17 |
17 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / param.py |
MindSporeHSDPParamV2.reduce_partial_output |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / param.py |
MindSporeHSDPParamV2.reduce_partial_output |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / param.py |
MindSporeHSDPParamV2.reduce_comm_dtype |
|
9 |
5 |
0 |
|
44% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / param.py |
MindSporeHSDPParamV2.reduce_scatter_output |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / param.py |
MindSporeHSDPParamV2.clear_reduce_scatter_output |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / param.py |
MindSporeHSDPParamV2.all_reduce_output |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / param.py |
MindSporeHSDPParamV2.clear_all_reduce_output |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / param.py |
MindSporeHSDPParamV2.clear_unsharded_source_grad |
|
4 |
2 |
0 |
|
50% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / param.py |
MindSporeHSDPParamV2.apply_reduced_grad |
|
28 |
3 |
0 |
|
89% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / param.py |
MindSporeHSDPParamV2._release_full_param_storage_if_safe |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / param.py |
MindSporeHSDPParamV2._resolve_hsdp_placement |
|
11 |
11 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / param.py |
MindSporeHSDPParamV2._init_shard_placements |
|
12 |
12 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / param.py |
MindSporeHSDPParamV2._init_sharded_param |
|
50 |
50 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / param.py |
MindSporeHSDPParamV2.init_dtype_attrs |
|
8 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / param.py |
MindSporeHSDPParamV2.init_unsharded_param_buffers |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / param.py |
MindSporeHSDPParamV2.init_unsharded_param |
|
31 |
31 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / param.py |
MindSporeHSDPParamV2.to_sharded |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / param.py |
MindSporeHSDPParamV2.to_unsharded |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / param.py |
MindSporeHSDPParamV2._setattr_on_modules |
|
9 |
9 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / param.py |
MindSporeHSDPParamV2.to_sharded_dtensor |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / param.py |
MindSporeHSDPParamV2._to_local_unsharded_grad |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / param.py |
MindSporeHSDPParamV2.to_accumulated_grad_if_needed |
|
9 |
2 |
0 |
|
78% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / param.py |
MindSporeHSDPParamV2.accumulate_unsharded_grad_if_needed |
|
6 |
1 |
0 |
|
83% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / param.py |
MindSporeHSDPParamV2.alloc_unsharded_param_buffers |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / param.py |
MindSporeHSDPParamV2.free_unsharded_param |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / param.py |
MindSporeHSDPParamV2.all_gather_inputs |
|
7 |
7 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / param.py |
MindSporeHSDPParamV2.unsharded_param |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / param.py |
MindSporeHSDPParamV2.unsharded_grad_data |
|
4 |
1 |
0 |
|
75% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / param.py |
MindSporeHSDPParamV2.unsharded_accumulated_grad_data |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / param.py |
MindSporeHSDPParamV2._sharded_local_tensor |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / param.py |
MindSporeHSDPParamV2._sharded_param_storage_dtype |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / param.py |
MindSporeHSDPParamV2._assert_in_states |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / param.py |
MindSporeHSDPParamV2._resolve_reset_param |
|
9 |
9 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / param.py |
MindSporeHSDPParamV2._is_same_sharded_local_tensor |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / param.py |
MindSporeHSDPParamV2._validate_reset_local_tensor |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / param.py |
MindSporeHSDPParamV2._pin_reset_local_tensor_if_needed |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / param.py |
MindSporeHSDPParamV2._refresh_sharded_local_tensor |
|
10 |
10 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / param.py |
MindSporeHSDPParamV2.reset_sharded_param |
|
16 |
16 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / param.py |
MindSporeHSDPParamV2._get_unsharded_param_data |
|
20 |
20 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / param.py |
MindSporeHSDPParamV2.unshard |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / param.py |
MindSporeHSDPParamV2.wait_for_unshard |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / param.py |
MindSporeHSDPParamV2.shard |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / param.py |
MindSporeHSDPParamV2.reduce_scatter_grad |
|
31 |
16 |
0 |
|
48% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / param.py |
MindSporeHSDPParamV2.zero_grad |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / param.py |
MindSporeHSDPParamV2.all_reduce_grad |
|
15 |
6 |
0 |
|
60% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / param.py |
MindSporeHSDPParamV2.all_reduce_source_replicate_grad_inplace |
|
13 |
13 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / param.py |
set_requires_grad_if_needed |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / param.py |
(no function) |
|
101 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / param_group.py |
_normalize_device |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / param_group.py |
_shape_numel |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / param_group.py |
AllGatherBucket.copy_out |
|
36 |
5 |
0 |
|
86% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / param_group.py |
ReduceScatterBucket.hsdp_params |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / param_group.py |
ReduceScatterBucket.param_offsets |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / param_group.py |
ReduceScatterBucket.bucket_key |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / param_group.py |
ReduceScatterBucket.uses_collective |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / param_group.py |
ReduceScatterBucket.move_reducescatter_output |
|
5 |
1 |
0 |
|
80% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / param_group.py |
AllReduceBucket.hsdp_params |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / param_group.py |
AllReduceBucket.param_numels |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / param_group.py |
AllReduceBucket.param_offsets |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / param_group.py |
AllReduceBucket.uses_collective |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / param_group.py |
get_all_gather_metadata |
|
20 |
1 |
0 |
|
95% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / param_group.py |
all_gather_copy_in |
|
7 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / param_group.py |
reduce_scatter_copy_in |
|
18 |
2 |
0 |
|
89% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / param_group.py |
HSDPParamGroup.__init__ |
|
10 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / param_group.py |
HSDPParamGroup._init_all_gather_buckets |
|
18 |
1 |
0 |
|
94% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / param_group.py |
HSDPParamGroup.unshard |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / param_group.py |
HSDPParamGroup.foreach_all_gather |
|
17 |
2 |
0 |
|
88% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / param_group.py |
HSDPParamGroup.wait_for_unshard |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / param_group.py |
HSDPParamGroup._build_reduce_scatter_buckets |
|
25 |
4 |
0 |
|
84% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / param_group.py |
HSDPParamGroup._build_gradient_bucket_layout_from |
|
9 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / param_group.py |
HSDPParamGroup._build_all_reduce_buckets |
|
14 |
3 |
0 |
|
79% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / param_group.py |
HSDPParamGroup._issue_reduce_scatter_buckets |
|
14 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / param_group.py |
HSDPParamGroup.foreach_reducescatter |
|
9 |
3 |
0 |
|
67% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / param_group.py |
HSDPParamGroup._wait_reduce_scatter_buckets |
|
9 |
2 |
0 |
|
78% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / param_group.py |
HSDPParamGroup._issue_all_reduce_buckets |
|
6 |
2 |
0 |
|
67% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / param_group.py |
HSDPParamGroup.wait_reduce_scatter_and_issue_all_reduce |
|
28 |
5 |
0 |
|
82% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / param_group.py |
HSDPParamGroup.wait_all_reduce_and_save_grad |
|
14 |
2 |
0 |
|
86% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / param_group.py |
HSDPParamGroup.reset_iter_state |
|
17 |
17 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / param_group.py |
AllReduceParamGroup.__init__ |
|
9 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / param_group.py |
AllReduceParamGroup.compute_aligned_layout |
|
12 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / param_group.py |
AllReduceParamGroup.allocate_fused_buffer |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / param_group.py |
AllReduceParamGroup.get_param_buffer_view |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / param_group.py |
AllReduceParamGroup.get_param_grad_view |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / param_group.py |
AllReduceParamGroup.accumulate_reduce_partial_outputs |
|
10 |
6 |
0 |
|
40% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / param_group.py |
AllReduceParamGroup.issue_async_allreduce |
|
3 |
1 |
0 |
|
67% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / param_group.py |
AllReduceParamGroup.wait_and_split_grads |
|
9 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / param_group.py |
(no function) |
|
119 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / scheduler.py |
MindSporeHSDPSchedulerV2.zero_grad |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / scheduler.py |
MindSporeHSDPSchedulerV2._register_hooks |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / scheduler.py |
MindSporeHSDPSchedulerV2._init_platform |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / scheduler.py |
MindSporeHSDPSchedulerV2._new_cell_state |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / scheduler.py |
MindSporeHSDPSchedulerV2._register_post_backward_hook |
|
21 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / scheduler.py |
MindSporeHSDPSchedulerV2._forward_pre_hook |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / scheduler.py |
MindSporeHSDPSchedulerV2._register_backward_pre_hook |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / scheduler.py |
MindSporeHSDPSchedulerV2._forward_hook |
|
7 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / scheduler.py |
MindSporeHSDPSchedulerV2._backward_pre_hook |
|
7 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / scheduler.py |
MindSporeHSDPSchedulerV2._root_backward_hook |
|
8 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / scheduler.py |
MindSporeHSDPSchedulerV2._finalize_comm_fusion_reductions |
|
12 |
2 |
0 |
|
83% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / scheduler.py |
MindSporeHSDPSchedulerV2._finalize_per_param_reductions |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / scheduler.py |
MindSporeHSDPSchedulerV2.launch_tp_replicate_reduce_and_apply |
|
16 |
10 |
0 |
|
38% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / scheduler.py |
MindSporeHSDPSchedulerV2.reset_iter_state |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / scheduler.py |
MindSporeHSDPSchedulerV2._backward_hook |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / scheduler.py |
MindSporeHSDPSchedulerV2._without_ms_dispatch_mode |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / scheduler.py |
MindSporeHSDPSchedulerV2._without_ms_dispatch_mode.wrapped_hook |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / scheduler.py |
MindSporeHSDPSchedulerV2._register_forward_backward_hooks |
|
8 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / scheduler.py |
(no function) |
|
30 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / state.py |
_to_dtype_if_needed |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / state.py |
MindSporeHSDPStateV2.__init__ |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / state.py |
MindSporeHSDPStateV2._init_param_group |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / state.py |
MindSporeHSDPStateV2._move_states_to_device |
|
12 |
12 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / state.py |
MindSporeHSDPStateV2._build_param_source_shard_info |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / state.py |
MindSporeHSDPStateV2._init_hsdp_params |
|
15 |
15 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / state.py |
MindSporeHSDPStateV2._build_param_mesh_info |
|
7 |
2 |
0 |
|
71% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / state.py |
MindSporeHSDPStateV2._init_mp_dtypes |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / state.py |
MindSporeHSDPStateV2._validate_cpu_offload_params |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / state.py |
MindSporeHSDPStateV2.lazy_init |
|
7 |
4 |
0 |
|
43% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / state.py |
MindSporeHSDPStateV2._validate_no_meta_params |
|
3 |
1 |
0 |
|
67% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / state.py |
MindSporeHSDPStateV2.zero_grad |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / state.py |
MindSporeHSDPStateV2.post_backward_for_comm_fusion |
|
13 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / state.py |
MindSporeHSDPStateV2.post_backward |
|
18 |
9 |
0 |
|
50% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / state.py |
MindSporeHSDPStateV2._issue_reduce_scatter_for_current_module |
|
27 |
1 |
0 |
|
96% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / state.py |
MindSporeHSDPStateV2._wait_prev_reduce_scatter |
|
14 |
14 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / state.py |
MindSporeHSDPStateV2._issue_prev_fused_all_reduce |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / state.py |
MindSporeHSDPStateV2._wait_prev_reduce_scatter_without_all_reduce |
|
16 |
4 |
0 |
|
75% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / state.py |
MindSporeHSDPStateV2.wait_and_split_all_reduce_work_groups |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / state.py |
MindSporeHSDPStateV2.reset_iter_state |
|
18 |
1 |
0 |
|
94% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / state.py |
MindSporeHSDPStateV2.set_requires_grad_sync |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / state.py |
MindSporeHSDPStateV2.set_reduce_op_type |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / state.py |
MindSporeHSDPStateV2._sync_current_stream_if_needed |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / state.py |
MindSporeHSDPStateV2.prefetch |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / state.py |
(no function) |
|
39 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / init_weights.py |
_cell_to_empty |
|
14 |
14 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / init_weights.py |
_install_cell_to_empty_patch |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / init_weights.py |
_check_valid_init_device |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / init_weights.py |
init_on_device |
|
9 |
9 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / init_weights.py |
init_on_device._insert_param_to_cell |
|
10 |
10 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / init_weights.py |
init_on_device._insert_param_to_cell._custom_kwargs |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / init_weights.py |
(no function) |
|
9 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / loss_parallel_ops.py |
_is_floating_ms |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / loss_parallel_ops.py |
_compute_vocab_start |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / loss_parallel_ops.py |
distributed_log_softmax |
|
8 |
8 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / loss_parallel_ops.py |
distributed_nll_loss_forward |
|
33 |
33 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / loss_parallel_ops.py |
DistributedCrossEntropyFunction.forward |
|
44 |
44 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / loss_parallel_ops.py |
DistributedCrossEntropyFunction.backward |
|
46 |
46 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / loss_parallel_ops.py |
distributed_cross_entropy |
|
24 |
24 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / loss_parallel_ops.py |
distributed_cross_entropy_from_op_call |
|
9 |
9 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / loss_parallel_ops.py |
(no function) |
|
23 |
23 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / parameter_init.py |
init_parameters |
|
31 |
31 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / parameter_init.py |
(no function) |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / pipeline_parallel / __init__.py |
(no function) |
|
0 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / pipeline_parallel / _utils.py |
_MicroBatch.__init__ |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / pipeline_parallel / _utils.py |
_MicroBatch.construct |
|
24 |
24 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / pipeline_parallel / _utils.py |
_MicroBatch.split_inputs_with_custom_shard |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / pipeline_parallel / _utils.py |
_MicroBatch.split_inputs |
|
14 |
14 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / pipeline_parallel / _utils.py |
send_object_list |
|
10 |
10 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / pipeline_parallel / _utils.py |
recv_object_list |
|
14 |
14 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / pipeline_parallel / _utils.py |
(no function) |
|
15 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / pipeline_parallel / backward.py |
_fill_grads |
|
5 |
2 |
0 |
|
60% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / pipeline_parallel / backward.py |
_validate_grad_config |
|
2 |
1 |
0 |
|
50% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / pipeline_parallel / backward.py |
_set_requires_grad |
|
17 |
8 |
0 |
|
53% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / pipeline_parallel / backward.py |
_get_grad_node |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / pipeline_parallel / backward.py |
_get_node_id |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / pipeline_parallel / backward.py |
_accumulate_grads |
|
9 |
3 |
0 |
|
67% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / pipeline_parallel / backward.py |
_compute_nodes_out_degree |
|
19 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / pipeline_parallel / backward.py |
_compute_reachable_nodes |
|
25 |
3 |
0 |
|
88% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / pipeline_parallel / backward.py |
_find_boundary_intermediates_with_edge_slots |
|
19 |
5 |
0 |
|
74% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / pipeline_parallel / backward.py |
_group_weights_by_intermediate |
|
32 |
7 |
0 |
|
78% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / pipeline_parallel / backward.py |
GradFunction.__init__ |
|
11 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / pipeline_parallel / backward.py |
GradFunction._clear_res |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / pipeline_parallel / backward.py |
GradFunction._collect_weight_tensors |
|
5 |
2 |
0 |
|
60% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / pipeline_parallel / backward.py |
GradFunction._setup_intermediate_hooks |
|
18 |
1 |
0 |
|
94% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / pipeline_parallel / backward.py |
GradFunction._setup_intermediate_hooks.make_hook |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / pipeline_parallel / backward.py |
GradFunction._setup_intermediate_hooks.make_hook.prehook_fn |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / pipeline_parallel / backward.py |
GradFunction._format_input_grads |
|
5 |
3 |
0 |
|
40% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / pipeline_parallel / backward.py |
GradFunction._prune_intermediate_edges |
|
9 |
1 |
0 |
|
89% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / pipeline_parallel / backward.py |
GradFunction._process_weight_group |
|
28 |
5 |
0 |
|
82% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / pipeline_parallel / backward.py |
GradFunction._prepare_output_and_sens |
|
13 |
7 |
0 |
|
46% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / pipeline_parallel / backward.py |
GradFunction._collect_input_tensors |
|
17 |
10 |
0 |
|
41% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / pipeline_parallel / backward.py |
GradFunction.__call__ |
|
24 |
24 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / pipeline_parallel / backward.py |
GradFunction.accumulate_grad |
|
15 |
15 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / pipeline_parallel / backward.py |
GradFunction.compute_input_grad |
|
17 |
3 |
0 |
|
82% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / pipeline_parallel / backward.py |
GradFunction.compute_weight_grad |
|
19 |
5 |
0 |
|
74% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / pipeline_parallel / backward.py |
forward_and_gradfn |
|
12 |
3 |
0 |
|
75% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / pipeline_parallel / backward.py |
(no function) |
|
36 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / pipeline_parallel / mpipe_transpose.py |
MPipeTransposeExecutor._broadcast_tensors |
|
0 |
0 |
3 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / pipeline_parallel / mpipe_transpose.py |
MPipeTransposeExecutor._detached_forward |
|
0 |
0 |
1 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / pipeline_parallel / mpipe_transpose.py |
MPipeTransposeExecutor._mark_requires_grad |
|
0 |
0 |
1 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / pipeline_parallel / mpipe_transpose.py |
MPipeTransposeExecutor._explicit_forward_before_backward |
|
0 |
0 |
10 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / pipeline_parallel / mpipe_transpose.py |
MPipeTransposeExecutor._connected_forward |
|
0 |
0 |
1 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / pipeline_parallel / mpipe_transpose.py |
MPipeTransposeExecutor._detach_for_wire |
|
0 |
0 |
1 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / pipeline_parallel / mpipe_transpose.py |
MPipeTransposeExecutor._zeros_like |
|
0 |
0 |
1 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / pipeline_parallel / mpipe_transpose.py |
MPipeTransposeExecutor._owner_transpose_backward |
|
0 |
0 |
2 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / pipeline_parallel / mpipe_transpose.py |
MPipeTransposeExecutor._snapshot_tower_grads |
|
0 |
0 |
2 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / pipeline_parallel / mpipe_transpose.py |
MPipeTransposeExecutor._reduce_grads |
|
0 |
0 |
1 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / pipeline_parallel / mpipe_transpose.py |
(no function) |
|
3 |
3 |
32 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / pipeline_parallel / stage.py |
PipelineStageBase.__init__ |
|
16 |
16 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / pipeline_parallel / stage.py |
PipelineStageBase.clear_cache |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / pipeline_parallel / stage.py |
PipelineStageBase._check_pp_group |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / pipeline_parallel / stage.py |
PipelineStageBase._clear_recv_buffer |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / pipeline_parallel / stage.py |
PipelineStageBase._grad_position_from_requires_grad |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / pipeline_parallel / stage.py |
PipelineStageBase.is_first_stage |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / pipeline_parallel / stage.py |
PipelineStageBase.is_last_stage |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / pipeline_parallel / stage.py |
PipelineStageBase.forward_one_chunk |
|
23 |
23 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / pipeline_parallel / stage.py |
PipelineStageBase.recompute_one_chunk |
|
10 |
10 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / pipeline_parallel / stage.py |
PipelineStageBase.backward_one_chunk |
|
24 |
24 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / pipeline_parallel / stage.py |
PipelineStageBase.backward_input_one_chunk |
|
15 |
15 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / pipeline_parallel / stage.py |
PipelineStageBase.backward_weight_one_chunk |
|
21 |
21 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / pipeline_parallel / stage.py |
PipelineStageBase._construct_backward_func |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / pipeline_parallel / stage.py |
(no function) |
|
23 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
_a2a_reconstruct_ms |
|
7 |
7 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
_normalize_dim |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
_move_dim_to_front |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
_move_dim_from_front |
|
8 |
8 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
_normalize_all_to_all_single_result |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
_normalize_all_gather_single_result |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
_normalize_reduce_scatter_single_result |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
_mindspore_all_to_all_single |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
_mindspore_all_gather_single |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
_mindspore_reduce_scatter_single |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
_mindspore_variable_all_gather |
|
15 |
2 |
0 |
|
87% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
_validate_variable_row_splits |
|
14 |
4 |
0 |
|
71% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
_mindspore_variable_all_to_all |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
_MSDifferentiableAllToAllSingle.forward |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
_MSDifferentiableAllToAllSingle.backward |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
AsyncCollectiveTensor.__new__ |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
AsyncCollectiveTensor.__init__ |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
AsyncCollectiveTensor._wait_and_unwrap |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
AsyncCollectiveTensor.__ms_dispatch__ |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
AsyncCollectiveTensor.asnumpy |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
AsyncCollectiveTensor.numpy |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
AsyncCollectiveTensor.__array__ |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
AsyncCollectiveTensor.get_bytes |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
AsyncCollectiveTensor.tolist |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
AsyncCollectiveTensor.item |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
AsyncCollectiveTensor.__bool__ |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
AsyncCollectiveTensor.__int__ |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
AsyncCollectiveTensor.__float__ |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
AsyncCollectiveTensor.__index__ |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
AsyncCollectiveTensor.__repr__ |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
AsyncCollectiveTensor.__str__ |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
AsyncCollectiveTensor.__iter__ |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
_MSAsyncA2ALazyBwd._issue_async_a2a |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
_MSAsyncA2ALazyBwd.forward |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
_MSAsyncA2ALazyBwd.backward |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
_MSSyncHookFunction._role_enum |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
_MSSyncHookFunction._passthrough |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
_MSSyncHookFunction.forward |
|
11 |
11 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
_MSSyncHookFunction.backward |
|
12 |
12 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
_MSAsyncA2AFunction.forward |
|
8 |
8 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
_MSAsyncA2AFunction.backward |
|
10 |
10 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
_MSAsyncAllGatherFunction.forward |
|
7 |
7 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
_MSAsyncAllGatherFunction.backward |
|
11 |
11 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
_ensure_contiguous |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
MindSporePlatform.get_swap_optimizer |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
MindSporePlatform.__init__ |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
MindSporePlatform.is_linear_module |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
MindSporePlatform.is_embedding_module |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
MindSporePlatform.device_count |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
MindSporePlatform.get_rng_state |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
MindSporePlatform.set_rng_state |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
MindSporePlatform.device_type |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
MindSporePlatform.device |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
MindSporePlatform.get_device_handle |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
MindSporePlatform.manual_seed |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
MindSporePlatform.ones |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
MindSporePlatform.zeros |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
MindSporePlatform.full |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
MindSporePlatform.empty |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
MindSporePlatform.rand |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
MindSporePlatform.randn |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
MindSporePlatform.get_rank |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
MindSporePlatform.get_global_rank |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
MindSporePlatform.get_world_size |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
MindSporePlatform.get_op_name |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
MindSporePlatform.differentiable_all_gather_concat |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
MindSporePlatform.chunk |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
MindSporePlatform.differentiable_all_to_all |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
MindSporePlatform.tensor_type_cast |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
MindSporePlatform.differentiable_all_reduce |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
MindSporePlatform.differentiable_reduce_scatter |
|
7 |
7 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
MindSporePlatform.init_parameters |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
MindSporePlatform.update_param_data |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
MindSporePlatform.load_into_param |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
MindSporePlatform.get_cell_construct |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
MindSporePlatform.get_cells_and_names |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
MindSporePlatform.get_modules |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
MindSporePlatform.search_parameter_by_name |
|
17 |
17 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
MindSporePlatform.update_parameter_by_name |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
MindSporePlatform.set_layout_into_parameter |
|
17 |
17 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
MindSporePlatform.get_param_local_shape |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
MindSporePlatform.get_param_local_data |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
MindSporePlatform.get_param_type_size |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
MindSporePlatform.is_tensor |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
MindSporePlatform.get_tensor_storage_size |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
MindSporePlatform.new_zero_parameter |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
MindSporePlatform.new_tensor |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
MindSporePlatform.full_like |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
MindSporePlatform.isend |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
MindSporePlatform.irecv |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
MindSporePlatform.p2p_op |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
MindSporePlatform.batch_isend_irecv |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
MindSporePlatform.prepare_batch_p2p_group |
|
0 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
MindSporePlatform.p2p_exchange |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
MindSporePlatform.send_object_list |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
MindSporePlatform.recv_object_list |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
MindSporePlatform.set_tensor_requires_grad |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
MindSporePlatform.detach |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
MindSporePlatform._normalize_group_options |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
MindSporePlatform._create_group_with_options |
|
7 |
7 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
MindSporePlatform.new_group |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
MindSporePlatform._create_group |
|
7 |
7 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
MindSporePlatform.create_p2p_multi_stream_groups |
|
16 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
MindSporePlatform.all_gather_into_tensor |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
MindSporePlatform.all_gather_single |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
MindSporePlatform.all_reduce |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
MindSporePlatform.broadcast |
|
7 |
7 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
MindSporePlatform.scatter |
|
12 |
1 |
0 |
|
92% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
MindSporePlatform.reduce_scatter_tensor |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
MindSporePlatform.reduce_scatter_single |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
MindSporePlatform.all_to_all_single |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
MindSporePlatform.differentiable_all_to_all_single |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
MindSporePlatform.differentiable_variable_all_gather |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
MindSporePlatform.differentiable_async_allgather_wait |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
MindSporePlatform.differentiable_async_a2a_wait |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
MindSporePlatform.differentiable_all_to_all_single_async |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
MindSporePlatform.differentiable_sync_hook |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
MindSporePlatform.parameters_dict |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
MindSporePlatform.buffers_dict |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
MindSporePlatform.get_tensor_transform |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
MindSporePlatform.construct_strided_slice |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
MindSporePlatform.micro_batch |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
MindSporePlatform.get_model_state_dict |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
MindSporePlatform.set_model_state_dict |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
MindSporePlatform.save_checkpoint |
|
10 |
4 |
0 |
|
60% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
MindSporePlatform.load_checkpoint |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
MindSporePlatform.new_stream |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
MindSporePlatform.get_stream_context |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
MindSporePlatform.all_gather_object |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
MindSporePlatform.barrier |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
MindSporePlatform.init_process_group |
|
8 |
8 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
MindSporePlatform.destroy_process_group |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
MindSporePlatform.get_process_group_ranks |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
MindSporePlatform.get_backend |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
MindSporePlatform.split_group |
|
16 |
16 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
MindSporePlatform.get_group_local_rank |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
MindSporePlatform.get_group_rank |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
MindSporePlatform.no_grad |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
MindSporePlatform.preserve_version_counter |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
MindSporePlatform.relu |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
MindSporePlatform.cat |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
MindSporePlatform.empty_like |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
MindSporePlatform.get_current_stream |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
MindSporePlatform.synchronize |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
MindSporePlatform.new_event |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
MindSporePlatform.tree_map |
|
7 |
3 |
0 |
|
57% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
MindSporePlatform.register_forward_pre_hook |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
MindSporePlatform.register_full_backward_hook |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
MindSporePlatform.register_full_backward_pre_hook |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
MindSporePlatform.checkpoint |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
MindSporePlatform.checkpoint_wrapper |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
MindSporePlatform.checkpoint_exclude_wrapper |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
MindSporePlatform.swap_wrapper |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
MindSporePlatform.swap_tensor_wrapper |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
MindSporePlatform.get_class_activation_wrapper |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
MindSporePlatform.noop_context_fn |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
MindSporePlatform.create_selective_checkpoint_contexts |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
MindSporePlatform.ignore_sac_ops |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
MindSporePlatform.async_save_on_cpu |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
MindSporePlatform.recompute_handle_collector_ctx |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
MindSporePlatform.recompute_handle |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
MindSporePlatform.recompute_session_ctx |
|
4 |
1 |
0 |
|
75% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
MindSporePlatform.clear_recompute_session |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
MindSporePlatform.alloc_tensor_buffer |
|
11 |
11 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
MindSporePlatform.get_element_size |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
MindSporePlatform.tensor_to_numpy |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
MindSporePlatform.from_numpy |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
MindSporePlatform.clip_grad_norm_ |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
MindSporePlatform.meta_device |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
MindSporePlatform.init_on_device |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
MindSporePlatform.cast_fp_tensor |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
MindSporePlatform.apply_to_tensors |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
MindSporePlatform.apply_to_tensors.apply |
|
19 |
19 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
MindSporePlatform.profiler_record |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
MindSporePlatform.str_to_dtype |
|
8 |
8 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
MindSporePlatform.list_to_size |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
MindSporePlatform._maybe_reuse_world_group |
|
6 |
2 |
0 |
|
67% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
(no function) |
|
363 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / swap_optimizer / __init__.py |
(no function) |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / swap_optimizer / adapters.py |
_to_tuple |
|
5 |
3 |
0 |
|
40% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / swap_optimizer / adapters.py |
MindSporeAdamBaseAdapter.__init__ |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / swap_optimizer / adapters.py |
MindSporeAdamBaseAdapter.validate |
|
2 |
1 |
0 |
|
50% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / swap_optimizer / adapters.py |
MindSporeAdamBaseAdapter.iter_update_units |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / swap_optimizer / adapters.py |
MindSporeAdamBaseAdapter.all_slots |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / swap_optimizer / adapters.py |
MindSporeAdamBaseAdapter.initial_slots |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / swap_optimizer / adapters.py |
MindSporeAdamBaseAdapter.packed_layout_units |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / swap_optimizer / adapters.py |
MindSporeAdamBaseAdapter.checkpoint_state_dict |
|
12 |
12 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / swap_optimizer / adapters.py |
MindSporeAdamBaseAdapter.load_checkpoint_state_dict |
|
19 |
4 |
0 |
|
79% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / swap_optimizer / adapters.py |
MindSporeAdamBaseAdapter._state_dict |
|
3 |
1 |
0 |
|
67% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / swap_optimizer / adapters.py |
MindSporeAdamBaseAdapter._load_state_dict |
|
3 |
1 |
0 |
|
67% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / swap_optimizer / adapters.py |
MindSporeAdamBaseAdapter._checkpoint_slots |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / swap_optimizer / adapters.py |
MindSporeAdamBaseAdapter._checkpoint_slot_map |
|
16 |
5 |
0 |
|
69% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / swap_optimizer / adapters.py |
MindSporeAdamBaseAdapter._is_checkpoint_swappable_slot |
|
21 |
4 |
0 |
|
81% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / swap_optimizer / adapters.py |
MindSporeAdamBaseAdapter._make_slot |
|
14 |
2 |
0 |
|
86% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / swap_optimizer / adapters.py |
MindSporeAdamBaseAdapter.publish_packed_state |
|
14 |
6 |
0 |
|
57% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / swap_optimizer / adapters.py |
MindSporeAdamBaseAdapter._state_parameter |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / swap_optimizer / adapters.py |
MindSporeAdamBaseAdapter._parameter_device |
|
7 |
7 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / swap_optimizer / adapters.py |
MindSporeAdamBaseAdapter._slot_tensor |
|
4 |
2 |
0 |
|
50% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / swap_optimizer / adapters.py |
MindSporeAdamBaseAdapter._selected_keys |
|
12 |
12 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / swap_optimizer / adapters.py |
MindSporeAdamBaseAdapter._validate_gradient_count |
|
5 |
1 |
0 |
|
80% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / swap_optimizer / adapters.py |
MindSporeNativeAdamAdapter.matches |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / swap_optimizer / adapters.py |
MindSporeNativeAdamAdapter.validate |
|
7 |
2 |
0 |
|
71% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / swap_optimizer / adapters.py |
MindSporeNativeAdamAdapter.prepare_step |
|
23 |
23 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / swap_optimizer / adapters.py |
MindSporeNativeAdamAdapter.step_batch |
|
9 |
9 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / swap_optimizer / adapters.py |
MindSporeNativeAdamAdapter._build_slots |
|
7 |
7 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / swap_optimizer / adapters.py |
MindSporeNativeAdamAdapter._state_parameter |
|
7 |
1 |
0 |
|
86% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / swap_optimizer / adapters.py |
MindSporeNativeAdamAdapter._checkpoint_slots |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / swap_optimizer / adapters.py |
MindSporeNativeAdamAdapter._index_lr |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / swap_optimizer / adapters.py |
MindSporeNativeAdamWAdapter.matches |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / swap_optimizer / adapters.py |
MindSporeNativeAdamWAdapter.validate |
|
5 |
1 |
0 |
|
80% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / swap_optimizer / adapters.py |
MindSporeNativeAdamWAdapter.prepare_step |
|
15 |
15 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / swap_optimizer / adapters.py |
MindSporeNativeAdamWAdapter.step_batch |
|
12 |
12 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / swap_optimizer / adapters.py |
MindSporeNativeAdamWAdapter._build_slots |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / swap_optimizer / adapters.py |
MindSporeNativeAdamWAdapter._state_parameter |
|
5 |
1 |
0 |
|
80% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / swap_optimizer / adapters.py |
MindSporeNativeAdamWAdapter._checkpoint_slots |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / swap_optimizer / adapters.py |
MindSporeNativeAdamWAdapter._indexed |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / swap_optimizer / adapters.py |
MindFormersAdamWAdapter.matches |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / swap_optimizer / adapters.py |
MindFormersAdamWAdapter.validate |
|
3 |
1 |
0 |
|
67% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / swap_optimizer / adapters.py |
MindFormersAdamWAdapter.packed_layout_units |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / swap_optimizer / adapters.py |
MindFormersAdamWAdapter.prepare_step |
|
17 |
1 |
0 |
|
94% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / swap_optimizer / adapters.py |
MindFormersAdamWAdapter.step_batch |
|
29 |
13 |
0 |
|
55% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / swap_optimizer / adapters.py |
MindFormersAdamWAdapter.finish_step |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / swap_optimizer / adapters.py |
MindFormersAdamWAdapter._build_slots |
|
21 |
21 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / swap_optimizer / adapters.py |
MindFormersAdamWAdapter._state_parameter |
|
10 |
1 |
0 |
|
90% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / swap_optimizer / adapters.py |
MindFormersAdamWAdapter._checkpoint_slots |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / swap_optimizer / adapters.py |
MindFormersAdamWAdapter._sync_batch_master_params |
|
9 |
9 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / swap_optimizer / adapters.py |
(no function) |
|
66 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / swap_optimizer / swap_optimizer.py |
MindSporeSwapRuntime.__init__ |
|
9 |
9 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / swap_optimizer / swap_optimizer.py |
MindSporeSwapRuntime.packed_enabled |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / swap_optimizer / swap_optimizer.py |
MindSporeSwapRuntime.populate_slot_metadata |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / swap_optimizer / swap_optimizer.py |
MindSporeSwapRuntime.prepare_packed_host |
|
53 |
4 |
0 |
|
92% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / swap_optimizer / swap_optimizer.py |
MindSporeSwapRuntime.is_swappable_tensor |
|
19 |
19 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / swap_optimizer / swap_optimizer.py |
MindSporeSwapRuntime.is_packable_tensor |
|
12 |
12 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / swap_optimizer / swap_optimizer.py |
MindSporeSwapRuntime.storage_nbytes |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / swap_optimizer / swap_optimizer.py |
MindSporeSwapRuntime.refresh_swappable_slots |
|
8 |
1 |
0 |
|
88% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / swap_optimizer / swap_optimizer.py |
MindSporeSwapRuntime.make_cpu_tensor |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / swap_optimizer / swap_optimizer.py |
MindSporeSwapRuntime.copy_cpu_tensor |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / swap_optimizer / swap_optimizer.py |
MindSporeSwapRuntime.copy_to_device |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / swap_optimizer / swap_optimizer.py |
MindSporeSwapRuntime.wait_prefetch_slot |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / swap_optimizer / swap_optimizer.py |
MindSporeSwapRuntime.copy_to_cpu |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / swap_optimizer / swap_optimizer.py |
MindSporeSwapRuntime.wait_offload_slot |
|
2 |
1 |
0 |
|
50% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / swap_optimizer / swap_optimizer.py |
MindSporeSwapRuntime.load_into_tensor |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / swap_optimizer / swap_optimizer.py |
MindSporeSwapRuntime.restore_device_storage |
|
9 |
1 |
0 |
|
89% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / swap_optimizer / swap_optimizer.py |
MindSporeSwapRuntime.release_device_storage |
|
9 |
1 |
0 |
|
89% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / swap_optimizer / swap_optimizer.py |
MindSporeSwapRuntime.supports_packed_pipeline |
|
14 |
6 |
0 |
|
57% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / swap_optimizer / swap_optimizer.py |
MindSporeSwapRuntime.begin_packed_step |
|
30 |
30 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / swap_optimizer / swap_optimizer.py |
MindSporeSwapRuntime.enqueue_packed_prefetch |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / swap_optimizer / swap_optimizer.py |
MindSporeSwapRuntime.wait_packed_prefetch |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / swap_optimizer / swap_optimizer.py |
MindSporeSwapRuntime.activate_packed_batch |
|
9 |
9 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / swap_optimizer / swap_optimizer.py |
MindSporeSwapRuntime.enqueue_packed_offload_prefetch |
|
12 |
12 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / swap_optimizer / swap_optimizer.py |
MindSporeSwapRuntime.wait_packed_offload |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / swap_optimizer / swap_optimizer.py |
MindSporeSwapRuntime.finish_packed_offload |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / swap_optimizer / swap_optimizer.py |
MindSporeSwapRuntime.release_packed_step_results |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / swap_optimizer / swap_optimizer.py |
MindSporeSwapRuntime.end_packed_step |
|
34 |
12 |
0 |
|
65% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / swap_optimizer / swap_optimizer.py |
MindSporeSwapRuntime.current_stream |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / swap_optimizer / swap_optimizer.py |
MindSporeSwapRuntime.new_stream |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / swap_optimizer / swap_optimizer.py |
MindSporeSwapRuntime.stream_context |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / swap_optimizer / swap_optimizer.py |
MindSporeSwapRuntime.record_event |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / swap_optimizer / swap_optimizer.py |
MindSporeSwapRuntime.wait_event |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / swap_optimizer / swap_optimizer.py |
MindSporeSwapRuntime._copy_storage |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / swap_optimizer / swap_optimizer.py |
MindSporeSwapRuntime._copy_tensor |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / swap_optimizer / swap_optimizer.py |
MindSporeSwapRuntime._packed_layout_signature |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / swap_optimizer / swap_optimizer.py |
MindSporeSwapRuntime._materialize_staging_arena |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / swap_optimizer / swap_optimizer.py |
MindSporeSwapRuntime._copy_packed_to_device |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / swap_optimizer / swap_optimizer.py |
MindSporeSwapRuntime._copy_packed_to_host |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / swap_optimizer / swap_optimizer.py |
MindSporeSwapRuntime._require_staging_arena |
|
4 |
1 |
0 |
|
75% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / swap_optimizer / swap_optimizer.py |
MindSporeSwapRuntime._make_cpu_storage_view |
|
10 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / swap_optimizer / swap_optimizer.py |
MindSporeSwapRuntime._copy_cpu_tensor |
|
12 |
12 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / swap_optimizer / swap_optimizer.py |
MindSporeSwapRuntime._storage_tensor |
|
11 |
3 |
0 |
|
73% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / swap_optimizer / swap_optimizer.py |
MindSporeSwapRuntime._debug_storage_size |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / swap_optimizer / swap_optimizer.py |
MindSporeSwapRuntime._align_bytes |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / swap_optimizer / swap_optimizer.py |
MindSporeSwapOptimizer.__init__ |
|
12 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / swap_optimizer / swap_optimizer.py |
MindSporeSwapOptimizer.__getattr__ |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / swap_optimizer / swap_optimizer.py |
MindSporeSwapOptimizer.__call__ |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / swap_optimizer / swap_optimizer.py |
MindSporeSwapOptimizer.construct |
|
9 |
9 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / swap_optimizer / swap_optimizer.py |
MindSporeSwapOptimizer.state_dict |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / swap_optimizer / swap_optimizer.py |
MindSporeSwapOptimizer.load_state_dict |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / swap_optimizer / swap_optimizer.py |
MindSporeSwapOptimizer._build_adapter |
|
4 |
1 |
0 |
|
75% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / swap_optimizer / swap_optimizer.py |
get_swap_optimizer |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / swap_optimizer / swap_optimizer.py |
_is_float_dtype |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / swap_optimizer / swap_optimizer.py |
_is_cpu_tensor |
|
4 |
1 |
0 |
|
75% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / swap_optimizer / swap_optimizer.py |
(no function) |
|
86 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / utils.py |
normalize_runtime_device |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / utils.py |
(no function) |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / platform.py |
AsyncHandle.__init__ |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / platform.py |
AsyncHandle.wait |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / platform.py |
get_mindspore_platform |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / platform.py |
get_torch_platform |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / platform.py |
get_platform |
|
13 |
4 |
0 |
|
69% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / platform.py |
_build_p2p_edge_rank_lists |
|
12 |
1 |
0 |
|
92% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / platform.py |
Platform.get_swap_optimizer |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / platform.py |
Platform.get_rank |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / platform.py |
Platform.get_global_rank |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / platform.py |
Platform.get_group_rank |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / platform.py |
Platform.get_world_size |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / platform.py |
Platform.get_op_name |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / platform.py |
Platform.differentiable_all_gather_concat |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / platform.py |
Platform.chunk |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / platform.py |
Platform.differentiable_all_to_all |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / platform.py |
Platform.tensor_type_cast |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / platform.py |
Platform.is_tensor |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / platform.py |
Platform.get_tensor_storage_size |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / platform.py |
Platform.copy_each |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / platform.py |
Platform.differentiable_all_reduce |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / platform.py |
Platform.differentiable_reduce_scatter |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / platform.py |
Platform.init_parameters |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / platform.py |
Platform.get_cell_construct |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / platform.py |
Platform.get_cells_and_names |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / platform.py |
Platform.get_modules |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / platform.py |
Platform.search_parameter_by_name |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / platform.py |
Platform.update_parameter_by_name |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / platform.py |
Platform.set_layout_into_parameter |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / platform.py |
Platform.get_param_local_shape |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / platform.py |
Platform.get_param_local_data |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / platform.py |
Platform.update_param_data |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / platform.py |
Platform.get_param_type_size |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / platform.py |
Platform.new_zero_parameter |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / platform.py |
Platform.new_tensor |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / platform.py |
Platform.full_like |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / platform.py |
Platform.set_tensor_requires_grad |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / platform.py |
Platform.detach |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / platform.py |
Platform.all_gather_into_tensor |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / platform.py |
Platform.all_reduce |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / platform.py |
Platform.broadcast |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / platform.py |
Platform.broadcast_async |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / platform.py |
Platform.scatter |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / platform.py |
Platform.isend |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / platform.py |
Platform.irecv |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / platform.py |
Platform.p2p_op |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / platform.py |
Platform.batch_isend_irecv |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / platform.py |
Platform.prepare_batch_p2p_group |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / platform.py |
Platform.p2p_exchange |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / platform.py |
Platform.send_object_list |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / platform.py |
Platform.recv_object_list |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / platform.py |
Platform.reduce_scatter_tensor |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / platform.py |
Platform.all_gather_single |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / platform.py |
Platform.reduce_scatter_single |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / platform.py |
Platform.all_to_all_single |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / platform.py |
Platform.differentiable_variable_all_gather |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / platform.py |
Platform.differentiable_async_allgather_wait |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / platform.py |
Platform.differentiable_async_a2a_wait |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / platform.py |
Platform.differentiable_sync_hook |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / platform.py |
Platform.differentiable_all_to_all_single |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / platform.py |
Platform.differentiable_all_to_all_single_async |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / platform.py |
Platform.wait_async_tensor |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / platform.py |
Platform.arange |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / platform.py |
Platform.zeros |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / platform.py |
Platform.parameters_dict |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / platform.py |
Platform.buffers_dict |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / platform.py |
Platform.get_model_state_dict |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / platform.py |
Platform.set_model_state_dict |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / platform.py |
Platform.get_optim_state_dict |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / platform.py |
Platform.set_optim_state_dict |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / platform.py |
Platform.save_checkpoint |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / platform.py |
Platform.load_checkpoint |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / platform.py |
Platform._create_group |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / platform.py |
Platform.new_stream |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / platform.py |
Platform.get_stream_context |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / platform.py |
Platform.get_tensor_transform |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / platform.py |
Platform.construct_strided_slice |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / platform.py |
Platform.micro_batch |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / platform.py |
Platform.load_into_param |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / platform.py |
Platform.new_group |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / platform.py |
Platform.create_group |
|
6 |
1 |
0 |
|
83% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / platform.py |
Platform.create_p2p_multi_stream_groups |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / platform.py |
Platform._process_current_handle |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / platform.py |
Platform.set_grad_reduce_handle |
|
7 |
7 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / platform.py |
Platform.wait_grad_handle |
|
11 |
11 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / platform.py |
Platform.all_gather_object |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / platform.py |
Platform.barrier |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / platform.py |
Platform.init_process_group |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / platform.py |
Platform.destroy_process_group |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / platform.py |
Platform.get_process_group_ranks |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / platform.py |
Platform.get_backend |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / platform.py |
Platform.split_group |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / platform.py |
Platform.get_group_local_rank |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / platform.py |
Platform.no_grad |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / platform.py |
Platform.preserve_version_counter |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / platform.py |
Platform.relu |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / platform.py |
Platform.cat |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / platform.py |
Platform.empty_like |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / platform.py |
Platform.get_current_stream |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / platform.py |
Platform.synchronize |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / platform.py |
Platform.new_event |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / platform.py |
Platform.tree_map |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / platform.py |
Platform.is_linear_module |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / platform.py |
Platform.is_embedding_module |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / platform.py |
Platform.register_forward_pre_hook |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / platform.py |
Platform.register_full_backward_hook |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / platform.py |
Platform.register_full_backward_pre_hook |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / platform.py |
Platform.checkpoint |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / platform.py |
Platform.checkpoint_wrapper |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / platform.py |
Platform.checkpoint_exclude_wrapper |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / platform.py |
Platform.swap_wrapper |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / platform.py |
Platform.swap_tensor_wrapper |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / platform.py |
Platform.get_class_activation_wrapper |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / platform.py |
Platform.noop_context_fn |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / platform.py |
Platform.ignore_sac_ops |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / platform.py |
Platform.create_selective_checkpoint_contexts |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / platform.py |
Platform.create_native_selective_checkpoint_contexts |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / platform.py |
Platform.async_save_on_cpu |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / platform.py |
Platform.recompute_handle_collector_ctx |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / platform.py |
Platform.recompute_handle |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / platform.py |
Platform.recompute_session_ctx |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / platform.py |
Platform.clear_recompute_session |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / platform.py |
Platform.get_element_size |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / platform.py |
Platform.alloc_tensor_buffer |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / platform.py |
Platform.tensor_to_numpy |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / platform.py |
Platform.from_numpy |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / platform.py |
Platform.profiler_record |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / platform.py |
Platform.cast_fp_tensor |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / platform.py |
Platform.apply_to_tensors |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / platform.py |
Platform.clip_grad_norm_ |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / platform.py |
Platform.get_world_group |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / platform.py |
Platform.get_created_group |
|
4 |
1 |
0 |
|
75% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / platform.py |
Platform.mark_created_groups |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / platform.py |
Platform.meta_device |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / platform.py |
Platform.init_on_device |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / platform.py |
Platform.str_to_dtype |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / platform.py |
Platform.list_to_size |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / platform.py |
(no function) |
|
269 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / __init__.py |
(no function) |
|
0 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / activation_checkpoint / __init__.py |
(no function) |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / activation_checkpoint / activation_swap.py |
_raise_if_compiling |
|
2 |
1 |
0 |
|
50% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / activation_checkpoint / activation_swap.py |
FuncModule.__init__ |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / activation_checkpoint / activation_swap.py |
FuncModule.forward |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / activation_checkpoint / activation_swap.py |
_is_callable_exempt_from_overlap_check |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / activation_checkpoint / activation_swap.py |
_iter_wrappable_callable_attrs |
|
7 |
2 |
0 |
|
71% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / activation_checkpoint / activation_swap.py |
_mark_wrapped |
|
4 |
2 |
0 |
|
50% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / activation_checkpoint / activation_swap.py |
_get_wrapped_callable |
|
6 |
2 |
0 |
|
67% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / activation_checkpoint / activation_swap.py |
_raise_callable_already_wrapped |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / activation_checkpoint / activation_swap.py |
_check_callable_attr_not_wrapped |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / activation_checkpoint / activation_swap.py |
_check_and_mark_callable |
|
5 |
3 |
0 |
|
40% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / activation_checkpoint / activation_swap.py |
_check_and_mark_wrapped |
|
21 |
5 |
0 |
|
76% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / activation_checkpoint / activation_swap.py |
base_check_fn |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / activation_checkpoint / activation_swap.py |
AsyncSaveOnCpu.__init__ |
|
10 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / activation_checkpoint / activation_swap.py |
AsyncSaveOnCpu.__init__.pack_to_cpu |
|
18 |
3 |
0 |
|
83% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / activation_checkpoint / activation_swap.py |
AsyncSaveOnCpu.__init__.unpack_from_cpu |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / activation_checkpoint / activation_swap.py |
ActivationWrapper.__init__ |
|
13 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / activation_checkpoint / activation_swap.py |
ActivationWrapper._wrapped_module |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / activation_checkpoint / activation_swap.py |
ActivationWrapper.forward |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / activation_checkpoint / activation_swap.py |
ActivationWrapper.__getattr__ |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / activation_checkpoint / activation_swap.py |
ActivationWrapper.__getitem__ |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / activation_checkpoint / activation_swap.py |
ActivationWrapper.named_modules |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / activation_checkpoint / activation_swap.py |
ActivationWrapper.named_parameters |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / activation_checkpoint / activation_swap.py |
ActivationWrapper._post_state_dict_hook |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / activation_checkpoint / activation_swap.py |
ActivationWrapper._pre_load_state_dict_hook |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / activation_checkpoint / activation_swap.py |
SwapWrapper.__init__ |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / activation_checkpoint / activation_swap.py |
SwapWrapper.forward |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / activation_checkpoint / activation_swap.py |
swap_wrapper |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / activation_checkpoint / activation_swap.py |
swap_tensor_wrapper |
|
15 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / activation_checkpoint / activation_swap.py |
swap_tensor_wrapper._register_tensor |
|
7 |
1 |
0 |
|
86% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / activation_checkpoint / activation_swap.py |
(no function) |
|
46 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / activation_checkpoint / checkpoint.py |
_Holder.__init__ |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / activation_checkpoint / checkpoint.py |
_SessionActivation.__init__ |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / activation_checkpoint / checkpoint.py |
_NoopSaveInputs.forward |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / activation_checkpoint / checkpoint.py |
_NoopSaveInputs.setup_context |
|
8 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / activation_checkpoint / checkpoint.py |
_NoopSaveInputs.setup_context.get_args |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / activation_checkpoint / checkpoint.py |
_NoopSaveInputs.backward |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / activation_checkpoint / checkpoint.py |
_CheckpointFrame.__init__ |
|
12 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / activation_checkpoint / checkpoint.py |
_CheckpointFrame.check_recomputed_tensors_match |
|
19 |
6 |
0 |
|
68% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / activation_checkpoint / checkpoint.py |
_CheckpointFrame.clear_session |
|
7 |
1 |
0 |
|
86% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / activation_checkpoint / checkpoint.py |
_bind_session_activation |
|
6 |
2 |
0 |
|
67% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / activation_checkpoint / checkpoint.py |
_register_session_frame |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / activation_checkpoint / checkpoint.py |
_activate_registered_frames |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / activation_checkpoint / checkpoint.py |
_deactivate_session |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / activation_checkpoint / checkpoint.py |
_internal_assert |
|
2 |
1 |
0 |
|
50% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / activation_checkpoint / checkpoint.py |
_noop_context_fn |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / activation_checkpoint / checkpoint.py |
_default_metadata_fn |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / activation_checkpoint / checkpoint.py |
_infer_device_type |
|
11 |
4 |
0 |
|
64% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / activation_checkpoint / checkpoint.py |
_infer_device_type.add_device_type |
|
2 |
1 |
0 |
|
50% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / activation_checkpoint / checkpoint.py |
_get_device_module |
|
3 |
1 |
0 |
|
67% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / activation_checkpoint / checkpoint.py |
_get_device_states |
|
9 |
9 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / activation_checkpoint / checkpoint.py |
_get_device_states.add_device_id |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / activation_checkpoint / checkpoint.py |
_set_device_states |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / activation_checkpoint / checkpoint.py |
_get_autocast_kwargs |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / activation_checkpoint / checkpoint.py |
_create_recomputation_hooks |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / activation_checkpoint / checkpoint.py |
_create_recomputation_hooks.pack_hook |
|
19 |
4 |
0 |
|
79% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / activation_checkpoint / checkpoint.py |
_create_recomputation_hooks.unpack_hook |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / activation_checkpoint / checkpoint.py |
_run_fn_with_dynamo_disabled |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / activation_checkpoint / checkpoint.py |
_run_recomputation |
|
21 |
2 |
0 |
|
90% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / activation_checkpoint / checkpoint.py |
_create_checkpoint_hooks |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / activation_checkpoint / checkpoint.py |
_create_checkpoint_hooks.pack_hook |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / activation_checkpoint / checkpoint.py |
_create_checkpoint_hooks.unpack_hook |
|
18 |
2 |
0 |
|
89% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / activation_checkpoint / checkpoint.py |
_is_compiling |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / activation_checkpoint / checkpoint.py |
_native_checkpoint |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / activation_checkpoint / checkpoint.py |
_checkpoint_without_reentrant_generator |
|
43 |
12 |
0 |
|
72% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / activation_checkpoint / checkpoint.py |
_checkpoint_without_reentrant_generator.recompute_fn |
|
12 |
1 |
0 |
|
92% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / activation_checkpoint / checkpoint.py |
checkpoint |
|
25 |
11 |
0 |
|
56% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / activation_checkpoint / checkpoint.py |
recompute_handle_collector_ctx |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / activation_checkpoint / checkpoint.py |
recompute_handle |
|
13 |
6 |
0 |
|
54% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / activation_checkpoint / checkpoint.py |
_validate_session_id |
|
6 |
3 |
0 |
|
50% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / activation_checkpoint / checkpoint.py |
recompute_session_ctx |
|
13 |
2 |
0 |
|
85% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / activation_checkpoint / checkpoint.py |
clear_recompute_session |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / activation_checkpoint / checkpoint.py |
(no function) |
|
64 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / activation_checkpoint / checkpoint_exclude_wrapper.py |
_RecomputedInputHandle.__init__ |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / activation_checkpoint / checkpoint_exclude_wrapper.py |
_RecomputedInputHandle.mark_used |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / activation_checkpoint / checkpoint_exclude_wrapper.py |
_RecomputedInputHandle.materialize |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / activation_checkpoint / checkpoint_exclude_wrapper.py |
_RecomputedInputHandle.get_recomputed_tensor |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / activation_checkpoint / checkpoint_exclude_wrapper.py |
_ExcludeCache.__init__ |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / activation_checkpoint / checkpoint_exclude_wrapper.py |
_ExcludeCache.save |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / activation_checkpoint / checkpoint_exclude_wrapper.py |
_ExcludeCache.pop |
|
7 |
1 |
0 |
|
86% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / activation_checkpoint / checkpoint_exclude_wrapper.py |
_ExcludeCache.clear |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / activation_checkpoint / checkpoint_exclude_wrapper.py |
_pack_saved_tensor |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / activation_checkpoint / checkpoint_exclude_wrapper.py |
_unpack_saved_tensor |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / activation_checkpoint / checkpoint_exclude_wrapper.py |
_saved_tensors_context |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / activation_checkpoint / checkpoint_exclude_wrapper.py |
_append_tensor_inputs |
|
10 |
3 |
0 |
|
70% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / activation_checkpoint / checkpoint_exclude_wrapper.py |
_collect_tensor_inputs |
|
6 |
1 |
0 |
|
83% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / activation_checkpoint / checkpoint_exclude_wrapper.py |
_mark_recompute_inputs |
|
17 |
3 |
0 |
|
82% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / activation_checkpoint / checkpoint_exclude_wrapper.py |
_restore_recompute_inputs |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / activation_checkpoint / checkpoint_exclude_wrapper.py |
_resolve_input |
|
5 |
1 |
0 |
|
80% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / activation_checkpoint / checkpoint_exclude_wrapper.py |
_materialize_recompute_inputs |
|
7 |
2 |
0 |
|
71% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / activation_checkpoint / checkpoint_exclude_wrapper.py |
_has_used_input |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / activation_checkpoint / checkpoint_exclude_wrapper.py |
_get_replay_placeholder |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / activation_checkpoint / checkpoint_exclude_wrapper.py |
_make_replay_placeholder_output |
|
6 |
2 |
0 |
|
67% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / activation_checkpoint / checkpoint_exclude_wrapper.py |
_get_recompute_trigger |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / activation_checkpoint / checkpoint_exclude_wrapper.py |
_get_recompute_boundary |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / activation_checkpoint / checkpoint_exclude_wrapper.py |
_get_recompute_boundary._RecomputeBoundary.forward |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / activation_checkpoint / checkpoint_exclude_wrapper.py |
_get_recompute_boundary._RecomputeBoundary.backward |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / activation_checkpoint / checkpoint_exclude_wrapper.py |
_finalize_save_outputs_impl |
|
18 |
5 |
0 |
|
72% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / activation_checkpoint / checkpoint_exclude_wrapper.py |
_finalize_save_outputs |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / activation_checkpoint / checkpoint_exclude_wrapper.py |
CheckpointExcludeWrapper.__init__ |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / activation_checkpoint / checkpoint_exclude_wrapper.py |
CheckpointExcludeWrapper.forward |
|
22 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / activation_checkpoint / checkpoint_exclude_wrapper.py |
checkpoint_exclude_wrapper |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / activation_checkpoint / checkpoint_exclude_wrapper.py |
(no function) |
|
55 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / activation_checkpoint / checkpoint_wrapper.py |
CheckpointWrapper.__init__ |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / activation_checkpoint / checkpoint_wrapper.py |
CheckpointWrapper._do_checkpoint |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / activation_checkpoint / checkpoint_wrapper.py |
CheckpointWrapper.forward |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / activation_checkpoint / checkpoint_wrapper.py |
ckpt_wrapper |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / activation_checkpoint / checkpoint_wrapper.py |
(no function) |
|
8 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / activation_checkpoint / native_compile.py |
_to_torch_checkpoint_policy |
|
12 |
12 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / activation_checkpoint / native_compile.py |
_torch_policy_adapter |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / activation_checkpoint / native_compile.py |
create_native_selective_checkpoint_contexts |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / activation_checkpoint / native_compile.py |
(no function) |
|
8 |
8 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / activation_checkpoint / sac.py |
_is_compiling |
|
4 |
1 |
0 |
|
75% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / activation_checkpoint / sac.py |
_VersionWrapper.__init__ |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / activation_checkpoint / sac.py |
_VersionWrapper.get_val |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / activation_checkpoint / sac.py |
_SwapCacheEntry.__init__ |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / activation_checkpoint / sac.py |
_maybe_detach |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / activation_checkpoint / sac.py |
SelectiveCheckpointContext.__init__ |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / activation_checkpoint / sac.py |
_policy_from_bool |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / activation_checkpoint / sac.py |
ignore_sac_ops |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / activation_checkpoint / sac.py |
_CachingTorchDispatchMode.__init__ |
|
8 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / activation_checkpoint / sac.py |
_CachingTorchDispatchMode.__torch_dispatch__ |
|
27 |
1 |
0 |
|
96% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / activation_checkpoint / sac.py |
_CachedTorchDispatchMode.__init__ |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / activation_checkpoint / sac.py |
_CachedTorchDispatchMode.__torch_dispatch__ |
|
19 |
2 |
0 |
|
89% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / activation_checkpoint / sac.py |
create_selective_checkpoint_contexts |
|
17 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / activation_checkpoint / sac.py |
create_selective_checkpoint_contexts.policy_fn |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / activation_checkpoint / sac.py |
(no function) |
|
28 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / clip_grad.py |
(no function) |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / common / __init__.py |
(no function) |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / common / moe.py |
_is_npu_tensor |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / common / moe.py |
_run_experts_for_loop |
|
15 |
1 |
0 |
|
93% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / common / moe.py |
_run_experts_grouped_mm_gpu |
|
10 |
10 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / common / moe.py |
_run_experts_grouped_mm_npu |
|
8 |
8 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / common / moe.py |
FeedForward.__init__ |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / common / moe.py |
FeedForward.forward |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / common / moe.py |
GroupedExperts.__init__ |
|
7 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / common / moe.py |
GroupedExperts._reset_parameters |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / common / moe.py |
GroupedExperts.forward |
|
10 |
5 |
0 |
|
50% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / common / moe.py |
TokenChoiceTopKRouter.__init__ |
|
12 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / common / moe.py |
TokenChoiceTopKRouter.forward |
|
16 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / common / moe.py |
TokenChoiceTopKRouter._get_node_limited_routing_scores |
|
8 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / common / moe.py |
_compute_load_balance_loss |
|
18 |
2 |
0 |
|
89% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / common / moe.py |
MoEAuxLossAutoScaler.forward |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / common / moe.py |
MoEAuxLossAutoScaler.backward |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / common / moe.py |
MoEAuxLossAutoScaler.set_loss_scale |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / common / moe.py |
MoE.__init__ |
|
14 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / common / moe.py |
MoE.permutation |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / common / moe.py |
MoE.unpermutation |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / common / moe.py |
MoE.forward |
|
28 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / common / moe.py |
MoE.update_expert_bias |
|
8 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / common / moe.py |
update_expert_bias |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / common / moe.py |
(no function) |
|
40 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / custom_ops / __init__.py |
(no function) |
|
0 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / custom_ops / gdn / __init__.py |
(no function) |
|
0 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / custom_ops / gdn / chunk_gated_delta_rule.py |
_l2norm |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / custom_ops / gdn / chunk_gated_delta_rule.py |
chunk_gated_delta_rule_fwd_prepare |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / custom_ops / gdn / chunk_gated_delta_rule.py |
chunk_gated_delta_rule_fwd_apply_state |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / custom_ops / gdn / chunk_gated_delta_rule.py |
chunk_gated_delta_rule_fwd_output |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / custom_ops / gdn / chunk_gated_delta_rule.py |
chunk_gated_delta_rule_fwd |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / custom_ops / gdn / chunk_gated_delta_rule.py |
chunk_gated_delta_rule_bwd_prepare |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / custom_ops / gdn / chunk_gated_delta_rule.py |
chunk_gated_delta_rule_bwd_state |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / custom_ops / gdn / chunk_gated_delta_rule.py |
chunk_gated_delta_rule_bwd_finish |
|
8 |
8 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / custom_ops / gdn / chunk_gated_delta_rule.py |
chunk_gated_delta_rule_bwd |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / custom_ops / gdn / chunk_gated_delta_rule.py |
chunk_gated_delta_rule_fwd_prepare_saved |
|
9 |
9 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / custom_ops / gdn / chunk_gated_delta_rule.py |
chunk_gated_delta_rule_fwd_apply_state_saved |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / custom_ops / gdn / chunk_gated_delta_rule.py |
chunk_gated_delta_rule_fwd_output_saved |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / custom_ops / gdn / chunk_gated_delta_rule.py |
chunk_gated_delta_rule_fwd_saved |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / custom_ops / gdn / chunk_gated_delta_rule.py |
chunk_gated_delta_rule_bwd_prepare_saved |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / custom_ops / gdn / chunk_gated_delta_rule.py |
chunk_gated_delta_rule_bwd_state_saved |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / custom_ops / gdn / chunk_gated_delta_rule.py |
chunk_gated_delta_rule_bwd_finish_saved |
|
10 |
10 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / custom_ops / gdn / chunk_gated_delta_rule.py |
chunk_gated_delta_rule_bwd_saved |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / custom_ops / gdn / chunk_gated_delta_rule.py |
ChunkGatedDeltaRuleFunction.forward |
|
9 |
9 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / custom_ops / gdn / chunk_gated_delta_rule.py |
ChunkGatedDeltaRuleFunction.backward |
|
7 |
7 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / custom_ops / gdn / chunk_gated_delta_rule.py |
chunk_gated_delta_rule |
|
22 |
22 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / custom_ops / gdn / chunk_gated_delta_rule.py |
(no function) |
|
54 |
54 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / custom_ops / gdn / state_summary.py |
_validate_fixed_summary_shape |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / custom_ops / gdn / state_summary.py |
chunk_gated_delta_rule_state_summary_fwd |
|
17 |
17 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / custom_ops / gdn / state_summary.py |
chunk_gated_delta_rule_state_gradient_summary_bwd |
|
13 |
13 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / custom_ops / gdn / state_summary.py |
apply_gdn_state_summary |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / custom_ops / gdn / state_summary.py |
apply_gdn_state_gradient_summary |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / custom_ops / gdn / state_summary.py |
(no function) |
|
12 |
12 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / custom_ops / gdn / triton / __init__.py |
(no function) |
|
0 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / custom_ops / gdn / triton / chunk_delta_h.py |
chunk_gated_delta_rule_fwd_kernel_h_blockdim64 |
|
154 |
154 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / custom_ops / gdn / triton / chunk_delta_h.py |
chunk_gated_delta_rule_fwd_h |
|
19 |
19 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / custom_ops / gdn / triton / chunk_delta_h.py |
chunk_gated_delta_rule_bwd_kernel_dhu_blockdim64 |
|
159 |
159 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / custom_ops / gdn / triton / chunk_delta_h.py |
chunk_gated_delta_rule_bwd_dhu |
|
15 |
15 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / custom_ops / gdn / triton / chunk_delta_h.py |
(no function) |
|
16 |
16 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / custom_ops / gdn / triton / chunk_o.py |
chunk_bwd_kernel_dqkwg |
|
122 |
122 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / custom_ops / gdn / triton / chunk_o.py |
chunk_bwd_kernel_dv_local |
|
38 |
38 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / custom_ops / gdn / triton / chunk_o.py |
chunk_fwd_kernel_o |
|
57 |
57 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / custom_ops / gdn / triton / chunk_o.py |
chunk_bwd_dqkwg |
|
35 |
35 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / custom_ops / gdn / triton / chunk_o.py |
chunk_bwd_dv_local |
|
11 |
11 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / custom_ops / gdn / triton / chunk_o.py |
chunk_fwd_o |
|
16 |
16 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / custom_ops / gdn / triton / chunk_o.py |
(no function) |
|
19 |
19 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / custom_ops / gdn / triton / chunk_scaled_dot_kkt.py |
chunk_scaled_dot_kkt_fwd_kernel |
|
52 |
52 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / custom_ops / gdn / triton / chunk_scaled_dot_kkt.py |
chunk_scaled_dot_kkt_fwd_kernel_intra_sub_inter |
|
33 |
33 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / custom_ops / gdn / triton / chunk_scaled_dot_kkt.py |
chunk_scaled_dot_kkt_fwd_kernel_intra_sub_intra |
|
31 |
31 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / custom_ops / gdn / triton / chunk_scaled_dot_kkt.py |
chunk_scaled_dot_kkt_fwd |
|
21 |
21 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / custom_ops / gdn / triton / chunk_scaled_dot_kkt.py |
(no function) |
|
16 |
16 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / custom_ops / gdn / triton / cumsum.py |
chunk_local_cumsum_scalar_kernel |
|
22 |
22 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / custom_ops / gdn / triton / cumsum.py |
chunk_local_cumsum_scalar |
|
10 |
10 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / custom_ops / gdn / triton / cumsum.py |
chunk_local_cumsum |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / custom_ops / gdn / triton / cumsum.py |
(no function) |
|
10 |
10 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / custom_ops / gdn / triton / solve_tril.py |
_ensure_slice_ops |
|
9 |
9 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / custom_ops / gdn / triton / solve_tril.py |
solve_tril_16x16_loop_kernel_paral_v3 |
|
58 |
58 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / custom_ops / gdn / triton / solve_tril.py |
merge_16x16_to_32x32_loop_inverse_kernel |
|
32 |
32 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / custom_ops / gdn / triton / solve_tril.py |
merge_32x32_to_64x64_loop_inverse_kernel |
|
32 |
32 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / custom_ops / gdn / triton / solve_tril.py |
solve_tril_64x64_kernel |
|
27 |
27 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / custom_ops / gdn / triton / solve_tril.py |
solve_tril_64 |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / custom_ops / gdn / triton / solve_tril.py |
solve_tril |
|
29 |
29 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / custom_ops / gdn / triton / solve_tril.py |
(no function) |
|
24 |
24 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / custom_ops / gdn / triton / state_summary.py |
gdn_packed_state_summary_kernel |
|
45 |
45 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / custom_ops / gdn / triton / state_summary.py |
gdn_state_grad_ext_kernel |
|
55 |
55 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / custom_ops / gdn / triton / state_summary.py |
(no function) |
|
10 |
10 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / custom_ops / gdn / triton / utils.py |
tensor_cache |
|
7 |
7 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / custom_ops / gdn / triton / utils.py |
tensor_cache._is_match |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / custom_ops / gdn / triton / utils.py |
tensor_cache._make_wrapper |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / custom_ops / gdn / triton / utils.py |
tensor_cache._make_wrapper.wrapper |
|
11 |
11 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / custom_ops / gdn / triton / utils.py |
prepare_lens |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / custom_ops / gdn / triton / utils.py |
prepare_chunk_indices |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / custom_ops / gdn / triton / utils.py |
get_abs_err |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / custom_ops / gdn / triton / utils.py |
get_err_ratio |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / custom_ops / gdn / triton / utils.py |
assert_close |
|
10 |
10 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / custom_ops / gdn / triton / utils.py |
get_available_device |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / custom_ops / gdn / triton / utils.py |
map_triton_backend_to_torch_device |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / custom_ops / gdn / triton / utils.py |
check_pytorch_version |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / custom_ops / gdn / triton / utils.py |
custom_device_ctx |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / custom_ops / gdn / triton / utils.py |
input_guard |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / custom_ops / gdn / triton / utils.py |
input_guard.wrapper |
|
17 |
17 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / custom_ops / gdn / triton / utils.py |
_cpu_device_warning |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / custom_ops / gdn / triton / utils.py |
prepare_chunk_offsets |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / custom_ops / gdn / triton / utils.py |
get_all_max_shared_mem |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / custom_ops / gdn / triton / utils.py |
Backend.get_shared_memory |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / custom_ops / gdn / triton / utils.py |
check_shared_mem |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / custom_ops / gdn / triton / utils.py |
get_autotune_config |
|
10 |
10 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / custom_ops / gdn / triton / utils.py |
get_npu_properties |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / custom_ops / gdn / triton / utils.py |
get_vector_num |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / custom_ops / gdn / triton / utils.py |
is_arch35 |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / custom_ops / gdn / triton / utils.py |
(no function) |
|
87 |
87 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / custom_ops / gdn / triton / wy_fast.py |
prepare_wy_repr_bwd_kernel |
|
84 |
84 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / custom_ops / gdn / triton / wy_fast.py |
recompute_w_u_fwd_kernel |
|
48 |
48 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / custom_ops / gdn / triton / wy_fast.py |
recompute_w_u_fwd |
|
13 |
13 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / custom_ops / gdn / triton / wy_fast.py |
prepare_wy_repr_bwd |
|
17 |
17 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / custom_ops / gdn / triton / wy_fast.py |
(no function) |
|
15 |
15 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / custom_ops / kda / __init__.py |
(no function) |
|
0 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / custom_ops / kda / chunk_kda.py |
_validate_local_inputs |
|
35 |
35 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / custom_ops / kda / chunk_kda.py |
_validate_p2p_inputs |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / custom_ops / kda / chunk_kda.py |
fused_chunk_kda |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / custom_ops / kda / chunk_kda.py |
_KDAStateP2PFunction.forward |
|
44 |
44 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / custom_ops / kda / chunk_kda.py |
_KDAStateP2PFunction.backward |
|
43 |
43 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / custom_ops / kda / chunk_kda.py |
fused_chunk_kda_p2p |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / custom_ops / kda / chunk_kda.py |
(no function) |
|
18 |
18 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / custom_ops / kda / fla_adapter.py |
_get_attribute |
|
14 |
14 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / custom_ops / kda / fla_adapter.py |
_parse_version |
|
4 |
1 |
0 |
|
75% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / custom_ops / kda / fla_adapter.py |
_require_triton_ascend |
|
7 |
7 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / custom_ops / kda / fla_adapter.py |
_require_npu_backends |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / custom_ops / kda / fla_adapter.py |
_require_fla_kda_runtime |
|
19 |
8 |
0 |
|
58% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / custom_ops / kda / fla_adapter.py |
is_fla_triton_kda_available |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / custom_ops / kda / fla_adapter.py |
get_fla_kda_staged_ops |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / custom_ops / kda / fla_adapter.py |
run_fla_chunk_kda |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / custom_ops / kda / fla_adapter.py |
(no function) |
|
41 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / custom_ops / kda / state_summary.py |
_validate_prepared_summary_inputs |
|
15 |
15 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / custom_ops / kda / state_summary.py |
_validate_gradient_summary_inputs |
|
20 |
20 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / custom_ops / kda / state_summary.py |
_launch_kda_state_summary |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / custom_ops / kda / state_summary.py |
_launch_kda_state_gradient_summary |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / custom_ops / kda / state_summary.py |
_launch_kda_mixed_state_summary |
|
8 |
8 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / custom_ops / kda / state_summary.py |
kda_state_summary_forward_from_prepared |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / custom_ops / kda / state_summary.py |
kda_state_gradient_summary_from_prepared |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / custom_ops / kda / state_summary.py |
apply_kda_state_summary |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / custom_ops / kda / state_summary.py |
apply_kda_state_gradient_summary |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / custom_ops / kda / state_summary.py |
(no function) |
|
16 |
16 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / custom_ops / kda / triton / __init__.py |
(no function) |
|
0 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / custom_ops / kda / triton / state_summary.py |
_exp2 |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / custom_ops / kda / triton / state_summary.py |
kda_split_state_summary_kernel |
|
61 |
61 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / custom_ops / kda / triton / state_summary.py |
kda_state_grad_ext_kernel |
|
52 |
52 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / custom_ops / kda / triton / state_summary.py |
(no function) |
|
9 |
9 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / dtensor.py |
DTensorBase.__new__ |
|
12 |
6 |
0 |
|
50% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / dtensor.py |
DTensorBase.__torch_function__ |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / dtensor.py |
DTensorBase.grad |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / dtensor.py |
DTensorBase.grad |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / dtensor.py |
DTensorBase.requires_grad |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / dtensor.py |
DTensorBase.requires_grad |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / dtensor.py |
DTensorBase.requires_grad_ |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / dtensor.py |
DTensorBase.grad_fn |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / dtensor.py |
DTensorBase.grad_zero_ |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / dtensor.py |
DTensorBase.detach |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / dtensor.py |
DTensorBase.detach_ |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / dtensor.py |
DTensorBase.is_leaf |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / dtensor.py |
DTensorBase.retains_grad |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / dtensor.py |
DTensorBase.retains_grad |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / dtensor.py |
DTensorBase.backward |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / dtensor.py |
DTensorBase.device |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / dtensor.py |
DTensorBase.data |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / dtensor.py |
DTensorBase.data |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / dtensor.py |
DTensorBase.dtype |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / dtensor.py |
DTensorBase.shape |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / dtensor.py |
DTensorBase.type |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / dtensor.py |
DTensorBase.size |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / dtensor.py |
DTensorBase.ndim |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / dtensor.py |
DTensorBase.data_ptr |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / dtensor.py |
DTensorBase.numel |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / dtensor.py |
DTensorBase._alias_placements |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / dtensor.py |
DTensorBase.to |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / dtensor.py |
DTensorBase.__repr__ |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / dtensor.py |
(no function) |
|
47 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / __init__.py |
(no function) |
|
0 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / hook_function.py |
PostBackwardFunction.forward |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / hook_function.py |
PostBackwardFunction.backward |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / hook_function.py |
PostBackwardFunction.apply |
|
26 |
2 |
0 |
|
92% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / hook_function.py |
(no function) |
|
8 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / param.py |
_copy_without_bumping_version |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / param.py |
ParameterHookMigrator.__init__ |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / param.py |
ParameterHookMigrator._save_backward_hooks |
|
7 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / param.py |
ParameterHookMigrator._migrate_backward_hooks |
|
9 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / param.py |
TorchHSDPParamV2.__init__ |
|
31 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / param.py |
TorchHSDPParamV2._get_base_spmd_placements |
|
7 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / param.py |
TorchHSDPParamV2._build_storage_source_layout |
|
16 |
1 |
0 |
|
94% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / param.py |
TorchHSDPParamV2._fsdp_domain_ranks |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / param.py |
TorchHSDPParamV2._source_dim_covered_by_fsdp |
|
7 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / param.py |
TorchHSDPParamV2._apply_data_parallel_placements |
|
13 |
1 |
0 |
|
92% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / param.py |
TorchHSDPParamV2._build_sharding_spec |
|
17 |
2 |
0 |
|
88% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / param.py |
TorchHSDPParamV2.reduce_partial_output |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / param.py |
TorchHSDPParamV2.reduce_partial_output |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / param.py |
TorchHSDPParamV2.reduce_comm_dtype |
|
9 |
5 |
0 |
|
44% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / param.py |
TorchHSDPParamV2.reduce_scatter_output |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / param.py |
TorchHSDPParamV2.clear_reduce_scatter_output |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / param.py |
TorchHSDPParamV2.all_reduce_output |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / param.py |
TorchHSDPParamV2.clear_all_reduce_output |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / param.py |
TorchHSDPParamV2.clear_unsharded_source_grad |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / param.py |
TorchHSDPParamV2.apply_reduced_grad |
|
29 |
7 |
0 |
|
76% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / param.py |
TorchHSDPParamV2._resolve_hsdp_placement |
|
10 |
3 |
0 |
|
70% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / param.py |
TorchHSDPParamV2._init_shard_placements |
|
12 |
2 |
0 |
|
83% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / param.py |
TorchHSDPParamV2._init_sharded_param |
|
54 |
6 |
0 |
|
89% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / param.py |
TorchHSDPParamV2.init_dtype_attrs |
|
8 |
8 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / param.py |
TorchHSDPParamV2.init_unsharded_param_buffers |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / param.py |
TorchHSDPParamV2.init_unsharded_param |
|
16 |
8 |
0 |
|
50% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / param.py |
TorchHSDPParamV2.to_sharded |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / param.py |
TorchHSDPParamV2.to_unsharded |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / param.py |
TorchHSDPParamV2._setattr_on_modules |
|
9 |
3 |
0 |
|
67% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / param.py |
TorchHSDPParamV2.to_sharded_dtensor |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / param.py |
TorchHSDPParamV2.to_accumulated_grad_if_needed |
|
9 |
3 |
0 |
|
67% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / param.py |
TorchHSDPParamV2.accumulate_unsharded_grad_if_needed |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / param.py |
TorchHSDPParamV2.alloc_unsharded_param_buffers |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / param.py |
TorchHSDPParamV2.free_unsharded_param |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / param.py |
TorchHSDPParamV2.all_gather_inputs |
|
7 |
2 |
0 |
|
71% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / param.py |
TorchHSDPParamV2._sharded_local_tensor |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / param.py |
TorchHSDPParamV2.unsharded_param |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / param.py |
TorchHSDPParamV2.unsharded_grad_data |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / param.py |
TorchHSDPParamV2.unsharded_accumulated_grad_data |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / param.py |
TorchHSDPParamV2._assert_in_states |
|
2 |
1 |
0 |
|
50% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / param.py |
TorchHSDPParamV2._resolve_reset_param |
|
11 |
1 |
0 |
|
91% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / param.py |
TorchHSDPParamV2._is_same_sharded_local_tensor |
|
4 |
1 |
0 |
|
75% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / param.py |
TorchHSDPParamV2._validate_reset_local_tensor |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / param.py |
TorchHSDPParamV2._pin_reset_local_tensor_if_needed |
|
3 |
1 |
0 |
|
67% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / param.py |
TorchHSDPParamV2._refresh_sharded_local_tensor |
|
17 |
2 |
0 |
|
88% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / param.py |
TorchHSDPParamV2.reset_sharded_param |
|
17 |
1 |
0 |
|
94% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / param.py |
TorchHSDPParamV2._update_shardedparam_storage_forcely |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / param.py |
TorchHSDPParamV2._get_unsharded_param_data |
|
19 |
9 |
0 |
|
53% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / param.py |
TorchHSDPParamV2.unshard |
|
3 |
1 |
0 |
|
67% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / param.py |
TorchHSDPParamV2.wait_for_unshard |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / param.py |
TorchHSDPParamV2.shard |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / param.py |
TorchHSDPParamV2.reduce_scatter_grad |
|
35 |
8 |
0 |
|
77% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / param.py |
TorchHSDPParamV2.all_reduce_grad |
|
13 |
1 |
0 |
|
92% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / param.py |
TorchHSDPParamV2.all_reduce_source_replicate_grad_inplace |
|
15 |
3 |
0 |
|
80% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / param.py |
set_requires_grad_if_needed |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / param.py |
(no function) |
|
92 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / param_group.py |
AllGatherBucket.init_flat_param_buffer |
|
25 |
2 |
0 |
|
92% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / param_group.py |
AllGatherBucket.is_flat_buffer_valid |
|
4 |
1 |
0 |
|
75% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / param_group.py |
AllGatherBucket.copy_out |
|
40 |
3 |
0 |
|
92% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / param_group.py |
ReduceScatterBucket.hsdp_params |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / param_group.py |
ReduceScatterBucket.param_offsets |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / param_group.py |
ReduceScatterBucket.bucket_key |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / param_group.py |
ReduceScatterBucket.uses_collective |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / param_group.py |
ReduceScatterBucket.move_reducescatter_output |
|
5 |
1 |
0 |
|
80% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / param_group.py |
AllReduceBucket.hsdp_params |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / param_group.py |
AllReduceBucket.param_numels |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / param_group.py |
AllReduceBucket.param_offsets |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / param_group.py |
AllReduceBucket.uses_collective |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / param_group.py |
get_all_gather_metadata |
|
20 |
1 |
0 |
|
95% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / param_group.py |
all_gather_copy_in |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / param_group.py |
reduce_scatter_copy_in |
|
17 |
2 |
0 |
|
88% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / param_group.py |
HSDPParamGroup.__init__ |
|
10 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / param_group.py |
HSDPParamGroup._init_all_gather_buckets |
|
21 |
1 |
0 |
|
95% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / param_group.py |
HSDPParamGroup.unshard |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / param_group.py |
HSDPParamGroup.foreach_all_gather |
|
22 |
5 |
0 |
|
77% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / param_group.py |
HSDPParamGroup.wait_for_unshard |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / param_group.py |
HSDPParamGroup._build_reduce_scatter_buckets |
|
38 |
4 |
0 |
|
89% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / param_group.py |
HSDPParamGroup._build_gradient_bucket_layout_from |
|
9 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / param_group.py |
HSDPParamGroup._build_all_reduce_buckets |
|
14 |
1 |
0 |
|
93% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / param_group.py |
HSDPParamGroup._issue_reduce_scatter_buckets |
|
17 |
1 |
0 |
|
94% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / param_group.py |
HSDPParamGroup.foreach_reducescatter |
|
10 |
3 |
0 |
|
70% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / param_group.py |
HSDPParamGroup._wait_reduce_scatter_buckets |
|
9 |
1 |
0 |
|
89% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / param_group.py |
HSDPParamGroup._issue_all_reduce_buckets |
|
7 |
1 |
0 |
|
86% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / param_group.py |
HSDPParamGroup.wait_reduce_scatter_and_issue_all_reduce |
|
29 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / param_group.py |
HSDPParamGroup.wait_all_reduce_and_save_grad |
|
14 |
1 |
0 |
|
93% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / param_group.py |
HSDPParamGroup.reset_iter_state |
|
21 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / param_group.py |
AllReduceParamGroup.__init__ |
|
9 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / param_group.py |
AllReduceParamGroup.compute_aligned_layout |
|
12 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / param_group.py |
AllReduceParamGroup.allocate_fused_buffer |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / param_group.py |
AllReduceParamGroup.get_param_buffer_view |
|
3 |
1 |
0 |
|
67% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / param_group.py |
AllReduceParamGroup.get_param_grad_view |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / param_group.py |
AllReduceParamGroup.accumulate_reduce_partial_outputs |
|
10 |
3 |
0 |
|
70% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / param_group.py |
AllReduceParamGroup.issue_async_allreduce |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / param_group.py |
AllReduceParamGroup.wait_and_split_grads |
|
9 |
2 |
0 |
|
78% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / param_group.py |
(no function) |
|
114 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / scheduler.py |
_dynamo_disable |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / scheduler.py |
_dynamo_disable.wrapper |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / scheduler.py |
TorchHSDPSchedulerV2.__init__ |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / scheduler.py |
TorchHSDPSchedulerV2._register_hooks |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / scheduler.py |
TorchHSDPSchedulerV2._init_platform |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / scheduler.py |
TorchHSDPSchedulerV2._new_cell_state |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / scheduler.py |
TorchHSDPSchedulerV2._register_post_backward_hook |
|
21 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / scheduler.py |
TorchHSDPSchedulerV2._forward_pre_hook |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / scheduler.py |
TorchHSDPSchedulerV2._register_backward_pre_hook |
|
8 |
8 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / scheduler.py |
TorchHSDPSchedulerV2._register_backward_pre_hook.wrapper_for_backward_pre_hook |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / scheduler.py |
TorchHSDPSchedulerV2._forward_hook |
|
7 |
2 |
0 |
|
71% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / scheduler.py |
TorchHSDPSchedulerV2._backward_pre_hook |
|
7 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / scheduler.py |
TorchHSDPSchedulerV2._root_backward_hook |
|
9 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / scheduler.py |
TorchHSDPSchedulerV2._finalize_comm_fusion_reductions |
|
12 |
8 |
0 |
|
33% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / scheduler.py |
TorchHSDPSchedulerV2._finalize_per_param_reductions |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / scheduler.py |
TorchHSDPSchedulerV2.launch_tp_replicate_reduce_and_apply |
|
16 |
1 |
0 |
|
94% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / scheduler.py |
TorchHSDPSchedulerV2.reset_iter_state |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / scheduler.py |
TorchHSDPSchedulerV2._backward_hook |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / scheduler.py |
TorchHSDPSchedulerV2._grouped_forward_pre_hook_skip |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / scheduler.py |
TorchHSDPSchedulerV2._grouped_forward_post_hook_skip |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / scheduler.py |
TorchHSDPSchedulerV2._grouped_forward_pre_hook |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / scheduler.py |
TorchHSDPSchedulerV2._register_forward_module_hook |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / scheduler.py |
TorchHSDPSchedulerV2._register_forward_backward_hooks |
|
9 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / scheduler.py |
(no function) |
|
46 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / state.py |
_to_dtype_if_needed |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / state.py |
TorchHSDPStateV2.__init__ |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / state.py |
TorchHSDPStateV2._init_param_group |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / state.py |
TorchHSDPStateV2._move_states_to_device |
|
11 |
11 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / state.py |
TorchHSDPStateV2._build_param_source_shard_info |
|
7 |
1 |
0 |
|
86% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / state.py |
TorchHSDPStateV2._init_hsdp_params |
|
15 |
2 |
0 |
|
87% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / state.py |
TorchHSDPStateV2._build_param_mesh_info |
|
7 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / state.py |
TorchHSDPStateV2._init_mp_dtypes |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / state.py |
TorchHSDPStateV2._validate_cpu_offload_params |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / state.py |
TorchHSDPStateV2.lazy_init |
|
7 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / state.py |
TorchHSDPStateV2._validate_no_meta_params |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / state.py |
TorchHSDPStateV2.post_backward_for_comm_fusion |
|
13 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / state.py |
TorchHSDPStateV2.post_backward |
|
17 |
1 |
0 |
|
94% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / state.py |
TorchHSDPStateV2._issue_reduce_scatter_for_current_module |
|
30 |
8 |
0 |
|
73% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / state.py |
TorchHSDPStateV2._wait_prev_reduce_scatter |
|
14 |
12 |
0 |
|
14% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / state.py |
TorchHSDPStateV2._issue_prev_fused_all_reduce |
|
5 |
4 |
0 |
|
20% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / state.py |
TorchHSDPStateV2._wait_prev_reduce_scatter_without_all_reduce |
|
16 |
2 |
0 |
|
88% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / state.py |
TorchHSDPStateV2.wait_and_split_all_reduce_work_groups |
|
4 |
2 |
0 |
|
50% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / state.py |
TorchHSDPStateV2.reset_iter_state |
|
18 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / state.py |
TorchHSDPStateV2.set_requires_grad_sync |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / state.py |
TorchHSDPStateV2.set_reduce_op_type |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / state.py |
TorchHSDPStateV2._sync_current_stream_if_needed |
|
6 |
5 |
0 |
|
17% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / state.py |
(no function) |
|
34 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / state_dict_utils.py |
_gather_full_state_dict |
|
15 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / state_dict_utils.py |
_offload_sharded_state_dict |
|
8 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / state_dict_utils.py |
get_model_state_dict |
|
15 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / state_dict_utils.py |
_scatter_model_state_dict |
|
21 |
1 |
0 |
|
95% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / state_dict_utils.py |
set_model_state_dict |
|
9 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / state_dict_utils.py |
_is_dtensor_param |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / state_dict_utils.py |
_is_replicate_group_root |
|
7 |
7 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / state_dict_utils.py |
_param_to_fqn |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / state_dict_utils.py |
_build_id_to_fqn |
|
15 |
15 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / state_dict_utils.py |
_get_param_dtensor_info |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / state_dict_utils.py |
_convert_state_tensor |
|
20 |
20 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / state_dict_utils.py |
_is_scalar_state |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / state_dict_utils.py |
_convert_state_scalar |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / state_dict_utils.py |
_determine_is_root |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / state_dict_utils.py |
_convert_state_entries |
|
16 |
16 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / state_dict_utils.py |
_build_result_param_groups |
|
9 |
9 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / state_dict_utils.py |
_check_chained_optimizer |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / state_dict_utils.py |
get_optim_state_dict |
|
22 |
22 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / state_dict_utils.py |
_check_strict_fqns |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / state_dict_utils.py |
_load_state_values |
|
8 |
8 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / state_dict_utils.py |
_merge_param_groups |
|
20 |
20 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / state_dict_utils.py |
_build_target_fqn_mappings |
|
18 |
18 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / state_dict_utils.py |
_load_fqn_states |
|
12 |
12 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / state_dict_utils.py |
set_optim_state_dict |
|
26 |
26 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / state_dict_utils.py |
_convert_input_tensor_to_target |
|
36 |
36 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / state_dict_utils.py |
_convert_input_scalar_to_target |
|
13 |
13 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / state_dict_utils.py |
_get_broadcast_groups |
|
27 |
27 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / state_dict_utils.py |
_broadcast_fqn_list |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / state_dict_utils.py |
_build_state_schema |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / state_dict_utils.py |
_broadcast_schema_per_fqn |
|
9 |
9 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / state_dict_utils.py |
_get_local_device |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / state_dict_utils.py |
_broadcast_scalar_entry |
|
13 |
13 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / state_dict_utils.py |
_broadcast_tensor_entry |
|
13 |
13 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / state_dict_utils.py |
_broadcast_tensor_data_per_fqn |
|
12 |
12 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / state_dict_utils.py |
_broadcast_state_from_rank0 |
|
14 |
14 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / state_dict_utils.py |
_flatten_optim_state_dict |
|
14 |
14 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / state_dict_utils.py |
_check_inconsistent_pg_fields |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / state_dict_utils.py |
_parse_unflatten_entries |
|
18 |
18 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / state_dict_utils.py |
_assemble_param_groups |
|
18 |
18 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / state_dict_utils.py |
_unflatten_optim_state_dict |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / state_dict_utils.py |
_build_optim_state_dict_load_template |
|
14 |
14 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / state_dict_utils.py |
_resolve_dtype |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / state_dict_utils.py |
_match_fqn_from_remainder |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / state_dict_utils.py |
_build_saved_id_to_fqn |
|
9 |
9 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / state_dict_utils.py |
_parse_nested_state_entry |
|
19 |
19 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / state_dict_utils.py |
_parse_nested_param_group_entry |
|
14 |
14 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / state_dict_utils.py |
_build_nested_template_from_metadata |
|
25 |
25 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / state_dict_utils.py |
_build_flatten_template_from_metadata |
|
25 |
25 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / state_dict_utils.py |
_create_empty_state_tensor |
|
11 |
11 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / state_dict_utils.py |
_infer_state_keys |
|
14 |
14 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / state_dict_utils.py |
(no function) |
|
63 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / function_override.py |
DTensorBackwardHookFunction.apply |
|
25 |
25 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / function_override.py |
ExtendBackwardHook.setup_output_hook |
|
5 |
3 |
0 |
|
40% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / function_override.py |
override_functions |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / function_override.py |
(no function) |
|
12 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / group_utils.py |
_validate_intra_step |
|
7 |
7 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / group_utils.py |
_compute_group_starts |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / group_utils.py |
_build_groups_for_blocks |
|
8 |
8 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / group_utils.py |
generate_groups_from_template |
|
33 |
33 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / group_utils.py |
create_sub_groups |
|
31 |
31 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / group_utils.py |
(no function) |
|
8 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / init_weights.py |
init_on_device |
|
12 |
12 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / init_weights.py |
init_on_device._register_parameter |
|
7 |
7 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / init_weights.py |
init_on_device._register_buffer |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / init_weights.py |
(no function) |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / loss_parallel_ops.py |
_is_floating_torch |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / loss_parallel_ops.py |
_compute_vocab_start |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / loss_parallel_ops.py |
distributed_log_softmax |
|
8 |
8 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / loss_parallel_ops.py |
distributed_nll_loss_forward |
|
32 |
32 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / loss_parallel_ops.py |
DistributedCrossEntropyFunction.forward |
|
46 |
46 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / loss_parallel_ops.py |
DistributedCrossEntropyFunction.backward |
|
45 |
45 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / loss_parallel_ops.py |
distributed_cross_entropy |
|
24 |
24 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / loss_parallel_ops.py |
distributed_cross_entropy_from_op_call |
|
9 |
9 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / loss_parallel_ops.py |
(no function) |
|
21 |
21 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / pipeline_parallel / __init__.py |
(no function) |
|
0 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / pipeline_parallel / _utils.py |
_MicroBatch.__init__ |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / pipeline_parallel / _utils.py |
_MicroBatch.forward |
|
24 |
24 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / pipeline_parallel / _utils.py |
_MicroBatch.split_inputs_with_custom_shard |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / pipeline_parallel / _utils.py |
_MicroBatch.split_inputs |
|
11 |
11 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / pipeline_parallel / _utils.py |
(no function) |
|
7 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / pipeline_parallel / mpipe_transpose.py |
MPipeTransposeExecutor._local |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / pipeline_parallel / mpipe_transpose.py |
MPipeTransposeExecutor._broadcast_tensors |
|
7 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / pipeline_parallel / mpipe_transpose.py |
MPipeTransposeExecutor._detached_forward |
|
5 |
1 |
0 |
|
80% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / pipeline_parallel / mpipe_transpose.py |
MPipeTransposeExecutor._connected_forward |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / pipeline_parallel / mpipe_transpose.py |
MPipeTransposeExecutor._mark_requires_grad |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / pipeline_parallel / mpipe_transpose.py |
MPipeTransposeExecutor._explicit_forward_before_backward |
|
7 |
1 |
0 |
|
86% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / pipeline_parallel / mpipe_transpose.py |
MPipeTransposeExecutor._contiguous |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / pipeline_parallel / mpipe_transpose.py |
MPipeTransposeExecutor._detach_for_wire |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / pipeline_parallel / mpipe_transpose.py |
MPipeTransposeExecutor._zeros_like |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / pipeline_parallel / mpipe_transpose.py |
MPipeTransposeExecutor._owner_transpose_backward |
|
5 |
1 |
0 |
|
80% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / pipeline_parallel / mpipe_transpose.py |
MPipeTransposeExecutor._snapshot_tower_grads |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / pipeline_parallel / mpipe_transpose.py |
MPipeTransposeExecutor._reduce_grads |
|
10 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / pipeline_parallel / mpipe_transpose.py |
(no function) |
|
21 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / pipeline_parallel / stage.py |
PipelineStageBase.__init__ |
|
13 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / pipeline_parallel / stage.py |
PipelineStageBase.clear_cache |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / pipeline_parallel / stage.py |
PipelineStageBase._clear_recv_buffer |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / pipeline_parallel / stage.py |
PipelineStageBase._check_pp_group |
|
5 |
3 |
0 |
|
40% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / pipeline_parallel / stage.py |
PipelineStageBase.is_first_stage |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / pipeline_parallel / stage.py |
PipelineStageBase.is_last_stage |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / pipeline_parallel / stage.py |
PipelineStageBase.forward_one_chunk |
|
13 |
1 |
0 |
|
92% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / pipeline_parallel / stage.py |
PipelineStageBase._filter_grad_outputs |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / pipeline_parallel / stage.py |
PipelineStageBase._build_last_stage_sens |
|
10 |
7 |
0 |
|
30% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / pipeline_parallel / stage.py |
PipelineStageBase._populate_bwd_cache |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / pipeline_parallel / stage.py |
PipelineStageBase._get_trainable_params |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / pipeline_parallel / stage.py |
PipelineStageBase.backward_one_chunk |
|
20 |
6 |
0 |
|
70% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / pipeline_parallel / stage.py |
PipelineStageBase.backward_input_one_chunk |
|
14 |
2 |
0 |
|
86% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / pipeline_parallel / stage.py |
PipelineStageBase.backward_weight_one_chunk |
|
13 |
4 |
0 |
|
69% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / pipeline_parallel / stage.py |
(no function) |
|
24 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / platform.py |
_a2a_reconstruct |
|
7 |
7 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / platform.py |
_normalize_dim |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / platform.py |
_move_dim_to_front |
|
5 |
2 |
0 |
|
60% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / platform.py |
_move_dim_from_front |
|
8 |
5 |
0 |
|
38% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / platform.py |
_TorchContiguousGrad.forward |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / platform.py |
_TorchContiguousGrad.backward |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / platform.py |
_TorchAsyncA2AFunction.forward |
|
8 |
8 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / platform.py |
_TorchAsyncA2AFunction.backward |
|
11 |
11 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / platform.py |
_TorchAsyncAllGatherFunction.forward |
|
7 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / platform.py |
_TorchAsyncAllGatherFunction.backward |
|
12 |
1 |
0 |
|
92% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / platform.py |
_AsyncA2ALazyBwd.forward |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / platform.py |
_AsyncA2ALazyBwd.backward |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / platform.py |
_TorchSyncHookFunction._role_enum |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / platform.py |
_TorchSyncHookFunction.forward |
|
14 |
14 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / platform.py |
_TorchSyncHookFunction.backward |
|
11 |
11 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / platform.py |
_TorchP2PExchangeFunction.forward |
|
8 |
8 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / platform.py |
_TorchP2PExchangeFunction.backward |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / platform.py |
_TorchDifferentiableVariableAllGather.forward |
|
30 |
6 |
0 |
|
80% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / platform.py |
_TorchDifferentiableVariableAllGather.backward |
|
13 |
3 |
0 |
|
77% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / platform.py |
_ensure_contiguous |
|
5 |
1 |
0 |
|
80% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / platform.py |
_TorchBatchP2PWork.__init__ |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / platform.py |
_TorchBatchP2PWork.wait |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / platform.py |
TorchPlatform.get_swap_optimizer |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / platform.py |
TorchPlatform.is_linear_module |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / platform.py |
TorchPlatform.is_embedding_module |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / platform.py |
TorchPlatform.device_count |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / platform.py |
TorchPlatform.device_type |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / platform.py |
TorchPlatform.device |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / platform.py |
TorchPlatform.get_rng_state |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / platform.py |
TorchPlatform.set_rng_state |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / platform.py |
TorchPlatform.manual_seed |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / platform.py |
TorchPlatform.ones |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / platform.py |
TorchPlatform.zeros |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / platform.py |
TorchPlatform.full |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / platform.py |
TorchPlatform.empty |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / platform.py |
TorchPlatform.rand |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / platform.py |
TorchPlatform.randn |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / platform.py |
TorchPlatform.get_rank |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / platform.py |
TorchPlatform.get_global_rank |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / platform.py |
TorchPlatform.get_group_rank |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / platform.py |
TorchPlatform.get_world_size |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / platform.py |
TorchPlatform.get_param_local_shape |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / platform.py |
TorchPlatform.get_param_local_data |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / platform.py |
TorchPlatform.update_param_data |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / platform.py |
TorchPlatform.load_into_param |
|
9 |
9 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / platform.py |
TorchPlatform.get_op_name |
|
14 |
12 |
0 |
|
14% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / platform.py |
TorchPlatform.differentiable_all_gather_concat |
|
8 |
4 |
0 |
|
50% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / platform.py |
TorchPlatform.chunk |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / platform.py |
TorchPlatform.differentiable_all_to_all |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / platform.py |
TorchPlatform.tensor_type_cast |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / platform.py |
TorchPlatform.differentiable_all_reduce |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / platform.py |
TorchPlatform.get_cell_construct |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / platform.py |
TorchPlatform.get_cells_and_names |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / platform.py |
TorchPlatform.get_modules |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / platform.py |
TorchPlatform.search_parameter_by_name |
|
17 |
3 |
0 |
|
82% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / platform.py |
TorchPlatform.update_parameter_by_name |
|
5 |
1 |
0 |
|
80% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / platform.py |
TorchPlatform.set_layout_into_parameter |
|
8 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / platform.py |
TorchPlatform.differentiable_reduce_scatter |
|
8 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / platform.py |
TorchPlatform.get_device_handle |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / platform.py |
TorchPlatform.get_param_type_size |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / platform.py |
TorchPlatform.is_tensor |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / platform.py |
TorchPlatform.get_tensor_storage_size |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / platform.py |
TorchPlatform.copy_each |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / platform.py |
TorchPlatform.parameters_dict |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / platform.py |
TorchPlatform.buffers_dict |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / platform.py |
TorchPlatform.get_model_state_dict |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / platform.py |
TorchPlatform.set_model_state_dict |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / platform.py |
TorchPlatform.get_optim_state_dict |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / platform.py |
TorchPlatform.set_optim_state_dict |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / platform.py |
TorchPlatform.save_checkpoint |
|
3 |
1 |
0 |
|
67% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / platform.py |
TorchPlatform.load_checkpoint |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / platform.py |
TorchPlatform.new_zero_parameter |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / platform.py |
TorchPlatform.new_tensor |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / platform.py |
TorchPlatform.full_like |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / platform.py |
TorchPlatform.set_tensor_requires_grad |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / platform.py |
TorchPlatform.new_group |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / platform.py |
TorchPlatform.detach |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / platform.py |
TorchPlatform._create_group |
|
8 |
8 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / platform.py |
TorchPlatform.create_p2p_multi_stream_groups |
|
19 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / platform.py |
TorchPlatform.all_gather_into_tensor |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / platform.py |
TorchPlatform.all_gather_single |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / platform.py |
TorchPlatform.all_reduce |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / platform.py |
TorchPlatform.broadcast |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / platform.py |
TorchPlatform.broadcast_async |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / platform.py |
TorchPlatform.scatter |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / platform.py |
TorchPlatform.isend |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / platform.py |
TorchPlatform.irecv |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / platform.py |
TorchPlatform.p2p_op |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / platform.py |
TorchPlatform.batch_isend_irecv |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / platform.py |
TorchPlatform.prepare_batch_p2p_group |
|
27 |
3 |
0 |
|
89% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / platform.py |
TorchPlatform.p2p_exchange |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / platform.py |
TorchPlatform.send_object_list |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / platform.py |
TorchPlatform.recv_object_list |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / platform.py |
TorchPlatform.reduce_scatter_tensor |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / platform.py |
TorchPlatform.reduce_scatter_single |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / platform.py |
TorchPlatform.all_to_all_single |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / platform.py |
TorchPlatform.differentiable_all_to_all_single |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / platform.py |
TorchPlatform.differentiable_all_to_all_single_async |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / platform.py |
TorchPlatform.differentiable_variable_all_gather |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / platform.py |
TorchPlatform.wait_async_tensor |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / platform.py |
TorchPlatform.differentiable_async_allgather_wait |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / platform.py |
TorchPlatform.arange |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / platform.py |
TorchPlatform.differentiable_async_a2a_wait |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / platform.py |
TorchPlatform.differentiable_sync_hook |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / platform.py |
TorchPlatform.get_tensor_transform |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / platform.py |
TorchPlatform.construct_strided_slice |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / platform.py |
TorchPlatform.micro_batch |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / platform.py |
TorchPlatform.new_stream |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / platform.py |
TorchPlatform.get_stream_context |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / platform.py |
TorchPlatform.all_gather_object |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / platform.py |
TorchPlatform.barrier |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / platform.py |
TorchPlatform.init_process_group |
|
6 |
1 |
0 |
|
83% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / platform.py |
TorchPlatform.get_world_group |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / platform.py |
TorchPlatform.destroy_process_group |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / platform.py |
TorchPlatform.get_process_group_ranks |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / platform.py |
TorchPlatform.get_backend |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / platform.py |
TorchPlatform.split_group |
|
11 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / platform.py |
TorchPlatform.get_group_local_rank |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / platform.py |
TorchPlatform.no_grad |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / platform.py |
TorchPlatform.preserve_version_counter |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / platform.py |
TorchPlatform.relu |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / platform.py |
TorchPlatform.cat |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / platform.py |
TorchPlatform.empty_like |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / platform.py |
TorchPlatform.get_current_stream |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / platform.py |
TorchPlatform.synchronize |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / platform.py |
TorchPlatform.new_event |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / platform.py |
TorchPlatform.tree_map |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / platform.py |
TorchPlatform.checkpoint |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / platform.py |
TorchPlatform.recompute_handle_collector_ctx |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / platform.py |
TorchPlatform.recompute_handle |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / platform.py |
TorchPlatform.recompute_session_ctx |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / platform.py |
TorchPlatform.clear_recompute_session |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / platform.py |
TorchPlatform.is_compiling |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / platform.py |
TorchPlatform.checkpoint_wrapper |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / platform.py |
TorchPlatform.checkpoint_exclude_wrapper |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / platform.py |
TorchPlatform.swap_wrapper |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / platform.py |
TorchPlatform.swap_tensor_wrapper |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / platform.py |
TorchPlatform.get_class_activation_wrapper |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / platform.py |
TorchPlatform.noop_context_fn |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / platform.py |
TorchPlatform.ignore_sac_ops |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / platform.py |
TorchPlatform.create_selective_checkpoint_contexts |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / platform.py |
TorchPlatform.create_native_selective_checkpoint_contexts |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / platform.py |
TorchPlatform.async_save_on_cpu |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / platform.py |
TorchPlatform.get_element_size |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / platform.py |
TorchPlatform.alloc_tensor_buffer |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / platform.py |
TorchPlatform.tensor_to_numpy |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / platform.py |
TorchPlatform.from_numpy |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / platform.py |
TorchPlatform.clip_grad_norm_ |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / platform.py |
TorchPlatform.profiler_record |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / platform.py |
TorchPlatform.cast_fp_tensor |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / platform.py |
TorchPlatform.apply_to_tensors |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / platform.py |
TorchPlatform.apply_to_tensors.apply |
|
22 |
11 |
0 |
|
50% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / platform.py |
TorchPlatform.meta_device |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / platform.py |
TorchPlatform.init_on_device |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / platform.py |
TorchPlatform.str_to_dtype |
|
10 |
3 |
0 |
|
70% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / platform.py |
TorchPlatform.list_to_size |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / platform.py |
(no function) |
|
343 |
1 |
0 |
|
99% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / swap_optimizer / __init__.py |
(no function) |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / swap_optimizer / adapters.py |
TorchAdamBaseAdapter.matches |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / swap_optimizer / adapters.py |
TorchAdamBaseAdapter.__init__ |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / swap_optimizer / adapters.py |
TorchAdamBaseAdapter.validate |
|
9 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / swap_optimizer / adapters.py |
TorchAdamBaseAdapter.prepare_step |
|
19 |
1 |
0 |
|
95% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / swap_optimizer / adapters.py |
TorchAdamBaseAdapter._prepare_packed_step |
|
28 |
4 |
0 |
|
86% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / swap_optimizer / adapters.py |
TorchAdamBaseAdapter.iter_update_units |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / swap_optimizer / adapters.py |
TorchAdamBaseAdapter.initial_slots |
|
7 |
7 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / swap_optimizer / adapters.py |
TorchAdamBaseAdapter.step_batch |
|
39 |
7 |
0 |
|
82% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / swap_optimizer / adapters.py |
TorchAdamBaseAdapter.all_slots |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / swap_optimizer / adapters.py |
TorchAdamBaseAdapter.checkpoint_state_dict |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / swap_optimizer / adapters.py |
TorchAdamBaseAdapter.load_checkpoint_state_dict |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / swap_optimizer / adapters.py |
TorchAdamBaseAdapter.publish_packed_state |
|
12 |
2 |
0 |
|
83% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / swap_optimizer / adapters.py |
TorchAdamBaseAdapter.export_swappable_state |
|
24 |
1 |
0 |
|
96% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / swap_optimizer / adapters.py |
TorchAdamBaseAdapter.strip_swappable_state |
|
10 |
1 |
0 |
|
90% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / swap_optimizer / adapters.py |
TorchAdamBaseAdapter.load_swappable_state |
|
42 |
16 |
0 |
|
62% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / swap_optimizer / adapters.py |
TorchAdamBaseAdapter._init_param_state |
|
26 |
10 |
0 |
|
62% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / swap_optimizer / adapters.py |
TorchAdamBaseAdapter._register_present_slots |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / swap_optimizer / adapters.py |
TorchAdamBaseAdapter._build_slots |
|
14 |
3 |
0 |
|
79% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / swap_optimizer / adapters.py |
TorchAdamBaseAdapter._make_slot |
|
10 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / swap_optimizer / adapters.py |
TorchAdamBaseAdapter._ordered_slots |
|
10 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / swap_optimizer / adapters.py |
TorchAdamBaseAdapter._state_keys_for_param |
|
9 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / swap_optimizer / adapters.py |
TorchAdamBaseAdapter._slot_tensor |
|
4 |
1 |
0 |
|
75% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / swap_optimizer / adapters.py |
TorchAdamBaseAdapter._configured_state_keys |
|
9 |
1 |
0 |
|
89% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / swap_optimizer / adapters.py |
TorchAdamBaseAdapter._cast_state_tensor_like_torch |
|
10 |
10 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / swap_optimizer / adapters.py |
TorchAdamBaseAdapter._cast_swappable_tensor_to_cpu |
|
5 |
1 |
0 |
|
80% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / swap_optimizer / adapters.py |
TorchAdamBaseAdapter._default_state_keys |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / swap_optimizer / adapters.py |
TorchNativeAdamAdapter.matches |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / swap_optimizer / adapters.py |
TorchNativeAdamWAdapter.matches |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / swap_optimizer / adapters.py |
TorchNewAdamWAdapter.matches |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / swap_optimizer / adapters.py |
(no function) |
|
59 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / swap_optimizer / swap_optimizer.py |
TorchSwapRuntime.__init__ |
|
10 |
10 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / swap_optimizer / swap_optimizer.py |
TorchSwapRuntime.packed_enabled |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / swap_optimizer / swap_optimizer.py |
TorchSwapRuntime.populate_slot_metadata |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / swap_optimizer / swap_optimizer.py |
TorchSwapRuntime.is_packable_template |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / swap_optimizer / swap_optimizer.py |
TorchSwapRuntime.is_distributed_tensor |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / swap_optimizer / swap_optimizer.py |
TorchSwapRuntime.prepare_packed_host |
|
33 |
33 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / swap_optimizer / swap_optimizer.py |
TorchSwapRuntime.is_swappable_tensor |
|
21 |
21 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / swap_optimizer / swap_optimizer.py |
TorchSwapRuntime.storage_nbytes |
|
7 |
7 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / swap_optimizer / swap_optimizer.py |
TorchSwapRuntime.make_cpu_tensor |
|
10 |
10 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / swap_optimizer / swap_optimizer.py |
TorchSwapRuntime.make_zero_cpu_tensor_like |
|
9 |
9 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / swap_optimizer / swap_optimizer.py |
TorchSwapRuntime.make_device_tensor_like |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / swap_optimizer / swap_optimizer.py |
TorchSwapRuntime.make_empty_device_tensor_like |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / swap_optimizer / swap_optimizer.py |
TorchSwapRuntime.copy_to_device |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / swap_optimizer / swap_optimizer.py |
TorchSwapRuntime.wait_prefetch_slot |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / swap_optimizer / swap_optimizer.py |
TorchSwapRuntime.copy_to_cpu |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / swap_optimizer / swap_optimizer.py |
TorchSwapRuntime.wait_offload_slot |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / swap_optimizer / swap_optimizer.py |
TorchSwapRuntime.restore_device_storage |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / swap_optimizer / swap_optimizer.py |
TorchSwapRuntime.release_device_storage |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / swap_optimizer / swap_optimizer.py |
TorchSwapRuntime.current_stream |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / swap_optimizer / swap_optimizer.py |
TorchSwapRuntime.new_stream |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / swap_optimizer / swap_optimizer.py |
TorchSwapRuntime.stream_context |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / swap_optimizer / swap_optimizer.py |
TorchSwapRuntime.record_event |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / swap_optimizer / swap_optimizer.py |
TorchSwapRuntime.wait_event |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / swap_optimizer / swap_optimizer.py |
TorchSwapRuntime.supports_packed_pipeline |
|
10 |
10 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / swap_optimizer / swap_optimizer.py |
TorchSwapRuntime._storage_tensor |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / swap_optimizer / swap_optimizer.py |
TorchSwapRuntime.begin_packed_step |
|
31 |
31 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / swap_optimizer / swap_optimizer.py |
TorchSwapRuntime.enqueue_packed_prefetch |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / swap_optimizer / swap_optimizer.py |
TorchSwapRuntime.wait_packed_prefetch |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / swap_optimizer / swap_optimizer.py |
TorchSwapRuntime.activate_packed_batch |
|
14 |
14 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / swap_optimizer / swap_optimizer.py |
TorchSwapRuntime.enqueue_packed_offload_prefetch |
|
12 |
12 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / swap_optimizer / swap_optimizer.py |
TorchSwapRuntime.wait_packed_offload |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / swap_optimizer / swap_optimizer.py |
TorchSwapRuntime.finish_packed_offload |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / swap_optimizer / swap_optimizer.py |
TorchSwapRuntime.end_packed_step |
|
26 |
26 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / swap_optimizer / swap_optimizer.py |
TorchSwapRuntime._build_packed_batch_plan |
|
19 |
19 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / swap_optimizer / swap_optimizer.py |
TorchSwapRuntime._materialize_staging_arena |
|
16 |
16 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / swap_optimizer / swap_optimizer.py |
TorchSwapRuntime._drop_packed_views |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / swap_optimizer / swap_optimizer.py |
TorchSwapRuntime._copy_packed_to_device |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / swap_optimizer / swap_optimizer.py |
TorchSwapRuntime._copy_packed_to_host |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / swap_optimizer / swap_optimizer.py |
TorchSwapRuntime._require_staging_arena |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / swap_optimizer / swap_optimizer.py |
TorchSwapRuntime._align_bytes |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / swap_optimizer / swap_optimizer.py |
TorchSwapOptimizer.__init__ |
|
9 |
9 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / swap_optimizer / swap_optimizer.py |
TorchSwapOptimizer.__getattr__ |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / swap_optimizer / swap_optimizer.py |
TorchSwapOptimizer.param_groups |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / swap_optimizer / swap_optimizer.py |
TorchSwapOptimizer.param_groups |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / swap_optimizer / swap_optimizer.py |
TorchSwapOptimizer.state |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / swap_optimizer / swap_optimizer.py |
TorchSwapOptimizer.defaults |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / swap_optimizer / swap_optimizer.py |
TorchSwapOptimizer.add_param_group |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / swap_optimizer / swap_optimizer.py |
TorchSwapOptimizer.zero_grad |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / swap_optimizer / swap_optimizer.py |
TorchSwapOptimizer.step |
|
8 |
8 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / swap_optimizer / swap_optimizer.py |
TorchSwapOptimizer.state_dict |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / swap_optimizer / swap_optimizer.py |
TorchSwapOptimizer.load_state_dict |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / swap_optimizer / swap_optimizer.py |
TorchSwapOptimizer._build_adapter |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / swap_optimizer / swap_optimizer.py |
TorchSwapOptimizer._no_grad_context |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / swap_optimizer / swap_optimizer.py |
get_swap_optimizer |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / swap_optimizer / swap_optimizer.py |
(no function) |
|
91 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / tools / __init__.py |
(no function) |
|
0 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / tools / logging.py |
_ContextFilter.__init__ |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / tools / logging.py |
_ContextFilter.filter |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / tools / logging.py |
_build_formatter |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / tools / logging.py |
_normalize_level |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / tools / logging.py |
_canonical_component |
|
7 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / tools / logging.py |
_parse_config |
|
13 |
1 |
0 |
|
92% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / tools / logging.py |
_env_levels |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / tools / logging.py |
_make_handler |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / tools / logging.py |
get_logger |
|
10 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / tools / logging.py |
set_level |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / tools / logging.py |
configure |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / tools / logging.py |
set_format |
|
8 |
1 |
0 |
|
88% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / tools / logging.py |
logging_enabled |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / tools / logging.py |
(no function) |
|
38 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / __init__.py |
__getattr__ |
|
6 |
4 |
0 |
|
33% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / __init__.py |
__dir__ |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / __init__.py |
(no function) |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / base.py |
BaseTrainer.__init__ |
|
14 |
14 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / base.py |
BaseTrainer._setup |
|
27 |
27 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / base.py |
BaseTrainer._build_model |
|
10 |
10 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / base.py |
BaseTrainer._build_loss |
|
7 |
7 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / base.py |
BaseTrainer._build_model_assets |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / base.py |
BaseTrainer._build_data_transform |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / base.py |
BaseTrainer._build_dataset |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / base.py |
BaseTrainer._build_collate_fn |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / base.py |
BaseTrainer._build_dataloader |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / base.py |
BaseTrainer._compute_train_iters |
|
22 |
22 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / base.py |
BaseTrainer._build_optimizer |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / base.py |
BaseTrainer._build_lr_scheduler |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / base.py |
BaseTrainer._build_training_context |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / base.py |
BaseTrainer._init_callbacks |
|
11 |
11 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / base.py |
BaseTrainer.on_train_begin |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / base.py |
BaseTrainer.on_train_end |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / base.py |
BaseTrainer.on_epoch_begin |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / base.py |
BaseTrainer.on_epoch_end |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / base.py |
BaseTrainer.on_step_begin |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / base.py |
BaseTrainer.on_step_end |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / base.py |
BaseTrainer.preforward |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / base.py |
BaseTrainer.preforward._to_device |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / base.py |
BaseTrainer.postforward |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / base.py |
BaseTrainer.forward_backward_step |
|
24 |
24 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / base.py |
BaseTrainer.model_reshard |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / base.py |
BaseTrainer._configure_fsdp_gradient_sync |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / base.py |
BaseTrainer.configure_fsdp_gradient_sync |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / base.py |
BaseTrainer.train_step |
|
29 |
29 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / base.py |
BaseTrainer.destroy_distributed |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / base.py |
BaseTrainer.train |
|
29 |
29 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / base.py |
(no function) |
|
92 |
92 |
2 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / callbacks / __init__.py |
(no function) |
|
10 |
10 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / callbacks / base.py |
Callback.__init__ |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / callbacks / base.py |
Callback.on_step_begin |
|
0 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / callbacks / base.py |
Callback.on_step_end |
|
0 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / callbacks / base.py |
Callback.on_micro_step_begin |
|
0 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / callbacks / base.py |
Callback.on_micro_step_end |
|
0 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / callbacks / base.py |
Callback.on_epoch_begin |
|
0 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / callbacks / base.py |
Callback.on_epoch_end |
|
0 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / callbacks / base.py |
Callback.on_train_begin |
|
0 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / callbacks / base.py |
Callback.on_train_end |
|
0 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / callbacks / base.py |
(no function) |
|
12 |
12 |
2 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / callbacks / checkpoint_callback.py |
_as_list |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / callbacks / checkpoint_callback.py |
_unwrap_single |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / callbacks / checkpoint_callback.py |
CheckpointerCallback.__init__ |
|
16 |
16 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / callbacks / checkpoint_callback.py |
CheckpointerCallback.on_train_begin |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / callbacks / checkpoint_callback.py |
CheckpointerCallback.on_step_end |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / callbacks / checkpoint_callback.py |
CheckpointerCallback.on_epoch_end |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / callbacks / checkpoint_callback.py |
CheckpointerCallback.on_train_end |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / callbacks / checkpoint_callback.py |
CheckpointerCallback.wait_for_pending_save |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / callbacks / checkpoint_callback.py |
CheckpointerCallback._model_state_dict |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / callbacks / checkpoint_callback.py |
CheckpointerCallback._collect_extra_state |
|
9 |
9 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / callbacks / checkpoint_callback.py |
CheckpointerCallback._steps_per_epoch |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / callbacks / checkpoint_callback.py |
CheckpointerCallback._save_checkpoint |
|
10 |
10 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / callbacks / checkpoint_callback.py |
CheckpointerCallback._resolve_restore_path |
|
15 |
15 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / callbacks / checkpoint_callback.py |
CheckpointerCallback._load_checkpoint |
|
27 |
27 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / callbacks / checkpoint_callback.py |
CheckpointerCallback._apply_extra_state |
|
28 |
28 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / callbacks / checkpoint_callback.py |
(no function) |
|
30 |
30 |
2 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / callbacks / environ_meter_callback.py |
EnvironMeterCallback.__init__ |
|
8 |
8 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / callbacks / environ_meter_callback.py |
EnvironMeterCallback._scalar |
|
7 |
7 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / callbacks / environ_meter_callback.py |
EnvironMeterCallback._tensor_numel |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / callbacks / environ_meter_callback.py |
EnvironMeterCallback._batch_tokens |
|
12 |
12 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / callbacks / environ_meter_callback.py |
EnvironMeterCallback._batch_samples |
|
9 |
9 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / callbacks / environ_meter_callback.py |
EnvironMeterCallback._batch_mapping |
|
7 |
7 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / callbacks / environ_meter_callback.py |
EnvironMeterCallback._micro_batches |
|
13 |
13 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / callbacks / environ_meter_callback.py |
EnvironMeterCallback._metric_group |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / callbacks / environ_meter_callback.py |
EnvironMeterCallback._reduce |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / callbacks / environ_meter_callback.py |
EnvironMeterCallback._current_lr |
|
10 |
10 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / callbacks / environ_meter_callback.py |
EnvironMeterCallback._memory_metrics |
|
7 |
7 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / callbacks / environ_meter_callback.py |
EnvironMeterCallback.state_dict |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / callbacks / environ_meter_callback.py |
EnvironMeterCallback.load_state_dict |
|
9 |
9 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / callbacks / environ_meter_callback.py |
EnvironMeterCallback.on_step_begin |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / callbacks / environ_meter_callback.py |
EnvironMeterCallback.on_step_end |
|
15 |
15 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / callbacks / environ_meter_callback.py |
(no function) |
|
30 |
30 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / callbacks / evaluate_callback.py |
EvaluateCallback.__init__ |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / callbacks / evaluate_callback.py |
EvaluateCallback._evaluate |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / callbacks / evaluate_callback.py |
EvaluateCallback._trigger_evaluation |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / callbacks / evaluate_callback.py |
EvaluateCallback.on_step_end |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / callbacks / evaluate_callback.py |
EvaluateCallback.on_epoch_end |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / callbacks / evaluate_callback.py |
(no function) |
|
10 |
10 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / callbacks / garbage_collection_callback.py |
GarbageCollectionCallback.__init__ |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / callbacks / garbage_collection_callback.py |
GarbageCollectionCallback.on_step_end |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / callbacks / garbage_collection_callback.py |
(no function) |
|
7 |
7 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / callbacks / logging_callback.py |
LoggingCallback.__init__ |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / callbacks / logging_callback.py |
LoggingCallback._format_value |
|
7 |
7 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / callbacks / logging_callback.py |
LoggingCallback._format_message |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / callbacks / logging_callback.py |
LoggingCallback._write |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / callbacks / logging_callback.py |
LoggingCallback.on_step_end |
|
8 |
8 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / callbacks / logging_callback.py |
(no function) |
|
12 |
12 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / callbacks / profiling_callback.py |
ProfilingCallback.__init__ |
|
13 |
13 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / callbacks / profiling_callback.py |
ProfilingCallback.on_train_begin |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / callbacks / profiling_callback.py |
ProfilingCallback.on_step_end |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / callbacks / profiling_callback.py |
ProfilingCallback.on_train_end |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / callbacks / profiling_callback.py |
(no function) |
|
8 |
8 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / callbacks / tqdm_callback.py |
TqdmCallback.__init__ |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / callbacks / tqdm_callback.py |
TqdmCallback._to_float |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / callbacks / tqdm_callback.py |
TqdmCallback._format_metric |
|
10 |
10 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / callbacks / tqdm_callback.py |
TqdmCallback._create_progress_bar |
|
8 |
8 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / callbacks / tqdm_callback.py |
TqdmCallback._postfix |
|
13 |
13 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / callbacks / tqdm_callback.py |
TqdmCallback.on_train_begin |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / callbacks / tqdm_callback.py |
TqdmCallback.on_step_end |
|
8 |
8 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / callbacks / tqdm_callback.py |
TqdmCallback.write |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / callbacks / tqdm_callback.py |
TqdmCallback.on_train_end |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / callbacks / tqdm_callback.py |
(no function) |
|
18 |
18 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / config / __init__.py |
(no function) |
|
9 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / config / data.py |
DatasetConfig.build |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / config / data.py |
DatasetConfig.__getattr__ |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / config / data.py |
DatasetConfig.to_dict |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / config / data.py |
DataLoaderConfig.build |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / config / data.py |
DataLoaderConfig.__getattr__ |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / config / data.py |
DataLoaderConfig.to_dict |
|
7 |
7 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / config / data.py |
(no function) |
|
26 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / config / manager.py |
_parse_override_tokens |
|
12 |
9 |
0 |
|
25% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / config / manager.py |
_parse_cli_scalar |
|
19 |
19 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / config / manager.py |
_target_hints |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / config / manager.py |
_replace_target_path |
|
25 |
25 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / config / manager.py |
_replace_path |
|
35 |
35 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / config / manager.py |
_apply_typed_overrides |
|
8 |
5 |
0 |
|
38% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / config / manager.py |
_load_training_config |
|
9 |
4 |
0 |
|
56% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / config / manager.py |
parse_training_args |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / config / manager.py |
(no function) |
|
22 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / config / optimization.py |
OptimizerConfig.to_dict |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / config / optimization.py |
(no function) |
|
12 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / config / parallelism.py |
ActivationCheckpointConfig.__post_init__ |
|
2 |
1 |
0 |
|
50% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / config / parallelism.py |
PlanOverride.to_override |
|
10 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / config / parallelism.py |
PlanOverride._validate_tp_divide_attrs |
|
13 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / config / parallelism.py |
PlanOverride._parse_inner_out_src |
|
14 |
1 |
0 |
|
93% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / config / parallelism.py |
PlanOverride._parse_contract_field |
|
12 |
2 |
0 |
|
83% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / config / parallelism.py |
_import_module_type |
|
21 |
5 |
0 |
|
76% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / config / parallelism.py |
_target_replacement_factory |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / config / parallelism.py |
_target_replacement_factory.factory |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / config / parallelism.py |
entries_to_module_replacements |
|
15 |
4 |
0 |
|
73% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / config / parallelism.py |
_has_sharding_action |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / config / parallelism.py |
_validate_when |
|
2 |
1 |
0 |
|
50% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / config / parallelism.py |
entries_to_plan_overrides |
|
25 |
3 |
0 |
|
88% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / config / parallelism.py |
normalize_distributed_setup_overrides |
|
7 |
7 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / config / parallelism.py |
(no function) |
|
54 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / config / resolver.py |
_fail |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / config / resolver.py |
import_target |
|
23 |
8 |
0 |
|
65% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / config / resolver.py |
_is_union |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / config / resolver.py |
_type_name |
|
5 |
2 |
0 |
|
60% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / config / resolver.py |
_normalize_list |
|
3 |
1 |
0 |
|
67% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / config / resolver.py |
_normalize_tuple |
|
9 |
9 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / config / resolver.py |
_coerce_none |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / config / resolver.py |
_coerce_union |
|
10 |
4 |
0 |
|
60% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / config / resolver.py |
_coerce_scalar |
|
9 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / config / resolver.py |
_coerce_literal |
|
9 |
6 |
0 |
|
33% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / config / resolver.py |
coerce_value |
|
29 |
12 |
0 |
|
59% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / config / resolver.py |
_resolve_union |
|
10 |
1 |
0 |
|
90% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / config / resolver.py |
_resolve_dataclass |
|
18 |
7 |
0 |
|
61% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / config / resolver.py |
_target_hints |
|
5 |
2 |
0 |
|
60% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / config / resolver.py |
_normalize_target_args |
|
19 |
5 |
0 |
|
74% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / config / resolver.py |
_resolve_target |
|
16 |
5 |
0 |
|
69% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / config / resolver.py |
_resolve_dataloader_config |
|
12 |
12 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / config / resolver.py |
_resolve_dataset_config |
|
9 |
9 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / config / resolver.py |
_resolve_optimizer_config |
|
5 |
1 |
0 |
|
80% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / config / resolver.py |
resolve_component |
|
16 |
4 |
0 |
|
75% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / config / resolver.py |
resolve_root |
|
15 |
5 |
0 |
|
67% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / config / resolver.py |
(no function) |
|
35 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / config / target.py |
_serialize_config_value |
|
11 |
11 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / config / target.py |
Target.__init__ |
|
7 |
2 |
0 |
|
71% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / config / target.py |
Target.__getattr__ |
|
5 |
2 |
0 |
|
60% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / config / target.py |
Target.build |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / config / target.py |
Target.replace |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / config / target.py |
Target.to_dict |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / config / target.py |
(no function) |
|
11 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / config / trainer.py |
TrainerConfig.__post_init__ |
|
5 |
2 |
0 |
|
60% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / config / trainer.py |
TrainerConfig.to_dict |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / config / trainer.py |
save_configs |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / config / trainer.py |
(no function) |
|
36 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / config / training.py |
(no function) |
|
43 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / runtime / __init__.py |
(no function) |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / runtime / data_iterator.py |
BackgroundPrefetcher.__init__ |
|
9 |
9 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / runtime / data_iterator.py |
BackgroundPrefetcher._worker |
|
11 |
11 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / runtime / data_iterator.py |
BackgroundPrefetcher.__iter__ |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / runtime / data_iterator.py |
BackgroundPrefetcher.__next__ |
|
14 |
14 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / runtime / data_iterator.py |
BackgroundPrefetcher.state_dict |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / runtime / data_iterator.py |
BackgroundPrefetcher.stop |
|
10 |
10 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / runtime / data_iterator.py |
HyperIter.__init__ |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / runtime / data_iterator.py |
HyperIter.__iter__ |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / runtime / data_iterator.py |
HyperIter.__next__ |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / runtime / data_iterator.py |
HyperIter.stop |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / runtime / data_iterator.py |
HyperIter.state_dict |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / runtime / data_iterator.py |
(no function) |
|
19 |
19 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / runtime / device.py |
get_device_name |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / runtime / device.py |
get_dist_comm_backend |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / runtime / device.py |
synchronize |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / runtime / device.py |
stream_synchronize |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / runtime / device.py |
empty_cache |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / runtime / device.py |
set_device |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / runtime / device.py |
get_device_rng_state |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / runtime / device.py |
set_device_rng_state |
|
7 |
7 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / runtime / device.py |
is_nccl_backend |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / runtime / device.py |
is_hccl_backend |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / runtime / device.py |
get_gpu_compute_capability |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / runtime / device.py |
is_sm90_or_above |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / runtime / device.py |
get_compute_units |
|
10 |
10 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / runtime / device.py |
(no function) |
|
19 |
19 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / runtime / distributed.py |
all_gather |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / runtime / distributed.py |
all_reduce |
|
11 |
11 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / runtime / distributed.py |
get_world_size_safe |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / runtime / distributed.py |
get_global_rank_safe |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / runtime / distributed.py |
get_local_rank_safe |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / runtime / distributed.py |
initialize_distributed |
|
14 |
14 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / runtime / distributed.py |
create_distributed_setup_from_config |
|
39 |
39 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / runtime / distributed.py |
create_distributed_setup_from_config._local_rank |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / runtime / distributed.py |
destroy_process_group |
|
17 |
17 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / runtime / distributed.py |
(no function) |
|
17 |
17 |
2 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / runtime / fsdp.py |
model_reshard |
|
7 |
7 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / runtime / fsdp.py |
configure_fsdp_gradient_sync |
|
9 |
9 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / runtime / fsdp.py |
(no function) |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / runtime / logging.py |
_is_rank_0 |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / runtime / logging.py |
setup_logging |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / runtime / logging.py |
_info_rank0 |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / runtime / logging.py |
create_logger |
|
8 |
8 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / runtime / logging.py |
enable_third_party_logging |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / runtime / logging.py |
disable_warning |
|
7 |
7 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / runtime / logging.py |
(no function) |
|
16 |
16 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / runtime / loss_aggregation.py |
count_loss_token |
|
8 |
8 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / runtime / loss_aggregation.py |
count_loss_token._count |
|
29 |
29 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / runtime / loss_aggregation.py |
(no function) |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / runtime / memory.py |
print_device_mem_info |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / runtime / memory.py |
print_cpu_memory_info |
|
7 |
7 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / runtime / memory.py |
empty_cache |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / runtime / memory.py |
(no function) |
|
8 |
8 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / runtime / metrics.py |
mean_global_loss |
|
31 |
31 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / runtime / metrics.py |
(no function) |
|
7 |
7 |
2 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / runtime / profiling.py |
get_cache_dir |
|
7 |
7 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / runtime / profiling.py |
ProfilerWithMem.__init__ |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / runtime / profiling.py |
ProfilerWithMem.__enter__ |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / runtime / profiling.py |
ProfilerWithMem.__exit__ |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / runtime / profiling.py |
ProfilerWithMem.start |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / runtime / profiling.py |
ProfilerWithMem.stop |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / runtime / profiling.py |
ProfilerWithMem.step |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / runtime / profiling.py |
create_profiler |
|
22 |
22 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / runtime / profiling.py |
create_profiler.handler_fn |
|
25 |
25 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / runtime / profiling.py |
(no function) |
|
20 |
20 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / runtime / random.py |
enable_high_precision_for_bf16 |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / runtime / random.py |
enable_full_determinism |
|
19 |
19 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / runtime / random.py |
set_seed |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / runtime / random.py |
(no function) |
|
12 |
12 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / state.py |
(no function) |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / text_trainer.py |
TextTrainer.__init__ |
|
16 |
16 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / text_trainer.py |
TextTrainer._build_model_assets |
|
22 |
22 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / text_trainer.py |
TextTrainer._build_data_transform |
|
7 |
7 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / text_trainer.py |
TextTrainer._build_collate_fn |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / text_trainer.py |
TextTrainer._build_get_batch |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / text_trainer.py |
TextTrainer.distributed_setup |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / text_trainer.py |
TextTrainer.mesh |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / text_trainer.py |
TextTrainer.dp_cp_mesh |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / text_trainer.py |
TextTrainer.on_train_begin |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / text_trainer.py |
TextTrainer.on_train_end |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / text_trainer.py |
TextTrainer.on_epoch_begin |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / text_trainer.py |
TextTrainer.on_epoch_end |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / text_trainer.py |
TextTrainer.on_step_begin |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / text_trainer.py |
TextTrainer.on_step_end |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / text_trainer.py |
TextTrainer.forward_backward_step |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / text_trainer.py |
TextTrainer.train_step |
|
28 |
28 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / text_trainer.py |
TextTrainer.train |
|
26 |
26 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / text_trainer.py |
(no function) |
|
37 |
37 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / vlm_trainer.py |
VLMTrainer.__init__ |
|
16 |
16 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / vlm_trainer.py |
VLMTrainer._build_model_assets |
|
12 |
12 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / vlm_trainer.py |
VLMTrainer._build_data_transform |
|
7 |
7 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / vlm_trainer.py |
VLMTrainer._build_collate_fn |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / vlm_trainer.py |
VLMTrainer._build_get_batch |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / vlm_trainer.py |
VLMTrainer.distributed_setup |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / vlm_trainer.py |
VLMTrainer.mesh |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / vlm_trainer.py |
VLMTrainer.dp_cp_mesh |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / vlm_trainer.py |
VLMTrainer.on_train_begin |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / vlm_trainer.py |
VLMTrainer.on_train_end |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / vlm_trainer.py |
VLMTrainer.on_epoch_begin |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / vlm_trainer.py |
VLMTrainer.on_epoch_end |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / vlm_trainer.py |
VLMTrainer.on_step_begin |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / vlm_trainer.py |
VLMTrainer.on_step_end |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / vlm_trainer.py |
VLMTrainer.train_step |
|
32 |
32 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / vlm_trainer.py |
VLMTrainer.train |
|
25 |
25 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / vlm_trainer.py |
(no function) |
|
35 |
35 |
0 |
|
0% |