提交 25ca446b authored 作者: Arnaud Bergeron's avatar Arnaud Bergeron

Fix syntax and flake8 errors in opt_util.py

上级 17f74589
...@@ -31,8 +31,8 @@ def grab_cpu_scalar(v, nd): ...@@ -31,8 +31,8 @@ def grab_cpu_scalar(v, nd):
""" """
if v.owner is not None: if v.owner is not None:
n = v.owner n = v.owner
elif (isinstance(n.op, (GpuDimShuffle, DimShuffle)) and if (isinstance(n.op, (GpuDimShuffle, DimShuffle)) and
n.op.new_order == ('x',) * nd): n.op.new_order == ('x',) * nd):
return grab_cpu_scalar(n.inputs[0], n.inputs[0].ndim) return grab_cpu_scalar(n.inputs[0], n.inputs[0].ndim)
elif isinstance(n.op, (GpuFromHost, HostFromGpu)): elif isinstance(n.op, (GpuFromHost, HostFromGpu)):
return grab_cpu_scalar(n.inputs[0], nd) return grab_cpu_scalar(n.inputs[0], nd)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论