From 3946ad2bfd2504947240548fa5a17d4279c50919 Mon Sep 17 00:00:00 2001 From: Frederic Bastien Date: Mon, 8 Jun 2020 13:57:12 -0700 Subject: [PATCH] Print the ptxas command line as higher VLOG level. --- 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 45c60dc29ee..cc16fadb817 100644 --- a/tensorflow/stream_executor/gpu/asm_compiler.cc +++ b/tensorflow/stream_executor/gpu/asm_compiler.cc @@ -214,6 +214,10 @@ port::StatusOr> CompileGpuAsm(int cc_major, int cc_minor, } ptxas_args.insert(ptxas_args.end(), options.extra_flags.begin(), 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.SetChannelAction(tensorflow::CHAN_STDERR, tensorflow::ACTION_PIPE);