diff options
| author | Clive Verghese <cliveverghese@google.com> | 2024-02-08 16:32:24 -0800 |
|---|---|---|
| committer | TensorFlower Gardener <gardener@tensorflow.org> | 2024-02-08 16:40:23 -0800 |
| commit | fa0f62b268dbd9687432e00d827db59cbe6c83f7 (patch) | |
| tree | 142ccb4f993bad928470a524a3f85f08cefdbce7 | |
| parent | 1a56d6f8f78824a10a266fc784674405e43fb0ab (diff) | |
| download | tensorflow-fa0f62b268dbd9687432e00d827db59cbe6c83f7.tar.gz | |
Add kTaskEnvPlaneName for task environment information.
PiperOrigin-RevId: 605461449
3 files changed, 6 insertions, 1 deletions
diff --git a/third_party/xla/third_party/tsl/tsl/profiler/convert/BUILD b/third_party/xla/third_party/tsl/tsl/profiler/convert/BUILD index a92c9f0cdf7..85e04b05b4f 100644 --- a/third_party/xla/third_party/tsl/tsl/profiler/convert/BUILD +++ b/third_party/xla/third_party/tsl/tsl/profiler/convert/BUILD @@ -1,9 +1,9 @@ load("//tsl:tsl.bzl", "set_external_visibility") +load("//tsl/platform:build_config.bzl", "tsl_cc_test") load( "@local_tsl//tsl/platform:rules_cc.bzl", "cc_library", ) -load("//tsl/platform:build_config.bzl", "tsl_cc_test") load("//tsl/profiler/builds:build_config.bzl", "tf_profiler_copts") package( diff --git a/third_party/xla/third_party/tsl/tsl/profiler/utils/xplane_schema.cc b/third_party/xla/third_party/tsl/tsl/profiler/utils/xplane_schema.cc index 0f330f7653c..1d2c123bdcf 100644 --- a/third_party/xla/third_party/tsl/tsl/profiler/utils/xplane_schema.cc +++ b/third_party/xla/third_party/tsl/tsl/profiler/utils/xplane_schema.cc @@ -59,6 +59,8 @@ const absl::string_view kCounterEventsLineName = "_counters_"; const absl::string_view kDeviceVendorNvidia = "Nvidia"; const absl::string_view kDeviceVendorAMD = "AMD"; +const absl::string_view kTaskEnvPlaneName = "Task Environment"; + namespace { constexpr int kNumHostEventTypes = diff --git a/third_party/xla/third_party/tsl/tsl/profiler/utils/xplane_schema.h b/third_party/xla/third_party/tsl/tsl/profiler/utils/xplane_schema.h index 6b8fa485ccf..df8084eb925 100644 --- a/third_party/xla/third_party/tsl/tsl/profiler/utils/xplane_schema.h +++ b/third_party/xla/third_party/tsl/tsl/profiler/utils/xplane_schema.h @@ -77,6 +77,9 @@ TF_CONST_INIT extern const absl::string_view kCounterEventsLineName; TF_CONST_INIT extern const absl::string_view kDeviceVendorNvidia; TF_CONST_INIT extern const absl::string_view kDeviceVendorAMD; +// Name of Xplane that contains environment information +TF_CONST_INIT extern const absl::string_view kTaskEnvPlaneName; + // Max collectives to display per TPU. // Since in most cases there will be more than 9 collectives, the last line // contains all collectives that did not qualify to get their own line. |