MCPcopy Create free account
hub / github.com/DeepRec-AI/DeepRec / TEST

Function TEST

tensorflow/core/distributed_runtime/rpc/grpc_channel_test.cc:29–56  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

27#define IsSameAddrSp DeviceNameUtils::IsSameAddressSpace
28
29TEST(GrpcChannelTest, IsSameAddressSpace) {
30 // Same.
31 EXPECT_TRUE(IsSameAddrSp("/job:mnist/replica:10/task:10/cpu:0",
32 "/job:mnist/replica:10/task:10/cpu:1"));
33 EXPECT_TRUE(IsSameAddrSp("/job:mnist/replica:10/task:10/cpu:0",
34 "/job:mnist/replica:10/task:10/device:GPU:2"));
35 EXPECT_TRUE(IsSameAddrSp("/job:mnist/replica:10/task:10",
36 "/job:mnist/replica:10/task:10/device:GPU:2"));
37 EXPECT_TRUE(IsSameAddrSp("/job:mnist/replica:10/task:10/cpu:1",
38 "/job:mnist/replica:10/task:10"));
39
40 // Different.
41 EXPECT_FALSE(IsSameAddrSp("/job:mnist/replica:10/task:9/cpu:0",
42 "/job:mnist/replica:10/task:10/cpu:0"));
43 EXPECT_FALSE(IsSameAddrSp("/job:mnist/replica:9/task:10/cpu:0",
44 "/job:mnist/replica:10/task:10/cpu:0"));
45 EXPECT_FALSE(IsSameAddrSp("/job:MNIST/replica:10/task:10/cpu:0",
46 "/job:mnist/replica:10/task:10/cpu:0"));
47
48 // Invalid names.
49 EXPECT_FALSE(IsSameAddrSp("random_invalid_target", "random_invalid_target"));
50 EXPECT_FALSE(IsSameAddrSp("/job:/replica:10/task:10/cpu:0",
51 "/job:/replica:10/task:10/cpu:1"));
52 EXPECT_FALSE(IsSameAddrSp("/job:mnist/replica:xx/task:10/cpu:0",
53 "/job:mnist/replica:xx/task:10/cpu:1"));
54 EXPECT_FALSE(IsSameAddrSp("/job:mnist/replica:10/task:yy/cpu:0",
55 "/job:mnist/replica:10/task:yy/cpu:1"));
56}
57
58TEST(GrpcChannelTest, HostPorts) {
59 GrpcChannelSpec spec;

Callers

nothing calls this directly

Calls 13

NewGrpcChannelCacheFunction · 0.85
sortFunction · 0.85
NewHostPortGrpcChannelFunction · 0.85
AddHostPortsJobMethod · 0.45
FindWorkerChannelMethod · 0.45
getMethod · 0.45
ListWorkersMethod · 0.45
ListWorkersInJobMethod · 0.45
emptyMethod · 0.45
beginMethod · 0.45
endMethod · 0.45

Tested by

no test coverage detected