提交 e7582189 authored 作者: Arnaud Bergeron's avatar Arnaud Bergeron

Make sure we don't remove the gpuarray kernel cache when loading the cache.

上级 1237167b
...@@ -750,6 +750,9 @@ class ModuleCache(object): ...@@ -750,6 +750,9 @@ class ModuleCache(object):
if subdirs_elem == 'lock_dir': if subdirs_elem == 'lock_dir':
continue continue
root = os.path.join(self.dirname, subdirs_elem) root = os.path.join(self.dirname, subdirs_elem)
# Don't delete the gpuarray kernel cache
if root == config.gpuarray.cache_path:
continue
key_pkl = os.path.join(root, 'key.pkl') key_pkl = os.path.join(root, 'key.pkl')
if key_pkl in self.loaded_key_pkl: if key_pkl in self.loaded_key_pkl:
continue continue
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论