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

7 8
if(WITH_XPU_BKCL)
  list(REMOVE_ITEM TEST_OPS "test_gen_bkcl_id_op")
9 10
endif()

11 12 13 14 15 16
file(
  GLOB DIST_TEST_OPS
  RELATIVE "${CMAKE_CURRENT_SOURCE_DIR}"
  "test_dist_*.py")
if(WITH_XPU_BKCL)
  list(APPEND DIST_TEST_OPS test_gen_bkcl_id_op)
17 18
endif()

T
tianshuo78520a 已提交
19 20 21
list(REMOVE_ITEM TEST_OPS test_concat_op_xpu)
list(REMOVE_ITEM TEST_OPS test_mean_op_xpu)

T
tianshuo78520a 已提交
22
foreach(TEST_OP ${TEST_OPS})
23
  py_test_modules(${TEST_OP} MODULES ${TEST_OP})
T
tianshuo78520a 已提交
24
endforeach(TEST_OP)
25

26
foreach(TEST_OP ${DIST_TEST_OPS})
27
  py_test_modules(${TEST_OP} MODULES ${TEST_OP})
28 29
endforeach(TEST_OP)

30 31
set_tests_properties(test_mul_op_xpu PROPERTIES TIMEOUT 120)
set_tests_properties(test_conv2d_op_xpu PROPERTIES TIMEOUT 120)