Skip to content

Commit f595696

Browse files
TelGomeTelGome
committed
Add missing reifier of model.
1 parent 1deb685 commit f595696

File tree

282 files changed

+2584
-281
lines changed

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

282 files changed

+2584
-281
lines changed

‎graph_net/torch/sym_dim_reifiers/naive_cv_sym_dim_reifier.py‎

Lines changed: 413 additions & 0 deletions
Large diffs are not rendered by default.

‎graph_net/torch/sym_dim_reifiers/naive_nlp_sym_dim_reifier.py‎

Lines changed: 15 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -40,6 +40,9 @@ def _get_map_nlp_sym_shapes_str2reifier(cls):
4040
"[(S0,S1),(S0,S1)]": cls.reify_batch_s0_seq_s1,
4141
"[(S0,S1,768)]": cls.reify_batch_s0_seq_s1,
4242
"[(S0,S1)]": cls.reify_nlp_or_gnn_batch_s0_seq_s1,
43+
"[(S0,1),(S0,64),(S0,64)]": cls.reify_s0_1_64,
44+
"[(S0,12),(S0,12,512),(S0,12,512)]": cls.reify_s0_1_64,
45+
"[(S0,16),(S0,16)]": cls.reify_s0_16,
4346
}
4447
return cls.g_nlp_sym_shapes_str2reifier
4548

@@ -74,3 +77,15 @@ def reify_nlp_or_gnn_batch_s0_seq_s1(self):
7477
[128, 64],
7578
],
7679
}
80+
81+
def reify_s0_1_64(self):
82+
S0 = (sympy.Symbol("S0"),)
83+
return {
84+
S0: [[1], [2], [4], [8], [16], [24], [32], [48], [64]],
85+
}
86+
87+
def reify_s0_16(self):
88+
S0 = (sympy.Symbol("S0"),)
89+
return {
90+
S0: [[1], [2], [4], [8], [16], [32], [48], [64], [128]],
91+
}

‎samples/cosyvoice/CosyVoice-300M/graph_net.json‎

Lines changed: 21 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1 +1,21 @@
1-
{"framework": "torch", "num_devices_required": 1, "num_nodes_required": 1, "dynamic": false, "source": "cosyvoice", "heuristic_tag": "audio", "dimension_generalization_passes": ["naive_call_method_view_pass", "naive_call_method_reshape_pass", "naive_call_method_expand_pass", "non_batch_call_method_expand_pass", "non_batch_call_function_arange_pass", "non_batch_call_function_getitem_slice_pass", "non_batch_call_function_full_pass", "non_batch_call_function_full_plus_one_pass", "non_batch_call_function_zeros_pass", "non_batch_call_function_arange_plus_one_pass"]}
1+
{
2+
"framework": "torch",
3+
"num_devices_required": 1,
4+
"num_nodes_required": 1,
5+
"dynamic": false,
6+
"source": "cosyvoice",
7+
"heuristic_tag": "audio",
8+
"dimension_generalization_passes": [
9+
"naive_call_method_view_pass",
10+
"naive_call_method_reshape_pass",
11+
"naive_call_method_expand_pass",
12+
"non_batch_call_method_expand_pass",
13+
"non_batch_call_function_arange_pass",
14+
"non_batch_call_function_getitem_slice_pass",
15+
"non_batch_call_function_full_pass",
16+
"non_batch_call_function_full_plus_one_pass",
17+
"non_batch_call_function_zeros_pass",
18+
"non_batch_call_function_arange_plus_one_pass"
19+
],
20+
"symbolic_dimension_reifier": "naive_cv_sym_dim_reifier"
21+
}

‎samples/cosyvoice/CosyVoice2-0.5B/graph_net.json‎

Lines changed: 21 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1 +1,21 @@
1-
{"framework": "torch", "num_devices_required": 1, "num_nodes_required": 1, "dynamic": false, "source": "cosyvoice", "heuristic_tag": "audio", "dimension_generalization_passes": ["naive_call_method_view_pass", "naive_call_method_reshape_pass", "naive_call_method_expand_pass", "non_batch_call_method_expand_pass", "non_batch_call_function_arange_pass", "non_batch_call_function_getitem_slice_pass", "non_batch_call_function_full_pass", "non_batch_call_function_full_plus_one_pass", "non_batch_call_function_zeros_pass", "non_batch_call_function_arange_plus_one_pass"]}
1+
{
2+
"framework": "torch",
3+
"num_devices_required": 1,
4+
"num_nodes_required": 1,
5+
"dynamic": false,
6+
"source": "cosyvoice",
7+
"heuristic_tag": "audio",
8+
"dimension_generalization_passes": [
9+
"naive_call_method_view_pass",
10+
"naive_call_method_reshape_pass",
11+
"naive_call_method_expand_pass",
12+
"non_batch_call_method_expand_pass",
13+
"non_batch_call_function_arange_pass",
14+
"non_batch_call_function_getitem_slice_pass",
15+
"non_batch_call_function_full_pass",
16+
"non_batch_call_function_full_plus_one_pass",
17+
"non_batch_call_function_zeros_pass",
18+
"non_batch_call_function_arange_plus_one_pass"
19+
],
20+
"symbolic_dimension_reifier": "naive_cv_sym_dim_reifier"
21+
}

‎samples/mmpose/SimpleBaseline3D/graph_net.json‎

Lines changed: 21 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1 +1,21 @@
1-
{"framework": "torch", "num_devices_required": 1, "num_nodes_required": 1, "dynamic": false, "source": "mmpose", "heuristic_tag": "computer_vision", "dimension_generalization_passes": ["naive_call_method_view_pass", "naive_call_method_reshape_pass", "naive_call_method_expand_pass", "non_batch_call_method_expand_pass", "non_batch_call_function_arange_pass", "non_batch_call_function_getitem_slice_pass", "non_batch_call_function_full_pass", "non_batch_call_function_full_plus_one_pass", "non_batch_call_function_zeros_pass", "non_batch_call_function_arange_plus_one_pass"]}
1+
{
2+
"framework": "torch",
3+
"num_devices_required": 1,
4+
"num_nodes_required": 1,
5+
"dynamic": false,
6+
"source": "mmpose",
7+
"heuristic_tag": "computer_vision",
8+
"dimension_generalization_passes": [
9+
"naive_call_method_view_pass",
10+
"naive_call_method_reshape_pass",
11+
"naive_call_method_expand_pass",
12+
"non_batch_call_method_expand_pass",
13+
"non_batch_call_function_arange_pass",
14+
"non_batch_call_function_getitem_slice_pass",
15+
"non_batch_call_function_full_pass",
16+
"non_batch_call_function_full_plus_one_pass",
17+
"non_batch_call_function_zeros_pass",
18+
"non_batch_call_function_arange_plus_one_pass"
19+
],
20+
"symbolic_dimension_reifier": "naive_cv_sym_dim_reifier"
21+
}

‎samples/mmpose/pose_swin_b/graph_net.json‎

Lines changed: 21 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1 +1,21 @@
1-
{"framework": "torch", "num_devices_required": 1, "num_nodes_required": 1, "dynamic": false, "source": "mmpose", "heuristic_tag": "computer_vision", "dimension_generalization_passes": ["naive_call_method_view_pass", "naive_call_method_reshape_pass", "naive_call_method_expand_pass", "non_batch_call_method_expand_pass", "non_batch_call_function_arange_pass", "non_batch_call_function_getitem_slice_pass", "non_batch_call_function_full_pass", "non_batch_call_function_full_plus_one_pass", "non_batch_call_function_zeros_pass", "non_batch_call_function_arange_plus_one_pass"]}
1+
{
2+
"framework": "torch",
3+
"num_devices_required": 1,
4+
"num_nodes_required": 1,
5+
"dynamic": false,
6+
"source": "mmpose",
7+
"heuristic_tag": "computer_vision",
8+
"dimension_generalization_passes": [
9+
"naive_call_method_view_pass",
10+
"naive_call_method_reshape_pass",
11+
"naive_call_method_expand_pass",
12+
"non_batch_call_method_expand_pass",
13+
"non_batch_call_function_arange_pass",
14+
"non_batch_call_function_getitem_slice_pass",
15+
"non_batch_call_function_full_pass",
16+
"non_batch_call_function_full_plus_one_pass",
17+
"non_batch_call_function_zeros_pass",
18+
"non_batch_call_function_arange_plus_one_pass"
19+
],
20+
"symbolic_dimension_reifier": "naive_cv_sym_dim_reifier"
21+
}

‎samples/mmpose/pose_swin_l/graph_net.json‎

Lines changed: 21 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1 +1,21 @@
1-
{"framework": "torch", "num_devices_required": 1, "num_nodes_required": 1, "dynamic": false, "source": "mmpose", "heuristic_tag": "computer_vision", "dimension_generalization_passes": ["naive_call_method_view_pass", "naive_call_method_reshape_pass", "naive_call_method_expand_pass", "non_batch_call_method_expand_pass", "non_batch_call_function_arange_pass", "non_batch_call_function_getitem_slice_pass", "non_batch_call_function_full_pass", "non_batch_call_function_full_plus_one_pass", "non_batch_call_function_zeros_pass", "non_batch_call_function_arange_plus_one_pass"]}
1+
{
2+
"framework": "torch",
3+
"num_devices_required": 1,
4+
"num_nodes_required": 1,
5+
"dynamic": false,
6+
"source": "mmpose",
7+
"heuristic_tag": "computer_vision",
8+
"dimension_generalization_passes": [
9+
"naive_call_method_view_pass",
10+
"naive_call_method_reshape_pass",
11+
"naive_call_method_expand_pass",
12+
"non_batch_call_method_expand_pass",
13+
"non_batch_call_function_arange_pass",
14+
"non_batch_call_function_getitem_slice_pass",
15+
"non_batch_call_function_full_pass",
16+
"non_batch_call_function_full_plus_one_pass",
17+
"non_batch_call_function_zeros_pass",
18+
"non_batch_call_function_arange_plus_one_pass"
19+
],
20+
"symbolic_dimension_reifier": "naive_cv_sym_dim_reifier"
21+
}

‎samples/mmpose/pose_swin_t/graph_net.json‎

Lines changed: 21 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1 +1,21 @@
1-
{"framework": "torch", "num_devices_required": 1, "num_nodes_required": 1, "dynamic": false, "source": "mmpose", "heuristic_tag": "computer_vision", "dimension_generalization_passes": ["naive_call_method_view_pass", "naive_call_method_reshape_pass", "naive_call_method_expand_pass", "non_batch_call_method_expand_pass", "non_batch_call_function_arange_pass", "non_batch_call_function_getitem_slice_pass", "non_batch_call_function_full_pass", "non_batch_call_function_full_plus_one_pass", "non_batch_call_function_zeros_pass", "non_batch_call_function_arange_plus_one_pass"]}
1+
{
2+
"framework": "torch",
3+
"num_devices_required": 1,
4+
"num_nodes_required": 1,
5+
"dynamic": false,
6+
"source": "mmpose",
7+
"heuristic_tag": "computer_vision",
8+
"dimension_generalization_passes": [
9+
"naive_call_method_view_pass",
10+
"naive_call_method_reshape_pass",
11+
"naive_call_method_expand_pass",
12+
"non_batch_call_method_expand_pass",
13+
"non_batch_call_function_arange_pass",
14+
"non_batch_call_function_getitem_slice_pass",
15+
"non_batch_call_function_full_pass",
16+
"non_batch_call_function_full_plus_one_pass",
17+
"non_batch_call_function_zeros_pass",
18+
"non_batch_call_function_arange_plus_one_pass"
19+
],
20+
"symbolic_dimension_reifier": "naive_cv_sym_dim_reifier"
21+
}

‎samples/nemo/parakeet-tdt-0.6b-v3/graph_net.json‎

Lines changed: 21 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1 +1,21 @@
1-
{"framework": "torch", "num_devices_required": 1, "num_nodes_required": 1, "dynamic": false, "source": "nemo", "heuristic_tag": "audio", "dimension_generalization_passes": ["naive_call_method_view_pass", "naive_call_method_reshape_pass", "naive_call_method_expand_pass", "non_batch_call_method_expand_pass", "non_batch_call_function_arange_pass", "non_batch_call_function_getitem_slice_pass", "non_batch_call_function_full_pass", "non_batch_call_function_full_plus_one_pass", "non_batch_call_function_zeros_pass", "non_batch_call_function_arange_plus_one_pass"]}
1+
{
2+
"framework": "torch",
3+
"num_devices_required": 1,
4+
"num_nodes_required": 1,
5+
"dynamic": false,
6+
"source": "nemo",
7+
"heuristic_tag": "audio",
8+
"dimension_generalization_passes": [
9+
"naive_call_method_view_pass",
10+
"naive_call_method_reshape_pass",
11+
"naive_call_method_expand_pass",
12+
"non_batch_call_method_expand_pass",
13+
"non_batch_call_function_arange_pass",
14+
"non_batch_call_function_getitem_slice_pass",
15+
"non_batch_call_function_full_pass",
16+
"non_batch_call_function_full_plus_one_pass",
17+
"non_batch_call_function_zeros_pass",
18+
"non_batch_call_function_arange_plus_one_pass"
19+
],
20+
"symbolic_dimension_reifier": "naive_cv_sym_dim_reifier"
21+
}

‎samples/nemo/stt_en_fastconformer_hybrid_large_streaming_multi/graph_net.json‎

Lines changed: 21 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1 +1,21 @@
1-
{"framework": "torch", "num_devices_required": 1, "num_nodes_required": 1, "dynamic": false, "source": "nemo", "heuristic_tag": "audio", "dimension_generalization_passes": ["naive_call_method_view_pass", "naive_call_method_reshape_pass", "naive_call_method_expand_pass", "non_batch_call_method_expand_pass", "non_batch_call_function_arange_pass", "non_batch_call_function_getitem_slice_pass", "non_batch_call_function_full_pass", "non_batch_call_function_full_plus_one_pass", "non_batch_call_function_zeros_pass", "non_batch_call_function_arange_plus_one_pass"]}
1+
{
2+
"framework": "torch",
3+
"num_devices_required": 1,
4+
"num_nodes_required": 1,
5+
"dynamic": false,
6+
"source": "nemo",
7+
"heuristic_tag": "audio",
8+
"dimension_generalization_passes": [
9+
"naive_call_method_view_pass",
10+
"naive_call_method_reshape_pass",
11+
"naive_call_method_expand_pass",
12+
"non_batch_call_method_expand_pass",
13+
"non_batch_call_function_arange_pass",
14+
"non_batch_call_function_getitem_slice_pass",
15+
"non_batch_call_function_full_pass",
16+
"non_batch_call_function_full_plus_one_pass",
17+
"non_batch_call_function_zeros_pass",
18+
"non_batch_call_function_arange_plus_one_pass"
19+
],
20+
"symbolic_dimension_reifier": "naive_cv_sym_dim_reifier"
21+
}

0 commit comments

Comments
 (0)