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

Print the verbose output of PTXAS.

Browse Source
This commit is contained in:
Frederic Bastien 2020-05-20 08:32:29 -07:00
parent 854e0bfdb2
commit ce806400a4
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

@ -228,6 +228,10 @@ port::StatusOr<std::vector<uint8>> CompileGpuAsm(int cc_major, int cc_minor,
absl::StrFormat("ptxas exited with non-zero error code %d, output: %s", absl::StrFormat("ptxas exited with non-zero error code %d, output: %s",
exit_status, stderr_output)); exit_status, stderr_output));
} }
// Print the verbose output of ptxas.
if (!stderr_output.empty()) {
VLOG(2) << stderr_output;
}
// Read in the result of compilation and return it as a byte vector. // Read in the result of compilation and return it as a byte vector.
std::string cubin; std::string cubin;