fix(mge/distributed): fix single_machine_cache error

GitOrigin-RevId: be57111408a7cb4861c5b8534e4305ab366ce5ba

fix(mge/distributed): fix single_machine_cache error
GitOrigin-RevId: be57111408a7cb4861c5b8534e4305ab366ce5ba
05ee6038 · Megvii Engine Team · 0a56a5b1 · 05ee6038
隐藏空白更改
内联并排

Showing with 1 addition and 0 deletion

imperative/python/megengine/distributed/group.py imperative/python/megengine/distributed/group.py +1 -0

未找到文件。
--- a/imperative/python/megengine/distributed/group.py
+++ b/imperative/python/megengine/distributed/group.py
@@ -56,6 +56,7 @@ class Group:
    def reset(self, proc_ranks):
        self.check(proc_ranks)
        self.proc_ranks = proc_ranks
+        self.is_single_machine_cache = None
        self.stream = _sh.get_next()

    def check(self, proc_ranks):