| load("//tensorflow:tensorflow.bzl", "tf_cc_test") |
| |
| package( |
| default_visibility = ["//tensorflow:internal"], |
| licenses = ["notice"], # Apache 2.0 |
| ) |
| |
| cc_library( |
| name = "host_tracer_utils", |
| srcs = ["host_tracer_utils.cc"], |
| hdrs = ["host_tracer_utils.h"], |
| visibility = ["//tensorflow/core/profiler:friends"], |
| deps = [ |
| "//tensorflow/core:lib", |
| "//tensorflow/core/profiler/internal:parse_annotation", |
| "//tensorflow/core/profiler/internal:traceme_recorder", |
| "//tensorflow/core/profiler/protobuf:xplane_proto_cc", |
| "//tensorflow/core/profiler/utils:xplane_builder", |
| "@com_google_absl//absl/container:flat_hash_map", |
| ], |
| ) |
| |
| cc_library( |
| name = "host_tracer", |
| srcs = ["host_tracer.cc"], |
| deps = [ |
| ":host_tracer_utils", |
| "//tensorflow/core:core_cpu_lib", |
| "//tensorflow/core:lib", |
| "//tensorflow/core:lib_internal", |
| "//tensorflow/core:protos_all_cc", |
| "//tensorflow/core/profiler/internal:profiler_factory", |
| "//tensorflow/core/profiler/internal:profiler_interface", |
| "//tensorflow/core/profiler/internal:traceme_recorder", |
| "//tensorflow/core/profiler/protobuf:xplane_proto_cc", |
| "//tensorflow/core/profiler/utils:xplane_schema", |
| "//tensorflow/core/profiler/utils:xplane_utils", |
| "@com_google_absl//absl/strings", |
| ], |
| alwayslink = True, |
| ) |
| |
| tf_cc_test( |
| name = "host_tracer_test", |
| srcs = ["host_tracer_test.cc"], |
| deps = [ |
| ":host_tracer", |
| "//tensorflow/core:core_cpu_lib", |
| "//tensorflow/core:lib", |
| "//tensorflow/core:protos_all_cc", |
| "//tensorflow/core:test", |
| "//tensorflow/core/profiler/internal:profiler_interface", |
| "//tensorflow/core/profiler/lib:traceme", |
| "//tensorflow/core/profiler/protobuf:xplane_proto_cc", |
| "//tensorflow/core/profiler/utils:xplane_schema", |
| "//tensorflow/core/profiler/utils:xplane_visitor", |
| "@com_google_absl//absl/types:optional", |
| "@com_google_googletest//:gtest_main", |
| ], |
| ) |
| |
| cc_library( |
| name = "python_tracer", |
| srcs = ["python_tracer.cc"], |
| copts = ["-fexceptions"], |
| features = ["-use_header_modules"], |
| deps = [ |
| "//tensorflow/core:core_cpu_lib", |
| "//tensorflow/core:lib", |
| "//tensorflow/core:lib_internal", |
| "//tensorflow/core:protos_all_cc", |
| "//tensorflow/core/profiler/internal:profiler_factory", |
| "//tensorflow/core/profiler/internal:profiler_interface", |
| "//tensorflow/core/profiler/protobuf:xplane_proto_cc", |
| "//tensorflow/core/profiler/utils:xplane_schema", |
| "//tensorflow/core/profiler/utils:xplane_utils", |
| "//tensorflow/python/profiler/internal:python_hooks", |
| "@com_google_absl//absl/strings", |
| ], |
| alwayslink = True, |
| ) |
| |
| cc_library( |
| name = "metadata_collector", |
| srcs = ["metadata_collector.cc"], |
| deps = [ |
| "//tensorflow/compiler/xla/service/gpu:gpu_debug_info_manager", |
| "//tensorflow/core:lib", |
| "//tensorflow/core:lib_internal", |
| "//tensorflow/core/profiler/internal:profiler_factory", |
| "//tensorflow/core/profiler/internal:profiler_interface", |
| "//tensorflow/core/profiler/protobuf:xplane_proto_cc", |
| "//tensorflow/core/profiler/utils:xplane_builder", |
| "//tensorflow/core/profiler/utils:xplane_schema", |
| "//tensorflow/core/profiler/utils:xplane_utils", |
| ], |
| alwayslink = True, |
| ) |