[incubator-mxnet] branch master updated: fix dot(csr.T, dns)=dns can't be called on cpu and gpu (#11087)

[haibin]

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

haibin 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 8be4b8e  fix dot(csr.T, dns)=dns can't be called on cpu and gpu 
(#11087)
8be4b8e is described below

commit 8be4b8ef9b30eb90aa3885f47be6691a2bbafa1b
Author: XiaotaoChen <chenxiaotao1...@gmail.com>
AuthorDate: Thu May 31 08:34:28 2018 +0800

    fix dot(csr.T, dns)=dns can't be called on cpu and gpu (#11087)
---
 src/operator/tensor/dot-inl.h | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/src/operator/tensor/dot-inl.h b/src/operator/tensor/dot-inl.h
index 2c9a483..ffdb706 100644
--- a/src/operator/tensor/dot-inl.h
+++ b/src/operator/tensor/dot-inl.h
@@ -246,6 +246,10 @@ inline bool DotForwardInferStorageType(const 
nnvm::NodeAttrs& attrs,
     if (target_stype == kRowSparseStorage) {
       dispatched = storage_type_assign(&out_stype, kRowSparseStorage,
                                        dispatch_mode, 
DispatchMode::kFComputeEx);
+    // csr.T, rsp/dns -> dns
+    } else if (target_stype == kDefaultStorage) {
+      dispatched = storage_type_assign(&out_stype, kDefaultStorage, 
dispatch_mode,
+                                       DispatchMode::kFComputeEx);
     }
   }
   if (!dispatched && lhs_stype == kCSRStorage && rhs_rsp_or_dns &&

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