File tree Expand file tree Collapse file tree 2 files changed +5
-18
lines changed
xla/backends/profiler/gpu Expand file tree Collapse file tree 2 files changed +5
-18
lines changed Original file line number Diff line number Diff line change @@ -14,6 +14,7 @@ limitations under the License.
1414==============================================================================*/
1515
1616#include " xla/backends/profiler/gpu/rocm_tracer_utils.h"
17+ #include " tsl/platform/logging.h"
1718
1819// for rocprofiler-sdk
1920namespace xla {
Original file line number Diff line number Diff line change @@ -16,29 +16,15 @@ limitations under the License.
1616#ifndef XLA_BACKENDS_PROFILER_GPU_ROCM_TRACER_UTILS_H_
1717#define XLA_BACKENDS_PROFILER_GPU_ROCM_TRACER_UTILS_H_
1818
19- #include < cstdint>
20- #include < cstring>
21- #include < map>
22- #include < mutex>
23- #include < sstream>
19+ #include < cstddef>
20+ #include < cstdint>
21+ #include < limits>
2422#include < string>
25- #include < unordered_map>
26- #include < vector>
27- #include < time.h>
28- #include < unistd.h>
29- #include < chrono>
30- #include < unistd.h>
3123
3224#include " absl/container/flat_hash_map.h"
33- #include " absl/strings/string_view.h"
3425#include " absl/container/node_hash_set.h"
35- #include " absl/container/node_hash_map .h"
26+ #include " absl/strings/string_view .h"
3627#include " absl/synchronization/mutex.h"
37- #include " xla/tsl/profiler/backends/cpu/annotation_stack.h"
38- #include " xla/tsl/profiler/utils/time_utils.h"
39-
40- #include " tsl/platform/errors.h"
41- #include " tsl/platform/macros.h"
4228
4329namespace xla {
4430namespace profiler {
You can’t perform that action at this time.
0 commit comments