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

add gpu_allocator_try_time config, test=develop (#19675)

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