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

Enable the condition to move

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