CMakeLists.txt 690 字节
Newer Older
D
Dang Qingqing 已提交
1 2 3
file(GLOB TEST_OPS RELATIVE "${CMAKE_CURRENT_SOURCE_DIR}" "test_*.py")
string(REPLACE ".py" "" TEST_OPS "${TEST_OPS}")

4 5
list(REMOVE_ITEM TEST_OPS test_multi_precision_fp16_train)

D
Dang Qingqing 已提交
6
foreach(src ${TEST_OPS})
Z
Zhang, Guoming 已提交
7
        py_test(${src} SRCS ${src}.py)
D
Dang Qingqing 已提交
8
endforeach()
9 10 11

py_test_modules(test_multi_precision_fp16_train MODULES test_multi_precision_fp16_train ENVS FLAGS_cudnn_deterministic=true FLAGS_cudnn_batchnorm_spatial_persistent=true FLAGS_conv_workspace_size_limit=1000)

Y
YUNSHEN XIE 已提交
12 13
set_tests_properties(test_image_classification_fp16 PROPERTIES TIMEOUT 120)
set_tests_properties(test_weight_decay_extend PROPERTIES TIMEOUT 120)
14
set_tests_properties(test_multi_precision_fp16_train PROPERTIES TIMEOUT 120)