提交 4b7ca419 authored 作者: abergeron's avatar abergeron

Merge pull request #1773 from nouiz/make_vector_bug

bugfix, for a case that don't happen right now.
...@@ -587,7 +587,7 @@ class MakeVector(T.Op): ...@@ -587,7 +587,7 @@ class MakeVector(T.Op):
out[0][...] = inputs out[0][...] = inputs
def c_code_cache_version(self): def c_code_cache_version(self):
return (1,) return (2,)
def c_code(self, node, name, inp, out_, sub): def c_code(self, node, name, inp, out_, sub):
out, = out_ out, = out_
...@@ -604,7 +604,7 @@ class MakeVector(T.Op): ...@@ -604,7 +604,7 @@ class MakeVector(T.Op):
ret = """ ret = """
npy_intp dims[1]; npy_intp dims[1];
dims[0] = %(out_shape)s; dims[0] = %(out_shape)s;
if(!%(out)s || PyArray_DIMS(%(out)s)[0] == %(out_shape)s){ if(!%(out)s || PyArray_DIMS(%(out)s)[0] != %(out_shape)s){
Py_XDECREF(%(out)s); Py_XDECREF(%(out)s);
%(out)s = (PyArrayObject*)PyArray_EMPTY(1, dims, %(out_dtype)s, 0); %(out)s = (PyArrayObject*)PyArray_EMPTY(1, dims, %(out_dtype)s, 0);
} }
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论