提交 41c26968 authored 作者: James Bergstra's avatar James Bergstra

gemm - adding some commented printfs that say when gemm allocates output

上级 0f5b3bf2
...@@ -452,6 +452,7 @@ class Gemm(GemmRelated): ...@@ -452,6 +452,7 @@ class Gemm(GemmRelated):
dims[0] = %(_z)s->dimensions[0]; dims[0] = %(_z)s->dimensions[0];
dims[1] = %(_z)s->dimensions[1]; dims[1] = %(_z)s->dimensions[1];
%(_zout)s = (PyArrayObject*)PyArray_SimpleNew(2, dims, type_num_%(_z)s); %(_zout)s = (PyArrayObject*)PyArray_SimpleNew(2, dims, type_num_%(_z)s);
//fprintf(stderr, "Gemm Allocating %%i %%i\\n", dims[0], dims[1]);
if(!%(_zout)s) { if(!%(_zout)s) {
PyErr_SetString(PyExc_MemoryError, "failed to alloc gemm_no_inplace output"); PyErr_SetString(PyExc_MemoryError, "failed to alloc gemm_no_inplace output");
%(fail)s %(fail)s
...@@ -827,6 +828,7 @@ class Dot22(GemmRelated): ...@@ -827,6 +828,7 @@ class Dot22(GemmRelated):
dims[0] = %(_x)s->dimensions[0]; dims[0] = %(_x)s->dimensions[0];
dims[1] = %(_y)s->dimensions[1]; dims[1] = %(_y)s->dimensions[1];
%(_zout)s = (PyArrayObject*)PyArray_SimpleNew(2, dims, type_num_%(_x)s); %(_zout)s = (PyArrayObject*)PyArray_SimpleNew(2, dims, type_num_%(_x)s);
//fprintf(stderr, "Dot Allocating %%i %%i\\n", dims[0], dims[1]);
if(!%(_zout)s) { if(!%(_zout)s) {
PyErr_SetString(PyExc_MemoryError, "failed to alloc dot22 output"); PyErr_SetString(PyExc_MemoryError, "failed to alloc dot22 output");
%(fail)s %(fail)s
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论