paddlepaddle--paddle
wehub-resource-sync
45 行
1.3 KiB
CMake
45 行
1.3 KiB
CMake
if(WITH_XPU)
|
|
add_subdirectory(amp)
|
|
add_subdirectory(cpp)
|
|
endif()
|
|
|
|
file(
|
|
GLOB TEST_OPS
|
|
RELATIVE "${CMAKE_CURRENT_SOURCE_DIR}"
|
|
"test_*.py")
|
|
string(REPLACE ".py" "" TEST_OPS "${TEST_OPS}")
|
|
|
|
list(FILTER TEST_OPS EXCLUDE REGEX "^test_dist_")
|
|
|
|
if(WITH_XPU_BKCL)
|
|
list(REMOVE_ITEM TEST_OPS "test_gen_bkcl_id_op")
|
|
endif()
|
|
|
|
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)
|
|
endif()
|
|
string(REPLACE ".py" "" DIST_TEST_OPS "${DIST_TEST_OPS}")
|
|
|
|
foreach(TEST_OP ${TEST_OPS})
|
|
py_test_modules(${TEST_OP} MODULES ${TEST_OP})
|
|
endforeach()
|
|
|
|
foreach(TEST_OP ${DIST_TEST_OPS})
|
|
py_test_modules(${TEST_OP} MODULES ${TEST_OP})
|
|
endforeach()
|
|
|
|
set_tests_properties(test_conv2d_op_xpu PROPERTIES TIMEOUT 120)
|
|
set_tests_properties(test_mul_op_xpu PROPERTIES TIMEOUT 120)
|
|
set_tests_properties(test_slice_op_xpu PROPERTIES TIMEOUT 40)
|
|
set_tests_properties(test_matmul_v2_op_xpu PROPERTIES TIMEOUT 900)
|
|
set_tests_properties(test_collective_identity_xpu
|
|
PROPERTIES LABELS "RUN_TYPE=DIST_KUNLUN")
|
|
set_tests_properties(test_collective_allgather_xpu
|
|
PROPERTIES LABELS "RUN_TYPE=DIST_KUNLUN")
|
|
set_tests_properties(test_collective_allreduce_xpu
|
|
PROPERTIES LABELS "RUN_TYPE=DIST_KUNLUN")
|