|
|
|
|
@ -212,10 +212,10 @@ if(WITH_DISTRIBUTE)
|
|
|
|
|
py_test_modules(test_dgc_op MODULES test_dgc_op)
|
|
|
|
|
endif()
|
|
|
|
|
if(NOT APPLE)
|
|
|
|
|
set_tests_properties(test_dist_mnist PROPERTIES TIMEOUT 300)
|
|
|
|
|
set_tests_properties(test_dist_mnist_nccl PROPERTIES TIMEOUT 300)
|
|
|
|
|
set_tests_properties(test_dist_mnist_lars PROPERTIES TIMEOUT 300)
|
|
|
|
|
set_tests_properties(test_dist_word2vec PROPERTIES TIMEOUT 300)
|
|
|
|
|
set_tests_properties(test_dist_mnist PROPERTIES TIMEOUT 350)
|
|
|
|
|
set_tests_properties(test_dist_mnist_nccl PROPERTIES TIMEOUT 350)
|
|
|
|
|
set_tests_properties(test_dist_mnist_lars PROPERTIES TIMEOUT 350)
|
|
|
|
|
set_tests_properties(test_dist_word2vec PROPERTIES TIMEOUT 350)
|
|
|
|
|
py_test_modules(test_dist_se_resnext MODULES test_dist_se_resnext)
|
|
|
|
|
py_test_modules(test_dist_se_resnext_nccl MODULES test_dist_se_resnext_nccl)
|
|
|
|
|
bash_test_modules(test_launch MODULES test_launch.sh)
|
|
|
|
|
|