未验证 提交 a7691603 编写于 作者: Z Zeng Jinle 提交者: GitHub

add gpu_allocator_try_time config, test=develop (#19675)

上级 0b06db94
...@@ -37,7 +37,7 @@ ...@@ -37,7 +37,7 @@
#endif #endif
DEFINE_int64( DEFINE_int64(
gpu_allocator_retry_time, 100, gpu_allocator_retry_time, 10000,
"The retry time (milliseconds) when allocator fails " "The retry time (milliseconds) when allocator fails "
"to allocate memory. No retry if this value is not greater than 0"); "to allocate memory. No retry if this value is not greater than 0");
......
...@@ -204,7 +204,7 @@ def __bootstrap__(): ...@@ -204,7 +204,7 @@ def __bootstrap__():
'cudnn_exhaustive_search', 'selected_gpus', 'sync_nccl_allreduce', 'cudnn_exhaustive_search', 'selected_gpus', 'sync_nccl_allreduce',
'limit_of_tmp_allocation', 'limit_of_tmp_allocation',
'times_excess_than_required_tmp_allocation', 'times_excess_than_required_tmp_allocation',
'cudnn_batchnorm_spatial_persistent' 'cudnn_batchnorm_spatial_persistent', 'gpu_allocator_retry_time'
] ]
core.init_gflags([sys.argv[0]] + core.init_gflags([sys.argv[0]] +
["--tryfromenv=" + ",".join(read_env_flags)]) ["--tryfromenv=" + ",".join(read_env_flags)])
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册