CMakeLists.txt

file(GLOB TEST_OPS RELATIVE "${CMAKE_CURRENT_SOURCE_DIR}" "test_*.py")
string(REPLACE ".py" "" TEST_OPS "${TEST_OPS}")

if (WITH_MLU)
    foreach(TEST_OP ${TEST_OPS})
        py_test_modules(${TEST_OP} MODULES ${TEST_OP})
    endforeach(TEST_OP)
    set_tests_properties(test_collective_broadcast PROPERTIES TIMEOUT 120)
    set_tests_properties(test_collective_allreduce PROPERTIES TIMEOUT 120)
endif()