| hyper_parallel / __init__.py |
|
12 |
0 |
0 |
|
100% |
| hyper_parallel / collectives / __init__.py |
|
0 |
0 |
0 |
|
100% |
| hyper_parallel / collectives / cc.py |
|
14 |
0 |
0 |
|
100% |
| hyper_parallel / core / __init__.py |
|
0 |
0 |
0 |
|
100% |
| hyper_parallel / core / activation_checkpoint / __init__.py |
|
2 |
0 |
0 |
|
100% |
| hyper_parallel / core / activation_checkpoint / activation_checkpoint.py |
|
17 |
0 |
0 |
|
100% |
| hyper_parallel / core / activation_checkpoint / swap.py |
|
252 |
89 |
0 |
|
65% |
| hyper_parallel / core / checkpoint / __init__.py |
|
11 |
0 |
0 |
|
100% |
| hyper_parallel / core / checkpoint / api.py |
|
83 |
11 |
0 |
|
87% |
| hyper_parallel / core / checkpoint / filesystem_storage.py |
|
172 |
23 |
0 |
|
87% |
| hyper_parallel / core / checkpoint / layout.py |
|
46 |
6 |
0 |
|
87% |
| hyper_parallel / core / checkpoint / loader.py |
|
11 |
1 |
0 |
|
91% |
| hyper_parallel / core / checkpoint / metadata.py |
|
29 |
0 |
0 |
|
100% |
| hyper_parallel / core / checkpoint / planner.py |
|
83 |
2 |
0 |
|
98% |
| hyper_parallel / core / checkpoint / reshard.py |
|
159 |
8 |
0 |
|
95% |
| hyper_parallel / core / checkpoint / saver.py |
|
11 |
1 |
0 |
|
91% |
| hyper_parallel / core / checkpoint / standard_planner.py |
|
212 |
23 |
0 |
|
89% |
| hyper_parallel / core / checkpoint / storage.py |
|
42 |
0 |
0 |
|
100% |
| hyper_parallel / core / checkpoint / util.py |
|
82 |
12 |
0 |
|
85% |
| hyper_parallel / core / device_mesh.py |
|
432 |
132 |
0 |
|
69% |
| hyper_parallel / core / dtensor.py |
|
105 |
7 |
0 |
|
93% |
| hyper_parallel / core / fully_shard / __init__.py |
|
2 |
0 |
0 |
|
100% |
| hyper_parallel / core / fully_shard / api.py |
|
245 |
158 |
0 |
|
36% |
| hyper_parallel / core / fully_shard / hsdp_async_grad_hook.py |
|
88 |
78 |
0 |
|
11% |
| hyper_parallel / core / fully_shard / hsdp_grad_buffer.py |
|
164 |
145 |
0 |
|
12% |
| hyper_parallel / core / fully_shard / hsdp_grad_hook.py |
|
113 |
102 |
0 |
|
10% |
| hyper_parallel / core / fully_shard / hsdp_param.py |
|
60 |
27 |
0 |
|
55% |
| hyper_parallel / core / fully_shard / hsdp_param_buffer.py |
|
84 |
74 |
0 |
|
12% |
| hyper_parallel / core / fully_shard / hsdp_scheduler.py |
|
118 |
50 |
0 |
|
58% |
| hyper_parallel / core / fully_shard / hsdp_state.py |
|
131 |
82 |
0 |
|
37% |
| hyper_parallel / core / fully_shard / hsdp_utils.py |
|
71 |
12 |
0 |
|
83% |
| hyper_parallel / core / hsdp / __init__.py |
|
2 |
0 |
0 |
|
100% |
| hyper_parallel / core / hsdp / api.py |
|
101 |
23 |
0 |
|
77% |
| hyper_parallel / core / hsdp / hsdp_async_grad_hook.py |
|
88 |
21 |
0 |
|
76% |
| hyper_parallel / core / hsdp / hsdp_grad_buffer.py |
|
164 |
76 |
0 |
|
54% |
| hyper_parallel / core / hsdp / hsdp_grad_hook.py |
|
113 |
39 |
0 |
|
65% |
| hyper_parallel / core / hsdp / hsdp_param.py |
|
239 |
16 |
0 |
|
93% |
| hyper_parallel / core / hsdp / hsdp_param_buffer.py |
|
66 |
10 |
0 |
|
85% |
| hyper_parallel / core / hsdp / hsdp_scheduler.py |
|
89 |
9 |
0 |
|
90% |
| hyper_parallel / core / hsdp / hsdp_state.py |
|
111 |
23 |
0 |
|
79% |
| hyper_parallel / core / hsdp / hsdp_utils.py |
|
23 |
0 |
0 |
|
100% |
| hyper_parallel / core / layout.py |
|
334 |
29 |
0 |
|
91% |
| hyper_parallel / core / parameter_init.py |
|
4 |
0 |
0 |
|
100% |
| hyper_parallel / core / pipeline_parallel / __init__.py |
|
4 |
0 |
0 |
|
100% |
| hyper_parallel / core / pipeline_parallel / scheduler.py |
|
476 |
232 |
0 |
|
51% |
| hyper_parallel / core / pipeline_parallel / stage.py |
|
229 |
64 |
0 |
|
72% |
| hyper_parallel / core / pipeline_parallel / utils.py |
|
33 |
11 |
0 |
|
67% |
| hyper_parallel / core / placement_types.py |
|
61 |
25 |
0 |
|
59% |
| hyper_parallel / core / random.py |
|
185 |
42 |
0 |
|
77% |
| hyper_parallel / core / redistribute_infer.py |
|
331 |
128 |
0 |
|
61% |
| hyper_parallel / core / shard / __init__.py |
|
0 |
0 |
0 |
|
100% |
| hyper_parallel / core / shard / _op_dispatch.py |
|
441 |
98 |
0 |
|
78% |
| hyper_parallel / core / shard / api.py |
|
305 |
69 |
0 |
|
77% |
| hyper_parallel / core / shard / local_func.py |
|
48 |
8 |
0 |
|
83% |
| hyper_parallel / core / shard / ops / __init__.py |
|
0 |
0 |
0 |
|
100% |
| hyper_parallel / core / shard / ops / parallel_activation_with_axis.py |
|
30 |
3 |
0 |
|
90% |
| hyper_parallel / core / shard / ops / parallel_argmax_with_value_ops.py |
|
34 |
31 |
0 |
|
9% |
| hyper_parallel / core / shard / ops / parallel_concat.py |
|
32 |
28 |
0 |
|
12% |
| hyper_parallel / core / shard / ops / parallel_cumsum.py |
|
30 |
4 |
0 |
|
87% |
| hyper_parallel / core / shard / ops / parallel_elementwise.py |
|
245 |
63 |
0 |
|
74% |
| hyper_parallel / core / shard / ops / parallel_embedding.py |
|
26 |
1 |
0 |
|
96% |
| hyper_parallel / core / shard / ops / parallel_expand.py |
|
88 |
12 |
1 |
|
86% |
| hyper_parallel / core / shard / ops / parallel_expand_dims.py |
|
26 |
6 |
0 |
|
77% |
| hyper_parallel / core / shard / ops / parallel_flash_attention_score.py |
|
111 |
21 |
0 |
|
81% |
| hyper_parallel / core / shard / ops / parallel_gather.py |
|
131 |
65 |
0 |
|
50% |
| hyper_parallel / core / shard / ops / parallel_isin.py |
|
24 |
2 |
0 |
|
92% |
| hyper_parallel / core / shard / ops / parallel_masked_scatter.py |
|
12 |
1 |
0 |
|
92% |
| hyper_parallel / core / shard / ops / parallel_matmul.py |
|
193 |
44 |
0 |
|
77% |
| hyper_parallel / core / shard / ops / parallel_multinomial.py |
|
19 |
1 |
0 |
|
95% |
| hyper_parallel / core / shard / ops / parallel_new_ones.py |
|
16 |
2 |
0 |
|
88% |
| hyper_parallel / core / shard / ops / parallel_norm.py |
|
70 |
38 |
0 |
|
46% |
| hyper_parallel / core / shard / ops / parallel_one_hot_ext.py |
|
169 |
149 |
0 |
|
12% |
| hyper_parallel / core / shard / ops / parallel_ops.py |
|
18 |
6 |
0 |
|
67% |
| hyper_parallel / core / shard / ops / parallel_ops_register.py |
|
5 |
0 |
0 |
|
100% |
| hyper_parallel / core / shard / ops / parallel_pad.py |
|
32 |
7 |
0 |
|
78% |
| hyper_parallel / core / shard / ops / parallel_reduce.py |
|
155 |
24 |
0 |
|
85% |
| hyper_parallel / core / shard / ops / parallel_repeat_interleave.py |
|
33 |
5 |
0 |
|
85% |
| hyper_parallel / core / shard / ops / parallel_reshape.py |
|
141 |
41 |
0 |
|
71% |
| hyper_parallel / core / shard / ops / parallel_scatter_update.py |
|
41 |
11 |
0 |
|
73% |
| hyper_parallel / core / shard / ops / parallel_slice.py |
|
34 |
29 |
0 |
|
15% |
| hyper_parallel / core / shard / ops / parallel_slice_ext.py |
|
9 |
6 |
0 |
|
33% |
| hyper_parallel / core / shard / ops / parallel_sort.py |
|
25 |
4 |
0 |
|
84% |
| hyper_parallel / core / shard / ops / parallel_split.py |
|
81 |
39 |
0 |
|
52% |
| hyper_parallel / core / shard / ops / parallel_squeeze.py |
|
91 |
20 |
0 |
|
78% |
| hyper_parallel / core / shard / ops / parallel_topk.py |
|
18 |
2 |
0 |
|
89% |
| hyper_parallel / core / shard / ops / parallel_transpose.py |
|
38 |
8 |
0 |
|
79% |
| hyper_parallel / core / shard / ops / parallel_tuple_elementwise.py |
|
6 |
1 |
0 |
|
83% |
| hyper_parallel / core / shard / ops / parallel_unbind.py |
|
23 |
1 |
0 |
|
96% |
| hyper_parallel / core / shard / sharding_plan.py |
|
8 |
0 |
0 |
|
100% |
| hyper_parallel / core / tensor_redistribution.py |
|
209 |
32 |
0 |
|
85% |
| hyper_parallel / core / utils.py |
|
14 |
0 |
0 |
|
100% |
| hyper_parallel / platform / __init__.py |
|
1 |
0 |
0 |
|
100% |
| hyper_parallel / platform / mindspore / __init__.py |
|
0 |
0 |
0 |
|
100% |
| hyper_parallel / platform / mindspore / custom_pass / __init__.py |
|
7 |
0 |
0 |
|
100% |
| hyper_parallel / platform / mindspore / custom_pass / build_plugin.py |
|
56 |
56 |
0 |
|
0% |
| hyper_parallel / platform / mindspore / dtensor.py |
|
89 |
19 |
0 |
|
79% |
| hyper_parallel / platform / mindspore / hsdp / __init__.py |
|
0 |
0 |
0 |
|
100% |
| hyper_parallel / platform / mindspore / hsdp / async_grad_hook.py |
|
9 |
6 |
0 |
|
33% |
| hyper_parallel / platform / mindspore / hsdp / grad_hook.py |
|
31 |
8 |
0 |
|
74% |
| hyper_parallel / platform / mindspore / hsdp / param.py |
|
29 |
0 |
0 |
|
100% |
| hyper_parallel / platform / mindspore / hsdp / scheduler.py |
|
109 |
55 |
0 |
|
50% |
| hyper_parallel / platform / mindspore / hsdp / state.py |
|
36 |
1 |
0 |
|
97% |
| hyper_parallel / platform / mindspore / parameter_init.py |
|
32 |
3 |
0 |
|
91% |
| hyper_parallel / platform / mindspore / pipeline_parallel / __init__.py |
|
0 |
0 |
0 |
|
100% |
| hyper_parallel / platform / mindspore / pipeline_parallel / _utils.py |
|
80 |
80 |
0 |
|
0% |
| hyper_parallel / platform / mindspore / pipeline_parallel / stage.py |
|
79 |
66 |
0 |
|
16% |
| hyper_parallel / platform / mindspore / platform.py |
|
347 |
66 |
0 |
|
81% |
| hyper_parallel / platform / mindspore / platform_graph.py |
|
27 |
10 |
0 |
|
63% |
| hyper_parallel / platform / platform.py |
|
259 |
69 |
0 |
|
73% |
| hyper_parallel / platform / torch / __init__.py |
|
0 |
0 |
0 |
|
100% |
| hyper_parallel / platform / torch / activation_checkpoint / __init__.py |
|
1 |
0 |
0 |
|
100% |
| hyper_parallel / platform / torch / activation_checkpoint / activation_swap.py |
|
86 |
17 |
0 |
|
80% |
| hyper_parallel / platform / torch / activation_checkpoint / sac.py |
|
105 |
41 |
0 |
|
61% |
| hyper_parallel / platform / torch / dtensor.py |
|
121 |
51 |
0 |
|
58% |
| hyper_parallel / platform / torch / fully_shard / __init__.py |
|
0 |
0 |
0 |
|
100% |
| hyper_parallel / platform / torch / fully_shard / async_grad_hook.py |
|
10 |
10 |
0 |
|
0% |
| hyper_parallel / platform / torch / fully_shard / grad_hook.py |
|
10 |
10 |
0 |
|
0% |
| hyper_parallel / platform / torch / fully_shard / hook_function.py |
|
38 |
11 |
0 |
|
71% |
| hyper_parallel / platform / torch / fully_shard / param.py |
|
336 |
162 |
0 |
|
52% |
| hyper_parallel / platform / torch / fully_shard / scheduler.py |
|
63 |
12 |
0 |
|
81% |
| hyper_parallel / platform / torch / fully_shard / state.py |
|
118 |
62 |
0 |
|
47% |
| hyper_parallel / platform / torch / fully_shard / utils.py |
|
46 |
9 |
0 |
|
80% |
| hyper_parallel / platform / torch / function_override.py |
|
44 |
12 |
0 |
|
73% |
| hyper_parallel / platform / torch / group_utils.py |
|
86 |
18 |
0 |
|
79% |
| hyper_parallel / platform / torch / hsdp / __init__.py |
|
0 |
0 |
0 |
|
100% |
| hyper_parallel / platform / torch / hsdp / async_grad_hook.py |
|
10 |
4 |
0 |
|
60% |
| hyper_parallel / platform / torch / hsdp / grad_hook.py |
|
10 |
4 |
0 |
|
60% |
| hyper_parallel / platform / torch / hsdp / hook_function.py |
|
37 |
7 |
0 |
|
81% |
| hyper_parallel / platform / torch / hsdp / param.py |
|
22 |
0 |
0 |
|
100% |
| hyper_parallel / platform / torch / hsdp / scheduler.py |
|
60 |
11 |
0 |
|
82% |
| hyper_parallel / platform / torch / hsdp / state.py |
|
14 |
1 |
0 |
|
93% |
| hyper_parallel / platform / torch / pipeline_parallel / __init__.py |
|
0 |
0 |
0 |
|
100% |
| hyper_parallel / platform / torch / pipeline_parallel / _utils.py |
|
49 |
10 |
0 |
|
80% |
| hyper_parallel / platform / torch / pipeline_parallel / stage.py |
|
75 |
10 |
0 |
|
87% |
| hyper_parallel / platform / torch / platform.py |
|
372 |
68 |
0 |
|
82% |