提交 58bcdcf3 authored 作者: Frederic Bastien's avatar Frederic Bastien

Fix a test crash when device=cuda in theano/gof/tests/test_lazy.py

上级 ae14cf0e
......@@ -217,7 +217,8 @@ def op_lifter(OP, cuda_only=False):
if (not replace or
(cuda_only and
get_context(context_name).kind != b'cuda') or
any(["complex" in i.dtype for i in node.inputs])):
any(["complex" in getattr(i, 'dtype', "")
for i in node.inputs])):
return False
# tag the inputs with the context in case
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论