提交 2c258623 authored 作者: Frederic's avatar Frederic

Don't try to move to the GPU alloc that aren't float32.

This is needed to make the new GpuImages2Neibs ignore_border mode pass.
上级 b21945df
...@@ -1337,6 +1337,8 @@ def local_gpualloc(node): ...@@ -1337,6 +1337,8 @@ def local_gpualloc(node):
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 a subtensor with input on gpu or alloc
replace = True replace = True
if replace and node.inputs[0].dtype != 'float32':
replace = False
if replace: if replace:
val = node.inputs[0] val = node.inputs[0]
shp = node.inputs[1:] shp = node.inputs[1:]
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论