Skip to content
Snippets Groups Projects
Commit d31fb9f4 authored by Eugene Zhulenev's avatar Eugene Zhulenev Committed by TensorFlower Gardener
Browse files

[xla:cpu] Remove TraceMe annotations from individual thunks

Because all thunks in XLA:CPU are non-blocking, TraceMe annotations are super confusing as they only capture the time that it takes to launch async tasks into the thread pool.

Instead ThunkExecutor should use TraceMe producers/consumers to correctly profile async thunks execution.

Also it's very easy to forget to add TraceMe to individual thunks (judging by the fact that not all thunks have TraceMe annotations), and it's better to annotate them in a single place.

PiperOrigin-RevId: 729734307
parent e78b8cc0
No related merge requests found
Showing
with 14 additions and 60 deletions
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment