This is an automated email from the ASF dual-hosted git repository.

jxie pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/incubator-mxnet.git


The following commit(s) were added to refs/heads/master by this push:
     new 51776aa  Fix test_sparse_operator failures due to out-of-bounds ref. 
(#9674)
51776aa is described below

commit 51776aa2384430d67bf3bd23d2c36059503fd3c8
Author: Dick Carter <dick.car...@comcast.net>
AuthorDate: Fri Feb 2 11:19:51 2018 -0800

    Fix test_sparse_operator failures due to out-of-bounds ref. (#9674)
---
 src/operator/tensor/broadcast_reduce_op.h | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/operator/tensor/broadcast_reduce_op.h 
b/src/operator/tensor/broadcast_reduce_op.h
index 76c78be..02d48b4 100644
--- a/src/operator/tensor/broadcast_reduce_op.h
+++ b/src/operator/tensor/broadcast_reduce_op.h
@@ -362,7 +362,7 @@ inline bool SumOpForwardInferStorageType(const 
nnvm::NodeAttrs& attrs,
                                      DispatchMode::kFCompute);
   }
 
-  if (!dispatched && in_stype == kCSRStorage &&
+  if (!dispatched && in_stype == kCSRStorage && param.axis.ndim() == 1 &&
       (param.axis[0] == 0 || param.axis[0] == 1) && !param.keepdims &&
       !param.exclude) {
     // If input is csr and axis is 0 or 1, and neither of keepdims or exclude

-- 
To stop receiving notification emails like this one, please contact
j...@apache.org.

Reply via email to