Print the verbose output of PTXAS.

2020-05-20 08:32:29 -07:00 · 2020-05-20 08:32:29 -07:00 · ce806400a4
commit ce806400a4
parent 854e0bfdb2
1 changed files with 4 additions and 0 deletions
--- a/tensorflow/stream_executor/gpu/asm_compiler.cc
+++ b/tensorflow/stream_executor/gpu/asm_compiler.cc
@ -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",
                        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.
  std::string cubin;