From 05ee6038435fd65bc957b9b6c4e4b893e32ff3a2 Mon Sep 17 00:00:00 2001
From: Megvii Engine Team <megengine@megvii.com>
Date: Thu, 18 Nov 2021 16:50:04 +0800
Subject: [PATCH] fix(mge/distributed): fix single_machine_cache error

GitOrigin-RevId: be57111408a7cb4861c5b8534e4305ab366ce5ba
---
 imperative/python/megengine/distributed/group.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/imperative/python/megengine/distributed/group.py b/imperative/python/megengine/distributed/group.py
index 94d725e0..3f4e300f 100644
--- 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):