run dist train test single process

7 years ago · 22f58382bc
parent d78cab1519
commit 22f58382bc
2 changed files with 2 additions and 0 deletions
--- a/python/paddle/fluid/tests/unittests/CMakeLists.txt
+++ b/python/paddle/fluid/tests/unittests/CMakeLists.txt
@ -62,6 +62,7 @@ list(REMOVE_ITEM TEST_OPS test_registry)
 list(REMOVE_ITEM TEST_OPS test_fetch_var)
 list(REMOVE_ITEM TEST_OPS test_parallel_op)
 list(REMOVE_ITEM TEST_OPS test_dynrnn_static_input)
+list(REMOVE_ITEM TEST_OPS test_dist_train)

 # tests that can be bundled together in one python process for speed.
 if(WITH_FAST_BUNDLE_TEST)
@ -100,3 +101,4 @@ py_test_modules(test_registry MODULES test_registry)
 py_test_modules(test_fetch_var MODULES test_fetch_var)
 py_test_modules(test_dynrnn_static_input MODULES test_dynrnn_static_input)
 py_test_modules(test_parallel_op MODULES test_parallel_op)
+py_test_modules(test_dist_train MODULES test_dist_train)
--- a/python/paddle/fluid/tests/unittests/test_dist_train_op.py
+++ b/python/paddle/fluid/tests/unittests/test_dist_train_op.py