其他
FLAGS_benchmark
(始于0.12.0)
用于基准测试。设置后,它将使局域删除同步,添加一些内存使用日志,并在内核启动后同步所有cuda内核。
取值范围
Bool型,缺省值为False。
示例
FLAGS_benchmark=True - 同步以测试基准。
FLAGS_inner_op_parallelism
(始于1.3.0)
大多数Operators都在单线程模式下工作,但对于某些Operators,使用多线程更合适。 例如,优化稀疏梯度的优化Op使用多线程工作会更快。该flag用于设置Op内的线程数。
取值范围
Int32型,缺省值为0,这意味着operator将不会在多线程模式下运行。
示例
FLAGS_inner_op_parallelism=5 - 将operator内的线程数设为5。
注意
目前只有稀疏的adam op支持inner_op_parallelism。
FLAGS_max_body_size
(始于1.0.0)
控制BRPC中的最大消息大小。
取值范围
Int32型,缺省值为2147483647。
示例
FLAGS_max_body_size=2147483647 - 将BRPC消息大小设为2147483647。
FLAGS_sync_nccl_allreduce
(始于1.3)
如果FLAGS_sync_nccl_allreduce为True,则会在allreduce_op_handle中调用 cudaStreamSynchronize(nccl_stream) ,这种模式在某些情况下可以获得更好的性能。
取值范围
Bool型,缺省值为True。
示例
FLAGS_sync_nccl_allreduce=True - 在allreduce_op_handle中调用 cudaStreamSynchronize(nccl_stream) 。
FLAGS_tracer_profile_fname
(始于1.4.0)
FLAGS_tracer_profile_fname表示由gperftools生成的命令式跟踪器的分析器文件名。仅在编译选项选择`WITH_PROFILER = ON`时有效。如果禁用则设为empty。
取值范围
String型,缺省值为(“gperf”)。
示例
FLAGS_tracer_profile_fname=”gperf_profile_file” - 将命令式跟踪器的分析器文件名设为”gperf_profile_file”。