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

Fix Python 3 error.

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