提交 caf6aff1 authored 作者: Pascal Lamblin's avatar Pascal Lamblin

Use _props_dict() from the right Op.

It was using the one from GpuFromHost instead of AdvancedIncSubtensor1.
上级 edd1c456
...@@ -1138,11 +1138,12 @@ def local_gpu_advanced_incsubtensor1(node): ...@@ -1138,11 +1138,12 @@ def local_gpu_advanced_incsubtensor1(node):
'least \'0.6\'.', stacklevel=1) 'least \'0.6\'.', stacklevel=1)
active_device_no = theano.sandbox.cuda.active_device_number() active_device_no = theano.sandbox.cuda.active_device_number()
compute_capability = device_properties(active_device_no)['major'] compute_capability = device_properties(active_device_no)['major']
props_dict = host_input.owner.op._props_dict()
if (compute_capability < 2 or y.ndim != 2 or x.ndim != 2): if (compute_capability < 2 or y.ndim != 2 or x.ndim != 2):
gpu_op = GpuAdvancedIncSubtensor1(**node.op._props_dict()) gpu_op = GpuAdvancedIncSubtensor1(**props_dict)
else: else:
gpu_op = GpuAdvancedIncSubtensor1_dev20(**node.op._props_dict()) gpu_op = GpuAdvancedIncSubtensor1_dev20(**props_dict)
return [gpu_op(as_cuda_ndarray_variable(x), return [gpu_op(as_cuda_ndarray_variable(x),
as_cuda_ndarray_variable(y), *coords)] as_cuda_ndarray_variable(y), *coords)]
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论