提交 651a9abf authored 作者: Razvan Pascanu's avatar Razvan Pascanu

fixed viewop to work on gpu

上级 f4c827ab
...@@ -257,16 +257,17 @@ class ViewOp(theano.gof.Op): ...@@ -257,16 +257,17 @@ class ViewOp(theano.gof.Op):
iname = inames[0] iname = inames[0]
oname = onames[0] oname = onames[0]
fail = sub['fail'] fail = sub['fail']
if isinstance(node.inputs[0].type, theano.tensor.TensorType): if isinstance(node.inputs[0].type, theano.scalar.Scalar):
return """ return """
Py_XDECREF(%(oname)s);
Py_XINCREF(%(iname)s);
%(oname)s = %(iname)s; %(oname)s = %(iname)s;
"""%locals() """ % locals()
else: else:
return """ return """
Py_XDECREF(%(oname)s);
%(oname)s = %(iname)s; %(oname)s = %(iname)s;
""" % locals() Py_XINCREF(%(iname)s);
"""%locals()
deep_copy_op = DeepCopyOp() deep_copy_op = DeepCopyOp()
view_op = ViewOp() view_op = ViewOp()
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论