| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / __init__.py |
(no function) |
|
19 |
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 |
init_process_group |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / collectives / cc.py |
destroy_process_group |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / collectives / cc.py |
get_process_group_ranks |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / collectives / cc.py |
get_backend |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / collectives / cc.py |
split_group |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / collectives / cc.py |
get_group_local_rank |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / collectives / cc.py |
mark_created_groups |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / collectives / cc.py |
(no function) |
|
11 |
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 / context_parallel / __init__.py |
(no function) |
|
3 |
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 |
9 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / async_context_parallel.py |
_a2a_reconstruct |
|
7 |
7 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / async_context_parallel.py |
AsyncContextParallel.__init__ |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / async_context_parallel.py |
AsyncContextParallel.apply |
|
25 |
25 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / async_context_parallel.py |
AsyncContextParallel._register_proj_hooks |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / async_context_parallel.py |
AsyncContextParallel._proj_post_hook |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / async_context_parallel.py |
AsyncContextParallel._wait_a2a |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / async_context_parallel.py |
AsyncContextParallel._attn_pre_hook_ulysses |
|
7 |
7 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / async_context_parallel.py |
AsyncContextParallel._attn_pre_hook_ulysses._local |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / async_context_parallel.py |
AsyncContextParallel._attn_pre_hook_hybrid |
|
14 |
14 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / async_context_parallel.py |
AsyncContextParallel._attn_pre_hook_hybrid._local |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / async_context_parallel.py |
AsyncContextParallel._attn_pre_hook_hybrid._local_dt |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / async_context_parallel.py |
AsyncContextParallel._attn_post_hook_ata |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / async_context_parallel.py |
AsyncContextParallel._attn_post_hook_ata._process |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / async_context_parallel.py |
AsyncContextParallel._proj_bwd_pre_hook |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / async_context_parallel.py |
(no function) |
|
22 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / context_parallel.py |
_ensure_1d |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / context_parallel.py |
_build_2d_mesh |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / context_parallel.py |
_scatter_seq_to_head |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / context_parallel.py |
_gather_head_to_seq |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / context_parallel.py |
_gather_seq |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / context_parallel.py |
ContextParallel.__init__ |
|
8 |
8 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / context_parallel.py |
ContextParallel.apply |
|
22 |
22 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / context_parallel.py |
ContextParallel._pre_hook_colossal |
|
16 |
16 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / context_parallel.py |
ContextParallel._pre_hook_ulysses |
|
9 |
9 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / context_parallel.py |
ContextParallel._ata_scatter_to_2d |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / context_parallel.py |
ContextParallel._pre_hook_hybrid |
|
17 |
17 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / context_parallel.py |
ContextParallel._post_hook_ata |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / context_parallel.py |
ContextParallel._post_hook_ata._process |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / context_parallel.py |
ContextParallel._post_hook_colossal |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / context_parallel.py |
ContextParallel._post_hook_colossal._process |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / context_parallel.py |
ContextParallel._apply_lb_colossal |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / context_parallel / context_parallel.py |
ContextParallel._lb_colossal_forward |
|
18 |
18 |
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) |
|
26 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / __init__.py |
(no function) |
|
11 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / api.py |
_gather_from_all_ranks |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / api.py |
_save_impl |
|
34 |
34 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / api.py |
_async_persist_worker |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / api.py |
_async_persist_wait_process |
|
13 |
13 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / api.py |
save |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / api.py |
async_save |
|
8 |
8 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / api.py |
load |
|
33 |
33 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / api.py |
(no function) |
|
30 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / async_staging.py |
_tensor_to_staging_cpu |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / async_staging.py |
_stage_leaf |
|
9 |
9 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / async_staging.py |
build_staged_state_dict |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / async_staging.py |
(no function) |
|
10 |
0 |
0 |
|
100% |
| / 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 |
1 |
0 |
|
93% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / filesystem_storage.py |
FileSystemWriter._collect_tensors |
|
9 |
5 |
0 |
|
44% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / filesystem_storage.py |
FileSystemWriter._write_tensors |
|
12 |
10 |
0 |
|
17% |
| / 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 |
|
12 |
1 |
0 |
|
92% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / filesystem_storage.py |
_copy_tensor_to_target |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / filesystem_storage.py |
_load_bytes_file |
|
8 |
1 |
0 |
|
88% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / filesystem_storage.py |
_get_tensor_size |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / filesystem_storage.py |
_load_tensor_file |
|
37 |
37 |
0 |
|
0% |
| / 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 |
2 |
0 |
|
78% |
| / 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.execute_read |
|
7 |
2 |
0 |
|
71% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / filesystem_storage.py |
(no function) |
|
37 |
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 |
9 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / layout.py |
get_global_layout |
|
10 |
10 |
0 |
|
0% |
| / 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 |
(no function) |
|
29 |
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 |
|
14 |
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 |
|
21 |
21 |
0 |
|
0% |
| / 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) |
|
62 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / reshard.py |
check_layout |
|
14 |
14 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / reshard.py |
check_layout.check_type_is_sequence |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / reshard.py |
rank_id_to_dev_id_list |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / reshard.py |
infer_intersection |
|
13 |
13 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / reshard.py |
infer_intersection.is_valid_axis_list |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / reshard.py |
infer_slice_area_by_rank |
|
25 |
25 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / reshard.py |
infer_slice_area_by_rank._get_dev_num_along_dim |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / reshard.py |
ReshardHandler.__init__ |
|
32 |
32 |
0 |
|
0% |
| / 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 |
11 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / reshard.py |
ReshardHandler.get_real_tensor |
|
17 |
17 |
0 |
|
0% |
| / 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 |
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 |
3 |
0 |
|
80% |
| / 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 |
|
32 |
19 |
0 |
|
41% |
| / 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 |
|
39 |
17 |
0 |
|
56% |
| / 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_result |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / standard_planner.py |
StandardSavePlanner.cache_result |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / standard_planner.py |
StandardSavePlanner.get_data |
|
15 |
8 |
0 |
|
47% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / standard_planner.py |
create_read_items_for_chunk_list |
|
16 |
16 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / standard_planner.py |
StandardLoadPlanner.__init__ |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / standard_planner.py |
StandardLoadPlanner.configure_planner |
|
9 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / standard_planner.py |
StandardLoadPlanner.build_local_plan |
|
26 |
17 |
0 |
|
35% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / standard_planner.py |
StandardLoadPlanner.build_global_plan |
|
1 |
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 |
|
8 |
8 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / standard_planner.py |
StandardLoadPlanner.apply_tensor |
|
8 |
8 |
0 |
|
0% |
| / 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 |
3 |
0 |
|
75% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / standard_planner.py |
(no function) |
|
40 |
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) |
|
42 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / util.py |
check_path |
|
6 |
1 |
0 |
|
83% |
| / 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 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / util.py |
chunk_to_area |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / util.py |
create_chunk_list_for_tensor |
|
24 |
24 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / util.py |
remove_redundant_plans |
|
28 |
28 |
0 |
|
0% |
| / 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 |
8 |
0 |
|
27% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / util.py |
traverse_state_dict._traverse_obj |
|
8 |
5 |
0 |
|
38% |
| / 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 |
1 |
0 |
|
80% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / util.py |
set_element |
|
18 |
11 |
0 |
|
39% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / util.py |
set_element.extend_list |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / distributed_checkpoint / util.py |
(no function) |
|
21 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / __init__.py |
(no function) |
|
0 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / _mesh_layout.py |
_flatten_tuple |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / _mesh_layout.py |
_contiguous_strides |
|
6 |
1 |
0 |
|
83% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / _mesh_layout.py |
_FlatLayout.__init__ |
|
23 |
2 |
0 |
|
91% |
| / 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 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / _mesh_layout.py |
_FlatLayout.check_orthogonal |
|
4 |
2 |
0 |
|
50% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / _mesh_layout.py |
_FlatLayout._offset_for_linear_index |
|
11 |
2 |
0 |
|
82% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / _mesh_layout.py |
_FlatLayout.all_ranks_from_zero |
|
3 |
1 |
0 |
|
67% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / _mesh_layout.py |
_MeshLayout.__init__ |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / _mesh_layout.py |
_MeshLayout.from_sizes_strides |
|
5 |
1 |
0 |
|
80% |
| / 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.__getitem__ |
|
3 |
1 |
0 |
|
67% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / _mesh_layout.py |
_MeshLayout.__iter__ |
|
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.numel |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / _mesh_layout.py |
_MeshLayout.cosize |
|
1 |
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.splice |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / _mesh_layout.py |
_MeshLayout.remap_to_numpy |
|
13 |
2 |
0 |
|
85% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / _mesh_layout.py |
(no function) |
|
34 |
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 |
DeviceMesh.__init__ |
|
13 |
1 |
0 |
|
92% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / device_mesh.py |
DeviceMesh._validate_device_type |
|
4 |
1 |
0 |
|
75% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / device_mesh.py |
DeviceMesh._resolve_mesh |
|
19 |
11 |
0 |
|
42% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / device_mesh.py |
DeviceMesh._set_layout |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / device_mesh.py |
DeviceMesh._init_shape_attributes |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / device_mesh.py |
DeviceMesh._init_mesh_dim_names |
|
13 |
4 |
0 |
|
69% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / device_mesh.py |
DeviceMesh._init_instance_state |
|
9 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / device_mesh.py |
DeviceMesh._compute_coordinate_on_dim |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / device_mesh.py |
DeviceMesh._compute_coordinates_from_mesh |
|
7 |
1 |
0 |
|
86% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / device_mesh.py |
DeviceMesh.size |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / device_mesh.py |
DeviceMesh.get_coordinate |
|
1 |
1 |
0 |
|
0% |
| / 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__ |
|
1 |
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 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / device_mesh.py |
DeviceMesh._init_process_groups |
|
14 |
5 |
0 |
|
64% |
| / 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 |
1 |
0 |
|
0% |
| / 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.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 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / device_mesh.py |
DeviceMesh.__getitem__ |
|
9 |
2 |
0 |
|
78% |
| / 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._validate_getitem_dimensions |
|
10 |
3 |
0 |
|
70% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / device_mesh.py |
DeviceMesh._get_or_create_original_sub_mesh |
|
8 |
3 |
0 |
|
62% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / device_mesh.py |
DeviceMesh._create_and_cache_sub_mesh |
|
18 |
4 |
0 |
|
78% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / device_mesh.py |
DeviceMesh.get_group |
|
9 |
9 |
0 |
|
0% |
| / 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 |
|
27 |
27 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / device_mesh.py |
DeviceMesh.get_local_rank |
|
20 |
20 |
0 |
|
0% |
| / 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._get_root_mesh |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / device_mesh.py |
DeviceMesh.concatenate |
|
20 |
4 |
0 |
|
80% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / device_mesh.py |
DeviceMesh._concatenate |
|
28 |
6 |
0 |
|
79% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / device_mesh.py |
DeviceMesh._create_flatten_mesh |
|
24 |
24 |
0 |
|
0% |
| / 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 |
29 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / device_mesh.py |
DeviceMesh.get_global_shape |
|
27 |
6 |
0 |
|
78% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / device_mesh.py |
DeviceMesh.get_comm_group_by_axis |
|
14 |
14 |
0 |
|
0% |
| / 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 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / device_mesh.py |
DeviceMesh.__str__ |
|
1 |
1 |
0 |
|
0% |
| / 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 |
_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 |
|
14 |
3 |
0 |
|
79% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / device_mesh.py |
(no function) |
|
94 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / dtensor.py |
SkipDTensorDispatch.__init__ |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / dtensor.py |
SkipDTensorDispatch.__enter__ |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / dtensor.py |
SkipDTensorDispatch.__exit__ |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / dtensor.py |
_is_alias_placements |
|
9 |
3 |
0 |
|
67% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / dtensor.py |
_build_layout |
|
14 |
3 |
0 |
|
79% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / dtensor.py |
DTensor.__init_data__ |
|
4 |
0 |
0 |
|
100% |
| / 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 |
|
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.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.to_local |
|
1 |
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 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / dtensor.py |
DTensor.local_shape |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / dtensor.py |
DTensor.redistribute |
|
4 |
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 |
10 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / dtensor.py |
distribute_tensor |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / dtensor.py |
_distribute_module_param_source |
|
3 |
1 |
0 |
|
67% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / dtensor.py |
_distribute_module_new_parameter |
|
3 |
1 |
0 |
|
67% |
| / 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 |
_replicate_submodule_params_buffers |
|
13 |
1 |
0 |
|
92% |
| / 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 |
|
6 |
2 |
0 |
|
67% |
| / 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 |
(no function) |
|
63 |
0 |
1 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / init_weights.py |
init_empty_weights |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / init_weights.py |
init_on_device |
|
2 |
2 |
0 |
|
0% |
| / 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 |
_infer_slice_area_by_rank |
|
24 |
24 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / layout.py |
_infer_slice_area_by_rank._get_dev_num_alone_dim |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / layout.py |
_infer_slice_area_by_rank._rank_id_to_dev_id_list |
|
6 |
6 |
0 |
|
0% |
| / 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 |
|
10 |
10 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / layout.py |
Layout.__init__ |
|
12 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / layout.py |
Layout.from_device_mesh |
|
11 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / layout.py |
Layout.__call__ |
|
8 |
1 |
0 |
|
88% |
| / 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 |
11 |
0 |
|
65% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / layout.py |
Layout.to_dict |
|
6 |
2 |
0 |
|
67% |
| / 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 |
1 |
0 |
|
83% |
| / 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 |
1 |
0 |
|
80% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / layout.py |
Layout.tensor_map_to_placement |
|
19 |
1 |
0 |
|
95% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / layout.py |
Layout.__setstate__ |
|
2 |
2 |
0 |
|
0% |
| / 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 |
1 |
0 |
|
0% |
| / 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 |
1 |
0 |
|
0% |
| / 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 |
|
3 |
1 |
0 |
|
67% |
| / 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 |
|
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 |
1 |
0 |
|
86% |
| / 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 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / layout.py |
Layout.reset_partial |
|
3 |
3 |
0 |
|
0% |
| / 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_global_shape |
|
1 |
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 |
13 |
0 |
|
0% |
| / 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 |
3 |
0 |
|
81% |
| / 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__ |
|
7 |
2 |
0 |
|
71% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / layout.py |
(no function) |
|
80 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / parameter_init.py |
init_parameters |
|
3 |
3 |
0 |
|
0% |
| / 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 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / placement_types.py |
Shard.__init__ |
|
2 |
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.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__ |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / placement_types.py |
Shard.__str__ |
|
1 |
1 |
0 |
|
0% |
| / 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__ |
|
1 |
1 |
0 |
|
0% |
| / 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 |
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 |
1 |
0 |
|
0% |
| / 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 |
1 |
0 |
|
67% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / placement_types.py |
Partial.__eq__ |
|
3 |
3 |
0 |
|
0% |
| / 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 |
2 |
0 |
|
0% |
| / 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) |
|
37 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / random.py |
is_rng_supported_mesh |
|
4 |
4 |
0 |
|
0% |
| / 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._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 |
|
7 |
7 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / random.py |
_calc_shard_info |
|
27 |
27 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / random.py |
_calc_shard_linear_idx |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / random.py |
_resolve_device |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / random.py |
_resolve_device.get_device |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / random.py |
local_shard_size_and_offset |
|
10 |
10 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / random.py |
fork_rng |
|
19 |
19 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / random.py |
(no function) |
|
53 |
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 |
8 |
0 |
|
0% |
| / 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 |
1 |
0 |
|
67% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / redistribute_infer.py |
DevMat._get_combined_size |
|
7 |
7 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / redistribute_infer.py |
DevMat._get_devices_along_dim |
|
20 |
17 |
0 |
|
15% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / redistribute_infer.py |
DevMat.get_devices_along_dim |
|
15 |
13 |
0 |
|
13% |
| / 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 |
10 |
0 |
|
60% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / redistribute_infer.py |
RedistributionOperatorInfer._handle_simple_split_case |
|
11 |
1 |
0 |
|
91% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / redistribute_infer.py |
RedistributionOperatorInfer._handle_tuple_split_case |
|
12 |
10 |
0 |
|
17% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / redistribute_infer.py |
RedistributionOperatorInfer.infer_split_by_axis |
|
16 |
3 |
0 |
|
81% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / redistribute_infer.py |
RedistributionOperatorInfer._handle_none_dim_permute_case |
|
24 |
9 |
0 |
|
62% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / redistribute_infer.py |
RedistributionOperatorInfer._handle_none_dim_tuple_permute_case |
|
29 |
27 |
0 |
|
7% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / redistribute_infer.py |
RedistributionOperatorInfer._handle_tuple_dim_permute_case |
|
29 |
27 |
0 |
|
7% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / redistribute_infer.py |
RedistributionOperatorInfer.infer_permute_by_axis |
|
16 |
3 |
0 |
|
81% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / redistribute_infer.py |
RedistributionOperatorInfer._handle_tuple_concat_case |
|
13 |
11 |
0 |
|
15% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / redistribute_infer.py |
RedistributionOperatorInfer._handle_simple_concat_case |
|
12 |
4 |
0 |
|
67% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / redistribute_infer.py |
RedistributionOperatorInfer.infer_concat_by_axis |
|
11 |
2 |
0 |
|
82% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / redistribute_infer.py |
RedistributionOperatorInfer.infer_ops_list |
|
28 |
3 |
0 |
|
89% |
| / 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 |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / tensor_redistribution.py |
TensorRedistribution._construct_strided_slice |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / tensor_redistribution.py |
TensorRedistribution._construct_all_concat_new |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / tensor_redistribution.py |
TensorRedistribution._construct_all_split |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / tensor_redistribution.py |
TensorRedistribution._construct_all_to_all |
|
51 |
45 |
0 |
|
12% |
| / 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.redistribution |
|
36 |
13 |
0 |
|
64% |
| / 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 |
|
16 |
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 |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / tensor_redistribution.py |
TensorRedistribution.reduce_partial |
|
42 |
33 |
0 |
|
21% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / dtensor / tensor_redistribution.py |
(no function) |
|
23 |
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 |
_generate_permute_indices |
|
20 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / expert_parallel / expert_parallel.py |
_permute |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / expert_parallel / expert_parallel.py |
_unpermute |
|
5 |
1 |
0 |
|
80% |
| / 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 |
ExpertParallel.__init__ |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / expert_parallel / expert_parallel.py |
ExpertParallel._partition_fn |
|
9 |
1 |
0 |
|
89% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / expert_parallel / expert_parallel.py |
ExpertParallel._token_dispatch |
|
13 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / expert_parallel / expert_parallel.py |
ExpertParallel._token_combine |
|
5 |
0 |
0 |
|
100% |
| / 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._partition_fn |
|
10 |
1 |
0 |
|
90% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / expert_parallel / expert_parallel.py |
ExpertTensorParallel._token_dispatch |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / expert_parallel / expert_parallel.py |
ExpertTensorParallel._token_combine |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / expert_parallel / expert_parallel.py |
(no function) |
|
33 |
0 |
0 |
|
100% |
| / 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 |
9 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / api.py |
_check_strict_keys |
|
10 |
10 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / api.py |
_resolve_local_tensor |
|
11 |
11 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / api.py |
_UnshardHandle.__init__ |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / api.py |
_UnshardHandle.wait |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / api.py |
HSDPModule.__init__ |
|
1 |
1 |
0 |
|
0% |
| / 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 |
7 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / api.py |
HSDPModule.zero_grad |
|
7 |
7 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / api.py |
HSDPModule.set_modules_to_forward_prefetch |
|
8 |
3 |
0 |
|
62% |
| / 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 |
3 |
0 |
|
70% |
| / 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.set_requires_all_reduce |
|
9 |
9 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / api.py |
HSDPModule.set_reshard_after_forward |
|
9 |
9 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / api.py |
HSDPModule.set_reshard_after_backward |
|
9 |
9 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / api.py |
HSDPModule.set_reduce_op_type |
|
2 |
2 |
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 |
1 |
0 |
|
67% |
| / 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 |
_check_hsdp_input_valid |
|
24 |
24 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / api.py |
_get_device_from_mesh |
|
13 |
13 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / api.py |
_normalize_replicate_params |
|
7 |
5 |
0 |
|
29% |
| / 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 |
fully_shard |
|
28 |
6 |
0 |
|
79% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / api.py |
get_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 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / api.py |
(no function) |
|
43 |
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 |
1 |
2 |
|
90% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / hsdp_param.py |
_build_group_info_from_process_group |
|
3 |
1 |
0 |
|
67% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / hsdp_param.py |
HSDPParamV2.__init__ |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / hsdp_param.py |
HSDPParamV2._init_sharded_param |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / hsdp_param.py |
HSDPParamV2.init_dtype_attrs |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / hsdp_param.py |
HSDPParamV2.init_all_gather_outputs |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / hsdp_param.py |
HSDPParamV2.init_unsharded_param |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / hsdp_param.py |
HSDPParamV2.to_sharded |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / hsdp_param.py |
HSDPParamV2.to_unsharded |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / hsdp_param.py |
HSDPParamV2.to_sharded_dtensor |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / hsdp_param.py |
HSDPParamV2.to_accumulated_grad_if_needed |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / hsdp_param.py |
HSDPParamV2.accumulate_unsharded_grad_if_needed |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / hsdp_param.py |
HSDPParamV2.alloc_all_gather_outputs |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / hsdp_param.py |
HSDPParamV2.free_unsharded_param |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / hsdp_param.py |
HSDPParamV2.all_gather_inputs |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / hsdp_param.py |
HSDPParamV2.unsharded_param |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / hsdp_param.py |
HSDPParamV2.unsharded_grad_data |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / hsdp_param.py |
HSDPParamV2.unsharded_accumulated_grad_data |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / hsdp_param.py |
HSDPParamV2._sharded_local_tensor |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / hsdp_param.py |
HSDPParamV2._get_unsharded_param_data |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / hsdp_param.py |
HSDPParamV2.unshard |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / hsdp_param.py |
HSDPParamV2.wait_for_unshard |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / hsdp_param.py |
HSDPParamV2.shard |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / hsdp_param.py |
HSDPParamV2.reduce_scatter_grad |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / hsdp_param.py |
HSDPParamV2.all_reduce_grad |
|
1 |
1 |
0 |
|
0% |
| / 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 |
2 |
0 |
|
78% |
| / 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 |
2 |
0 |
|
78% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / hsdp_param.py |
HSDPParamV2._init_group_infos |
|
8 |
1 |
0 |
|
88% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / hsdp_param.py |
HSDPParamV2._build_layout_driven_group_info |
|
29 |
6 |
0 |
|
79% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / hsdp_param.py |
HSDPParamV2._normalize_unsharded_grad_to_local |
|
9 |
1 |
0 |
|
89% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / hsdp_param.py |
(no function) |
|
46 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / hsdp_scheduler.py |
HSDPSchedulerContext.__init__ |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / hsdp_scheduler.py |
HSDPSchedulerV2.__init__ |
|
21 |
21 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / hsdp_scheduler.py |
HSDPSchedulerV2._init_platform |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / hsdp_scheduler.py |
HSDPSchedulerV2._new_cell_state |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / hsdp_scheduler.py |
HSDPSchedulerV2._register_hooks |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / hsdp_scheduler.py |
HSDPSchedulerV2._register_forward_backward_hooks |
|
1 |
1 |
0 |
|
0% |
| / 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 |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / hsdp_scheduler.py |
HSDPSchedulerV2.set_reshard_after_backward |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / hsdp_scheduler.py |
HSDPSchedulerV2.set_requires_all_reduce |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / hsdp_scheduler.py |
HSDPSchedulerV2.set_requires_grad_sync |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / hsdp_scheduler.py |
HSDPSchedulerV2._hsdp_forward_pre_hook |
|
31 |
18 |
0 |
|
42% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / hsdp_scheduler.py |
HSDPSchedulerV2._lazy_init_all_states |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / hsdp_scheduler.py |
HSDPSchedulerV2._init_params_fqn |
|
19 |
2 |
0 |
|
89% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / hsdp_scheduler.py |
HSDPSchedulerV2._hsdp_forward_hook |
|
9 |
9 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / hsdp_scheduler.py |
HSDPSchedulerV2._hsdp_backward_pre_hook |
|
7 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / hsdp_scheduler.py |
HSDPSchedulerV2._hsdp_backward_hook |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / hsdp_scheduler.py |
HSDPSchedulerV2._grouped_forward_pre_hook_skip |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / hsdp_scheduler.py |
HSDPSchedulerV2._grouped_forward_post_hook_skip |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / hsdp_scheduler.py |
HSDPSchedulerV2._grouped_forward_pre_hook |
|
7 |
7 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / hsdp_scheduler.py |
HSDPSchedulerV2._make_grouped_forward_post_hook |
|
2 |
2 |
0 |
|
0% |
| / 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 |
7 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / hsdp_scheduler.py |
HSDPSchedulerV2.set_forward_prefetch_cells |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / hsdp_scheduler.py |
HSDPSchedulerV2.set_backward_prefetch_cells |
|
1 |
1 |
0 |
|
0% |
| / 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 |
1 |
0 |
|
80% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / hsdp_scheduler.py |
(no function) |
|
34 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / hsdp_state.py |
HSDPState.__init__ |
|
15 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / hsdp_state.py |
HSDPState._init_hsdp_params |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / hsdp_state.py |
HSDPState._move_states_to_device |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / hsdp_state.py |
HSDPState.shard |
|
9 |
9 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / hsdp_state.py |
HSDPState.unshard |
|
11 |
11 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / hsdp_state.py |
HSDPState.prefetch |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / hsdp_state.py |
HSDPState.wait_for_unshard |
|
10 |
10 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / hsdp_state.py |
HSDPState._iter_managed_params |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / hsdp_state.py |
(no function) |
|
16 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / hsdp_utils.py |
HSDPConfigV2.__init__ |
|
10 |
10 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / hsdp_utils.py |
_named_parameters_with_duplicates |
|
10 |
3 |
0 |
|
70% |
| / 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 |
2 |
0 |
|
86% |
| / 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 |
2 |
0 |
|
75% |
| / 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 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / hsdp_utils.py |
get_rank_list_for_axes |
|
19 |
2 |
0 |
|
89% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / hsdp_utils.py |
get_split_rank_lists_for_axes |
|
20 |
2 |
0 |
|
90% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / hsdp_utils.py |
get_hsdp_state |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / fully_shard / hsdp_utils.py |
(no function) |
|
44 |
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) |
|
33 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / __init__.py |
(no function) |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / scheduler.py |
MetaStep.__init__ |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / scheduler.py |
MetaStep.micro_index |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / scheduler.py |
MetaStep.stage_index |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / scheduler.py |
MetaStep.type |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / scheduler.py |
MetaStep.__eq__ |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / scheduler.py |
MetaStep.__ne__ |
|
3 |
3 |
0 |
|
0% |
| / 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__ |
|
1 |
1 |
0 |
|
0% |
| / 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 |
PipelineScheduleRuntime.__init__ |
|
15 |
15 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / scheduler.py |
PipelineScheduleRuntime.convert_stages_dict |
|
4 |
4 |
0 |
|
0% |
| / 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._check_stages |
|
8 |
8 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / scheduler.py |
PipelineScheduleRuntime._init_stages |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / scheduler.py |
PipelineScheduleRuntime.run |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / scheduler.py |
PipelineScheduleRuntime.sync_shared_parameters_grad |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / scheduler.py |
PipelineScheduleRuntime.update_losses |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / scheduler.py |
PipelineScheduleRuntime._wait_p2p |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / scheduler.py |
PipelineScheduleRuntime.run_microbatches |
|
48 |
48 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / scheduler.py |
add_send_recv |
|
19 |
19 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / scheduler.py |
add_send_recv._need_com |
|
13 |
13 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / scheduler.py |
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 |
add_send_recv.process_rank_communication |
|
11 |
11 |
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.construct_exec_order |
|
15 |
15 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / scheduler.py |
Schedule1F1B.__init__ |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / scheduler.py |
Schedule1F1B.construct_exec_order |
|
42 |
42 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / scheduler.py |
ScheduleInterleaved1F1B.__init__ |
|
14 |
14 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / scheduler.py |
ScheduleInterleaved1F1B.warmup_ops |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / scheduler.py |
ScheduleInterleaved1F1B.forward_stage_index |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / scheduler.py |
ScheduleInterleaved1F1B.backward_stage_index |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / scheduler.py |
ScheduleInterleaved1F1B.construct_stage_exec_order |
|
39 |
39 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / scheduler.py |
detect_cycle_in_graph |
|
39 |
39 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / scheduler.py |
output_cycle_results |
|
8 |
8 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / scheduler.py |
parse_and_validate |
|
23 |
23 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / scheduler.py |
parse_and_validate.parse_elements |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / scheduler.py |
generate_operations |
|
35 |
35 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / scheduler.py |
generate_operations.stage_to_rank |
|
7 |
7 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / scheduler.py |
generate_operations.find_send_target |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / scheduler.py |
generate_operations.find_recv_source |
|
8 |
8 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / scheduler.py |
validate_pipeline_execution |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / scheduler.py |
(no function) |
|
62 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / stage.py |
SharedParameterInfo.__init__ |
|
7 |
7 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / stage.py |
SharedParameterInfo.parameter |
|
1 |
1 |
0 |
|
0% |
| / 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__ |
|
19 |
19 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / stage.py |
PipelineStage.init |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / stage.py |
PipelineStage._init_pp_group |
|
11 |
11 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / stage.py |
PipelineStage.clear_states |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / stage.py |
PipelineStage._check_shared_parameters |
|
10 |
10 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / stage.py |
PipelineStage._sync_shared_parameters |
|
8 |
8 |
0 |
|
0% |
| / 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 |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / stage.py |
PipelineStage.sync_shared_parameters_grad |
|
10 |
10 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / stage.py |
PipelineStage._check_src_stage |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / stage.py |
PipelineStage._check_dst_stage |
|
5 |
5 |
0 |
|
0% |
| / 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_within_stage_ranks |
|
11 |
11 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / stage.py |
PipelineStage.get_last_stage_sens |
|
14 |
14 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / stage.py |
PipelineStage._construct_forward_recv_info |
|
10 |
10 |
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.exec_fwd_recv_ops |
|
14 |
14 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / stage.py |
PipelineStage._construct_backward_recv_info |
|
8 |
8 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / stage.py |
PipelineStage._extract_meta_from_tensor |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / stage.py |
PipelineStage.exec_fwd_send_ops |
|
18 |
18 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / stage.py |
PipelineStage.exec_bwd_recv_ops |
|
8 |
8 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / stage.py |
PipelineStage.exec_bwd_send_ops |
|
10 |
10 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / stage.py |
(no function) |
|
39 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / utils.py |
BatchDimSpec.__init__ |
|
3 |
3 |
0 |
|
0% |
| / 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 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / utils.py |
BatchDimSpec.from_dict |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / utils.py |
_RecvInfo.__init__ |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / utils.py |
_RecvInfo.global_rank |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / utils.py |
_RecvInfo.buffer |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / utils.py |
_RecvInfo.buffer |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / pipeline_parallel / utils.py |
(no function) |
|
17 |
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 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / _op_dispatch.py |
add_no_skip_ops |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / _op_dispatch.py |
remove_no_skip_ops |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / _op_dispatch.py |
enable_dtensor_dispatch |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / _op_dispatch.py |
disable_dtensor_dispatch |
|
1 |
1 |
0 |
|
0% |
| / 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 |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / _op_dispatch.py |
LayoutCacheKey.__eq__ |
|
3 |
1 |
0 |
|
67% |
| / 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 |
1 |
0 |
|
0% |
| / 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 |
1 |
0 |
|
0% |
| / 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 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / _op_dispatch.py |
OpDispatcher.__init__ |
|
13 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / _op_dispatch.py |
OpDispatcher._setup_paths_from_env |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / _op_dispatch.py |
OpDispatcher._setup_yaml_dir |
|
8 |
2 |
0 |
|
75% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / _op_dispatch.py |
OpDispatcher._extend_sys_path |
|
6 |
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._register_single_distributed_op |
|
19 |
6 |
0 |
|
68% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / _op_dispatch.py |
OpDispatcher._process_args_and_kwargs |
|
43 |
43 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / _op_dispatch.py |
OpDispatcher._with_layout_infer |
|
34 |
34 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / _op_dispatch.py |
OpDispatcher._extract_single_arg_layout |
|
16 |
16 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / _op_dispatch.py |
OpDispatcher._pack_infer_output |
|
7 |
7 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / _op_dispatch.py |
OpDispatcher._with_layout_infer_with_tuple_expand |
|
28 |
28 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / _op_dispatch.py |
OpDispatcher._with_layout_infer_reshape |
|
26 |
26 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / _op_dispatch.py |
OpDispatcher._process_args_and_kwargs_with_shape |
|
55 |
55 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / _op_dispatch.py |
OpDispatcher._with_layout_infer_with_shape |
|
35 |
35 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / _op_dispatch.py |
OpDispatcher._with_layout_infer_slice |
|
34 |
34 |
0 |
|
0% |
| / 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 |
2 |
0 |
|
85% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / _op_dispatch.py |
OpDispatcher._dispatch_random_op |
|
23 |
23 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / _op_dispatch.py |
OpDispatcher._unwrap_args |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / _op_dispatch.py |
OpDispatcher._unwrap_args.unwrap |
|
7 |
2 |
0 |
|
71% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / _op_dispatch.py |
OpDispatcher._should_bypass_dispatch |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / _op_dispatch.py |
OpDispatcher._dispatch_layout_infer |
|
14 |
14 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / _op_dispatch.py |
OpDispatcher._wrap_output |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / _op_dispatch.py |
OpDispatcher._dispatch_new |
|
17 |
17 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / _op_dispatch.py |
OpDispatcher.dispatch |
|
8 |
5 |
0 |
|
38% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / _op_dispatch.py |
(no function) |
|
71 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / api.py |
_has_kwargs |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / api.py |
_get_param_name |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / api.py |
_convert_sharding_plan |
|
20 |
20 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / api.py |
_convert_sharding_plan._is_placement_tuple |
|
15 |
15 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / api.py |
_convert_sharding_plan._to_layout |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / api.py |
_convert_sharding_plan._convert_value |
|
15 |
15 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / api.py |
_convert_sharding_plan._convert_forward_plan |
|
16 |
16 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / api.py |
_parallel_in |
|
19 |
19 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / api.py |
_parallel_in._get_layout |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / api.py |
_parallel_out |
|
18 |
18 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / api.py |
_forward_pre_hook |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / api.py |
_forward_pre_with_kwargs_hook |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / api.py |
_forward_hook |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / api.py |
_forward_with_kwargs_hook |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / api.py |
_register_hook |
|
20 |
20 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / api.py |
_register_hook._register_cell_hook |
|
7 |
7 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / api.py |
_register_hook._set_layouts |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / api.py |
_register_local_tensor_hook |
|
7 |
7 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / api.py |
_register_local_tensor_hook.hook_func |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / api.py |
_register_local_tensor_hook.hook_func._recursive_to_local |
|
8 |
8 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / api.py |
_shard_callable |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / api.py |
_shard_callable._shard_wrapper |
|
8 |
8 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / api.py |
shard_module |
|
43 |
43 |
0 |
|
0% |
| / 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 |
custom_shard |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / custom_shard.py |
custom_shard.wrapped |
|
39 |
39 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / custom_shard.py |
(no function) |
|
9 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / dfunction.py |
_LocalCallable.__init__ |
|
3 |
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 |
ActivationWithAxisDistributedOp.infer_layout |
|
12 |
1 |
0 |
|
92% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_activation_with_axis.py |
ActivationWithAxisDistributedOp.check_layout |
|
14 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_activation_with_axis.py |
(no function) |
|
4 |
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 |
|
27 |
2 |
0 |
|
93% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_argmax_with_value_ops.py |
ArgMaxWithValueDistributedOp.infer_layout.is_shard |
|
10 |
5 |
0 |
|
50% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_argmax_with_value_ops.py |
(no function) |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_argsort.py |
ArgsortDistributedOp.infer_layout |
|
19 |
3 |
0 |
|
84% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_argsort.py |
(no function) |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_atleast_1d.py |
Atleast1DDistributedOp.infer_layout |
|
25 |
4 |
0 |
|
84% |
| / 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) |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_chunk_view.py |
ChunkViewDistributedOp.infer_layout |
|
38 |
4 |
0 |
|
89% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_chunk_view.py |
(no function) |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_concat.py |
ConcatDistributedOp.infer_layout |
|
19 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_concat.py |
(no function) |
|
3 |
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 |
|
13 |
1 |
0 |
|
92% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_conv3d.py |
Conv3dDistributedOp.infer_layout |
|
24 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_conv3d.py |
Conv3dDistributedOp.infer_layout.idx_to_alias |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_conv3d.py |
Conv3dDistributedOp.get_expand_impl |
|
15 |
15 |
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 |
|
8 |
8 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_conv3d.py |
(no function) |
|
8 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_cumsum.py |
CumsumDistributedOp.infer_layout |
|
23 |
1 |
0 |
|
96% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_cumsum.py |
CumsumDistributedOp.infer_layout.idx_to_alias |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_cumsum.py |
(no function) |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_elementwise.py |
ElementWiseDistributedOp.infer_layout |
|
19 |
3 |
0 |
|
84% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_elementwise.py |
ElementWiseDistributedOp._handle_no_input_shapes |
|
5 |
1 |
0 |
|
80% |
| / 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._extract_input_shapes |
|
7 |
5 |
0 |
|
29% |
| / 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._merge_tensor_maps_without_shape |
|
23 |
23 |
0 |
|
0% |
| / 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 |
|
16 |
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.get_expand_impl |
|
15 |
1 |
0 |
|
93% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_elementwise.py |
AddDistributedOp.get_expand_impl.expand_impl1 |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_elementwise.py |
AddDistributedOp.get_expand_impl.expand_impl2 |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_elementwise.py |
(no function) |
|
23 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_embedding.py |
EmbeddingDistributedOp.infer_layout |
|
19 |
0 |
0 |
|
100% |
| / 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 |
20 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_embedding.py |
EmbeddingDistributedOp.get_expand_impl |
|
7 |
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 |
7 |
0 |
|
65% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_embedding.py |
(no function) |
|
7 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_expand.py |
ExpandDistributedOp.infer_layout |
|
38 |
3 |
1 |
|
92% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_expand.py |
ExpandDistributedOp.infer_layout.idx_to_alias |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_expand.py |
ExpandAsDistributedOp.infer_layout |
|
38 |
4 |
0 |
|
89% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_expand.py |
ExpandAsDistributedOp.infer_layout.idx_to_alias |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_expand.py |
(no function) |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_expand_dims.py |
ExpandDimsDistributedOp.infer_layout |
|
22 |
5 |
0 |
|
77% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_expand_dims.py |
(no function) |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_flatten.py |
FlattenDistributedOp.infer_layout |
|
33 |
2 |
0 |
|
94% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_flatten.py |
(no function) |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_gather.py |
IndexSelectDistributedOp.infer_layout |
|
27 |
3 |
0 |
|
89% |
| / 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.infer_layout |
|
38 |
3 |
0 |
|
92% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_gather.py |
GatherDDistributedOp.get_expand_impl |
|
7 |
1 |
0 |
|
86% |
| / 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.infer_layout |
|
12 |
1 |
0 |
|
92% |
| / 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) |
|
17 |
0 |
0 |
|
100% |
| / 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.infer_layout |
|
29 |
2 |
0 |
|
93% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_histc_ext.py |
(no function) |
|
7 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_inplace_scatter_value.py |
InplaceScatterValueDistributedOp.infer_layout |
|
28 |
2 |
0 |
|
93% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_inplace_scatter_value.py |
(no function) |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_isin.py |
IsinDistributedOp.infer_layout |
|
17 |
1 |
0 |
|
94% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_isin.py |
IsinDistributedOp.infer_layout.idx_to_alias |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_isin.py |
(no function) |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_masked_scatter.py |
MaskedScatterDistributedOp.infer_layout |
|
9 |
1 |
0 |
|
89% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_masked_scatter.py |
(no function) |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_matmul.py |
MatMulExtDistributedOp.infer_layout |
|
26 |
26 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_matmul.py |
MatMulDistributedOp.infer_layout |
|
33 |
6 |
0 |
|
82% |
| / 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 |
|
9 |
2 |
0 |
|
78% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_matmul.py |
BatchMatMulExtDistributedOp.infer_layout |
|
15 |
3 |
0 |
|
80% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_matmul.py |
BatchMatMulDistributedOp.infer_layout |
|
24 |
6 |
0 |
|
75% |
| / 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) |
|
21 |
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 |
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._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._tensor_map_to_placements |
|
19 |
4 |
0 |
|
79% |
| / 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 |
|
27 |
27 |
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 |
|
18 |
4 |
0 |
|
78% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_ms_flash_attention_score.py |
FlashAttentionScoreDistributedOp._infer_softmax_layout_conservatively |
|
11 |
1 |
0 |
|
91% |
| / 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 |
|
6 |
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 |
|
26 |
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 |
10 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_ms_flash_attention_score.py |
FlashAttentionScoreDistributedOp._check_batch_consistency |
|
9 |
9 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_ms_flash_attention_score.py |
FlashAttentionScoreDistributedOp._check_hidden_consistency |
|
9 |
9 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_ms_flash_attention_score.py |
FlashAttentionScoreDistributedOp._check_dim_consistency |
|
9 |
9 |
0 |
|
0% |
| / 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 |
|
7 |
7 |
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 |
|
16 |
7 |
0 |
|
56% |
| / 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 |
|
39 |
39 |
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 |
3 |
0 |
|
0% |
| / 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_scalar_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) |
|
69 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_multinomial.py |
MultinomialDistributedOp.infer_layout |
|
17 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_multinomial.py |
(no function) |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_new_ones.py |
NewOnesDistributedOp.infer_layout |
|
12 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_new_ones.py |
(no function) |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_nonzero.py |
NonzeroDistributedOp.infer_layout |
|
24 |
1 |
0 |
|
96% |
| / 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) |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_norm.py |
NormDistributedOp.infer_layout |
|
33 |
6 |
0 |
|
82% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_norm.py |
LayerNormDistributedOp.infer_layout |
|
28 |
1 |
0 |
|
96% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_norm.py |
LayerNormDistributedOp.infer_layout.idx_to_alias |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_norm.py |
(no function) |
|
6 |
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 |
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._tensor_map_to_placements |
|
19 |
4 |
0 |
|
79% |
| / 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 |
1 |
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_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 |
23 |
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_dynamic |
|
27 |
27 |
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 |
14 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_npu_flash_attention_score.py |
NPUFlashAttentionScoreDistributedOp._validate_pse_configuration |
|
7 |
7 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_npu_flash_attention_score.py |
NPUFlashAttentionScoreDistributedOp.infer_layout |
|
28 |
7 |
0 |
|
75% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_npu_flash_attention_score.py |
NPUFlashAttentionScoreDistributedOp._infer_softmax_layout_conservatively |
|
11 |
11 |
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 |
|
7 |
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 |
|
26 |
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 |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_npu_flash_attention_score.py |
NPUFlashAttentionScoreDistributedOp._validate_sharding_consistency |
|
10 |
10 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_npu_flash_attention_score.py |
NPUFlashAttentionScoreDistributedOp._check_batch_consistency |
|
9 |
9 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_npu_flash_attention_score.py |
NPUFlashAttentionScoreDistributedOp._check_hidden_consistency |
|
9 |
9 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_npu_flash_attention_score.py |
NPUFlashAttentionScoreDistributedOp._check_dim_consistency |
|
9 |
9 |
0 |
|
0% |
| / 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 |
17 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_npu_flash_attention_score.py |
NPUFlashAttentionScoreDistributedOp._truncate_result |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_npu_flash_attention_score.py |
NPUFlashAttentionScoreDistributedOp._adjust_head_num |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_npu_flash_attention_score.py |
NPUFlashAttentionScoreDistributedOp._compute_adjusted_sparse_params |
|
7 |
7 |
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 |
|
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_expand_impl.expanded_impl |
|
16 |
16 |
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 |
|
28 |
28 |
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 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_npu_flash_attention_score.py |
NPUFlashAttentionScoreDistributedOp._normalize_dim_map |
|
3 |
1 |
0 |
|
67% |
| / 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_scalar_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) |
|
64 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_one_hot_ext.py |
OneHotExtDistributedOp.infer_layout |
|
20 |
4 |
0 |
|
80% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_one_hot_ext.py |
OneHotExtDistributedOp.get_expand_impl |
|
13 |
13 |
0 |
|
0% |
| / 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._get_num_classes |
|
5 |
1 |
0 |
|
80% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_one_hot_ext.py |
OneHotExtDistributedOp._validate_num_classes |
|
4 |
1 |
0 |
|
75% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_one_hot_ext.py |
OneHotExtDistributedOp._validate_indices_dtype |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_one_hot_ext.py |
OneHotExtDistributedOp._get_sharded_axes |
|
12 |
12 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_one_hot_ext.py |
OneHotExtDistributedOp._get_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_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 |
5 |
0 |
|
17% |
| / 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 |
|
5 |
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 |
OneHotExtDistributedOp._tensor_map_to_placements |
|
19 |
4 |
0 |
|
79% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_one_hot_ext.py |
(no function) |
|
21 |
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 |
5 |
0 |
|
0% |
| / 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 |
(no function) |
|
8 |
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 |
OuterDistributedOp.infer_layout |
|
25 |
1 |
0 |
|
96% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_outer.py |
OuterDistributedOp.infer_layout._get_flattened_map |
|
3 |
1 |
0 |
|
67% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_outer.py |
OuterDistributedOp.infer_layout.idx_to_alias |
|
5 |
1 |
0 |
|
80% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_outer.py |
(no function) |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_pad.py |
PadDistributedOp.infer_layout |
|
29 |
5 |
0 |
|
83% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_pad.py |
(no function) |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_reduce.py |
ReduceExtDistributedOpBase.__init__ |
|
4 |
1 |
0 |
|
75% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_reduce.py |
ReduceExtDistributedOpBase.infer_layout |
|
16 |
7 |
0 |
|
56% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_reduce.py |
ReduceExtDistributedOpBase._infer_output_layout |
|
8 |
0 |
0 |
|
100% |
| / 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.infer_layout |
|
21 |
1 |
0 |
|
95% |
| / 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) |
|
34 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_repeat.py |
RepeatDistributedOp.infer_layout |
|
37 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_repeat.py |
RepeatDistributedOp.infer_layout.idx_to_alias |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_repeat.py |
(no function) |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_repeat_interleave.py |
RepeatInterleaveDistributedOp.infer_layout |
|
26 |
4 |
0 |
|
85% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_repeat_interleave.py |
RepeatInterleaveDistributedOp.infer_layout.idx_to_alias |
|
3 |
1 |
0 |
|
67% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_repeat_interleave.py |
(no function) |
|
4 |
0 |
0 |
|
100% |
| / 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._parse_shape_args |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_reshape.py |
ReshapeDistributedOp._parse_torch_shape_args |
|
13 |
4 |
0 |
|
69% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_reshape.py |
ReshapeDistributedOp._parse_mindspore_shape_args |
|
3 |
1 |
0 |
|
67% |
| / 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 |
|
14 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_reshape.py |
ReshapeDistributedOp._handle_sharded_axis |
|
12 |
3 |
0 |
|
75% |
| / 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.infer_layout |
|
14 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_reshape.py |
(no function) |
|
20 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_scaled_dot_product_attention.py |
ScaledDotProductAttentionDistributedOp._tensor_map_to_placements |
|
19 |
19 |
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 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_scaled_dot_product_attention.py |
ScaledDotProductAttentionDistributedOp._get_dims |
|
4 |
4 |
0 |
|
0% |
| / 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 |
17 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_scaled_dot_product_attention.py |
ScaledDotProductAttentionDistributedOp._get_split_info |
|
8 |
8 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_scaled_dot_product_attention.py |
ScaledDotProductAttentionDistributedOp._get_split_id |
|
26 |
26 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_scaled_dot_product_attention.py |
ScaledDotProductAttentionDistributedOp._validate_sharding_consistency |
|
16 |
16 |
0 |
|
0% |
| / 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 |
5 |
0 |
|
0% |
| / 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 |
17 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_scaled_dot_product_attention.py |
ScaledDotProductAttentionDistributedOp.infer_layout |
|
8 |
2 |
0 |
|
75% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_scaled_dot_product_attention.py |
ScaledDotProductAttentionDistributedOp.get_expand_impl |
|
12 |
12 |
0 |
|
0% |
| / 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 |
|
23 |
23 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_scaled_dot_product_attention.py |
(no function) |
|
25 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_scatter.py |
ScatterDistributedOp.infer_layout |
|
29 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_scatter.py |
(no function) |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_scatter_update.py |
ScatterUpdateDistributedOp.infer_layout |
|
13 |
2 |
0 |
|
85% |
| / 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) |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_slice.py |
SliceDistributedOp._is_shard_dim |
|
14 |
5 |
0 |
|
64% |
| / 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 |
|
7 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_slice.py |
(no function) |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_slice_ext.py |
SliceExtDistributedOp.infer_layout |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_slice_ext.py |
(no function) |
|
3 |
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 |
SplitWithSizeDistributedOp.infer_layout |
|
9 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_split.py |
SplitWithSizeViewDistributedOp.infer_layout |
|
9 |
9 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_split.py |
SplitDistributedOp.infer_layout |
|
27 |
7 |
0 |
|
74% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_split.py |
SplitTensorDistributedOp.infer_layout |
|
12 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_split.py |
SplitTensorViewDistributedOp.infer_layout |
|
12 |
12 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_split.py |
TensorSplitDistributedOp.infer_layout |
|
27 |
2 |
0 |
|
93% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_split.py |
(no function) |
|
14 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_squeeze.py |
SqueezeDistributedOp.infer_layout |
|
9 |
3 |
0 |
|
67% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_squeeze.py |
SqueezeDistributedOp._extract_args |
|
13 |
3 |
0 |
|
77% |
| / 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 |
|
5 |
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 |
|
7 |
0 |
0 |
|
100% |
| / 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 |
|
11 |
1 |
0 |
|
91% |
| / 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) |
|
15 |
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 |
|
26 |
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) |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_topk.py |
TopKDistributedOp.infer_layout |
|
15 |
1 |
0 |
|
93% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_topk.py |
(no function) |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_transpose.py |
TransposeDistributedOp.infer_layout |
|
34 |
2 |
0 |
|
94% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_transpose.py |
(no function) |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_tuple_elementwise.py |
TupleElementWiseDistributedOp.infer_layout |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_tuple_elementwise.py |
(no function) |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_unbind.py |
UnbindDistributedOp.infer_layout |
|
19 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / shard / ops / parallel_unbind.py |
(no function) |
|
4 |
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 |
(no function) |
|
3 |
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) |
|
15 |
0 |
0 |
|
100% |
| / 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__ |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / tensor_parallel / style.py |
ColwiseParallel.__repr__ |
|
1 |
1 |
0 |
|
0% |
| / 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 |
1 |
0 |
|
88% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / tensor_parallel / style.py |
ColwiseParallel._partition_embedding_fn |
|
8 |
1 |
0 |
|
88% |
| / 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 |
1 |
0 |
|
89% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / tensor_parallel / style.py |
ColwiseParallel.apply.partition_fn |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / tensor_parallel / style.py |
ColwiseParallel.apply.input_fn |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / tensor_parallel / style.py |
ColwiseParallel.apply.output_fn |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / tensor_parallel / style.py |
RowwiseParallel.__init__ |
|
5 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / tensor_parallel / style.py |
RowwiseParallel.__repr__ |
|
1 |
1 |
0 |
|
0% |
| / 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 |
1 |
0 |
|
88% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / tensor_parallel / style.py |
RowwiseParallel._prepare_output_fn |
|
9 |
3 |
0 |
|
67% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / tensor_parallel / style.py |
RowwiseParallel.apply |
|
11 |
1 |
0 |
|
91% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / tensor_parallel / style.py |
RowwiseParallel.apply.partition_fn |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / tensor_parallel / style.py |
RowwiseParallel.apply.input_fn |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / tensor_parallel / style.py |
RowwiseParallel.apply.output_fn |
|
1 |
1 |
0 |
|
0% |
| / 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 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / tensor_parallel / style.py |
SequenceParallel.apply.input_fn |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / tensor_parallel / style.py |
SequenceParallel.apply.output_fn |
|
1 |
1 |
0 |
|
0% |
| / 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 |
2 |
0 |
|
80% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / tensor_parallel / style.py |
PrepareModuleInput._prepare_input_fn |
|
10 |
3 |
0 |
|
70% |
| / 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__ |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / tensor_parallel / style.py |
PrepareModuleOutput._prepare_out_fn |
|
17 |
1 |
0 |
|
94% |
| / 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 |
1 |
0 |
|
0% |
| / 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 |
(no function) |
|
57 |
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 |
clip_grad_norm_ |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / utils / clip_grad.py |
(no function) |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / utils / shape_utils.py |
compute_local_shape_and_global_offset |
|
21 |
2 |
0 |
|
90% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / core / utils / shape_utils.py |
(no function) |
|
2 |
0 |
0 |
|
100% |
| / 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 / autograd_compat.py |
requires_grad |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / autograd_compat.py |
requires_grad |
|
3 |
3 |
0 |
|
0% |
| / 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 |
7 |
0 |
|
50% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / autograd_compat.py |
is_leaf |
|
1 |
1 |
0 |
|
0% |
| / 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 |
13 |
0 |
|
0% |
| / 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 |
DTensorBase.__new__ |
|
23 |
23 |
0 |
|
0% |
| / 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.__copy__ |
|
16 |
16 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / dtensor.py |
DTensorBase.__fallback__ |
|
6 |
6 |
0 |
|
0% |
| / 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 |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / dtensor.py |
DTensorBase.set_data |
|
15 |
9 |
0 |
|
40% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / dtensor.py |
DTensorBase.has_init |
|
3 |
3 |
0 |
|
0% |
| / 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 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / dtensor.py |
(no function) |
|
27 |
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 / pack_utils.py |
_shape_tuple |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / pack_utils.py |
_has_strided_shard_layout |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / pack_utils.py |
_resolve_same_dim_strided_context |
|
19 |
19 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / pack_utils.py |
_placements_match_target_dim_only |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / pack_utils.py |
_orig_layout_is_supported |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / pack_utils.py |
_current_strided_layout_is_supported |
|
12 |
12 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / pack_utils.py |
supports_same_dim_strided_layout |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / pack_utils.py |
_resolve_unpacked_shape |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / pack_utils.py |
_get_packed_tensor_shape |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / pack_utils.py |
build_rs_plan |
|
30 |
30 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / pack_utils.py |
pack_for_reduce_scatter |
|
11 |
11 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / pack_utils.py |
unpack_from_all_gather |
|
7 |
7 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / pack_utils.py |
(no function) |
|
33 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / param.py |
_pack_for_reduce_scatter |
|
4 |
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 |
make_contiguous_strides_for |
|
18 |
18 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / param.py |
MindSporeHSDPParamV2.__init__ |
|
29 |
29 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / param.py |
MindSporeHSDPParamV2.uses_param_shard |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / param.py |
MindSporeHSDPParamV2.is_dtensor_compat_mode |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / param.py |
MindSporeHSDPParamV2._get_data_parallel_shard_placement |
|
10 |
10 |
0 |
|
0% |
| / 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._init_sharded_param |
|
42 |
42 |
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_all_gather_outputs |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / param.py |
MindSporeHSDPParamV2.init_unsharded_param |
|
13 |
8 |
0 |
|
38% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / param.py |
MindSporeHSDPParamV2._get_unsharded_param_from_all_gather_output |
|
8 |
8 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / param.py |
MindSporeHSDPParamV2.to_sharded |
|
7 |
3 |
0 |
|
57% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / param.py |
MindSporeHSDPParamV2.to_unsharded |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / param.py |
MindSporeHSDPParamV2._setattr_on_modules |
|
7 |
7 |
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 |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / param.py |
MindSporeHSDPParamV2.alloc_all_gather_outputs |
|
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 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / param.py |
MindSporeHSDPParamV2.unsharded_grad_data |
|
4 |
4 |
0 |
|
0% |
| / 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 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / param.py |
MindSporeHSDPParamV2.shard_world_size |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / param.py |
MindSporeHSDPParamV2.replicate_world_size |
|
3 |
3 |
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.reset_sharded_param |
|
35 |
35 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / param.py |
MindSporeHSDPParamV2._get_unsharded_param_data |
|
15 |
15 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / param.py |
MindSporeHSDPParamV2.unshard |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / param.py |
MindSporeHSDPParamV2.wait_for_unshard |
|
6 |
0 |
0 |
|
100% |
| / 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_output |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / param.py |
MindSporeHSDPParamV2.clear_reduce_scatter_output |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / param.py |
MindSporeHSDPParamV2.reduce_scatter_grad |
|
23 |
5 |
0 |
|
78% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / param.py |
MindSporeHSDPParamV2.zero_grad |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / param.py |
MindSporeHSDPParamV2.all_reduce_grad |
|
16 |
5 |
0 |
|
69% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / param.py |
MindSporeHSDPParamV2.all_reduce_output |
|
4 |
0 |
0 |
|
100% |
| / 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.apply_reduced_grad |
|
17 |
0 |
0 |
|
100% |
| / 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) |
|
73 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / param_group.py |
_normalize_device |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / param_group.py |
_shape_numel |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / param_group.py |
get_all_gather_metadata |
|
18 |
18 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / param_group.py |
AllGatherMetadata.__post_init__ |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / param_group.py |
get_comm_ctx |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / param_group.py |
AllGatherMetadataCache.get_metadata |
|
7 |
7 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / param_group.py |
all_gather_copy_in |
|
7 |
7 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / param_group.py |
split_with_sizes_copy |
|
7 |
7 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / param_group.py |
reduce_scatter_copy_in |
|
14 |
14 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / param_group.py |
HSDPParamGroup.__init__ |
|
31 |
31 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / param_group.py |
HSDPParamGroup._infer_layout_replicate_group |
|
10 |
10 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / param_group.py |
HSDPParamGroup._build_active_replicate_buckets |
|
13 |
13 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / param_group.py |
HSDPParamGroup._init_flat_param_buffer |
|
39 |
39 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / param_group.py |
HSDPParamGroup._is_flat_buffer_valid |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / param_group.py |
HSDPParamGroup._allocate_bucket_buffers_if_needed |
|
7 |
7 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / param_group.py |
HSDPParamGroup._pack_bucket_from_reduce_output |
|
12 |
12 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / param_group.py |
HSDPParamGroup._unpack_bucket_to_reduce_output |
|
11 |
11 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / param_group.py |
HSDPParamGroup.unshard |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / param_group.py |
HSDPParamGroup._init_mp_dtypes |
|
11 |
11 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / param_group.py |
HSDPParamGroup.wait_for_unshard |
|
13 |
13 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / param_group.py |
HSDPParamGroup.alloc_all_gather_output |
|
8 |
8 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / param_group.py |
HSDPParamGroup.free_all_gather_output |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / param_group.py |
HSDPParamGroup.foreach_all_gather |
|
27 |
27 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / param_group.py |
HSDPParamGroup.foreach_all_gather_copy_out |
|
14 |
14 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / param_group.py |
HSDPParamGroup.foreach_reduce |
|
50 |
50 |
0 |
|
0% |
| / 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 |
|
14 |
14 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / param_group.py |
HSDPParamGroup.wait_all_reduce_and_apply_grad |
|
9 |
9 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / param_group.py |
HSDPParamGroup.apply_fusion_reduced_grad |
|
12 |
12 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / param_group.py |
HSDPParamGroup._apply_reduced_grad |
|
13 |
13 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / param_group.py |
(no function) |
|
83 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / scheduler.py |
MindSporeHSDPSchedulerV2.zero_grad |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / scheduler.py |
MindSporeHSDPSchedulerV2._register_hooks |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / scheduler.py |
MindSporeHSDPSchedulerV2._init_platform |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / scheduler.py |
MindSporeHSDPSchedulerV2._new_cell_state |
|
18 |
6 |
0 |
|
67% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / scheduler.py |
MindSporeHSDPSchedulerV2._register_post_backward_hook |
|
21 |
4 |
0 |
|
81% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / scheduler.py |
MindSporeHSDPSchedulerV2._forward_pre_hook |
|
2 |
2 |
0 |
|
0% |
| / 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 |
7 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / scheduler.py |
MindSporeHSDPSchedulerV2._backward_pre_hook |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / scheduler.py |
MindSporeHSDPSchedulerV2._root_backward_hook |
|
13 |
13 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / scheduler.py |
MindSporeHSDPSchedulerV2._backward_hook |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / scheduler.py |
MindSporeHSDPSchedulerV2._register_forward_backward_hooks |
|
8 |
8 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / scheduler.py |
(no function) |
|
23 |
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._get_pending_unsharded_grad |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / state.py |
MindSporeHSDPStateV2._has_pending_unsharded_grad |
|
5 |
1 |
0 |
|
80% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / state.py |
MindSporeHSDPStateV2._synchronize_current_stream_if_needed |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / state.py |
MindSporeHSDPStateV2.__init__ |
|
12 |
12 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / state.py |
MindSporeHSDPStateV2._iter_managed_params |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / state.py |
MindSporeHSDPStateV2._comm_fusion_unsupported_reason |
|
17 |
17 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / state.py |
MindSporeHSDPStateV2._init_param_group |
|
9 |
9 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / state.py |
MindSporeHSDPStateV2.zero_grad |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / state.py |
MindSporeHSDPStateV2._div_if_needed |
|
5 |
3 |
0 |
|
40% |
| / 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._init_hsdp_params |
|
24 |
2 |
0 |
|
92% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / state.py |
MindSporeHSDPStateV2._init_mp_dtypes |
|
13 |
2 |
0 |
|
85% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / state.py |
MindSporeHSDPStateV2.lazy_init |
|
8 |
1 |
0 |
|
88% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / state.py |
MindSporeHSDPStateV2._validate_cpu_offload_params |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / state.py |
MindSporeHSDPStateV2._validate_no_meta_params |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / state.py |
MindSporeHSDPStateV2._allreduce_replicate_params |
|
15 |
2 |
0 |
|
87% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / state.py |
MindSporeHSDPStateV2._finish_ignored_allreduce |
|
11 |
9 |
0 |
|
18% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / state.py |
MindSporeHSDPStateV2.reduce_params |
|
14 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / state.py |
MindSporeHSDPStateV2.post_backward_for_comm_fusion |
|
11 |
11 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / state.py |
MindSporeHSDPStateV2._post_backward_without_reduce |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / state.py |
MindSporeHSDPStateV2._should_run_all_reduce |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / state.py |
MindSporeHSDPStateV2._queue_reduce_scatter_then_all_reduce |
|
11 |
1 |
0 |
|
91% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / state.py |
MindSporeHSDPStateV2._queue_compat_all_reduce |
|
4 |
1 |
0 |
|
75% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / state.py |
MindSporeHSDPStateV2.post_backward |
|
25 |
11 |
0 |
|
56% |
| / 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 |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / fully_shard / state.py |
(no function) |
|
43 |
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 / 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 / parameter_param_info_patch.py |
patch_mindspore_parameter_param_info_cycle_if_needed |
|
6 |
1 |
0 |
|
83% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / parameter_param_info_patch.py |
patch_mindspore_parameter_param_info_cycle_if_needed._patched_setter |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / parameter_param_info_patch.py |
(no function) |
|
3 |
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 |
|
20 |
20 |
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 |
|
12 |
12 |
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 / stage.py |
PipelineStageBase.__init__ |
|
13 |
13 |
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.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 |
|
13 |
13 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / pipeline_parallel / stage.py |
PipelineStageBase.backward_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 |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / pipeline_parallel / stage.py |
(no function) |
|
16 |
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_all_to_all_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 |
_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 |
MindSporePlatform.__init__ |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
MindSporePlatform.apply_runtime_patches_if_needed |
|
2 |
2 |
0 |
|
0% |
| / 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 |
|
1 |
1 |
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.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 |
1 |
0 |
|
0% |
| / 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 |
|
5 |
5 |
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 |
|
2 |
2 |
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 |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
MindSporePlatform.differentiable_reduce_scatter |
|
6 |
6 |
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.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_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._create_group |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
MindSporePlatform.all_gather_into_tensor |
|
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 |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
MindSporePlatform.reduce_scatter_tensor |
|
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_async_a2a_wait |
|
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.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.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.get_symmetric_memory_handler |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
MindSporePlatform.get_multicore_handler |
|
2 |
2 |
0 |
|
0% |
| / 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 |
|
13 |
13 |
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.no_grad |
|
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.new_event |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
MindSporePlatform.tree_map |
|
7 |
7 |
0 |
|
0% |
| / 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 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / platform.py |
MindSporePlatform.ckpt_wrapper |
|
2 |
2 |
0 |
|
0% |
| / 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.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.async_save_on_cpu |
|
2 |
2 |
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 |
1 |
0 |
|
0% |
| / 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 |
(no function) |
|
232 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / utils.py |
normalize_runtime_device |
|
3 |
3 |
0 |
|
0% |
| / 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 |
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 |
apply_runtime_patches_if_needed |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / platform.py |
Platform.apply_runtime_patches_if_needed |
|
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_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.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.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.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.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_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_to_all_single |
|
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_all_to_all_single |
|
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.get_model_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.get_symmetric_memory_handler |
|
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.create_group |
|
6 |
1 |
0 |
|
83% |
| / 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.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.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 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / platform.py |
Platform.register_full_backward_pre_hook |
|
1 |
1 |
0 |
|
0% |
| / 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.ckpt_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.noop_context_fn |
|
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.async_save_on_cpu |
|
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.tensor_to_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_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) |
|
195 |
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 / common / __init__.py |
(no function) |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / common / moe.py |
_run_experts_for_loop |
|
11 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / common / moe.py |
_run_experts_grouped_mm_gpu |
|
8 |
8 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / common / moe.py |
_run_experts_grouped_mm_npu |
|
12 |
12 |
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 |
|
8 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / common / moe.py |
MoE.__init__ |
|
11 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / common / moe.py |
MoE.forward |
|
31 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / common / moe.py |
update_expert_bias |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / common / moe.py |
(no function) |
|
26 |
0 |
0 |
|
100% |
| / 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 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / dtensor.py |
DTensorBase.grad |
|
1 |
1 |
0 |
|
0% |
| / 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 |
|
3 |
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 |
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.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 |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / dtensor.py |
DTensorBase.data |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / dtensor.py |
DTensorBase.dtype |
|
1 |
1 |
0 |
|
0% |
| / 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 |
|
5 |
5 |
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 |
0 |
0 |
|
100% |
| / 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.zero_ |
|
7 |
7 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / dtensor.py |
DTensorBase.copy_ |
|
8 |
8 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / dtensor.py |
DTensorBase.fill_ |
|
7 |
7 |
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) |
|
48 |
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 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / hook_function.py |
PostBackwardFunction.backward |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / hook_function.py |
PostBackwardFunction.apply |
|
26 |
26 |
0 |
|
0% |
| / 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 / pack_utils.py |
_has_strided_shard_layout |
|
2 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / pack_utils.py |
_resolve_same_dim_strided_context |
|
19 |
7 |
0 |
|
63% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / pack_utils.py |
_placements_match_target_dim_only |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / pack_utils.py |
_orig_layout_is_supported |
|
3 |
1 |
0 |
|
67% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / pack_utils.py |
_current_strided_layout_is_supported |
|
12 |
3 |
0 |
|
75% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / pack_utils.py |
supports_same_dim_strided_layout |
|
6 |
2 |
0 |
|
67% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / pack_utils.py |
_resolve_unpacked_shape |
|
3 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / pack_utils.py |
_get_packed_tensor_shape |
|
6 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / pack_utils.py |
build_rs_plan |
|
30 |
10 |
0 |
|
67% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / pack_utils.py |
pack_for_reduce_scatter |
|
11 |
3 |
0 |
|
73% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / pack_utils.py |
unpack_from_all_gather |
|
7 |
2 |
0 |
|
71% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / pack_utils.py |
(no function) |
|
32 |
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 |
_build_group_info_from_rank_list |
|
10 |
1 |
2 |
|
90% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / param.py |
_build_group_info_from_process_group |
|
8 |
2 |
1 |
|
75% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / param.py |
TorchHSDPParamV2.__init__ |
|
30 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / param.py |
TorchHSDPParamV2.uses_param_shard |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / param.py |
TorchHSDPParamV2.is_dtensor_compat_mode |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / param.py |
TorchHSDPParamV2._get_base_spmd_placements |
|
9 |
3 |
0 |
|
67% |
| / 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._init_group_infos |
|
7 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / param.py |
TorchHSDPParamV2._build_layout_driven_group_info |
|
26 |
4 |
0 |
|
85% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / param.py |
TorchHSDPParamV2._to_local_unsharded_grad |
|
7 |
1 |
0 |
|
86% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / param.py |
TorchHSDPParamV2.reduce_scatter_output |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / param.py |
TorchHSDPParamV2.clear_reduce_scatter_output |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / param.py |
TorchHSDPParamV2.all_reduce_output |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / param.py |
TorchHSDPParamV2.clear_all_reduce_output |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / param.py |
TorchHSDPParamV2.apply_reduced_grad |
|
31 |
31 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / param.py |
TorchHSDPParamV2._init_sharded_param |
|
44 |
9 |
0 |
|
80% |
| / 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_all_gather_outputs |
|
3 |
1 |
0 |
|
67% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / param.py |
TorchHSDPParamV2.init_unsharded_param |
|
7 |
4 |
0 |
|
43% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / param.py |
TorchHSDPParamV2._get_unsharded_param_from_all_gather_output |
|
8 |
1 |
0 |
|
88% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / param.py |
TorchHSDPParamV2.to_sharded |
|
7 |
3 |
0 |
|
57% |
| / 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 |
|
7 |
4 |
0 |
|
43% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / param.py |
TorchHSDPParamV2.to_sharded_dtensor |
|
1 |
1 |
0 |
|
0% |
| / 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_all_gather_outputs |
|
5 |
1 |
0 |
|
80% |
| / 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.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 |
|
4 |
1 |
0 |
|
75% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / param.py |
TorchHSDPParamV2.unsharded_accumulated_grad_data |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / param.py |
TorchHSDPParamV2._sharded_local_tensor |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / param.py |
TorchHSDPParamV2.shard_world_size |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / param.py |
TorchHSDPParamV2.replicate_world_size |
|
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.reset_sharded_param |
|
37 |
14 |
0 |
|
62% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / param.py |
TorchHSDPParamV2._get_unsharded_param_data |
|
14 |
7 |
0 |
|
50% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / param.py |
TorchHSDPParamV2.unshard |
|
4 |
1 |
0 |
|
75% |
| / 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 |
|
17 |
3 |
0 |
|
82% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / param.py |
TorchHSDPParamV2.all_reduce_grad |
|
11 |
4 |
0 |
|
64% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / param.py |
set_requires_grad_if_needed |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / param.py |
(no function) |
|
73 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / param_group.py |
get_all_gather_metadata |
|
18 |
18 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / param_group.py |
AllGatherMetadata.__post_init__ |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / param_group.py |
get_comm_ctx |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / param_group.py |
AllGatherMetadataCache.get_metadata |
|
7 |
7 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / param_group.py |
all_gather_copy_in |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / param_group.py |
reduce_scatter_copy_in |
|
14 |
14 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / param_group.py |
HSDPParamGroup.__init__ |
|
33 |
33 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / param_group.py |
HSDPParamGroup._infer_layout_replicate_group |
|
11 |
11 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / param_group.py |
HSDPParamGroup._build_active_replicate_buckets |
|
16 |
2 |
0 |
|
88% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / param_group.py |
HSDPParamGroup._allocate_bucket_buffers_if_needed |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / param_group.py |
HSDPParamGroup._pack_bucket_from_reduce_output |
|
12 |
12 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / param_group.py |
HSDPParamGroup._unpack_bucket_to_reduce_output |
|
11 |
11 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / param_group.py |
HSDPParamGroup._init_flat_param_buffer |
|
28 |
28 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / param_group.py |
HSDPParamGroup._is_flat_buffer_valid |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / param_group.py |
HSDPParamGroup.unshard |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / param_group.py |
HSDPParamGroup._init_mp_dtypes |
|
11 |
11 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / param_group.py |
HSDPParamGroup.wait_for_unshard |
|
13 |
7 |
0 |
|
46% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / param_group.py |
HSDPParamGroup.alloc_all_gather_output |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / param_group.py |
HSDPParamGroup.free_all_gather_output |
|
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 |
|
27 |
27 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / param_group.py |
HSDPParamGroup.foreach_all_gather_copy_out |
|
21 |
2 |
0 |
|
90% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / param_group.py |
HSDPParamGroup.foreach_reduce |
|
44 |
44 |
0 |
|
0% |
| / 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 |
|
14 |
14 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / param_group.py |
HSDPParamGroup.wait_all_reduce_and_apply_grad |
|
9 |
9 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / param_group.py |
HSDPParamGroup.apply_fusion_reduced_grad |
|
12 |
12 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / param_group.py |
HSDPParamGroup._apply_reduced_grad |
|
44 |
21 |
0 |
|
52% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / param_group.py |
(no function) |
|
73 |
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 |
|
17 |
1 |
0 |
|
94% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / scheduler.py |
TorchHSDPSchedulerV2._register_post_backward_hook |
|
21 |
21 |
0 |
|
0% |
| / 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 |
|
5 |
5 |
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 |
|
6 |
6 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / scheduler.py |
TorchHSDPSchedulerV2._root_backward_hook |
|
13 |
13 |
0 |
|
0% |
| / 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 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / scheduler.py |
TorchHSDPSchedulerV2._grouped_forward_post_hook_skip |
|
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 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / scheduler.py |
TorchHSDPSchedulerV2._register_forward_backward_hooks |
|
8 |
8 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / scheduler.py |
(no function) |
|
28 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / state.py |
_to_dtype_if_needed |
|
3 |
1 |
0 |
|
67% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / state.py |
TorchHSDPStateV2._get_pending_unsharded_grad |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / state.py |
TorchHSDPStateV2._has_pending_unsharded_grad |
|
5 |
2 |
0 |
|
60% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / state.py |
TorchHSDPStateV2.__init__ |
|
12 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / state.py |
TorchHSDPStateV2._comm_fusion_unsupported_reason |
|
17 |
7 |
0 |
|
59% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / state.py |
TorchHSDPStateV2._init_param_group |
|
9 |
3 |
0 |
|
67% |
| / 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._init_hsdp_params |
|
24 |
2 |
0 |
|
92% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / state.py |
TorchHSDPStateV2._init_mp_dtypes |
|
13 |
1 |
0 |
|
92% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / state.py |
TorchHSDPStateV2._validate_cpu_offload_params |
|
5 |
1 |
0 |
|
80% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / state.py |
TorchHSDPStateV2.lazy_init |
|
8 |
8 |
0 |
|
0% |
| / 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 |
|
19 |
7 |
0 |
|
63% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / state.py |
TorchHSDPStateV2._resolve_default_reduce_op |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / state.py |
TorchHSDPStateV2._resolve_reduce_op |
|
3 |
1 |
0 |
|
67% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / state.py |
TorchHSDPStateV2._should_run_all_reduce |
|
1 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / state.py |
TorchHSDPStateV2._queue_reduce_scatter_then_all_reduce |
|
9 |
9 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / state.py |
TorchHSDPStateV2._queue_compat_all_reduce |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / state.py |
TorchHSDPStateV2.post_backward |
|
25 |
9 |
0 |
|
64% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / state.py |
TorchHSDPStateV2.reduce_params |
|
17 |
17 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / state.py |
TorchHSDPStateV2.set_requires_grad_sync |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / fully_shard / state.py |
TorchHSDPStateV2.set_reduce_op_type |
|
6 |
6 |
0 |
|
0% |
| / 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 / 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 |
5 |
0 |
|
0% |
| / 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 |
|
8 |
8 |
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 / pipeline_parallel / __init__.py |
(no function) |
|
0 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / pipeline_parallel / stage.py |
PipelineStageBase.__init__ |
|
11 |
11 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / pipeline_parallel / stage.py |
PipelineStageBase.clear_cache |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / pipeline_parallel / stage.py |
PipelineStageBase._clear_recv_buffer |
|
4 |
4 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / pipeline_parallel / stage.py |
PipelineStageBase._check_pp_group |
|
5 |
5 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / pipeline_parallel / stage.py |
PipelineStageBase.is_first_stage |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / pipeline_parallel / stage.py |
PipelineStageBase.is_last_stage |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / pipeline_parallel / stage.py |
PipelineStageBase.forward_one_chunk |
|
13 |
13 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / pipeline_parallel / stage.py |
PipelineStageBase.backward_one_chunk |
|
24 |
24 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / pipeline_parallel / stage.py |
(no function) |
|
16 |
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 |
_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 |
_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 |
TorchPlatform.apply_runtime_patches_if_needed |
|
1 |
0 |
0 |
|
100% |
| / 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 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / platform.py |
TorchPlatform.empty |
|
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_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 |
|
2 |
0 |
0 |
|
100% |
| / 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 |
|
3 |
3 |
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 |
|
2 |
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.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 |
|
7 |
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.parameters_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.save_checkpoint |
|
3 |
3 |
0 |
|
0% |
| / 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 |
1 |
0 |
|
0% |
| / 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._create_group |
|
2 |
2 |
0 |
|
0% |
| / 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_reduce |
|
4 |
0 |
0 |
|
100% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / platform.py |
TorchPlatform.broadcast |
|
3 |
3 |
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_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.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.arange |
|
3 |
1 |
0 |
|
67% |
| / 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.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 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / platform.py |
TorchPlatform.get_symmetric_memory_handler |
|
3 |
3 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / platform.py |
TorchPlatform.get_multicore_handler |
|
2 |
2 |
0 |
|
0% |
| / 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.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 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / platform.py |
TorchPlatform.get_backend |
|
2 |
2 |
0 |
|
0% |
| / 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.cat |
|
1 |
1 |
0 |
|
0% |
| / 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.new_event |
|
2 |
2 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / platform.py |
TorchPlatform.tree_map |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / platform.py |
TorchPlatform.checkpoint |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / platform.py |
TorchPlatform.ckpt_wrapper |
|
4 |
4 |
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.noop_context_fn |
|
1 |
1 |
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.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.tensor_to_numpy |
|
1 |
0 |
0 |
|
100% |
| / 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 |
10 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / platform.py |
TorchPlatform.list_to_size |
|
1 |
1 |
0 |
|
0% |
| / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / torch / platform.py |
(no function) |
|
229 |
1 |
0 |
|
99% |