提交 75657df1 authored 作者: lamblin's avatar lamblin

Merge pull request #587 from nouiz/test_fix

Test fix
......@@ -87,7 +87,8 @@ class InputToGpuOptimizer(Optimizer):
# This happen frequently as we do 2 pass of the gpu optimizations
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
try:
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论