Skip to content

Commit 584a493

Browse files
loislocopybara-github
authored andcommitted
Add aggregation mode to cupti collector.
The goal of the pr is to generate a small, lightweight aggregated profile that could be collected frequently and would not slow down the execution. PiperOrigin-RevId: 885476831
1 parent fa6d938 commit 584a493

1 file changed

Lines changed: 4 additions & 1 deletion

File tree

tsl/profiler/protobuf/profiler_options.proto

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,7 @@ syntax = "proto3";
22

33
package tensorflow;
44

5-
// Next ID: 16
5+
// Next ID: 17
66
message ProfileOptions {
77
// Some default value of option are not proto3 default value. Use this version
88
// to determine if we should use default option value instead of proto3
@@ -103,6 +103,9 @@ message ProfileOptions {
103103

104104
// If set, this hostname will be used to name the profile file.
105105
string override_hostname = 15;
106+
107+
// If set, the profiler will aggregate the collected events at export time.
108+
bool aggregate_working_set_profile = 16;
106109
}
107110

108111
// Options for remote profiler session manager.

0 commit comments

Comments
 (0)