提交 d68e0bfc authored 作者: Arnaud Bergeron's avatar Arnaud Bergeron

Fix crash with GpuGemm.

上级 3bfef993
...@@ -146,7 +146,6 @@ class GpuGemm(BlasOp, Gemm): ...@@ -146,7 +146,6 @@ class GpuGemm(BlasOp, Gemm):
""" % vars """ % vars
else: else:
code = """ code = """
Py_XDECREF(%(out)s);
%(out)s = gpublas_try_copy(%(out)s, %(C)s); %(out)s = gpublas_try_copy(%(out)s, %(C)s);
if (%(out)s == NULL) { if (%(out)s == NULL) {
%(fail)s %(fail)s
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论