提交 98ec1243 authored 作者: Olivier Delalleau's avatar Olivier Delalleau

PEP8

上级 7be047f7
...@@ -259,6 +259,7 @@ gemv_inplace = Gemv(inplace=True) ...@@ -259,6 +259,7 @@ gemv_inplace = Gemv(inplace=True)
# For the user interface. Opt will make them inplace later # For the user interface. Opt will make them inplace later
gemv = gemv_no_inplace gemv = gemv_no_inplace
class Ger(Op): class Ger(Op):
""" """
BLAS defines general rank-1 update GER as A <- A + alpha x y' BLAS defines general rank-1 update GER as A <- A + alpha x y'
...@@ -367,7 +368,6 @@ def default_blas_ldflags(): ...@@ -367,7 +368,6 @@ def default_blas_ldflags():
"said value, this will disable this warning." "said value, this will disable this warning."
% new_path) % new_path)
use_unix_epd = False use_unix_epd = False
if use_unix_epd: if use_unix_epd:
return ' '.join( return ' '.join(
...@@ -729,17 +729,28 @@ class GemmRelated(Op): ...@@ -729,17 +729,28 @@ class GemmRelated(Op):
//); //);
switch(unit) switch(unit)
{ {
case 0x000: dgemm_(&N, &N, &Nz1, &Nz0, &Nx1, &a, y, &sy_0, x, &sx_0, &b, z, &sz_0); break; case 0x000: dgemm_(&N, &N, &Nz1, &Nz0, &Nx1, &a, y,
case 0x100: dgemm_(&N, &T, &Nz1, &Nz0, &Nx1, &a, y, &sy_0, x, &sx_1, &b, z, &sz_0); break; &sy_0, x, &sx_0, &b, z, &sz_0); break;
case 0x010: dgemm_(&T, &N, &Nz1, &Nz0, &Nx1, &a, y, &sy_1, x, &sx_0, &b, z, &sz_0); break; case 0x100: dgemm_(&N, &T, &Nz1, &Nz0, &Nx1, &a, y,
case 0x110: dgemm_(&T, &T, &Nz1, &Nz0, &Nx1, &a, y, &sy_1, x, &sx_1, &b, z, &sz_0); break; &sy_0, x, &sx_1, &b, z, &sz_0); break;
case 0x001: dgemm_(&T, &T, &Nz0, &Nz1, &Nx1, &a, x, &sx_0, y, &sy_0, &b, z, &sz_1); break; case 0x010: dgemm_(&T, &N, &Nz1, &Nz0, &Nx1, &a, y,
case 0x101: dgemm_(&N, &T, &Nz0, &Nz1, &Nx1, &a, x, &sx_1, y, &sy_0, &b, z, &sz_1); break; &sy_1, x, &sx_0, &b, z, &sz_0); break;
case 0x011: dgemm_(&T, &N, &Nz0, &Nz1, &Nx1, &a, x, &sx_0, y, &sy_1, &b, z, &sz_1); break; case 0x110: dgemm_(&T, &T, &Nz1, &Nz0, &Nx1, &a, y,
case 0x111: dgemm_(&N, &N, &Nz0, &Nz1, &Nx1, &a, x, &sx_1, y, &sy_1, &b, z, &sz_1); break; &sy_1, x, &sx_1, &b, z, &sz_0); break;
default: PyErr_SetString(PyExc_ValueError, "some matrix has no unit stride"); %(fail)s; case 0x001: dgemm_(&T, &T, &Nz0, &Nz1, &Nx1, &a, x,
&sx_0, y, &sy_0, &b, z, &sz_1); break;
case 0x101: dgemm_(&N, &T, &Nz0, &Nz1, &Nx1, &a, x,
&sx_1, y, &sy_0, &b, z, &sz_1); break;
case 0x011: dgemm_(&T, &N, &Nz0, &Nz1, &Nx1, &a, x,
&sx_0, y, &sy_1, &b, z, &sz_1); break;
case 0x111: dgemm_(&N, &N, &Nz0, &Nz1, &Nx1, &a, x,
&sx_1, y, &sy_1, &b, z, &sz_1); break;
default: PyErr_SetString(PyExc_ValueError,
"some matrix has no unit stride");
%(fail)s;
}; };
//fprintf(stderr, "Calling dgemm %%i %%i %%i %%i took %%f\\n", unit, Nz1, Nz0, Nx1, time_time()- t0); //fprintf(stderr, "Calling dgemm %%i %%i %%i %%i took %%f\\n",
// unit, Nz1, Nz0, Nx1, time_time()- t0);
""" """
end_switch_typenum = """ end_switch_typenum = """
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论