提交 2eb92518 authored 作者: Arnaud Bergeron's avatar Arnaud Bergeron

Fixed the python version warning to show the expected version.

上级 9c409e1b
......@@ -50,8 +50,13 @@ def init_dev(dev, name=None):
if v[1] < 0:
raise RuntimeError("Wrong minor API version for gpuarray:", v[1],
"Please update libgpuarray/pygpu.")
if len(v) < 3 or v[2] < 0:
print("Wrong python API version for gpuarray:", v[2],
if len(v) < 3:
vpy = -1
else:
vpy = v[2]
vpye = 0
if vpy < vpye:
print("Wrong python API version for gpuarray:", vpy, "expected:", vpye,
"Some python ops may not work correctly and/or crash. "
"Consider updating pygpu.", file=sys.stderr)
global pygpu_activated
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论