提交 1ac274a9 authored 作者: Tim Cooijmans's avatar Tim Cooijmans

BatchedDot: fix bad pointer arithmetic

上级 c56c2f70
...@@ -3667,7 +3667,7 @@ class BatchedDot(Op): ...@@ -3667,7 +3667,7 @@ class BatchedDot(Op):
case 0x111: sgemm_(&N, &N, &Nz1, &Nz2, &Nx2, &a, x, &sx_2, y, &sy_2, &b, z, &sz_2); break; case 0x111: sgemm_(&N, &N, &Nz1, &Nz2, &Nx2, &a, x, &sx_2, y, &sy_2, &b, z, &sz_2); break;
default: PyErr_SetString(PyExc_ValueError, "some matrix has no unit stride"); %(fail)s; default: PyErr_SetString(PyExc_ValueError, "some matrix has no unit stride"); %(fail)s;
}; };
x += Sz[0]; y += Sy[0]; z += Sz[0]; x += Sx[0]; y += Sy[0]; z += Sz[0];
} }
} }
break; break;
...@@ -3697,7 +3697,7 @@ class BatchedDot(Op): ...@@ -3697,7 +3697,7 @@ class BatchedDot(Op):
"some matrix has no unit stride"); "some matrix has no unit stride");
%(fail)s; %(fail)s;
}; };
x += Sz[0]; y += Sy[0]; z += Sz[0]; x += Sx[0]; y += Sy[0]; z += Sz[0];
} }
} }
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论