fix the C version
Former-commit-id: 470dfcd9336bed06916463770a8a144d67e0fd95
This commit is contained in:
parent
019304dcb4
commit
b6b701bd42
|
@ -682,6 +682,7 @@ int qpms_cython_trans_calculator_get_AB_arrays_loop(
|
||||||
for (int ax = 0; ax < resnd; ++ax){
|
for (int ax = 0; ax < resnd; ++ax){
|
||||||
assert(A_shape[ax] == B_shape[ax]);
|
assert(A_shape[ax] == B_shape[ax]);
|
||||||
assert(A_strides[ax] == B_strides[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;
|
if (A_shape[ax] > A_shape[longest_axis]) longest_axis = ax;
|
||||||
}
|
}
|
||||||
const npy_intp longlen = resultshape[longest_axis];
|
const npy_intp longlen = resultshape[longest_axis];
|
||||||
|
|
Loading…
Reference in New Issue