|
|
@ -385,8 +385,12 @@ function(cc_test_run TARGET_NAME)
|
|
|
|
set_property(TEST ${TARGET_NAME} PROPERTY ENVIRONMENT FLAGS_init_allocated_mem=true)
|
|
|
|
set_property(TEST ${TARGET_NAME} PROPERTY ENVIRONMENT FLAGS_init_allocated_mem=true)
|
|
|
|
set_property(TEST ${TARGET_NAME} PROPERTY ENVIRONMENT FLAGS_cudnn_deterministic=true)
|
|
|
|
set_property(TEST ${TARGET_NAME} PROPERTY ENVIRONMENT FLAGS_cudnn_deterministic=true)
|
|
|
|
# No unit test should exceed 2 minutes.
|
|
|
|
# No unit test should exceed 2 minutes.
|
|
|
|
|
|
|
|
if (APPLE OR WIN32)
|
|
|
|
|
|
|
|
set_tests_properties(${TARGET_NAME} PROPERTIES TIMEOUT 600)
|
|
|
|
|
|
|
|
else()
|
|
|
|
set_tests_properties(${TARGET_NAME} PROPERTIES TIMEOUT 120)
|
|
|
|
set_tests_properties(${TARGET_NAME} PROPERTIES TIMEOUT 120)
|
|
|
|
endif()
|
|
|
|
endif()
|
|
|
|
|
|
|
|
endif()
|
|
|
|
endfunction()
|
|
|
|
endfunction()
|
|
|
|
|
|
|
|
|
|
|
|
function(cc_test TARGET_NAME)
|
|
|
|
function(cc_test TARGET_NAME)
|
|
|
@ -743,9 +747,14 @@ function(py_test TARGET_NAME)
|
|
|
|
WORKING_DIRECTORY ${CMAKE_CURRENT_BINARY_DIR})
|
|
|
|
WORKING_DIRECTORY ${CMAKE_CURRENT_BINARY_DIR})
|
|
|
|
endif()
|
|
|
|
endif()
|
|
|
|
|
|
|
|
|
|
|
|
# No unit test should exceed 2 minutes.
|
|
|
|
if (APPLE OR WIN32)
|
|
|
|
|
|
|
|
set_tests_properties(${TARGET_NAME} PROPERTIES TIMEOUT 600)
|
|
|
|
|
|
|
|
else()
|
|
|
|
|
|
|
|
# No unit test should exceed 2 minutes in Linux.
|
|
|
|
set_tests_properties(${TARGET_NAME} PROPERTIES TIMEOUT 120)
|
|
|
|
set_tests_properties(${TARGET_NAME} PROPERTIES TIMEOUT 120)
|
|
|
|
endif()
|
|
|
|
endif()
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
endif()
|
|
|
|
endfunction()
|
|
|
|
endfunction()
|
|
|
|
|
|
|
|
|
|
|
|
# grpc_library generate grpc code using grpc_cpp_plugin and protoc
|
|
|
|
# grpc_library generate grpc code using grpc_cpp_plugin and protoc
|
|
|
|