You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@mxnet.apache.org by GitBox <gi...@apache.org> on 2017/12/13 14:29:19 UTC
[GitHub] ZiyueHuang commented on issue #9057: fix multiple kTempSpace resources needed in a single operator
ZiyueHuang commented on issue #9057: fix multiple kTempSpace resources needed in a single operator
URL: https://github.com/apache/incubator-mxnet/pull/9057#issuecomment-351406993
Test by adding following codes into sparse embedding forward,
```
mshadow::Stream<cpu>* s = ctx.get_stream<cpu>();
mshadow::Tensor<cpu, 1, char> w1 = ctx.requested[0]
.get_space_typed<cpu, 1, char>(mshadow::Shape1(10), s);
mshadow::Tensor<cpu, 1, char> w2 = ctx.requested[1]
.get_space_typed<cpu, 1, char>(mshadow::Shape1(10), s);
LOG(INFO) << " equal " << (w1.dptr_ == w2.dptr_);
```
Then run test of sparse embedding.
Output,
```
/home/ubuntu/fix_rsc_mx/python/mxnet/ndarray/sparse.py:664: RuntimeWarning: Assigning non-NDArray object to RowSparseNDArray is not efficient
RuntimeWarning)
[13:41:19] src/operator/tensor/./indexing_op.h:415: equal 0
[13:41:19] src/operator/tensor/./indexing_op.h:415: equal 0
[13:41:19] src/operator/tensor/./indexing_op.h:415: equal 0
```
----------------------------------------------------------------
This is an automated message from the Apache Git Service.
To respond to the message, please log on GitHub and use the
URL above to go to the specific comment.
For queries about this service, please contact Infrastructure at:
users@infra.apache.org
With regards,
Apache Git Services