From f27eb9794a0e328964f1af323d899c9a2f4ca9f9 Mon Sep 17 00:00:00 2001 From: Fan Date: Wed, 23 Oct 2019 15:15:14 +0800 Subject: [PATCH] reduce NPY_MAXARGS --- src/operator/numpy/np_einsum_op-inl.h | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/operator/numpy/np_einsum_op-inl.h b/src/operator/numpy/np_einsum_op-inl.h index f7935cef2a95..550b531e81a6 100644 --- a/src/operator/numpy/np_einsum_op-inl.h +++ b/src/operator/numpy/np_einsum_op-inl.h @@ -73,8 +73,8 @@ namespace mxnet { namespace op { -#define NPY_MAXDIMS 32 -#define NPY_MAXARGS 32 +#define NPY_MAXDIMS 16 +#define NPY_MAXARGS 16 inline TShape get_stride(const TShape& shape) { int ndim = shape.ndim(), prod = 1;