[ut bug fix opencl]opencl fix reshape/reshape2 bug (#8364)

Author: daming5432

lite/backends/opencl/cl_kernel/image/reshape_kernel.cl

@@ -50,8 +50,8 @@ __kernel void reshape(__read_only image2d_t input_image,
 
   int in_n0 = count0 / in_Stride2;
   int in_n1 = count1 / in_Stride2;
-  int in_n2 = count1 / in_Stride2;
-  int in_n3 = count2 / in_Stride2;
+  int in_n2 = count2 / in_Stride2;
+  int in_n3 = count3 / in_Stride2;
 
   count0 = count0 % in_Stride2;
   count1 = count1 % in_Stride2;
@@ -132,7 +132,7 @@ __kernel void reshape(__read_only image2d_t input_image,
     if (in_c2 % 4 == 0) {
       output.z = input2.x;
     } else if (in_c2 % 4 == 1) {
-      output.z = input1.y;
+      output.z = input2.y;
     } else if (in_c2 % 4 == 2) {
       output.z = input2.z;
     } else {

lite/tests/unittest_py/op/test_reshape2_op.py

@@ -26,6 +26,7 @@
 
 import numpy as np
 from functools import partial
+from functools import reduce
 
 
 class TestReshape2Op(AutoScanTest):
@@ -77,12 +78,17 @@ def sample_program_configs(self, draw):
             st.lists(
                 st.integers(
                     min_value=1, max_value=10), min_size=4, max_size=4))
+
         attr_shape = draw(
             st.lists(
                 st.integers(
-                    min_value=0, max_value=4),
-                min_size=len(in_shape),
+                    min_value=1, max_value=max(in_shape)),
+                min_size=1,
                 max_size=len(in_shape)))
+        assume(
+            reduce(lambda x, y: x * y, attr_shape) == reduce(
+                lambda x, y: x * y, in_shape))
+
         with_shape = draw(st.sampled_from([True, False]))
 
         def generate_input(*args, **kwargs):
@@ -95,7 +101,7 @@ def generate_input(*args, **kwargs):
                 "Out": ["output_data"],
                 "XShape": ["x_shape"],
             },
-            attrs={"shape": in_shape, })
+            attrs={"shape": attr_shape, })
         program_config = ProgramConfig(
             ops=[build_ops],
             weights={},
@@ -125,7 +131,7 @@ def _teller1(program_config, predictor_config):
         )
 
     def test(self, *args, **kwargs):
-        self.run_and_statis(quant=False, max_examples=25)
+        self.run_and_statis(quant=False, max_examples=200)
 
 
 if __name__ == "__main__":

lite/tests/unittest_py/op/test_reshape_op.py

@@ -26,6 +26,7 @@
 
 import numpy as np
 from functools import partial
+from functools import reduce
 
 
 class TestReshapeOp(AutoScanTest):
@@ -66,12 +67,17 @@ def sample_program_configs(self, draw):
             st.lists(
                 st.integers(
                     min_value=1, max_value=10), min_size=4, max_size=4))
+
         attr_shape = draw(
             st.lists(
                 st.integers(
-                    min_value=0, max_value=4),
-                min_size=len(in_shape),
+                    min_value=1, max_value=max(in_shape)),
+                min_size=1,
                 max_size=len(in_shape)))
+        assume(
+            reduce(lambda x, y: x * y, attr_shape) == reduce(
+                lambda x, y: x * y, in_shape))
+
         with_shape = draw(st.sampled_from([True, False]))
 
         def generate_input(*args, **kwargs):
@@ -81,7 +87,7 @@ def generate_input(*args, **kwargs):
             type="reshape",
             inputs={"X": ["input_data"], },
             outputs={"Out": ["output_data"], },
-            attrs={"shape": in_shape, })
+            attrs={"shape": attr_shape, })
         program_config = ProgramConfig(
             ops=[build_ops],
             weights={},
@@ -105,7 +111,7 @@ def _teller1(program_config, predictor_config):
         )
 
     def test(self, *args, **kwargs):
-        self.run_and_statis(quant=False, max_examples=25)
+        self.run_and_statis(quant=False, max_examples=200)
 
 
 if __name__ == "__main__":