Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

add mkldnn surport for concat #3

Merged
merged 12 commits into from
Dec 8, 2017
Merged

Conversation

wentingj
Copy link

@wentingj wentingj commented Dec 7, 2017

Description

Add mkldnn surport for concat, which passed unit test. First epoch of inception-bn has "Epoch[0] Train-accuracy=0.573438"

Checklist

Essentials

  • Passed code style checking (make lint)
  • Changes are complete (i.e. I finished coding on this PR)
  • All changes have test coverage
  • For user-facing API changes, API doc string has been updated. For new C++ functions in header files, their functionalities and arguments are well-documented.
  • To my best knowledge, examples are either not affected by this change, or have been fixed to be compatible with this change

Changes

  • Feature1, tests, (and when applicable, API doc)
  • Feature2, tests, (and when applicable, API doc)

Comments

  • If this change is a backward incompatible change, why must this change be made.
  • Interesting edge cases to note here

@zheng-da
Copy link
Owner

zheng-da commented Dec 8, 2017

@wentingj could you run "make lint" and fix the coding style, please? Thanks.
I just pushed new commits to fix the coding style in my code.

@wentingj
Copy link
Author

wentingj commented Dec 8, 2017

@zheng-da Thanks for your kind reminder. I have run "make lint" and passed coding style checking in my branch "wentingj:mkldnn-concat".

std::vector<nnvm::NodeEntry> heads(ograds.begin(), ograds.end());
for (size_t i = 0; i < n->inputs.size(); i++) {
Copy link
Owner

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

this makes native concat fail. The native concat only expects out_grad.

@zheng-da zheng-da merged commit 8758290 into zheng-da:mkldnn Dec 8, 2017
xinyu-intel pushed a commit to wentingj/incubator-mxnet-int8 that referenced this pull request May 6, 2018
* Updated tutorials page.

* Combined tutorial links

Added "alternative" links.

* Corrected typo

* Force build.

* Force build zheng-da#2

* Force build zheng-da#3

* Force zheng-da#4
jinhuang415 pushed a commit to jinhuang415/incubator-mxnet that referenced this pull request May 29, 2018
* Updated tutorials page.

* Combined tutorial links

Added "alternative" links.

* Corrected typo

* Force build.

* Force build zheng-da#2

* Force build zheng-da#3

* Force zheng-da#4
zheng-da pushed a commit that referenced this pull request Jun 28, 2018
* Updated tutorials page.

* Combined tutorial links

Added "alternative" links.

* Corrected typo

* Force build.

* Force build #2

* Force build #3

* Force #4
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants