Skip to content

Commit e8c1dca

Browse files
sannidhyachauhancopybara-github
authored andcommitted
Add Python bindings for continuous profiling and snapshot retrieval.
PiperOrigin-RevId: 843151720
1 parent 62db3b3 commit e8c1dca

File tree

1 file changed

+9
-0
lines changed

1 file changed

+9
-0
lines changed

tsl/profiler/protobuf/profiler_service.proto

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -17,6 +17,11 @@ service ProfilerService {
1717
rpc Terminate(TerminateRequest) returns (TerminateResponse) {}
1818
// Collects profiling data and returns user-friendly metrics.
1919
rpc Monitor(MonitorRequest) returns (MonitorResponse) {}
20+
// Starts a continuous profiling session.
21+
rpc ContinuousProfiling(ProfileRequest)
22+
returns (ContinuousProfilingResponse) {}
23+
// Gets a snapshot of an ongoing profiling session.
24+
rpc GetSnapShot(GetSnapShotRequest) returns (ProfileResponse) {}
2025
}
2126

2227
message ToolRequestOptions {
@@ -99,6 +104,10 @@ message TerminateRequest {
99104

100105
message TerminateResponse {}
101106

107+
message GetSnapShotRequest {}
108+
109+
message ContinuousProfilingResponse {}
110+
102111
// Next-ID: 4
103112
message MonitorRequest {
104113
// Duration for which to profile between each update.

0 commit comments

Comments
 (0)