[XPU] fix scale op test (#65251)

lj970926 · web-flow · commit db13846d3a52 · 2024-06-19T15:31:52.000+08:00
diff --git a/test/xpu/test_range_xpu.py b/test/xpu/test_range_xpu.py
@@ -10,7 +10,7 @@
 # distributed under the License is distributed on an "AS IS" BASIS,
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
-# limitations under the License.z
+# limitations under the License.
 
 import unittest
 
diff --git a/test/xpu/test_scale_op_xpu.py b/test/xpu/test_scale_op_xpu.py
@@ -17,6 +17,7 @@
 import numpy as np
 from get_test_cover_info import (
     XPUOpTestWrapper,
+    check_run_big_shape_test,
     create_test_class,
     get_xpu_op_support_types,
 )
@@ -50,11 +51,11 @@ def set_xpu(self):
 
         def set_inputs(self):
             if self.dtype == np.uint16:
-                x = np.random.random((10, 10)).astype('float32')
+                x = np.random.random(self.shape).astype('float32')
                 self.inputs = {'X': convert_float_to_uint16(x)}
             else:
                 self.inputs = {
-                    'X': np.random.random((10, 10)).astype(self.dtype)
+                    'X': np.random.random(self.shape).astype(self.dtype)
                 }
 
         def set_output(self):
@@ -102,29 +103,35 @@ class TestScaleOp5(TestScaleOp):
         def set_attrs(self):
             self.attrs = {'scale': -0.003}
 
-    # class TestScaleOpLargeShape1(TestScaleOp):
-    #     def set_shape(self):
-    #         self.shape = [64]
+    @check_run_big_shape_test()
+    class TestScaleOpLargeShape1(TestScaleOp):
+        def set_shape(self):
+            self.shape = [64]
 
-    # class TestScaleOpLargeShape2(TestScaleOp):
-    #     def set_shape(self):
-    #         self.shape = [8192, 1]
+    @check_run_big_shape_test()
+    class TestScaleOpLargeShape2(TestScaleOp):
+        def set_shape(self):
+            self.shape = [8192, 1]
 
-    # class TestScaleOpLargeShape3(TestScaleOp):
-    #     def set_shape(self):
-    #         self.shape = [1, 8192, 5, 64]
+    @check_run_big_shape_test()
+    class TestScaleOpLargeShape3(TestScaleOp):
+        def set_shape(self):
+            self.shape = [1, 8192, 5, 64]
 
-    # class TestScaleOpLargeShape4(TestScaleOp):
-    #     def set_shape(self):
-    #         self.shape = [8192, 1920]
+    @check_run_big_shape_test()
+    class TestScaleOpLargeShape4(TestScaleOp):
+        def set_shape(self):
+            self.shape = [8192, 1920]
 
-    # class TestScaleOpLargeShape5(TestScaleOp):
-    #     def set_shape(self):
-    #         self.shape = [1024, 5120]
+    @check_run_big_shape_test()
+    class TestScaleOpLargeShape5(TestScaleOp):
+        def set_shape(self):
+            self.shape = [1024, 5120]
 
-    # class TestScaleOpLargeShape6(TestScaleOp):
-    #     def set_shape(self):
-    #         self.shape = [8192, 3456]
+    @check_run_big_shape_test()
+    class TestScaleOpLargeShape6(TestScaleOp):
+        def set_shape(self):
+            self.shape = [8192, 3456]
 
 
 class TestScaleApiStatic(unittest.TestCase):
