提交 8db03cf1 authored 作者: Frederic Bastien's avatar Frederic Bastien

Port the fix to the new back-end.

上级 3f5e2f7f
......@@ -546,7 +546,12 @@ def local_gpua_lazy_ifelse(node, context_name):
if node.op.gpu:
return
c = node.inputs[0]
inps = [as_gpuarray_variable(v, context_name) for v in node.inputs[1:]]
inps = []
for v in node.inputs[1:]:
if isinstance(v.type, (tensor.TensorType, GpuArrayType)):
inps.append(as_gpuarray_variable(v, context_name))
else:
inps.append(v)
return IfElse(node.op.n_outs, gpu=True)(c, *inps, return_list=True)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论