Fix issues in point of MultiClientSession #5469
Conversation
| if oneflow._oneflow_internal.CurrentMachineId() == 0: | ||
| scope_util.InitScopeStack() | ||
| else: | ||
| exit(0) |
There was a problem hiding this comment.
如果在 Single-Client 下，worker 的 python 进程没有在 env.init 里 exit(0) 的话，会在 cluster 训练完退出以后，尝试执行 env.init 后面的 python 代码然后挂掉吧。这里肯定不能这样删除。
我觉得保险起见，要不提供两个接口？ flow.env.init 就强制是 Single-Client 的启动方式。如果在 is_multi_client 下调用 flow.env.init ，就直接弹出提示用户这个接口在 Multi-Client 下没有效果。 然后 Multi-Client 下的 env 初始化（用户不可控）就另外启用一个？
这里的逻辑需要跟 @clackhan @daquexian 讨论一下再确定。或者这个 PR 先不动这里的逻辑。
There was a problem hiding this comment.
我觉得可以按你提的建议先修改了，flow.env.init 里先保持在 single-client 下初始化 scope stack，然后 multi-client 下 scope stack 的初始化直接放在 MultiClientSession 被创建后。
…ter MultiClientSession created in multi-client
| @@ -40,6 +41,19 @@ def test(self): | |||
| sess.TryInit() | |||
| self.assertEqual(sess.status, sess.Status.INITED) | |||
|
|
|||
| # sess.TryClose() | |||
| # self.assertEqual(sess.status, sess.Status.CLOSED) | |||
There was a problem hiding this comment.
如果支持MultiClientSession的clear，这里是不是就可以写上了
There was a problem hiding this comment.
光 MultiClientSession 支持 clear 不行，还需要在 test_case2 里面重新创建一个 MultiClientSession，而不能用 GetDefaultSession。
不过之前讨论的结果不是 MultiClientSession 永不关闭吗？所有的测试都共用一个 session。
oneflow-ci-bot
requested review from
oneflow-ci-bot
and removed request for
oneflow-ci-bot
No description provided.