提交 6178fc1a authored 作者: sentient07's avatar sentient07

Revert "Handling tag 'cpu' in the middle of graph"

This reverts commit 1736a6ce4ae50c6b3a6b4c286e1f23b8a43a0f33.
上级 dd544cba
...@@ -326,8 +326,7 @@ class GraphToGPU(NavigatorOptimizer): ...@@ -326,8 +326,7 @@ class GraphToGPU(NavigatorOptimizer):
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]:
target = getattr(i.tag, 'target', None) if isinstance(i.type, GpuArrayType):
if (isinstance(i.type, GpuArrayType) and target != 'cpu'):
context_name = i.type.context_name context_name = i.type.context_name
move_to_GPU = True move_to_GPU = True
break break
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论