CMakeLists.txt 723 字节
Newer Older
1
if(WITH_IPU)
2 3 4 5 6 7 8 9 10
  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})
    # set all UTs timeout to 200s
    set_tests_properties(${TEST_OP} PROPERTIES TIMEOUT 200)
W
Wilber 已提交
11
  endforeach()
12

13 14 15 16
  set_tests_properties(test_conv_op_ipu PROPERTIES TIMEOUT 300)
  set_tests_properties(test_elemetwise_x_op_ipu PROPERTIES TIMEOUT 300)
  set_tests_properties(test_reduce_x_op_ipu PROPERTIES TIMEOUT 600)
  set_tests_properties(test_save_load_ipu PROPERTIES TIMEOUT 600)
17
  set_tests_properties(test_activation_ops_ipu PROPERTIES TIMEOUT 600)
A
Allen Guo 已提交
18 19 20

  add_subdirectory(custom_ops)

21
endif()