From ce806400a4c80c2bdb76a63dc29a8741d0ddc356 Mon Sep 17 00:00:00 2001
From: Frederic Bastien <fbastien@nvidia.com>
Date: Wed, 20 May 2020 08:32:29 -0700
Subject: [PATCH] Print the verbose output of PTXAS.

---
 tensorflow/stream_executor/gpu/asm_compiler.cc | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/tensorflow/stream_executor/gpu/asm_compiler.cc b/tensorflow/stream_executor/gpu/asm_compiler.cc
index 6122ec85c63..45c60dc29ee 100644
--- 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;