[PIR] Fix slice_dense_array and open test (#61864)

chen2016013 · web-flow · commit f774dedf9ae6 · 2024-02-21T10:27:05.000+08:00
* fic slice_dense_array and open test

* Update test_for_enumerate.py
diff --git a/paddle/fluid/pir/dialect/operator/ir/manual_op.cc b/paddle/fluid/pir/dialect/operator/ir/manual_op.cc
@@ -2991,7 +2991,7 @@ void SliceArrayDenseOp::VerifySig() {
   {
     auto input_size = num_operands();
     IR_ENFORCE(input_size == 2u,
-               "The size %d of inputs must be equal to 1.",
+               "The size %d of inputs must be equal to 2.",
                input_size);
     IR_ENFORCE((*this)
                    ->operand_source(0)
diff --git a/paddle/phi/infermeta/unary.cc b/paddle/phi/infermeta/unary.cc
@@ -3806,6 +3806,7 @@ void SliceArrayDenseInferMeta(const MetaTensor& input,
   }
   // out->set_dims(input.dims());
   out->set_dtype(input.dtype());
+  out->set_dims(input.dims());
 }
 
 void SliceRawInferMeta(const MetaTensor& input,
diff --git a/test/dygraph_to_static/test_for_enumerate.py b/test/dygraph_to_static/test_for_enumerate.py
@@ -172,7 +172,7 @@ def for_iter_var_list(x):
     # 1. prepare data, ref test_list.py
     x = paddle.to_tensor(x)
     iter_num = paddle.tensor.fill_constant(shape=[1], value=5, dtype="int32")
-    a = []
+    a = paddle.tensor.create_array("int32")
     for i in range(iter_num):
         a.append(x + i)
     # 2. iter list[var]
@@ -187,7 +187,7 @@ def for_enumerate_var_list(x):
     # 1. prepare data, ref test_list.py
     x = paddle.to_tensor(x)
     iter_num = paddle.tensor.fill_constant(shape=[1], value=5, dtype="int32")
-    a = []
+    a = paddle.tensor.create_array("int32")
     for i in range(iter_num):
         a.append(x + i)
     # 2. iter list[var]
@@ -389,7 +389,7 @@ def transformed_error(self, etype):
 
 class TestForInRangeConfig(TestTransform):
     def set_input(self):
-        self.input = np.array([5])
+        self.input = np.array([5]).astype("int32")
 
     def set_test_func(self):
         self.dygraph_func = for_in_range
@@ -489,6 +489,7 @@ class TestForIterVarList(TestForInRangeConfig):
     def set_test_func(self):
         self.dygraph_func = for_iter_var_list
 
+    @test_legacy_and_pt_and_pir
     def test_transformed_result_compare(self):
         self.set_test_func()
         self.transformed_result_compare()
@@ -498,6 +499,7 @@ class TestForEnumerateVarList(TestForInRangeConfig):
     def set_test_func(self):
         self.dygraph_func = for_enumerate_var_list
 
+    @test_legacy_and_pt_and_pir
     def test_transformed_result_compare(self):
         self.set_test_func()
         self.transformed_result_compare()

Original file line number	Diff line number	Diff line change
`@@ -2991,7 +2991,7 @@ void SliceArrayDenseOp::VerifySig() {`
`2991`	`2991`	`{`
`2992`	`2992`	`auto input_size = num_operands();`
`2993`	`2993`	`IR_ENFORCE(input_size == 2u,`
`2994`		`- "The size %d of inputs must be equal to 1.",`
	`2994`	`+ "The size %d of inputs must be equal to 2.",`
`2995`	`2995`	`input_size);`
`2996`	`2996`	`IR_ENFORCE((*this)`
`2997`	`2997`	`->operand_source(0)`
Original file line number	Diff line number	Diff line change
`@@ -3806,6 +3806,7 @@ void SliceArrayDenseInferMeta(const MetaTensor& input,`
`3806`	`3806`	`}`
`3807`	`3807`	`// out->set_dims(input.dims());`
`3808`	`3808`	`out->set_dtype(input.dtype());`
	`3809`	`+ out->set_dims(input.dims());`
`3809`	`3810`	`}`
`3810`	`3811`
`3811`	`3812`	`void SliceRawInferMeta(const MetaTensor& input,`