|
|
@ -119,7 +119,7 @@ def __bootstrap__():
|
|
|
|
read_env_flags = [
|
|
|
|
read_env_flags = [
|
|
|
|
'use_pinned_memory', 'check_nan_inf', 'benchmark', 'warpctc_dir',
|
|
|
|
'use_pinned_memory', 'check_nan_inf', 'benchmark', 'warpctc_dir',
|
|
|
|
'eager_delete_scope', 'use_mkldnn', 'initial_cpu_memory_in_mb',
|
|
|
|
'eager_delete_scope', 'use_mkldnn', 'initial_cpu_memory_in_mb',
|
|
|
|
'init_allocated_mem'
|
|
|
|
'init_allocated_mem', 'rpc_deadline'
|
|
|
|
]
|
|
|
|
]
|
|
|
|
if core.is_compiled_with_cuda():
|
|
|
|
if core.is_compiled_with_cuda():
|
|
|
|
read_env_flags += [
|
|
|
|
read_env_flags += [
|
|
|
|