提交 9683d861 authored 作者: Frederic's avatar Frederic

Fix comment

上级 4fbb0ef6
...@@ -2146,7 +2146,7 @@ def local_gpualloc(node): ...@@ -2146,7 +2146,7 @@ def local_gpualloc(node):
i.owner.op in [host_from_gpu, tensor.alloc] i.owner.op in [host_from_gpu, tensor.alloc]
for i in c.inputs[1:]]) for i in c.inputs[1:]])
for c, idx in node.outputs[0].clients]): for c, idx in node.outputs[0].clients]):
# if the client is a subtensor with input on gpu or alloc # if the client is on gpu or alloc
replace = True replace = True
if replace and node.inputs[0].dtype != 'float32': if replace and node.inputs[0].dtype != 'float32':
replace = False replace = False
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论