Update input constraints and json.

Xreki · Xreki · commit 7fd77022718b · 2025-12-30T23:26:47.000+08:00
diff --git a/graph_net/tools/init_input_tensor_constraints.sh b/graph_net/tools/init_input_tensor_constraints.sh
@@ -15,7 +15,7 @@ python3 -m graph_net.model_path_handler \
     "handler_path": "$GRAPH_NET_ROOT/constraint_util.py",
     "handler_class_name": "UpdateInputTensorConstraints",
     "handler_config": {
-        "resume": true,
+        "resume": false,
         "model_path_prefix": "$GRAPH_NET_ROOT/../",
         "data_input_predicator_filepath": "$GRAPH_NET_ROOT/torch/constraint_util.py",
         "data_input_predicator_class_name": "NaiveDataInputPredicator",
@@ -30,6 +30,7 @@ python3 -m graph_net.model_path_handler \
                 "naive_call_method_reshape_pass",
                 "naive_call_method_expand_pass",
                 "non_batch_call_method_expand_pass",
+                "non_batch_call_method_view_pass",
                 "non_batch_call_function_arange_pass",
                 "non_batch_call_function_getitem_slice_pass",
                 "non_batch_call_function_full_pass",
diff --git a/samples/mmseg/PoolFormer_m36/graph_net.json b/samples/mmseg/PoolFormer_m36/graph_net.json
@@ -5,6 +5,6 @@
     "dynamic": false,
     "source": "mmseg",
     "heuristic_tag": "computer_vision",
-    "dimension_generalization_passes": [],
-    "symbolic_dimension_reifier": "naive_cv_sym_dim_reifier"
+    "symbolic_dimension_reifier": "naive_cv_sym_dim_reifier",
+    "dimension_generalization_passes": []
 }
diff --git a/samples/mmseg/PoolFormer_m36/input_tensor_constraints.py b/samples/mmseg/PoolFormer_m36/input_tensor_constraints.py
@@ -1,4 +1,4 @@
-from sympy import Symbol
+from sympy import Symbol, Expr, Rel, Eq
 
 S0 = Symbol("S0")
 S1 = Symbol("S1")
diff --git a/samples/mmseg/Twins_pcpvt-b_uperhead/graph_net.json b/samples/mmseg/Twins_pcpvt-b_uperhead/graph_net.json
@@ -5,17 +5,19 @@
     "dynamic": false,
     "source": "mmseg",
     "heuristic_tag": "computer_vision",
+    "symbolic_dimension_reifier": "naive_cv_sym_dim_reifier",
     "dimension_generalization_passes": [
-        "naive_call_method_view_pass",
+        "batch_call_method_view_pass",
+        "tuple_arg_call_method_view_pass",
         "naive_call_method_reshape_pass",
         "naive_call_method_expand_pass",
         "non_batch_call_method_expand_pass",
+        "non_batch_call_method_view_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"
-    ],
-    "symbolic_dimension_reifier": "naive_cv_sym_dim_reifier"
+    ]
 }
diff --git a/samples/mmseg/Twins_pcpvt-b_uperhead/input_tensor_constraints.py b/samples/mmseg/Twins_pcpvt-b_uperhead/input_tensor_constraints.py
@@ -1,4 +1,4 @@
-from sympy import Symbol
+from sympy import Symbol, Expr, Rel, Eq
 
 S0 = Symbol("S0")
 S1 = Symbol("S1")
diff --git a/samples/timm/eva02_base_patch16_clip_224.merged2b/graph_net.json b/samples/timm/eva02_base_patch16_clip_224.merged2b/graph_net.json
@@ -4,5 +4,19 @@
     "num_nodes_required": 1,
     "source": "timm",
     "heuristic_tag": "computer_vision",
-    "symbolic_dimension_reifier": "naive_cv_sym_dim_reifier"
+    "symbolic_dimension_reifier": "naive_cv_sym_dim_reifier",
+    "dimension_generalization_passes": [
+        "batch_call_method_view_pass",
+        "tuple_arg_call_method_view_pass",
+        "naive_call_method_reshape_pass",
+        "naive_call_method_expand_pass",
+        "non_batch_call_method_expand_pass",
+        "non_batch_call_method_view_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"
+    ]
 }
diff --git a/samples/timm/eva02_base_patch16_clip_224.merged2b/input_tensor_constraints.py b/samples/timm/eva02_base_patch16_clip_224.merged2b/input_tensor_constraints.py
@@ -1,4 +1,4 @@
-from sympy import Symbol
+from sympy import Symbol, Expr, Rel, Eq
 
 S0 = Symbol("S0")
 
diff --git a/samples/timm/vit_base_patch16_224_miil.in21k/graph_net.json b/samples/timm/vit_base_patch16_224_miil.in21k/graph_net.json
@@ -5,5 +5,19 @@
     "dynamic": true,
     "source": "timm",
     "heuristic_tag": "computer_vision",
-    "symbolic_dimension_reifier": "naive_cv_sym_dim_reifier"
+    "symbolic_dimension_reifier": "naive_cv_sym_dim_reifier",
+    "dimension_generalization_passes": [
+        "batch_call_method_view_pass",
+        "tuple_arg_call_method_view_pass",
+        "naive_call_method_reshape_pass",
+        "naive_call_method_expand_pass",
+        "non_batch_call_method_expand_pass",
+        "non_batch_call_method_view_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"
+    ]
 }
diff --git a/samples/timm/vit_base_patch16_224_miil.in21k/input_tensor_constraints.py b/samples/timm/vit_base_patch16_224_miil.in21k/input_tensor_constraints.py
@@ -1,4 +1,4 @@
-from sympy import Symbol
+from sympy import Symbol, Expr, Rel, Eq
 
 S0 = Symbol("S0")
 
diff --git a/samples/timm/vit_small_patch16_rope_mixed_ape_224.naver_in1k/graph_net.json b/samples/timm/vit_small_patch16_rope_mixed_ape_224.naver_in1k/graph_net.json
@@ -5,5 +5,19 @@
     "dynamic": true,
     "source": "timm",
     "heuristic_tag": "computer_vision",
-    "symbolic_dimension_reifier": "naive_cv_sym_dim_reifier"
+    "symbolic_dimension_reifier": "naive_cv_sym_dim_reifier",
+    "dimension_generalization_passes": [
+        "batch_call_method_view_pass",
+        "tuple_arg_call_method_view_pass",
+        "naive_call_method_reshape_pass",
+        "naive_call_method_expand_pass",
+        "non_batch_call_method_expand_pass",
+        "non_batch_call_method_view_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"
+    ]
 }
diff --git a/samples/timm/vit_small_patch16_rope_mixed_ape_224.naver_in1k/input_tensor_constraints.py b/samples/timm/vit_small_patch16_rope_mixed_ape_224.naver_in1k/input_tensor_constraints.py
@@ -1,4 +1,4 @@
-from sympy import Symbol
+from sympy import Symbol, Expr, Rel, Eq
 
 S0 = Symbol("S0")
 
diff --git a/samples/torchaudio/wavlm_base/graph_net.json b/samples/torchaudio/wavlm_base/graph_net.json
@@ -4,18 +4,19 @@
     "num_nodes_required": 1,
     "source": "torchaudio",
     "heuristic_tag": "audio",
+    "symbolic_dimension_reifier": "naive_nlp_sym_dim_reifier",
     "dimension_generalization_passes": [
         "batch_call_method_view_pass",
         "tuple_arg_call_method_view_pass",
         "naive_call_method_reshape_pass",
         "naive_call_method_expand_pass",
         "non_batch_call_method_expand_pass",
+        "non_batch_call_method_view_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"
-    ],
-    "symbolic_dimension_reifier": "naive_nlp_sym_dim_reifier"
+    ]
 }
diff --git a/samples/torchvision/vit_l_16/graph_net.json b/samples/torchvision/vit_l_16/graph_net.json
@@ -5,17 +5,19 @@
     "dynamic": false,
     "source": "torchvision",
     "heuristic_tag": "computer_vision",
+    "symbolic_dimension_reifier": "naive_cv_sym_dim_reifier",
     "dimension_generalization_passes": [
-        "naive_call_method_view_pass",
+        "batch_call_method_view_pass",
+        "tuple_arg_call_method_view_pass",
         "naive_call_method_reshape_pass",
         "naive_call_method_expand_pass",
         "non_batch_call_method_expand_pass",
+        "non_batch_call_method_view_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"
-    ],
-    "symbolic_dimension_reifier": "naive_cv_sym_dim_reifier"
+    ]
 }
diff --git a/samples/torchvision/vit_l_16/input_tensor_constraints.py b/samples/torchvision/vit_l_16/input_tensor_constraints.py
@@ -1,4 +1,4 @@
-from sympy import Symbol
+from sympy import Symbol, Expr, Rel, Eq
 
 S0 = Symbol("S0")
 
diff --git a/samples/transformers-auto-model/Lin-Chen_ShareGPT4V-7B_Pretrained_vit-large336-l12/graph_net.json b/samples/transformers-auto-model/Lin-Chen_ShareGPT4V-7B_Pretrained_vit-large336-l12/graph_net.json
@@ -13,5 +13,19 @@
         "region:us"
     ],
     "heuristic_tag": "other",
-    "symbolic_dimension_reifier": "naive_cv_sym_dim_reifier"
+    "symbolic_dimension_reifier": "naive_cv_sym_dim_reifier",
+    "dimension_generalization_passes": [
+        "batch_call_method_view_pass",
+        "tuple_arg_call_method_view_pass",
+        "naive_call_method_reshape_pass",
+        "naive_call_method_expand_pass",
+        "non_batch_call_method_expand_pass",
+        "non_batch_call_method_view_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"
+    ]
 }
diff --git a/samples/transformers-auto-model/Lin-Chen_ShareGPT4V-7B_Pretrained_vit-large336-l12/input_tensor_constraints.py b/samples/transformers-auto-model/Lin-Chen_ShareGPT4V-7B_Pretrained_vit-large336-l12/input_tensor_constraints.py
@@ -1,4 +1,4 @@
-from sympy import Symbol
+from sympy import Symbol, Expr, Rel, Eq
 
 S0 = Symbol("S0")
 
diff --git a/samples/transformers-auto-model/OFA-Sys_chinese-clip-vit-large-patch14/graph_net.json b/samples/transformers-auto-model/OFA-Sys_chinese-clip-vit-large-patch14/graph_net.json
@@ -16,5 +16,19 @@
         "region:us"
     ],
     "heuristic_tag": "multimodal",
-    "symbolic_dimension_reifier": "naive_cv_sym_dim_reifier"
+    "symbolic_dimension_reifier": "naive_cv_sym_dim_reifier",
+    "dimension_generalization_passes": [
+        "batch_call_method_view_pass",
+        "tuple_arg_call_method_view_pass",
+        "naive_call_method_reshape_pass",
+        "naive_call_method_expand_pass",
+        "non_batch_call_method_expand_pass",
+        "non_batch_call_method_view_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"
+    ]
 }
diff --git a/samples/transformers-auto-model/OFA-Sys_chinese-clip-vit-large-patch14/input_tensor_constraints.py b/samples/transformers-auto-model/OFA-Sys_chinese-clip-vit-large-patch14/input_tensor_constraints.py
@@ -1,18 +1,20 @@
-from sympy import Symbol
+from sympy import Symbol, Expr, Rel, Eq
 
 S0 = Symbol("S0")
+S1 = Symbol("S1")
+S2 = Symbol("S2")
 
-dynamic_dim_constraint_symbols = [S0]
+dynamic_dim_constraint_symbols = [S0, S1, S2]
 
-dynamic_dim_constraint_symbol2example_value = {S0: 224}
+dynamic_dim_constraint_symbol2example_value = {S0: 2, S1: 7, S2: 224}
 
 dynamic_dim_constraint_relations = []
 
 dynamic_dim_constraint_input_shapes = [
-    ([2, 7], "L_kwargs_attention_mask_"),
-    ([2, 7], "L_kwargs_input_ids_"),
-    ([1, 3, S0, S0], "L_kwargs_pixel_values_"),
-    ([2, 7], "L_kwargs_token_type_ids_"),
+    ([S0, S1], "L_kwargs_attention_mask_"),
+    ([S0, S1], "L_kwargs_input_ids_"),
+    ([1, 3, S2, S2], "L_kwargs_pixel_values_"),
+    ([S0, S1], "L_kwargs_token_type_ids_"),
     ([1, 512], "L_self_modules_text_model_modules_embeddings_buffers_position_ids_"),
     (
         [768],
diff --git a/samples/transformers-auto-model/apple_aimv2-huge-patch14-224/graph_net.json b/samples/transformers-auto-model/apple_aimv2-huge-patch14-224/graph_net.json
@@ -22,5 +22,19 @@
         "region:us"
     ],
     "heuristic_tag": "other",
-    "symbolic_dimension_reifier": "naive_cv_sym_dim_reifier"
+    "symbolic_dimension_reifier": "naive_cv_sym_dim_reifier",
+    "dimension_generalization_passes": [
+        "batch_call_method_view_pass",
+        "tuple_arg_call_method_view_pass",
+        "naive_call_method_reshape_pass",
+        "naive_call_method_expand_pass",
+        "non_batch_call_method_expand_pass",
+        "non_batch_call_method_view_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"
+    ]
 }
diff --git a/samples/transformers-auto-model/apple_aimv2-huge-patch14-224/input_tensor_constraints.py b/samples/transformers-auto-model/apple_aimv2-huge-patch14-224/input_tensor_constraints.py
@@ -1,15 +1,16 @@
-from sympy import Symbol
+from sympy import Symbol, Expr, Rel, Eq
 
 S0 = Symbol("S0")
+S1 = Symbol("S1")
 
-dynamic_dim_constraint_symbols = [S0]
+dynamic_dim_constraint_symbols = [S0, S1]
 
-dynamic_dim_constraint_symbol2example_value = {S0: 224}
+dynamic_dim_constraint_symbol2example_value = {S0: 1, S1: 224}
 
 dynamic_dim_constraint_relations = []
 
 dynamic_dim_constraint_input_shapes = [
-    ([1, 3, S0, S0], "L_pixel_values_"),
+    ([S0, 3, S1, S1], "L_pixel_values_"),
     (
         [1536],
         "L_self_modules_preprocessor_modules_patchifier_modules_norm_parameters_weight_",
diff --git a/samples/transformers-auto-model/bge-small-en-v1.5/graph_net.json b/samples/transformers-auto-model/bge-small-en-v1.5/graph_net.json
@@ -29,18 +29,19 @@
         "region:us"
     ],
     "heuristic_tag": "other",
+    "symbolic_dimension_reifier": "naive_nlp_sym_dim_reifier",
     "dimension_generalization_passes": [
         "batch_call_method_view_pass",
         "tuple_arg_call_method_view_pass",
         "naive_call_method_reshape_pass",
         "naive_call_method_expand_pass",
         "non_batch_call_method_expand_pass",
+        "non_batch_call_method_view_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"
-    ],
-    "symbolic_dimension_reifier": "naive_nlp_sym_dim_reifier"
+    ]
 }
diff --git a/samples/transformers-auto-model/facebook_sam-vit-large/graph_net.json b/samples/transformers-auto-model/facebook_sam-vit-large/graph_net.json
@@ -19,5 +19,19 @@
         "region:us"
     ],
     "heuristic_tag": "other",
-    "symbolic_dimension_reifier": "naive_cv_sym_dim_reifier"
+    "symbolic_dimension_reifier": "naive_cv_sym_dim_reifier",
+    "dimension_generalization_passes": [
+        "batch_call_method_view_pass",
+        "tuple_arg_call_method_view_pass",
+        "naive_call_method_reshape_pass",
+        "naive_call_method_expand_pass",
+        "non_batch_call_method_expand_pass",
+        "non_batch_call_method_view_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"
+    ]
 }
diff --git a/samples/transformers-auto-model/facebook_sam-vit-large/input_tensor_constraints.py b/samples/transformers-auto-model/facebook_sam-vit-large/input_tensor_constraints.py
@@ -1,4 +1,4 @@
-from sympy import Symbol
+from sympy import Symbol, Expr, Rel, Eq
 
 S0 = Symbol("S0")
 
diff --git a/samples/transformers-auto-model/hf-tiny-model-private_tiny-random-Swinv2ForImageClassification/graph_net.json b/samples/transformers-auto-model/hf-tiny-model-private_tiny-random-Swinv2ForImageClassification/graph_net.json
@@ -15,8 +15,6 @@
         "region:us"
     ],
     "heuristic_tag": "computer_vision",
-    "dimension_generalization_passes": [
-        "non_batch_call_method_view_pass"
-    ],
-    "symbolic_dimension_reifier": "naive_cv_sym_dim_reifier"
-}
+    "symbolic_dimension_reifier": "naive_cv_sym_dim_reifier",
+    "dimension_generalization_passes": []
+}
diff --git a/samples/transformers-auto-model/hf-tiny-model-private_tiny-random-Swinv2ForImageClassification/input_tensor_constraints.py b/samples/transformers-auto-model/hf-tiny-model-private_tiny-random-Swinv2ForImageClassification/input_tensor_constraints.py
diff --git a/samples/transformers-auto-model/internetoftim_dinov2-base/graph_net.json b/samples/transformers-auto-model/internetoftim_dinov2-base/graph_net.json
diff --git a/samples/transformers-auto-model/internetoftim_dinov2-base/input_tensor_constraints.py b/samples/transformers-auto-model/internetoftim_dinov2-base/input_tensor_constraints.py
diff --git a/samples/transformers-auto-model/microsoft_beit-base-patch16-384/graph_net.json b/samples/transformers-auto-model/microsoft_beit-base-patch16-384/graph_net.json
diff --git a/samples/transformers-auto-model/microsoft_swin-base-patch4-window12-384-in22k/graph_net.json b/samples/transformers-auto-model/microsoft_swin-base-patch4-window12-384-in22k/graph_net.json
diff --git a/samples/transformers-auto-model/microsoft_swin-base-patch4-window12-384-in22k/input_tensor_constraints.py b/samples/transformers-auto-model/microsoft_swin-base-patch4-window12-384-in22k/input_tensor_constraints.py
diff --git a/samples/transformers-auto-model/ydshieh_tiny-random-ViTModel/graph_net.json b/samples/transformers-auto-model/ydshieh_tiny-random-ViTModel/graph_net.json
diff --git a/samples/transformers-auto-model/ydshieh_tiny-random-ViTModel/input_tensor_constraints.py b/samples/transformers-auto-model/ydshieh_tiny-random-ViTModel/input_tensor_constraints.py
diff --git a/samples/ultralytics/yolov3-tinyu/graph_net.json b/samples/ultralytics/yolov3-tinyu/graph_net.json
diff --git a/samples/ultralytics/yolov3-tinyu/input_tensor_constraints.py b/samples/ultralytics/yolov3-tinyu/input_tensor_constraints.py
diff --git a/samples/ultralytics/yolov9c/graph_net.json b/samples/ultralytics/yolov9c/graph_net.json
diff --git a/samples/ultralytics/yolov9c/input_tensor_constraints.py b/samples/ultralytics/yolov9c/input_tensor_constraints.py

Original file line number	Diff line number	Diff line change
`@@ -5,6 +5,6 @@`
`5`	`5`	`"dynamic": false,`
`6`	`6`	`"source": "mmseg",`
`7`	`7`	`"heuristic_tag": "computer_vision",`
`8`		`- "dimension_generalization_passes": [],`
`9`		`- "symbolic_dimension_reifier": "naive_cv_sym_dim_reifier"`
	`8`	`+ "symbolic_dimension_reifier": "naive_cv_sym_dim_reifier",`
	`9`	`+ "dimension_generalization_passes": []`
`10`	`10`	`}`
Original file line number	Diff line number	Diff line change
`@@ -1,4 +1,4 @@`
`1`		`-from sympy import Symbol`
	`1`	`+from sympy import Symbol, Expr, Rel, Eq`
`2`	`2`
`3`	`3`	`S0 = Symbol("S0")`
`4`	`4`	`S1 = Symbol("S1")`