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 <[email protected]>
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
[email protected].
