fix the C version

Former-commit-id: 470dfcd9336bed06916463770a8a144d67e0fd95
This commit is contained in:
Marek Nečada 2017-05-03 06:02:01 +03:00
parent 019304dcb4
commit b6b701bd42
1 changed files with 1 additions and 0 deletions

View File

@ -682,6 +682,7 @@ int qpms_cython_trans_calculator_get_AB_arrays_loop(
for (int ax = 0; ax < resnd; ++ax){
assert(A_shape[ax] == B_shape[ax]);
assert(A_strides[ax] == B_strides[ax]);
if (daxis == ax || saxis == ax) continue;
if (A_shape[ax] > A_shape[longest_axis]) longest_axis = ax;
}
const npy_intp longlen = resultshape[longest_axis];