Skip to content

Commit 1634999

Browse files
Added ConsumeResult struct to profiler interface for continuous profiling
PiperOrigin-RevId: 937131223
1 parent 62c96f1 commit 1634999

1 file changed

Lines changed: 8 additions & 0 deletions

File tree

third_party/xla/third_party/tsl/tsl/profiler/lib/profiler_interface.h

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -15,12 +15,20 @@ limitations under the License.
1515
#ifndef TENSORFLOW_TSL_PROFILER_LIB_PROFILER_INTERFACE_H_
1616
#define TENSORFLOW_TSL_PROFILER_LIB_PROFILER_INTERFACE_H_
1717

18+
#include <any>
19+
#include <cstddef>
20+
1821
#include "xla/tsl/platform/status.h"
1922
#include "tsl/profiler/protobuf/xplane.pb.h"
2023

2124
namespace tsl {
2225
namespace profiler {
2326

27+
struct ConsumeResult {
28+
std::any data;
29+
size_t estimated_size_bytes = 0;
30+
};
31+
2432
// Interface for tensorflow profiler plugins.
2533
//
2634
// ProfileSession calls each of these methods at most once per instance, and

0 commit comments

Comments
 (0)