[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
Showing
- third_party/xla/xla/backends/cpu/runtime/all_gather_thunk.cc 0 additions, 3 deletionsthird_party/xla/xla/backends/cpu/runtime/all_gather_thunk.cc
- third_party/xla/xla/backends/cpu/runtime/all_reduce_thunk.cc 0 additions, 2 deletionsthird_party/xla/xla/backends/cpu/runtime/all_reduce_thunk.cc
- third_party/xla/xla/backends/cpu/runtime/all_to_all_thunk.cc 0 additions, 2 deletionsthird_party/xla/xla/backends/cpu/runtime/all_to_all_thunk.cc
- third_party/xla/xla/backends/cpu/runtime/call_thunk.cc 0 additions, 2 deletionsthird_party/xla/xla/backends/cpu/runtime/call_thunk.cc
- third_party/xla/xla/backends/cpu/runtime/collective_permute_thunk.cc 0 additions, 3 deletions.../xla/xla/backends/cpu/runtime/collective_permute_thunk.cc
- third_party/xla/xla/backends/cpu/runtime/convolution_thunk.cc 0 additions, 3 deletions...d_party/xla/xla/backends/cpu/runtime/convolution_thunk.cc
- third_party/xla/xla/backends/cpu/runtime/copy_thunk.cc 0 additions, 2 deletionsthird_party/xla/xla/backends/cpu/runtime/copy_thunk.cc
- third_party/xla/xla/backends/cpu/runtime/custom_call_thunk.cc 0 additions, 3 deletions...d_party/xla/xla/backends/cpu/runtime/custom_call_thunk.cc
- third_party/xla/xla/backends/cpu/runtime/dot_thunk.cc 0 additions, 2 deletionsthird_party/xla/xla/backends/cpu/runtime/dot_thunk.cc
- third_party/xla/xla/backends/cpu/runtime/fft_thunk.cc 0 additions, 2 deletionsthird_party/xla/xla/backends/cpu/runtime/fft_thunk.cc
- third_party/xla/xla/backends/cpu/runtime/infeed_thunk.cc 0 additions, 3 deletionsthird_party/xla/xla/backends/cpu/runtime/infeed_thunk.cc
- third_party/xla/xla/backends/cpu/runtime/kernel_thunk.cc 6 additions, 10 deletionsthird_party/xla/xla/backends/cpu/runtime/kernel_thunk.cc
- third_party/xla/xla/backends/cpu/runtime/logical_id_thunk.cc 0 additions, 3 deletionsthird_party/xla/xla/backends/cpu/runtime/logical_id_thunk.cc
- third_party/xla/xla/backends/cpu/runtime/outfeed_thunk.cc 0 additions, 2 deletionsthird_party/xla/xla/backends/cpu/runtime/outfeed_thunk.cc
- third_party/xla/xla/backends/cpu/runtime/reduce_scatter_thunk.cc 0 additions, 3 deletions...arty/xla/xla/backends/cpu/runtime/reduce_scatter_thunk.cc
- third_party/xla/xla/backends/cpu/runtime/rng_state_thunk.cc 0 additions, 2 deletionsthird_party/xla/xla/backends/cpu/runtime/rng_state_thunk.cc
- third_party/xla/xla/backends/cpu/runtime/sort_thunk.cc 0 additions, 2 deletionsthird_party/xla/xla/backends/cpu/runtime/sort_thunk.cc
- third_party/xla/xla/backends/cpu/runtime/thunk.h 8 additions, 8 deletionsthird_party/xla/xla/backends/cpu/runtime/thunk.h
- third_party/xla/xla/backends/cpu/runtime/while_thunk.cc 0 additions, 3 deletionsthird_party/xla/xla/backends/cpu/runtime/while_thunk.cc
Please register or sign in to comment