Skip to content
This repository has been archived by the owner on Nov 17, 2023. It is now read-only.

Commit

Permalink
* Fix sanity
Browse files Browse the repository at this point in the history
  • Loading branch information
hanke580 committed Jun 1, 2020
1 parent 5f83e7d commit acda642
Showing 1 changed file with 2 additions and 4 deletions.
6 changes: 2 additions & 4 deletions src/operator/numpy/np_einsum_op-inl.h
Original file line number Diff line number Diff line change
Expand Up @@ -724,10 +724,8 @@ inline void NumpyEinsumProcess(const std::vector<TBlob>& inputs,
int j = 0;
for (idim = 0; idim < ndim_iter; ++idim) {
if (op_axes_arrays[i][idim] == -1 ||
(iop!=nop &&
opshape[i][op_axes_arrays[i][idim]] == 1 &&
opshape[iop][op_axes_arrays[iop][idim]] != 1)
){
(iop != nop && opshape[i][op_axes_arrays[i][idim]] == 1 &&
opshape[iop][op_axes_arrays[iop][idim]] != 1)) {
remainstride[iop][j++] = iterstride[iop][idim];
} else {
opstride[iop][op_axes_arrays[i][idim]] = iterstride[iop][idim];
Expand Down

0 comments on commit acda642

Please sign in to comment.