提交 323230b1 authored 作者: Frederic Bastien's avatar Frederic Bastien

Fix crash introduced in this PR.

上级 c4a65fcc
......@@ -346,7 +346,8 @@ class GraphToGPU(Optimizer):
self.local_optimizers_map.get(type(c.op), []))):
move_to_GPU = True
new_ops = None
if move_to_GPU and any(["complex" in i.dtype for i in node.inputs]):
if move_to_GPU and any(["complex" in getattr(i, 'dtype', "")
for i in node.inputs]):
move_to_GPU = False
# Apply the lifter
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论