Merge branch 'develop' into add_original_names

Author: Xreki

graph_net/config/small100_torch_samples_list.txt

@@ -0,0 +1,101 @@
+# shuf -n 100 torch_samples_list.txt > small_torch_samples_list.txt
+samples/ultralytics/yolov3-tinyu
+samples/transformers-auto-model/hf-tiny-model-private_tiny-random-Swinv2ForImageClassification
+samples/torchgeometric/RECT_L
+samples/transformers-auto-model/opus-mt-en-gv
+samples/transformers-auto-model/Visual-Attention-Network_van-tiny
+samples/timm/convnextv2_base.fcmae_ft_in1k
+samples/torchvision/vgg16_bn
+samples/timm/regnety_080_tv.tv2_in1k
+samples/transformers-auto-model/internetoftim_dinov2-base
+samples/timm/mobilenetv4_conv_aa_large.e230_r384_in12k
+samples/ultralytics/yolo12x
+samples/transformers-auto-model/opus-mt-eo-it
+samples/transformers-auto-model/opus-mt-es-csg
+samples/transformers-auto-model/google_byt5_base
+samples/transformers-auto-model/Fsoft-AIC_videberta-base
+samples/timm/mixer_l16_224.goog_in21k
+samples/transformers-auto-model/hf-tiny-model-private_tiny-random-EfficientFormerModel
+samples/transformers-auto-model/distilbert_distilbert-base-multilingual-cased
+samples/transformers-auto-model/ydshieh_tiny-random-ViTModel
+samples/transformers-auto-model/microsoft_beit-base-patch16-384
+samples/transformers-auto-model/hf-tiny-model-private_tiny-random-Data2VecVisionModel
+samples/transformers-auto-model/kanarya-750m
+samples/transformers-auto-model/Helsinki-NLP_opus-mt-zh-en
+samples/transformers-auto-model/opus-mt-de-tl
+samples/timm/vit_base_patch16_224_miil.in21k
+samples/mmseg/SegNeXt-b
+samples/transformers-auto-model/OFA-Sys_chinese-clip-vit-large-patch14
+samples/mmpose/ViTPose-B
+samples/transformers-auto-model/Sayan01_tiny-bert-sst2-distilled
+samples/timm/regnety_080.pycls_in1k
+samples/ultralytics/yolov9c
+samples/transformers-auto-model/opus-mt-bcl-fr
+samples/transformers-auto-model/opus-mt-de-ase
+samples/timm/eva02_base_patch16_clip_224.merged2b
+samples/mmpose/S-ViPNAS-Res50
+samples/transformers-auto-model/Lin-Chen_ShareGPT4V-7B_Pretrained_vit-large336-l12
+samples/transformers-auto-model/opus-mt-en-tw
+samples/timm/hgnet_tiny.paddle_in1k
+samples/transformers-auto-model/opus-mt-ase-es
+samples/transformers-auto-model/MIT_ast-finetuned-audioset-10-10-0.4593
+samples/transformers-auto-model/opus-mt-en-kqn
+samples/transformers-auto-model/opus-mt-en-guw
+samples/timm/vit_small_patch16_rope_mixed_ape_224.naver_in1k
+samples/timm/repvgg_b0.rvgg_in1k
+samples/transformers-auto-model/opus-mt-en-crs
+samples/transformers-auto-model/opus-mt-en-bat
+samples/transformers-auto-model/Francesco_resnet101-224-1k
+samples/timm/dpn107.mx_in1k
+samples/transformers-auto-model/google/t5-efficient-large-kv128
+samples/transformers-auto-model/aimarsg_prueba5
+samples/torchvision/resnext50_32x4d
+samples/timm/inception_v3.gluon_in1k
+samples/transformers-auto-model/mohsenfayyaz_BERT_Warmup
+samples/transformers-auto-model/MoritzLaurer_xtremedistil-l6-h256-mnli-fever-anli-ling-binary
+samples/transformers-auto-model/pipecat-ai_smart-turn-v2
+samples/timm/poolformerv2_m48.sail_in1k
+samples/timm/res2net101_26w_4s
+samples/transformers-auto-model/opus-mt-ar-es
+samples/timm/regnety_002.pycls_in1k
+samples/transformers-auto-model/simjo_dummy-model
+samples/mmseg/CCNet_R101
+samples/transformers-auto-model/opus-mt-tc-bible-big-deu_eng_fra_por_spa-bat
+samples/transformers-auto-model/facebook_sam-vit-large
+samples/transformers-auto-model/distilbert_distilbert-base-uncased
+samples/mmseg/PoolFormer_m36
+samples/timm/hgnetv2_b4.ssld_stage1_in22k_in1k
+samples/timm/regnetx_160.pycls_in1k
+samples/transformers-auto-model/google/bert_uncased_L-6_H-512_A-8
+samples/transformers-auto-model/ogoshi2000_stance-nystromformer
+samples/timm/resnetv2_18.ra4_e3600_r224_in1k
+samples/transformers-auto-model/TinyLlama/TinyLlama-1.1B-Chat-v0.4
+samples/timm/resnetaa50d.d_in12k
+samples/transformers-auto-model/bge-small-en-v1.5
+samples/timm/coatnet_rmlp_0_rw_224
+samples/transformers-auto-model/opus-mt-en-sal
+samples/mmseg/UPerNet_R101
+samples/transformers-auto-model/sentence-transformers/paraphrase-distilroberta-base-v1
+samples/transformers-auto-model/opus-mt-en-phi
+samples/transformers-auto-model/opus-mt-fi-niu
+samples/torchvision/vit_l_16
+samples/timm/res2net50_14w_8s.in1k
+samples/mmseg/Twins_pcpvt-b_uperhead
+samples/transformers-auto-model/google-t5_t5-large
+samples/transformers-auto-model/TrustSafeAI_RADAR_Vicuna_7B
+samples/torchaudio/wavlm_base
+samples/transformers-auto-model/opus-mt-NORTH_EU-NORTH_EU
+samples/transformers-auto-model/apple_aimv2-huge-patch14-224
+samples/timm/resnetrs270.tf_in1k
+samples/transformers-auto-model/opus-mt-de-no
+samples/transformers-auto-model/opus-mt-tc-bible-big-gmw-deu_eng_fra_por_spa
+samples/timm/repghostnet_111.in1k
+samples/mmseg/DeepLabV3+_R101
+samples/transformers-auto-model/microsoft_swin-base-patch4-window12-384-in22k
+samples/transformers-auto-model/opus-mt-es-de
+samples/transformers-auto-model/nvidia_mit-b4
+samples/transformers-auto-model/opus-mt-tc-bible-big-ine-deu_eng_fra_por_spa
+samples/mmpose/pose_swin_b
+samples/transformers-auto-model/opus-mt-bi-en
+samples/transformers-auto-model/all-mpnet-base-v2
+samples/transformers-auto-model/Neurora_opus-tatoeba-heb-eng

graph_net/config/small_torch_samples_list.txt

@@ -178,7 +178,7 @@ def run_decomposer_for_single_model(
         "decorator_path": f"{graphnet_root}/graph_net/{framework}/extractor.py",
         "decorator_config": {
             "name": model_name,
-            "custom_extractor_path": f"{graphnet_root}/graph_net/{framework}/naive_graph_decomposer.py",
+            "custom_extractor_path": f"{graphnet_root}/graph_net/{framework}/graph_decomposer.py",
             "custom_extractor_config": {
                 "output_dir": output_dir,
                 "split_positions": split_positions,

…aph_net/paddle/naive_graph_decomposer.py graph_net/paddle/graph_decomposer.pygraph_net/paddle/naive_graph_decomposer.py renamed to graph_net/paddle/graph_decomposer.py graph_net/paddle/naive_graph_decomposer.py renamed to graph_net/paddle/graph_decomposer.py

@@ -58,6 +58,11 @@ def serialize_to_py_str(self) -> str:
         lines = [
             (f"class {self.record_class_name}:"),
             (f'\tname = "{self.name}"'),
+            *(
+                [f'\toriginal_name = "{self.original_name}"']
+                if self.original_name is not None
+                else []
+            ),
             (f"\tshape = {self.shape}"),
             (f'\tdtype = "{self.dtype}"'),
             (f'\tdevice = "{self.device}"'),

graph_net/subgraph_decompose_and_evaluation_step.py

@@ -11,7 +11,7 @@ decorator_config_json_str=$(cat <<EOF
     "decorator_path": "$GRAPH_NET_ROOT/torch/extractor.py",
     "decorator_config": {
         "name": "$MODEL_NAME",
-        "custom_extractor_path": "$GRAPH_NET_ROOT/torch/naive_graph_decomposer.py",
+        "custom_extractor_path": "$GRAPH_NET_ROOT/torch/graph_decomposer.py",
         "custom_extractor_config": {
             "output_dir": "/tmp/chain_naive_decompose_workspace",
             "split_positions": [8, 16, 32],

graph_net/tensor_meta.py

@@ -16,7 +16,7 @@ extractor_config_json_str=$(cat <<EOF
     "decorator_path": "$GRAPH_NET_ROOT/torch/extractor.py",
     "decorator_config": {
         "name": "$MODEL_NAME",
-        "custom_extractor_path": "$GRAPH_NET_ROOT/torch/naive_graph_decomposer.py",
+        "custom_extractor_path": "$GRAPH_NET_ROOT/torch/graph_decomposer.py",
         "custom_extractor_config": {
             "output_dir": "$OUTPUT_DIR/${MODEL_NAME}_decomposed",
             "split_positions": [8, 16, 32],