提交 3f5e2f7f authored 作者: Frederic Bastien's avatar Frederic Bastien

[OPT FAILURE] for ifelse to GPU when some variable don't have the dtype attribute

上级 b0ce28cd
......@@ -548,7 +548,7 @@ def local_gpu_lazy_ifelse(node):
for i in range(len(outs)):
if (not isinstance(outs[i].type, CudaNdarrayType) and
outs[i].dtype == 'float32'):
getattr(outs[i], 'dtype', None) == 'float32'):
outs[i] = as_cuda_ndarray_variable(outs[i])
outs = gpu_ifelse(c, *outs, return_list=True)
for i in range(len(outs)):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论