diff options
author | 2018-01-24 16:10:12 -0800 | |
---|---|---|
committer | 2018-01-24 16:23:00 -0800 | |
commit | ffa63e57bdd703ae051ae849af5b5a272fca2223 (patch) | |
tree | bf95aadf07a2a8c590a238fbc9227d7cdf7f4360 /tensorflow/compiler/xla/service/executable.cc | |
parent | 7bf8ccdb4ef5b0b28c1cf0d5084e07ffbf0e2703 (diff) |
[TF:XLA] Replace most of HloProfilePrinter by a protocol buffer
This change replaces the meat of HloProfilePrinter with a protobuf
HloProfilePrinterData. The original plan was to serialize HloProfilePrinter
into C++ source code and put that in a .cc file along with the string for the
xla::ProgramShape. However, since we now directly serialize xla::ProgramShape
into a .o file, for consistency I think we should do the same thing for
HloProfilePrinter (instead of adding yet another output file to tfcompile).
The change itself is fairly simple, it is large mostly due to the mass renaming
I had to do.
PiperOrigin-RevId: 183158192
Diffstat (limited to 'tensorflow/compiler/xla/service/executable.cc')
-rw-r--r-- | tensorflow/compiler/xla/service/executable.cc | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/compiler/xla/service/executable.cc b/tensorflow/compiler/xla/service/executable.cc index 21e7fbea29..90481c7a88 100644 --- a/tensorflow/compiler/xla/service/executable.cc +++ b/tensorflow/compiler/xla/service/executable.cc @@ -73,7 +73,7 @@ StatusOr<std::unique_ptr<ShapedBuffer>> Executable::ExecuteOnStreamWrapper( std::unique_ptr<HloExecutionProfile> profile_ptr = module_config().debug_options().xla_hlo_profile() && hlo_profiling_enabled() - ? MakeUnique<HloExecutionProfile>(&hlo_profile_printer(), + ? MakeUnique<HloExecutionProfile>(&hlo_profile_printer_data(), &hlo_profile_index_map()) : nullptr; |