experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Add a debug string to executable.

Browse Source 
PiperOrigin-RevId: 341403664
Change-Id: I50211247e449a1715c11be844a4d48d5456c9d23
This commit is contained in:
Yunxing Dai 2020-11-09 07:45:24 -08:00 committed by TensorFlower Gardener
parent 9619f1cba1
commit c86b90dc9e
2 changed files with 8 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tensorflow/compiler/xla/service/compiler.h
View File

@ -135,7 +135,7 @@ class AotCompilationMetadata {
public:
AotCompilationMetadata(const AotCompilationMetadata&) = delete;
AotCompilationMetadata& operator=(AotCompilationMetadata const&) = delete;
virtual std::string ToString() const { return ""; }
virtual ~AotCompilationMetadata() = default;
protected:

7
tensorflow/compiler/xla/service/executable.h
View File

@ -375,6 +375,10 @@ class Executable {
bool dumping_snapshot() const { return hlo_proto_ != nullptr; }
HloProto const* hlo_proto() const { return hlo_proto_.get(); }
std::string& debug_info() { return debug_info_; }
void set_debug_info(const std::string& debug_info) {
debug_info_ = debug_info;
}
// Gather unused but donated buffers, return them to the caller of this API.
// We don't free buffers inside this function since the caller could have
// different preferences for buffer deallocation. For example, in TensorFlow,
@ -399,6 +403,9 @@ class Executable {
std::unique_ptr<HloProfilePrinterData> hlo_profile_printer_data_;
std::unique_ptr<HloProfileIndexMap> hlo_profile_index_map_;
// Generic debug information as a string.
std::string debug_info_;
};
} // namespace xla