Fix test_compiler and several sample errors.

Xreki · Xreki · commit 5799419ec2df · 2025-09-26T17:39:10.000+08:00
diff --git a/graph_net/paddle/test_compiler.py b/graph_net/paddle/test_compiler.py
@@ -7,6 +7,7 @@
 from dataclasses import dataclass
 from contextlib import contextmanager
 import time
+import math
 import numpy as np
 import random
 import platform
@@ -283,8 +284,16 @@ def test_single_model(args):
 
 
 def get_cmp_equal(expected_out, compiled_out):
+    def convert(x):
+        if x.dtype in [paddle.float16, paddle.bfloat16]:
+            return x.astype("float32")
+        elif x.dtype in [paddle.uint8, paddle.int8, paddle.int16]:
+            return x.astype("int32")
+        return x
+
     return " ".join(
-        str(int(paddle.equal_all(a, b))) for a, b in zip(expected_out, compiled_out)
+        str(int(paddle.equal_all(convert(a), convert(b))))
+        for a, b in zip(expected_out, compiled_out)
     )
 
 
diff --git a/paddle_samples/PaddleX/Co-DINO-R50/subgraph_2/model.py b/paddle_samples/PaddleX/Co-DINO-R50/subgraph_2/model.py
@@ -645,30 +645,6 @@ def forward(
         )
 
         return (
-            group_norm_0,
-            group_norm_1,
-            group_norm_2,
-            group_norm_3,
-            group_norm_4,
-            group_norm_5,
-            group_norm_6,
-            group_norm_7,
-            group_norm_8,
-            group_norm_9,
-            group_norm_10,
-            group_norm_11,
-            group_norm_12,
-            group_norm_13,
-            group_norm_14,
-            group_norm_15,
-            group_norm_16,
-            group_norm_17,
-            group_norm_18,
-            group_norm_19,
-            group_norm_20,
-            group_norm_21,
-            group_norm_22,
-            group_norm_23,
             add_0,
             add_1,
             add_2,
diff --git a/paddle_samples/PaddleX/Co-DINO-R50/subgraph_20/model.py b/paddle_samples/PaddleX/Co-DINO-R50/subgraph_20/model.py
@@ -1135,8 +1135,6 @@ def forward(
         )
 
         return (
-            group_norm_0,
-            group_norm_1,
             stack_0,
             stack_1,
             reshape_0,
diff --git a/paddle_samples/PaddleX/FCOS-ResNet50/subgraph_1/model.py b/paddle_samples/PaddleX/FCOS-ResNet50/subgraph_1/model.py
@@ -5079,86 +5079,4 @@ def forward(
             transpose_9,
         )
 
-        return (
-            group_norm_0,
-            group_norm_1,
-            group_norm_2,
-            group_norm_3,
-            group_norm_4,
-            group_norm_5,
-            group_norm_6,
-            group_norm_7,
-            group_norm_8,
-            group_norm_9,
-            group_norm_10,
-            group_norm_11,
-            group_norm_12,
-            group_norm_13,
-            group_norm_14,
-            group_norm_15,
-            group_norm_16,
-            group_norm_17,
-            group_norm_18,
-            group_norm_19,
-            group_norm_20,
-            group_norm_21,
-            group_norm_22,
-            group_norm_23,
-            group_norm_24,
-            group_norm_25,
-            group_norm_26,
-            group_norm_27,
-            group_norm_28,
-            group_norm_29,
-            group_norm_30,
-            group_norm_31,
-            group_norm_32,
-            group_norm_33,
-            group_norm_34,
-            group_norm_35,
-            group_norm_36,
-            group_norm_37,
-            group_norm_38,
-            group_norm_39,
-            group_norm_40,
-            group_norm_41,
-            group_norm_42,
-            group_norm_43,
-            group_norm_44,
-            group_norm_45,
-            group_norm_46,
-            group_norm_47,
-            group_norm_48,
-            group_norm_49,
-            group_norm_50,
-            group_norm_51,
-            group_norm_52,
-            group_norm_53,
-            group_norm_54,
-            group_norm_55,
-            group_norm_56,
-            group_norm_57,
-            group_norm_58,
-            group_norm_59,
-            group_norm_60,
-            group_norm_61,
-            group_norm_62,
-            group_norm_63,
-            group_norm_64,
-            group_norm_65,
-            group_norm_66,
-            group_norm_67,
-            group_norm_68,
-            group_norm_69,
-            group_norm_70,
-            group_norm_71,
-            group_norm_72,
-            group_norm_73,
-            group_norm_74,
-            group_norm_75,
-            group_norm_76,
-            group_norm_77,
-            group_norm_78,
-            group_norm_79,
-            add_n_0,
-        )
+        return add_n_0
diff --git a/paddle_samples/PaddleX/FCOS-ResNet50/subgraph_4/model.py b/paddle_samples/PaddleX/FCOS-ResNet50/subgraph_4/model.py
@@ -5066,86 +5066,4 @@ def forward(
             transpose_9,
         )
 
-        return (
-            group_norm_0,
-            group_norm_1,
-            group_norm_2,
-            group_norm_3,
-            group_norm_4,
-            group_norm_5,
-            group_norm_6,
-            group_norm_7,
-            group_norm_8,
-            group_norm_9,
-            group_norm_10,
-            group_norm_11,
-            group_norm_12,
-            group_norm_13,
-            group_norm_14,
-            group_norm_15,
-            group_norm_16,
-            group_norm_17,
-            group_norm_18,
-            group_norm_19,
-            group_norm_20,
-            group_norm_21,
-            group_norm_22,
-            group_norm_23,
-            group_norm_24,
-            group_norm_25,
-            group_norm_26,
-            group_norm_27,
-            group_norm_28,
-            group_norm_29,
-            group_norm_30,
-            group_norm_31,
-            group_norm_32,
-            group_norm_33,
-            group_norm_34,
-            group_norm_35,
-            group_norm_36,
-            group_norm_37,
-            group_norm_38,
-            group_norm_39,
-            group_norm_40,
-            group_norm_41,
-            group_norm_42,
-            group_norm_43,
-            group_norm_44,
-            group_norm_45,
-            group_norm_46,
-            group_norm_47,
-            group_norm_48,
-            group_norm_49,
-            group_norm_50,
-            group_norm_51,
-            group_norm_52,
-            group_norm_53,
-            group_norm_54,
-            group_norm_55,
-            group_norm_56,
-            group_norm_57,
-            group_norm_58,
-            group_norm_59,
-            group_norm_60,
-            group_norm_61,
-            group_norm_62,
-            group_norm_63,
-            group_norm_64,
-            group_norm_65,
-            group_norm_66,
-            group_norm_67,
-            group_norm_68,
-            group_norm_69,
-            group_norm_70,
-            group_norm_71,
-            group_norm_72,
-            group_norm_73,
-            group_norm_74,
-            group_norm_75,
-            group_norm_76,
-            group_norm_77,
-            group_norm_78,
-            group_norm_79,
-            add_n_0,
-        )
+        return add_n_0
diff --git a/paddle_samples/PaddleX/MaskFormer_small/subgraph_2/model.py b/paddle_samples/PaddleX/MaskFormer_small/subgraph_2/model.py
@@ -14274,24 +14274,6 @@ def forward(
         )
 
         return (
-            group_norm_0,
-            group_norm_1,
-            group_norm_2,
-            group_norm_3,
-            group_norm_4,
-            group_norm_5,
-            group_norm_6,
-            group_norm_7,
-            group_norm_8,
-            group_norm_9,
-            group_norm_10,
-            group_norm_11,
-            group_norm_12,
-            group_norm_13,
-            split_0,
-            split_1,
-            split_2,
-            split_3,
             slice_0,
             slice_1,
             slice_2,
diff --git a/paddle_samples/PaddleX/MaskFormer_tiny/subgraph_0/model.py b/paddle_samples/PaddleX/MaskFormer_tiny/subgraph_0/model.py
@@ -9546,24 +9546,6 @@ def forward(
         )
 
         return (
-            group_norm_0,
-            group_norm_1,
-            group_norm_2,
-            group_norm_3,
-            group_norm_4,
-            group_norm_5,
-            group_norm_6,
-            group_norm_7,
-            group_norm_8,
-            group_norm_9,
-            group_norm_10,
-            group_norm_11,
-            group_norm_12,
-            group_norm_13,
-            split_0,
-            split_1,
-            split_2,
-            split_3,
             slice_0,
             slice_1,
             slice_2,
diff --git a/paddle_samples/PaddleX/Nonstationary/subgraph_0/model.py b/paddle_samples/PaddleX/Nonstationary/subgraph_0/model.py
@@ -131,4 +131,4 @@ def forward(self, data_0, data_1, data_2):
             softmax_0,
         )
 
-        return split_0, split_1, split_2, split_3, einsum_0
+        return einsum_0
diff --git a/paddle_samples/PaddleX/Nonstationary/subgraph_10/model.py b/paddle_samples/PaddleX/Nonstationary/subgraph_10/model.py
@@ -1565,22 +1565,6 @@ def forward(
         )
 
         return (
-            split_0,
-            split_1,
-            split_2,
-            split_3,
-            split_4,
-            split_5,
-            split_6,
-            split_7,
-            split_8,
-            split_9,
-            split_10,
-            split_11,
-            split_12,
-            split_13,
-            split_14,
-            split_15,
             dropout_0,
             layer_norm_0,
         )
diff --git a/paddle_samples/PaddleX/Nonstationary/subgraph_2/model.py b/paddle_samples/PaddleX/Nonstationary/subgraph_2/model.py
@@ -531,14 +531,4 @@ def forward(
             unsqueeze_7,
         )
 
-        return (
-            split_0,
-            split_1,
-            split_2,
-            split_3,
-            split_4,
-            split_5,
-            split_6,
-            split_7,
-            layer_norm_0,
-        )
+        return layer_norm_0
diff --git a/paddle_samples/PaddleX/Nonstationary/subgraph_3/model.py b/paddle_samples/PaddleX/Nonstationary/subgraph_3/model.py
@@ -431,14 +431,4 @@ def forward(
             unsqueeze_7,
         )
 
-        return (
-            split_0,
-            split_1,
-            split_2,
-            split_3,
-            split_4,
-            split_5,
-            split_6,
-            split_7,
-            layer_norm_0,
-        )
+        return layer_norm_0
diff --git a/paddle_samples/PaddleX/Nonstationary/subgraph_8/model.py b/paddle_samples/PaddleX/Nonstationary/subgraph_8/model.py
@@ -1344,22 +1344,6 @@ def forward(
         )
 
         return (
-            split_0,
-            split_1,
-            split_2,
-            split_3,
-            split_4,
-            split_5,
-            split_6,
-            split_7,
-            split_8,
-            split_9,
-            split_10,
-            split_11,
-            split_12,
-            split_13,
-            split_14,
-            split_15,
             dropout_0,
             layer_norm_0,
         )
diff --git a/paddle_samples/PaddleX/Nonstationary/subgraph_9/model.py b/paddle_samples/PaddleX/Nonstationary/subgraph_9/model.py
@@ -64,4 +64,4 @@ def forward(self, data_0, data_1, data_2):
             unsqueeze_1,
         )
 
-        return split_0, split_1, split_2, split_3, scale_0
+        return scale_0
diff --git a/paddle_samples/PaddleX/Nonstationary_ad/subgraph_1/model.py b/paddle_samples/PaddleX/Nonstationary_ad/subgraph_1/model.py
diff --git a/paddle_samples/PaddleX/Nonstationary_ad/subgraph_2/model.py b/paddle_samples/PaddleX/Nonstationary_ad/subgraph_2/model.py
diff --git a/paddle_samples/PaddleX/PP-OCRv3_mobile_rec/subgraph_1/model.py b/paddle_samples/PaddleX/PP-OCRv3_mobile_rec/subgraph_1/model.py

Original file line number	Diff line number	Diff line change
`@@ -1135,8 +1135,6 @@ def forward(`
`1135`	`1135`	`)`
`1136`	`1136`
`1137`	`1137`	`return (`
`1138`		`- group_norm_0,`
`1139`		`- group_norm_1,`
`1140`	`1138`	`stack_0,`
`1141`	`1139`	`stack_1,`
`1142`	`1140`	`reshape_0,`
Original file line number	Diff line number	Diff line change
`@@ -131,4 +131,4 @@ def forward(self, data_0, data_1, data_2):`
`131`	`131`	`softmax_0,`
`132`	`132`	`)`
`133`	`133`
`134`		`- return split_0, split_1, split_2, split_3, einsum_0`
	`134`	`+ return einsum_0`