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/gpu/gpu_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/gpu/gpu_executable.cc')
-rw-r--r-- | tensorflow/compiler/xla/service/gpu/gpu_executable.cc | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/tensorflow/compiler/xla/service/gpu/gpu_executable.cc b/tensorflow/compiler/xla/service/gpu/gpu_executable.cc index 51d164cdf4..f5d67b9ea9 100644 --- a/tensorflow/compiler/xla/service/gpu/gpu_executable.cc +++ b/tensorflow/compiler/xla/service/gpu/gpu_executable.cc @@ -116,9 +116,9 @@ GpuExecutable::GpuExecutable( std::unique_ptr<const ThunkSchedule> thunk_schedule, std::unique_ptr<const HloModule> hlo_module, std::unique_ptr<const BufferAssignment> assignment, - std::unique_ptr<HloProfilePrinter> hlo_profile_printer, + std::unique_ptr<HloProfilePrinterData> hlo_profile_printer_data, std::unique_ptr<HloProfileIndexMap> hlo_profile_index_map) - : Executable(std::move(hlo_module), std::move(hlo_profile_printer), + : Executable(std::move(hlo_module), std::move(hlo_profile_printer_data), std::move(hlo_profile_index_map)), ptx_(ptx), cubin_(cubin), |