|
|
@ -58,7 +58,7 @@ def __bootstrap__():
|
|
|
|
|
|
|
|
|
|
|
|
read_env_flags = ['use_pinned_memory', 'check_nan_inf']
|
|
|
|
read_env_flags = ['use_pinned_memory', 'check_nan_inf']
|
|
|
|
if core.is_compile_gpu():
|
|
|
|
if core.is_compile_gpu():
|
|
|
|
read_env_flags.append(['fraction_of_gpu_memory_to_use', 'op_sync'])
|
|
|
|
read_env_flags += ['fraction_of_gpu_memory_to_use', 'op_sync']
|
|
|
|
core.init_gflags([sys.argv[0]] +
|
|
|
|
core.init_gflags([sys.argv[0]] +
|
|
|
|
["--tryfromenv=" + ",".join(read_env_flags)])
|
|
|
|
["--tryfromenv=" + ",".join(read_env_flags)])
|
|
|
|
core.init_glog(sys.argv[0])
|
|
|
|
core.init_glog(sys.argv[0])
|
|
|
|