提交 7a1c7fcb authored 作者: Ziye Fan's avatar Ziye Fan

Remove useless gpu_from_host(host_from_gpu(x)) op.

上级 4cf7afb4
...@@ -34,6 +34,12 @@ _logger = logging.getLogger(_logger_name) ...@@ -34,6 +34,12 @@ _logger = logging.getLogger(_logger_name)
def as_cuda_ndarray_variable(x): def as_cuda_ndarray_variable(x):
if \
x.owner is not None and \
isinstance(x.owner.op, GpuFromHost) and \
hasattr(x.owner.inputs[0].owner, 'op') and \
isinstance(x.owner.inputs[0].owner.op, HostFromGpu):
return x.owner.inputs[0].owner.inputs[0]
if hasattr(x, '_as_CudaNdarrayVariable'): if hasattr(x, '_as_CudaNdarrayVariable'):
return x._as_CudaNdarrayVariable() return x._as_CudaNdarrayVariable()
tensor_x = tensor.as_tensor_variable(x) tensor_x = tensor.as_tensor_variable(x)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论