提交 274955e4 authored 作者: notoraptor's avatar notoraptor

Fix Python 3 error.

上级 c89e4cca
...@@ -702,9 +702,9 @@ def local_gpua_elemwise(op, context_name, inputs, outputs): ...@@ -702,9 +702,9 @@ def local_gpua_elemwise(op, context_name, inputs, outputs):
have_opencl = False have_opencl = False
if inputs and isinstance(inputs[0].type, GpuArrayType): if inputs and isinstance(inputs[0].type, GpuArrayType):
kind = inputs[0].type.context.kind kind = inputs[0].type.context.kind
if kind.startswith('opencl'): if kind.startswith(b'opencl'):
have_opencl = True have_opencl = True
elif kind.startswith('cuda'): elif kind.startswith(b'cuda'):
have_cuda = True have_cuda = True
opname = False opname = False
if isinstance(scal_op, Erfinv): if isinstance(scal_op, Erfinv):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论