提交 94d9e1ca authored 作者: Thomas George's avatar Thomas George

fixed incorrect if statement

上级 87e06fdb
...@@ -296,7 +296,7 @@ class GpuGer(BlasOp): ...@@ -296,7 +296,7 @@ class GpuGer(BlasOp):
vars = dict(out=out[0], A=inp[0], alpha=inp[1], x=inp[2], y=inp[3], vars = dict(out=out[0], A=inp[0], alpha=inp[1], x=inp[2], y=inp[3],
fail=sub['fail'], name=name, params=sub['params']) fail=sub['fail'], name=name, params=sub['params'])
code = """ code = """
if (%(params)s->inplace || !GpuArray_ISONESEGMENT(&%(A)s->ga)) { if (!%(params)s->inplace || !GpuArray_ISONESEGMENT(&%(A)s->ga)) {
%(out)s = theano_try_copy(%(out)s, %(A)s); %(out)s = theano_try_copy(%(out)s, %(A)s);
if (%(out)s == NULL) { if (%(out)s == NULL) {
%(fail)s %(fail)s
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论