experiments
/
STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Print the ptxas command line as higher VLOG level.

This commit is contained in:
Frederic Bastien 2020-06-08 13:57:12 -07:00
parent 256cfa1823
commit 3946ad2bfd
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tensorflow/stream_executor/gpu/asm_compiler.cc
View File

@ -214,6 +214,10 @@ port::StatusOr<std::vector<uint8>> CompileGpuAsm(int cc_major, int cc_minor,
} }
ptxas_args.insert(ptxas_args.end(), options.extra_flags.begin(), ptxas_args.insert(ptxas_args.end(), options.extra_flags.begin(),
options.extra_flags.end()); options.extra_flags.end());
if (VLOG_IS_ON(3)) {
VLOG(3) << absl::StrJoin(ptxas_args, " ");
}
ptxas_info_dumper.SetProgram(ptxas_path, ptxas_args); ptxas_info_dumper.SetProgram(ptxas_path, ptxas_args);
ptxas_info_dumper.SetChannelAction(tensorflow::CHAN_STDERR, ptxas_info_dumper.SetChannelAction(tensorflow::CHAN_STDERR,
tensorflow::ACTION_PIPE); tensorflow::ACTION_PIPE);