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

Rework the gpuarray api version check.

上级 8fb5d66c
...@@ -39,11 +39,14 @@ register_transfer(transfer) ...@@ -39,11 +39,14 @@ register_transfer(transfer)
def init_dev(dev, name=None): def init_dev(dev, name=None):
if pygpu.gpuarray.api_version() != (-10000, 0): v = pygpu.gpuarray.api_version()
raise RuntimeError("Wrong API version for gpuarray:", if v[0] != -10000:
pygpu.gpuarray.api_version(), raise RuntimeError("Wrong major API version for gpuarray:", v[0],
"Make sure Theano and libgpuarray/pygpu " "Make sure Theano and libgpuarray/pygpu "
"are in sync.") "are in sync.")
if v[1] < 0:
raise RuntimeError("Wrong minor API version for gpuarray:", v[1],
"Please update libgpuarray/pygpu.")
global pygpu_activated global pygpu_activated
if dev not in init_dev.devmap: if dev not in init_dev.devmap:
ctx = pygpu.init(dev) ctx = pygpu.init(dev)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论