fix tests

test/dygraph_to_static/seq2seq_dygraph_model.py

@@ -297,7 +297,7 @@ def forward(self, inputs):
         loss = paddle.nn.functional.softmax_with_cross_entropy(
             logits=dec_output, label=label, soft_label=False
         )
-        loss = paddle.squeeze(loss, axes=[2])
+        loss = paddle.squeeze(loss, axis=[2])
         max_tar_seq_len = paddle.shape(tar)[1]
         tar_mask = paddle.static.nn.sequence_lod.sequence_mask(
             tar_sequence_length, maxlen=max_tar_seq_len, dtype='float32'
@@ -835,13 +835,13 @@ def forward(self, inputs):
         loss = paddle.nn.functional.softmax_with_cross_entropy(
             logits=dec_output, label=label, soft_label=False
         )
-        loss = paddle.squeeze(loss, axes=[2])
+        loss = paddle.squeeze(loss, axis=[2])
         max_tar_seq_len = paddle.shape(tar)[1]
         tar_mask = paddle.static.nn.sequence_lod.sequence_mask(
             tar_sequence_length, maxlen=max_tar_seq_len, dtype='float32'
         )
         loss = loss * tar_mask
         loss = paddle.mean(loss, axis=[0])
-        loss = fluid.layers.reduce_sum(loss)
+        loss = paddle.sum(loss)
 
         return loss

test/dygraph_to_static/test_se_resnet.py

@@ -20,6 +20,7 @@
 import unittest
 
 import numpy as np
+from dygraph_to_static_util import ast_only_test
 from predictor_utils import PredictorTools
 
 import paddle
@@ -561,6 +562,7 @@ def verify_predict(self):
                 ),
             )
 
+    @ast_only_test
     def test_check_result(self):
         pred_1, loss_1, acc1_1, acc5_1 = self.train(
             self.train_reader, to_static=False

test/dygraph_to_static/test_spec_names.py

@@ -14,6 +14,8 @@
 
 import unittest
 
+from dygraph_to_static_util import enable_fallback_guard
+
 import paddle
 from paddle.nn import Layer
 
@@ -101,4 +103,5 @@ def to_idx(name):
 
 
 if __name__ == '__main__':
-    unittest.main()
+    with enable_fallback_guard("False"):
+        unittest.main()

test/dygraph_to_static/test_to_tensor.py

@@ -15,6 +15,7 @@
 import unittest
 
 import numpy
+from dygraph_to_static_util import ast_only_test, sot_only_test
 
 import paddle
 from paddle.fluid import core
@@ -148,6 +149,7 @@ def test_to_tensor_badreturn(self):
         self.assertTrue(a.stop_gradient == b.stop_gradient)
         self.assertTrue(a.place._equals(b.place))
 
+    @ast_only_test
     def test_to_tensor_err_log(self):
         paddle.disable_static()
         x = paddle.to_tensor([3])
@@ -159,6 +161,18 @@ def test_to_tensor_err_log(self):
                 in str(e)
             )
 
+    @sot_only_test
+    def test_to_tensor_err_log_sot(self):
+        paddle.disable_static()
+        x = paddle.to_tensor([3])
+        try:
+            a = paddle.jit.to_static(case8)(x)
+        except Exception as e:
+            self.assertTrue(
+                "Can't constructs a 'paddle.Tensor' with data type <class 'dict'>"
+                in str(e)
+            )
+
 
 class TestStatic(unittest.TestCase):
     def test_static(self):

test/dygraph_to_static/test_train_step.py

@@ -17,6 +17,7 @@
 from functools import partial
 
 import numpy as np
+from dygraph_to_static_util import enable_fallback_guard
 
 import paddle
 
@@ -433,4 +434,5 @@ def setUp(self):
 
 
 if __name__ == "__main__":
-    unittest.main()
+    with enable_fallback_guard("False"):
+        unittest.main()

test/dygraph_to_static/test_train_step_resnet18_adam.py

@@ -15,6 +15,7 @@
 import platform
 import unittest
 
+from dygraph_to_static_util import enable_fallback_guard
 from test_train_step import (
     TestTrainStepTinyModel,
     loss_fn_tiny_model,
@@ -40,4 +41,5 @@ def setUp(self):
 
 
 if __name__ == "__main__":
-    unittest.main()
+    with enable_fallback_guard("False"):
+        unittest.main()

test/dygraph_to_static/test_train_step_resnet18_sgd.py

@@ -15,6 +15,7 @@
 import platform
 import unittest
 
+from dygraph_to_static_util import enable_fallback_guard
 from test_train_step import (
     TestTrainStepTinyModel,
     loss_fn_tiny_model,
@@ -40,4 +41,5 @@ def setUp(self):
 
 
 if __name__ == "__main__":
-    unittest.main()
+    with enable_fallback_guard("False"):
+        unittest.main()