提交 b5737084 authored 作者: Frederic's avatar Frederic

Make c code depend of the execution dtype and not compile time dtype.

上级 dbffc96d
...@@ -71,7 +71,7 @@ def ger_c_code(A, a, x, y, Z, destructive, fail): ...@@ -71,7 +71,7 @@ def ger_c_code(A, a, x, y, Z, destructive, fail):
|| (%(Z)s->dimensions[1] != %(A)s->dimensions[1])) || (%(Z)s->dimensions[1] != %(A)s->dimensions[1]))
{ {
if (%(Z)s) Py_XDECREF(%(Z)s); if (%(Z)s) Py_XDECREF(%(Z)s);
%(Z)s = (PyArrayObject*)PyArray_SimpleNew(2, dims, type_num_%(A)s); %(Z)s = (PyArrayObject*)PyArray_SimpleNew(2, dims, PyArray_TYPE(%(A)s));
if(!%(Z)s) { if(!%(Z)s) {
PyErr_SetString(PyExc_MemoryError, "failed to alloc ger output"); PyErr_SetString(PyExc_MemoryError, "failed to alloc ger output");
%(fail)s %(fail)s
...@@ -198,7 +198,7 @@ class CGer(BaseBLAS, Ger): ...@@ -198,7 +198,7 @@ class CGer(BaseBLAS, Ger):
return code return code
def c_code_cache_version(self): def c_code_cache_version(self):
return (1,) return (2,)
def make_thunk(*args, **kwargs): def make_thunk(*args, **kwargs):
# skip over Ger.make_thunk # skip over Ger.make_thunk
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论