diff options
author | A. Unique TensorFlower <gardener@tensorflow.org> | 2024-02-08 15:49:50 -0800 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2024-02-08 15:54:10 -0800 |
commit | 7212125a9eeb52f4b2188a1f9e36679020a046a6 (patch) | |
tree | 658e9b8a2626d7848e8f7a13f8ee02c5a83cc5da | |
parent | 50236fa45e774f6dfcc71b0dac814f9b6e5353b6 (diff) | |
download | tensorflow-7212125a9eeb52f4b2188a1f9e36679020a046a6.tar.gz |
Add reduce-scatter resource name.
PiperOrigin-RevId: 605450393
-rw-r--r-- | third_party/xla/xla/service/latency_hiding_scheduler.cc | 2 |
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"; } |