CMakeLists.txt 428 字节
Newer Older
1 2 3 4 5 6
file(GLOB TEST_OPS RELATIVE "${CMAKE_CURRENT_SOURCE_DIR}" "test_*.py")
string(REPLACE ".py" "" TEST_OPS "${TEST_OPS}")

foreach(TEST_OP ${TEST_OPS})
    py_test_modules(${TEST_OP} MODULES ${TEST_OP})
endforeach(TEST_OP)
Y
YUNSHEN XIE 已提交
7
set_tests_properties(test_concat_mkldnn_op PROPERTIES TIMEOUT 120)
8
set_tests_properties(test_conv3d_mkldnn_op PROPERTIES TIMEOUT 120)
9
set_tests_properties(test_flags_mkldnn_ops_on_off PROPERTIES TIMEOUT 120)