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

Use the versioning info from gpuarray to avoid unloadable cached modules.

上级 9f6f3579
...@@ -198,7 +198,10 @@ class GpuArrayType(Type): ...@@ -198,7 +198,10 @@ class GpuArrayType(Type):
return ['compyte'] return ['compyte']
def c_code_cache_version(self): def c_code_cache_version(self):
return (1,) ver = pygpu.gpuarray.api_version()
# we only use the major version since the minor revision are
# API-compatible.
return (1, ver[0])
class _operators(_tensor_py_operators): class _operators(_tensor_py_operators):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论