anko-intel opened a new pull request #20928:
URL: https://github.com/apache/incubator-mxnet/pull/20928
## Description ##
Improve performance of random.shuffle implementation.
Results from xxx instance of the follwing script
```
import mxnet as mx
from benchmark.opperf.utils.benchmark_utils import run_performance_test
shapes = [
( 10, 1),
( 32, 9999),
(4096, 1),
(4096, 500),
(4096, 4096),
(4096, 1024, 32)]
print("\nnot-in-place (mx.nd.random.shuffle):")
for shape in shapes:
s_res = run_performance_test(mx.nd.random.shuffle, run_backward=False,
inputs=[{"data": (shape)}])
print(s_res)
print("\nin-place (mx.np.random.shuffle):")
for shape in shapes:
s_res = run_performance_test(mx.np.random.shuffle, run_backward=False,
inputs=[{"x": (shape)}])
print(s_res)
```
shows following improvments:
## Checklist ##
### Essentials ###
- [ ] PR's title starts with a category (e.g. [BUGFIX], [MODEL], [TUTORIAL],
[FEATURE], [DOC], etc)
- [ ] Changes are complete (i.e. I finished coding on this PR)
- [ ] All changes have test coverage
- [ ] Code is well-documented
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]
