提交 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):
if subdirs_elem == 'lock_dir':
continue
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')
if key_pkl in self.loaded_key_pkl:
continue
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论