From 584a49317e4e339ff33041463a20b75f749c759d Mon Sep 17 00:00:00 2001 From: Ilya Tikhonovskiy Date: Wed, 18 Mar 2026 02:49:39 -0700 Subject: [PATCH] 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 --- tsl/profiler/protobuf/profiler_options.proto | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/tsl/profiler/protobuf/profiler_options.proto b/tsl/profiler/protobuf/profiler_options.proto index 04c46af2d..8820abe6a 100644 --- a/tsl/profiler/protobuf/profiler_options.proto +++ b/tsl/profiler/protobuf/profiler_options.proto @@ -2,7 +2,7 @@ syntax = "proto3"; package tensorflow; -// Next ID: 16 +// Next ID: 17 message ProfileOptions { // Some default value of option are not proto3 default value. Use this version // to determine if we should use default option value instead of proto3 @@ -103,6 +103,9 @@ message ProfileOptions { // If set, this hostname will be used to name the profile file. string override_hostname = 15; + + // If set, the profiler will aggregate the collected events at export time. + bool aggregate_working_set_profile = 16; } // Options for remote profiler session manager.