提交 b0e55935 authored 作者: Frederic's avatar Frederic

Fix opt crash newlly introduced.

上级 e254a7f8
...@@ -87,7 +87,8 @@ class InputToGpuOptimizer(Optimizer): ...@@ -87,7 +87,8 @@ class InputToGpuOptimizer(Optimizer):
# This happen frequently as we do 2 pass of the gpu optimizations # This happen frequently as we do 2 pass of the gpu optimizations
if (len(input.clients) == 1 and if (len(input.clients) == 1 and
input.clients[0][0].op == gpu_from_host): (input.clients[0][0] == 'output' or
input.clients[0][0].op == gpu_from_host)):
return return
try: try:
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论