提交 3c334179 编写于 作者: S silingtong123 提交者: liuwei1031

modify the method of skipping CI in distributed unittests (#21764)

上级 fb067fa4
...@@ -22,17 +22,6 @@ import os ...@@ -22,17 +22,6 @@ import os
flag_name = os.path.splitext(__file__)[0] flag_name = os.path.splitext(__file__)[0]
def skip_ci(func):
on_ci = bool(int(os.environ.get("SKIP_UNSTABLE_CI", '0')))
def __func__(*args, **kwargs):
if on_ci:
return
return func(*args, **kwargs)
return __func__
class TestDistCTR2x2(TestDistBase): class TestDistCTR2x2(TestDistBase):
def _setup_config(self): def _setup_config(self):
self._sync_mode = True self._sync_mode = True
...@@ -43,7 +32,7 @@ class TestDistCTR2x2(TestDistBase): ...@@ -43,7 +32,7 @@ class TestDistCTR2x2(TestDistBase):
"dist_ctr.py", delta=1e-2, check_error_log=True, log_name=flag_name) "dist_ctr.py", delta=1e-2, check_error_log=True, log_name=flag_name)
@skip_ci @unittest.skip(reason="Skip unstable ci")
class TestDistCTRWithL2Decay2x2(TestDistBase): class TestDistCTRWithL2Decay2x2(TestDistBase):
def _setup_config(self): def _setup_config(self):
self._sync_mode = True self._sync_mode = True
......
...@@ -19,17 +19,6 @@ import unittest ...@@ -19,17 +19,6 @@ import unittest
from test_dist_fleet_base import TestFleetBase from test_dist_fleet_base import TestFleetBase
def skip_ci(func):
on_ci = bool(int(os.environ.get("SKIP_UNSTABLE_CI", '0')))
def __func__(*args, **kwargs):
if on_ci:
return
return func(*args, **kwargs)
return __func__
class TestDistMnist2x2(TestFleetBase): class TestDistMnist2x2(TestFleetBase):
def _setup_config(self): def _setup_config(self):
self._sync_mode = False self._sync_mode = False
......
...@@ -24,17 +24,6 @@ from test_dist_fleet_base import TestFleetBase ...@@ -24,17 +24,6 @@ from test_dist_fleet_base import TestFleetBase
from dist_simnet_bow import train_network from dist_simnet_bow import train_network
def skip_ci(func):
on_ci = bool(int(os.environ.get("SKIP_UNSTABLE_CI", '0')))
def __func__(*args, **kwargs):
if on_ci:
return
return func(*args, **kwargs)
return __func__
class TestDistGeoCtr_2x2(TestFleetBase): class TestDistGeoCtr_2x2(TestFleetBase):
def _setup_config(self): def _setup_config(self):
self._sync_mode = False self._sync_mode = False
......
...@@ -20,23 +20,12 @@ import os ...@@ -20,23 +20,12 @@ import os
flag_name = os.path.splitext(__file__)[0] flag_name = os.path.splitext(__file__)[0]
def skip_ci(func):
on_ci = bool(int(os.environ.get("SKIP_UNSTABLE_CI", '0')))
def __func__(*args, **kwargs):
if on_ci:
return
return func(*args, **kwargs)
return __func__
class TestDistSeResneXt2x2Async(TestDistBase): class TestDistSeResneXt2x2Async(TestDistBase):
def _setup_config(self): def _setup_config(self):
self._sync_mode = False self._sync_mode = False
self._use_reader_alloc = False self._use_reader_alloc = False
@skip_ci @unittest.skip(reason="Skip unstable ci")
def test_dist_train(self): def test_dist_train(self):
self.check_with_place( self.check_with_place(
"dist_se_resnext.py", "dist_se_resnext.py",
......
...@@ -21,17 +21,6 @@ import os ...@@ -21,17 +21,6 @@ import os
flag_name = os.path.splitext(__file__)[0] flag_name = os.path.splitext(__file__)[0]
def skip_ci(func):
on_ci = bool(int(os.environ.get("SKIP_UNSTABLE_CI", '0')))
def __func__(*args, **kwargs):
if on_ci:
return
return func(*args, **kwargs)
return __func__
class TestDistSeResnetNCCL2DGC(TestDistBase): class TestDistSeResnetNCCL2DGC(TestDistBase):
def _setup_config(self): def _setup_config(self):
self._sync_mode = True self._sync_mode = True
...@@ -40,7 +29,7 @@ class TestDistSeResnetNCCL2DGC(TestDistBase): ...@@ -40,7 +29,7 @@ class TestDistSeResnetNCCL2DGC(TestDistBase):
self._nccl2_mode = True self._nccl2_mode = True
self._use_dgc = True self._use_dgc = True
@skip_ci @unittest.skip(reason="Skip unstable ci")
def test_dist_train(self): def test_dist_train(self):
import paddle.fluid as fluid import paddle.fluid as fluid
if fluid.core.is_compiled_with_cuda(): if fluid.core.is_compiled_with_cuda():
......
...@@ -21,24 +21,13 @@ import os ...@@ -21,24 +21,13 @@ import os
flag_name = os.path.splitext(__file__)[0] flag_name = os.path.splitext(__file__)[0]
def skip_ci(func):
on_ci = bool(int(os.environ.get("SKIP_UNSTABLE_CI", '0')))
def __func__(*args, **kwargs):
if on_ci:
return
return func(*args, **kwargs)
return __func__
class TestDistSeResneXtNCCL(TestDistBase): class TestDistSeResneXtNCCL(TestDistBase):
def _setup_config(self): def _setup_config(self):
self._sync_mode = True self._sync_mode = True
self._use_reader_alloc = False self._use_reader_alloc = False
self._nccl2_mode = True self._nccl2_mode = True
@skip_ci @unittest.skip(reason="Skip unstable ci")
def test_dist_train(self): def test_dist_train(self):
import paddle.fluid as fluid import paddle.fluid as fluid
if fluid.core.is_compiled_with_cuda(): if fluid.core.is_compiled_with_cuda():
...@@ -56,7 +45,7 @@ class TestDistSeResneXtNCCLMP(TestDistBase): ...@@ -56,7 +45,7 @@ class TestDistSeResneXtNCCLMP(TestDistBase):
self._nccl2_mode = True self._nccl2_mode = True
self._mp_mode = True self._mp_mode = True
@skip_ci @unittest.skip(reason="Skip unstable ci")
def test_dist_train(self): def test_dist_train(self):
import paddle.fluid as fluid import paddle.fluid as fluid
if fluid.core.is_compiled_with_cuda(): if fluid.core.is_compiled_with_cuda():
......
...@@ -21,23 +21,12 @@ import os ...@@ -21,23 +21,12 @@ import os
flag_name = os.path.splitext(__file__)[0] flag_name = os.path.splitext(__file__)[0]
def skip_ci(func):
on_ci = bool(int(os.environ.get("SKIP_UNSTABLE_CI", '0')))
def __func__(*args, **kwargs):
if on_ci:
return
return func(*args, **kwargs)
return __func__
class TestDistSeResneXt2x2(TestDistBase): class TestDistSeResneXt2x2(TestDistBase):
def _setup_config(self): def _setup_config(self):
self._sync_mode = True self._sync_mode = True
self._use_reader_alloc = False self._use_reader_alloc = False
@skip_ci @unittest.skip(reason="Skip unstable ci")
def test_dist_train(self): def test_dist_train(self):
self.check_with_place( self.check_with_place(
"dist_se_resnext.py", "dist_se_resnext.py",
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册