diff --git a/python/paddle/fluid/tests/unittests/test_clip_by_norm_op.py b/python/paddle/fluid/tests/unittests/test_clip_by_norm_op.py index 46433d78252219fe02c3c4b5ddfc525bda177f18..d78cb9b0321e80f3579c5697abcc7c870ab3179a 100644 --- a/python/paddle/fluid/tests/unittests/test_clip_by_norm_op.py +++ b/python/paddle/fluid/tests/unittests/test_clip_by_norm_op.py @@ -65,7 +65,7 @@ class TestCase3(TestClipByNormOp): self.max_norm = 1.0 -class TestClipByNormOpWithSelectedRows(OpTest): +class TestClipByNormOpWithSelectedRows(unittest.TestCase): def check_with_place(self, place): self.config_test_case() scope = core.Scope() diff --git a/python/paddle/fluid/tests/unittests/test_dropout_op.py b/python/paddle/fluid/tests/unittests/test_dropout_op.py index 7047e5a2d876ad97254cd3ba747c117d0de12ec1..29460d07dfd41724092ca51de3f7a5571ea4e7ae 100644 --- a/python/paddle/fluid/tests/unittests/test_dropout_op.py +++ b/python/paddle/fluid/tests/unittests/test_dropout_op.py @@ -171,6 +171,9 @@ class TestDropoutOpWithSeed(OpTest): self.check_grad(['X'], 'Out', max_relative_error=0.05) +@unittest.skipIf( + not core.is_compiled_with_cuda() or not core.op_support_gpu("dropout"), + "core is not compiled with CUDA or core is not support dropout") class TestFP16DropoutOp(OpTest): def setUp(self): self.op_type = "dropout" @@ -192,10 +195,12 @@ class TestFP16DropoutOp(OpTest): self.fix_seed = True def test_check_output(self): - if core.is_compiled_with_cuda() and core.op_support_gpu("dropout"): - self.check_output_with_place(core.CUDAPlace(0), atol=1e-3) + self.check_output_with_place(core.CUDAPlace(0), atol=1e-3) +@unittest.skipIf( + not core.is_compiled_with_cuda() or not core.op_support_gpu("dropout"), + "core is not compiled with CUDA or core is not support dropout") class TestFP16DropoutOp2(TestFP16DropoutOp): def init_test_case(self): self.input_size = [32, 64, 3] diff --git a/python/paddle/fluid/tests/unittests/test_fill_constant_op.py b/python/paddle/fluid/tests/unittests/test_fill_constant_op.py index 1263657025a5dee5d9cfff510c63385b4a0fe08d..2baec782c42203a3e94e11c165179ff83fbc20fe 100644 --- a/python/paddle/fluid/tests/unittests/test_fill_constant_op.py +++ b/python/paddle/fluid/tests/unittests/test_fill_constant_op.py @@ -81,7 +81,7 @@ class TestFillConstantOp4(OpTest): self.check_output() -class TestFillConstantOpWithSelectedRows(OpTest): +class TestFillConstantOpWithSelectedRows(unittest.TestCase): def check_with_place(self, place): scope = core.Scope() # create Out Variable diff --git a/python/paddle/fluid/tests/unittests/test_fill_op.py b/python/paddle/fluid/tests/unittests/test_fill_op.py index 96af96f6c7c4477ffa5afd822e629bfee5803d2e..36b56ab92801c270552c100d1b68d48678e3320d 100644 --- a/python/paddle/fluid/tests/unittests/test_fill_op.py +++ b/python/paddle/fluid/tests/unittests/test_fill_op.py @@ -55,7 +55,7 @@ class TestFillOp2(OpTest): self.check_output(check_dygraph=False) -class TestFillOp3(OpTest): +class TestFillOp3(unittest.TestCase): def check_with_place(self, place, f_cpu): scope = core.Scope() # create Out Variable diff --git a/python/paddle/fluid/tests/unittests/test_lookup_sparse_table_op.py b/python/paddle/fluid/tests/unittests/test_lookup_sparse_table_op.py index c7f4f3e913bfd66cbbb703c0e73336f9a3563507..a2a036e02a1c889705274fd1b9947a0b962c42b2 100644 --- a/python/paddle/fluid/tests/unittests/test_lookup_sparse_table_op.py +++ b/python/paddle/fluid/tests/unittests/test_lookup_sparse_table_op.py @@ -21,7 +21,7 @@ import paddle.fluid.core as core from paddle.fluid.op import Operator -class TestLookupSpraseTable(OpTest): +class TestLookupSpraseTable(unittest.TestCase): def check_with_place(self, place): scope = core.Scope() diff --git a/python/paddle/fluid/tests/unittests/test_lookup_table_op.py b/python/paddle/fluid/tests/unittests/test_lookup_table_op.py index 5c0c155b1d93f15bd0c881fbd91cb9bb64afc1f0..3442cd71d90a18cdd7e9992c8c71cb9e40f9015b 100644 --- a/python/paddle/fluid/tests/unittests/test_lookup_table_op.py +++ b/python/paddle/fluid/tests/unittests/test_lookup_table_op.py @@ -85,7 +85,7 @@ class TestLookupTableOpWithTensorIdsAndPadding(TestLookupTableOpWithTensorIds): pass -class TestLookupTableWIsSelectedRows(OpTest): +class TestLookupTableWIsSelectedRows(unittest.TestCase): def prepare_ids(self, scope, place): ids_tensor = scope.var('Ids').get_tensor() ids_array = np.array([[0], [4], [3], [5]]).astype("int64") @@ -250,7 +250,7 @@ class TestLookupTableOpWithTensorIdsAndPaddingInt8( pass -class TestLookupTableWIsSelectedRowsInt8(OpTest): +class TestLookupTableWIsSelectedRowsInt8(unittest.TestCase): def prepare_ids(self, scope, place): ids_tensor = scope.var('Ids').get_tensor() ids_array = np.array([[0], [4], [3], [5]]).astype("int64") diff --git a/python/paddle/fluid/tests/unittests/test_lookup_table_v2_op.py b/python/paddle/fluid/tests/unittests/test_lookup_table_v2_op.py index 811d48452e3a0a6878fcd771debafb2639f922d4..29772cd40043c124fa45f1112948b77037303dbe 100644 --- a/python/paddle/fluid/tests/unittests/test_lookup_table_v2_op.py +++ b/python/paddle/fluid/tests/unittests/test_lookup_table_v2_op.py @@ -84,7 +84,7 @@ class TestLookupTableOpWithTensorIdsAndPadding(TestLookupTableOpWithTensorIds): pass -class TestLookupTableWIsSelectedRows(OpTest): +class TestLookupTableWIsSelectedRows(unittest.TestCase): def prepare_ids(self, scope, place): ids_tensor = scope.var('Ids').get_tensor() ids_array = np.array([0, 4, 3, 5]).astype("int64") diff --git a/python/paddle/fluid/tests/unittests/test_sum_op.py b/python/paddle/fluid/tests/unittests/test_sum_op.py index 0be5be6e97d26c6ec42471d078e8e5995727e594..ebf153eb2c45f46314b9a1f28b9b6a0d00affe96 100644 --- a/python/paddle/fluid/tests/unittests/test_sum_op.py +++ b/python/paddle/fluid/tests/unittests/test_sum_op.py @@ -48,7 +48,7 @@ class TestSumOp(OpTest): pass -class TestSelectedRowsSumOp(OpTest): +class TestSelectedRowsSumOp(unittest.TestCase): def setUp(self): self.height = 10 self.row_numel = 12 @@ -144,6 +144,7 @@ class TestLoDTensorAndSelectedRowsOp(TestSelectedRowsSumOp): self.height = 10 self.row_numel = 12 self.rows = [0, 1, 2, 2, 4, 5, 6] + self.dtype = np.float32 def check_with_place(self, place, inplace): scope = core.Scope()