提交 c9752903 authored 作者: Frederic Bastien's avatar Frederic Bastien 提交者: sentient07

Enable the condition to move

上级 1bc89154
......@@ -308,17 +308,14 @@ class GraphToGPU(NavigatorOptimizer):
continue
# Move only if any of the inputs are on the GPU.
move_to_GPU = True
'''
import GpuArrayVariable and GpuArraySharedVariable when you uncomment
move_to_GPU = False
from .type import GpuArrayVariable, GpuArraySharedVariable #when you uncomment
if any([isinstance(i, GpuArrayVariable) or
isinstance(i, GpuArraySharedVariable)
for i in [mapping[v] for v in node.inputs] +
node.outputs]):
move_to_GPU = True
'''
context_name = None
for i in [mapping[i] for i in node.inputs]:
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论