reduce NPY_MAXARGS

apache · Oct 23, 2019 · f27eb97 · f27eb97
1 parent 663ba6c
commit f27eb97
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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;