aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorA. Unique TensorFlower <gardener@tensorflow.org>2024-02-08 15:49:50 -0800
committerTensorFlower Gardener <gardener@tensorflow.org>2024-02-08 15:54:10 -0800
commit7212125a9eeb52f4b2188a1f9e36679020a046a6 (patch)
tree658e9b8a2626d7848e8f7a13f8ee02c5a83cc5da
parent50236fa45e774f6dfcc71b0dac814f9b6e5353b6 (diff)
downloadtensorflow-7212125a9eeb52f4b2188a1f9e36679020a046a6.tar.gz
Add reduce-scatter resource name.
PiperOrigin-RevId: 605450393
-rw-r--r--third_party/xla/xla/service/latency_hiding_scheduler.cc2
1 files changed, 2 insertions, 0 deletions
diff --git a/third_party/xla/xla/service/latency_hiding_scheduler.cc b/third_party/xla/xla/service/latency_hiding_scheduler.cc
index eca6ba372b9..8ba9d68143d 100644
--- a/third_party/xla/xla/service/latency_hiding_scheduler.cc
+++ b/third_party/xla/xla/service/latency_hiding_scheduler.cc
@@ -380,6 +380,8 @@ absl::string_view AsyncTracker::GetResourceName(int64_t resource_type) const {
return "kSendHost";
case ResourceTypeToIndex(ResourceType::kRecvHost):
return "kRecvHost";
+ case ResourceTypeToIndex(ResourceType::kReduceScatter):
+ return "kReduceScatter";
default:
return "Not a valid default resource";
}