【Fix PIR Unittest No.43 BUAA】Fix test_tensor_array_to_tensor (#66337)

Author: Guanhuachen2003

test/deprecated/legacy_test/CMakeLists.txt

@@ -709,7 +709,6 @@ set(STATIC_BUILD_TESTS
     test_nce_deprecated
     test_sparse_conv_op
     test_sparse_norm_op
-    test_tensor_array_to_tensor
     test_tensor_array_to_tensor_deprecated
     test_unique
     test_one_hot_v2_op)

test/legacy_test/CMakeLists.txt

@@ -1107,7 +1107,8 @@ set(STATIC_BUILD_TESTS
     test_shuffle_batch_op
     test_update_loss_scaling_op
     test_segment_ops
-    test_while_op)
+    test_while_op
+    test_tensor_array_to_tensor)
 
 if(NOT WITH_GPU)
   list(REMOVE_ITEM STATIC_BUILD_TESTS test_fused_attention_op)

test/deprecated/legacy_test/test_tensor_array_to_tensor.py renamed to test/legacy_test/test_tensor_array_to_tensor.py

@@ -79,16 +79,17 @@ def set_program(self):
         self.output_vars = [output]
 
     def run_check(self, executor, scope):
-        executor.run(self.program, scope=scope)
+        result = executor.run(
+            self.program, fetch_list=self.output_vars, scope=scope
+        )
         for i, output in enumerate(self.outputs):
-            np.allclose(
-                np.array(scope.var(self.output_vars[i].name).get_tensor()),
-                output,
-                atol=0,
-            )
-        tensor_array_grad = scope.var(self.array.name).get_lod_tensor_array()
-        for i, input_grad in enumerate(self.input_grads):
-            np.allclose(np.array(tensor_array_grad[i]), input_grad, atol=0)
+            np.allclose(result[i], output, atol=0)
+        if not paddle.framework.use_pir_api():
+            tensor_array_grad = scope.var(
+                self.array.name
+            ).get_lod_tensor_array()
+            for i, input_grad in enumerate(self.input_grads):
+                np.allclose(np.array(tensor_array_grad[i]), input_grad, atol=0)
 
     def test_cpu(self):
         scope = core.Scope()