diff --git a/src/operator/numpy/np_matrix_op-inl.h b/src/operator/numpy/np_matrix_op-inl.h index 41c60442e763..38af423b3d3a 100644 --- a/src/operator/numpy/np_matrix_op-inl.h +++ b/src/operator/numpy/np_matrix_op-inl.h @@ -1316,8 +1316,8 @@ void NumpyDiagonalOpImpl(const TBlob& in_data, stride1 + stride2, offset, oshape[odim - 1]); } else { Kernel, xpu>::Launch(s, dsize, out_data.dptr(), - in_data.dptr(), Shape3(oleading, obody, otrailing), Shape3(ileading, ibody, itrailing), - stride1 + stride2, offset, oshape[odim - 1]); + in_data.dptr(), Shape3(oleading, obody, otrailing), + Shape3(ileading, ibody, itrailing), stride1 + stride2, offset, oshape[odim - 1]); } }); }); @@ -1342,7 +1342,8 @@ void NumpyDiagonalOpForward(const nnvm::NodeAttrs& attrs, const mxnet::TShape& oshape = outputs[0].shape_; const NumpyDiagonalParam& param = nnvm::get(attrs.parsed); - NumpyDiagonalOpImpl(in_data, out_data, ishape, oshape, out_data.Size(), param, s, req); + NumpyDiagonalOpImpl(in_data, out_data, ishape, oshape, + out_data.Size(), param, s, req); } template @@ -1363,7 +1364,8 @@ void NumpyDiagonalOpBackward(const nnvm::NodeAttrs& attrs, const mxnet::TShape& oshape = outputs[0].shape_; const NumpyDiagonalParam& param = nnvm::get(attrs.parsed); - NumpyDiagonalOpImpl(in_data, out_data, oshape, ishape, in_data.Size(), param, s, req); + NumpyDiagonalOpImpl(in_data, out_data, oshape, ishape, + in_data.Size(), param, s, req); } struct NumpyDiagflatParam : public dmlc::Parameter {