From b588d93eb05f75e4243a1a10ce10cf03304c91ed Mon Sep 17 00:00:00 2001
From: Megvii Engine Team <megengine@megvii.com>
Date: Wed, 8 Apr 2020 18:24:41 +0800
Subject: [PATCH] fix(mge): fix fast-run cache mismatch

GitOrigin-RevId: a41150f71ab46ec343209c4ac93bc8165aefa26c
---
 python_module/megengine/_internal/persistent_cache.py | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/python_module/megengine/_internal/persistent_cache.py b/python_module/megengine/_internal/persistent_cache.py
index a6fd45839..cbd0027f6 100644
--- a/python_module/megengine/_internal/persistent_cache.py
+++ b/python_module/megengine/_internal/persistent_cache.py
@@ -15,6 +15,7 @@ import shelve
 
 from .logconf import get_logger
 from .mgb import _PersistentCache
+from .version import __version__
 
 
 class _FakeRedisConn:
@@ -85,7 +86,10 @@ class PersistentCacheOnServer(_PersistentCache):
         return conn
 
     def _make_key(self, category, key):
-        return b"@".join((self._prefix.encode("ascii"), category.encode("ascii"), key))
+        prefix_with_version = "{}:MGB{}".format(self._prefix, __version__)
+        return b"@".join(
+            (prefix_with_version.encode("ascii"), category.encode("ascii"), key)
+        )
 
     def put(self, category, key, value):
         conn = self._conn
-- 
GitLab