MCPcopy Create free account
hub / github.com/Oneflow-Inc/oneflow / TEST

Function TEST

oneflow/core/control/ctrl_test.cpp:58–82  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

56
57#ifdef RPC_BACKEND_GRPC
58TEST(CtrlServer, new_delete) {
59 int port = CtrlUtil().FindAvailablePort();
60 if (port == -1) { return; }
61 EnvProto env_proto = GetEnvProto(port);
62 Singleton<EnvDesc>::New(env_proto);
63 Singleton<CtrlServer>::New();
64 Singleton<ProcessCtx>::New();
65 CHECK_JUST(
66 HostListCtrlBootstrap(*Singleton<EnvDesc>::Get())
67 .InitProcessCtx(Singleton<CtrlServer>::Get()->port(), Singleton<ProcessCtx>::Get()));
68 auto* client = new GrpcCtrlClient(*Singleton<ProcessCtx>::Get());
69 Singleton<CtrlClient>::SetAllocated(client);
70 Singleton<ResourceDesc, ForEnv>::New(GetResource(), GlobalProcessCtx::NumOfProcessPerNode());
71 Singleton<ResourceDesc, ForSession>::New(GetResource(), GlobalProcessCtx::NumOfProcessPerNode());
72
73 // do test
74 // OF_ENV_BARRIER();
75
76 Singleton<ResourceDesc, ForSession>::Delete();
77 Singleton<ResourceDesc, ForEnv>::Delete();
78 Singleton<CtrlClient>::Delete();
79 Singleton<ProcessCtx>::Delete();
80 Singleton<CtrlServer>::Delete();
81 Singleton<EnvDesc>::Delete();
82}
83#endif // RPC_BACKEND_GRPC
84
85} // namespace oneflow

Callers

nothing calls this directly

Calls 10

CtrlUtilClass · 0.85
GetFunction · 0.85
DeleteFunction · 0.85
FindAvailablePortMethod · 0.80
InitProcessCtxMethod · 0.80
GetEnvProtoFunction · 0.70
GetResourceFunction · 0.70
NewFunction · 0.50
portMethod · 0.45

Tested by

no test coverage detected