wehub-resource-sync
41 lines
1.4 KiB
CMake
41 lines
1.4 KiB
CMake
if(WITH_CUSTOM_DEVICE AND NOT WITH_GPU)
|
|
set(PLUGIN_URL https://github.com/PaddlePaddle/PaddleCustomDevice.git)
|
|
set(PLUGIN_TAG develop)
|
|
|
|
file(
|
|
GLOB TEST_OPS
|
|
RELATIVE "${CMAKE_CURRENT_SOURCE_DIR}"
|
|
"test_*.py")
|
|
string(REPLACE ".py" "" TEST_OPS "${TEST_OPS}")
|
|
|
|
list(REMOVE_ITEM TEST_OPS test_custom_cpu_profiler_plugin)
|
|
list(REMOVE_ITEM TEST_OPS test_custom_cpu_to_static)
|
|
list(REMOVE_ITEM TEST_OPS test_collective_process_group_xccl)
|
|
|
|
foreach(TEST_OP ${TEST_OPS})
|
|
py_test(
|
|
${TEST_OP}
|
|
SRCS ${TEST_OP}.py ENVS FLAGS_allocator_strategy=naive_best_fit
|
|
PLUGIN_URL=${PLUGIN_URL} PLUGIN_TAG=${PLUGIN_TAG}
|
|
FLAGS_enable_pir_with_pt_in_dy2st=False)
|
|
endforeach()
|
|
|
|
bash_test_modules(
|
|
test_fleet_launch_custom_device
|
|
START_BASH
|
|
test_fleet_launch_custom_device.sh
|
|
ENVS
|
|
PYTHONPATH=""
|
|
FLAGS_allocator_strategy=naive_best_fit
|
|
PADDLE_BINARY_DIR=${PADDLE_BINARY_DIR}
|
|
PLUGIN_URL=${PLUGIN_URL}
|
|
PLUGIN_TAG=${PLUGIN_TAG}
|
|
PYTHON_EXECUTABLE=${PYTHON_EXECUTABLE})
|
|
set_tests_properties(test_fleet_launch_custom_device PROPERTIES TIMEOUT 120)
|
|
set_tests_properties(test_custom_cpu_plugin PROPERTIES TIMEOUT 120)
|
|
set_tests_properties(test_custom_op_setup PROPERTIES TIMEOUT 120)
|
|
# cpp testing
|
|
paddle_test(extension_header_test SRCS extension_header_test.cc DEPS phi
|
|
common)
|
|
endif()
|