Thanks to visit codestin.com
Credit goes to github.com

Skip to content

Commit 1384e14

Browse files
Automated Code Change
PiperOrigin-RevId: 809844920
1 parent a26feac commit 1384e14

File tree

3 files changed

+10
-10
lines changed

3 files changed

+10
-10
lines changed

tensorflow/core/distributed_runtime/cluster_function_library_runtime.cc

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -74,7 +74,7 @@ absl::Status ClusterFunctionLibraryRuntime::ConstructFunctionGraph(
7474
}
7575

7676
auto input_node_builder =
77-
NodeDefBuilder(strings::StrCat("_recv_", in.name(), "_", i), "_Recv")
77+
NodeDefBuilder(absl::StrCat("_recv_", in.name(), "_", i), "_Recv")
7878
.Attr("tensor_type", dtypes[0])
7979
.Attr("tensor_name", in.name())
8080
.Attr("send_device", target)
@@ -125,7 +125,7 @@ absl::Status ClusterFunctionLibraryRuntime::ConstructFunctionGraph(
125125
}
126126

127127
auto output_node_builder =
128-
NodeDefBuilder(strings::StrCat("_send_", out.name(), "_", i), "_Send")
128+
NodeDefBuilder(absl::StrCat("_send_", out.name(), "_", i), "_Send")
129129
.Input(func_name, i, dtypes[0])
130130
.Attr("tensor_name", out.name())
131131
.Attr("send_device", target)

tensorflow/core/distributed_runtime/collective_param_resolver_distributed_test.cc

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -116,7 +116,7 @@ class DeviceResDistTest : public ::testing::Test {
116116
void DefineWorkers(int num_workers, int num_devices,
117117
const string& device_type, bool nccl) {
118118
for (int w = 0; w < num_workers; ++w) {
119-
string name = strings::StrCat("/job:worker/replica:0/task:", w);
119+
string name = absl::StrCat("/job:worker/replica:0/task:", w);
120120
DefineWorker(name, device_type, num_devices, nccl);
121121
}
122122
}
@@ -164,7 +164,7 @@ class DeviceResDistTest : public ::testing::Test {
164164
CollectiveType coll_type = REDUCTION_COLLECTIVE,
165165
int source_rank = 0) {
166166
for (int wi = 0; wi < num_workers; ++wi) {
167-
string task_name = strings::StrCat("/job:worker/replica:0/task:", wi);
167+
string task_name = absl::StrCat("/job:worker/replica:0/task:", wi);
168168
for (int di = 0; di < num_devices; ++di) {
169169
int idx = wi * num_devices + di;
170170
string device_name =
@@ -203,9 +203,9 @@ class DeviceResDistTest : public ::testing::Test {
203203
}
204204
int group_size = num_workers * num_devices;
205205
for (int wi = 0; wi < num_workers; ++wi) {
206-
string task_name = strings::StrCat("/job:worker/replica:0/task:", wi);
206+
string task_name = absl::StrCat("/job:worker/replica:0/task:", wi);
207207
for (int di = 0; di < num_devices; ++di) {
208-
string device_name = strings::StrCat(task_name, "/device:CPU:", di);
208+
string device_name = absl::StrCat(task_name, "/device:CPU:", di);
209209
IssueRequest(task_name, device_name, group_size);
210210
}
211211
}
@@ -245,9 +245,9 @@ class DeviceResDistTest : public ::testing::Test {
245245
const int dev_count = num_workers * num_devices;
246246
string dev0 = "/job:worker/replica:0/task:0/device:CPU:0";
247247
for (int wi = 0; wi < num_workers; ++wi) {
248-
string task_name = strings::StrCat("/job:worker/replica:0/task:", wi);
248+
string task_name = absl::StrCat("/job:worker/replica:0/task:", wi);
249249
for (int di = 0; di < num_devices; ++di) {
250-
string device_name = strings::StrCat(task_name, "/device:CPU:", di);
250+
string device_name = absl::StrCat(task_name, "/device:CPU:", di);
251251
int idx = wi * num_devices + di;
252252
TF_ASSERT_OK(status_[device_name]);
253253
EXPECT_EQ(cp_[device_name]->default_rank, idx);
@@ -283,7 +283,7 @@ class DeviceResDistTest : public ::testing::Test {
283283
CollectiveType coll_type = REDUCTION_COLLECTIVE,
284284
bool is_source = false) {
285285
string worker_name =
286-
strings::StrCat("/job:worker/replica:0/task:", worker_idx);
286+
absl::StrCat("/job:worker/replica:0/task:", worker_idx);
287287
DefineWorker(worker_name, device_type, num_devices, nccl);
288288
for (int i = 0; i < num_devices; ++i) {
289289
string device_name =

tensorflow/core/distributed_runtime/collective_rma_distributed_test.cc

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -249,7 +249,7 @@ class CollRMADistTest
249249
string device_type = "CPU";
250250
string dev0_worker_name;
251251
for (int w = 0; w < num_workers; ++w) {
252-
string name = strings::StrCat("/job:worker/replica:0/task:", w);
252+
string name = absl::StrCat("/job:worker/replica:0/task:", w);
253253
if (w == 0) {
254254
dev0_worker_name = name;
255255
}

0 commit comments

Comments
 (0)