【AMP OP&Test】update bfp16 & fp16 unit test for concat op. (#52357)

* update bfp16 & fp16 test. * update. --------- Co-authored-by: Roc <[email protected]>
PaddlePaddle · Mar 31, 2023 · 496bbea · 496bbea
1 parent 8372329
commit 496bbea
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/python/paddle/fluid/tests/unittests/test_concat_op.py b/python/paddle/fluid/tests/unittests/test_concat_op.py
@@ -296,7 +296,13 @@ def get_dtype(self):
     globals()[cls_name] = TestConcatBf16
 
 
+# add all unit test maybe timeout.
 create_test_bf16(TestConcatOp)
+create_test_bf16(TestConcatOp2)
+# create_test_bf16(TestConcatOp3)
+create_test_bf16(TestConcatOp4)
+# create_test_bf16(TestConcatOp5)
+# create_test_bf16(TestConcatOp6)
 
 
 class TestConcatOpError(unittest.TestCase):