提交 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):
|| (%(Z)s->dimensions[1] != %(A)s->dimensions[1]))
{
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) {
PyErr_SetString(PyExc_MemoryError, "failed to alloc ger output");
%(fail)s
......@@ -198,7 +198,7 @@ class CGer(BaseBLAS, Ger):
return code
def c_code_cache_version(self):
return (1,)
return (2,)
def make_thunk(*args, **kwargs):
# skip over Ger.make_thunk
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论