Skip to content

uninitialize output and bag_size in the fast path of EmbeddingBag to save overhead - #36681

Closed
xing-liu wants to merge 4 commits into
gh/xing-liu/2/basefrom
gh/xing-liu/2/head
Closed

xing-liu wants to merge 4 commits into
gh/xing-liu/2/basefrom
gh/xing-liu/2/head

Conversation

@xing-liu

@xing-liu xing-liu commented Apr 15, 2020 •

Copy link
Copy Markdown
Contributor

Stack from ghstack:

Differential Revision: D21058006

xing-liu pushed a commit that referenced this pull request Apr 15, 2020
…save overhead

ghstack-source-id: 1bec2ec
Pull Request resolved: #36681
…dingBag to save overhead"

[ghstack-poisoned]
xing-liu added a commit that referenced this pull request Apr 16, 2020
…save overhead

ghstack-source-id: 7fa0564
Pull Request resolved: #36681
@dr-ci

dr-ci Bot commented Apr 16, 2020 •

Copy link
Copy Markdown

💊 Build failures summary and remediations

As of commit 8b3c877 (more details on the Dr. CI page):


  • 1/1 failures introduced in this PR

XLA failure

Job pytorch_xla_linux_bionic_py3_6_clang9_build is failing. Please create an issue with title prefixed by [PT_BREAK] in pytorch/xla and link to to this PR. If you have questions, please reach out to @ailzhang / @dlibenzi / @JackCaoG.


This comment was automatically generated by Dr. CI (expand for details).Follow this link to opt-out of these comments for your Pull Requests.

Please report bugs/suggestions on the GitHub issue tracker.

See how this bot performed.

This comment has been revised 6 times.

…dingBag to save overhead"

Differential Revision: [D21058006](https://our.internmc.facebook.com/intern/diff/D21058006)

[ghstack-poisoned]
xing-liu added a commit that referenced this pull request Apr 16, 2020
…save overhead

ghstack-source-id: eb38595
Pull Request resolved: #36681
@jianyuh
jianyuh requested a review from jamesr66a April 17, 2020 02:41
@jianyuh

jianyuh commented Apr 17, 2020

Copy link
Copy Markdown
Member

Hi @jamesr66a could you take a quick look at this? It looks good to me. Thanks!

…dingBag to save overhead"

Differential Revision: [D21058006](https://our.internmc.facebook.com/intern/diff/D21058006)

[ghstack-poisoned]
xing-liu added a commit that referenced this pull request Apr 17, 2020
…save overhead

ghstack-source-id: 9cab932
Pull Request resolved: #36681
@jianyuh
jianyuh requested review from supriyar and yinghai April 24, 2020 02:55
bharatr21 pushed a commit to bharatr21/pytorch that referenced this pull request May 5, 2020
…save overhead (pytorch#36681)

Summary: Pull Request resolved: pytorch#36681

Test Plan:
Imported from OSS

Unit tests:
python test/run_test.py -i test_nn -- TestNNDeviceTypeCPU.test_EmbeddingBag_per_sample_weights_failures_cpu
python test/run_test.py -i test_nn -- TestNNDeviceTypeCPU.test_EmbeddingBag_per_sample_weights_and_offsets_cpu
python test/run_test.py -i test_nn -- TestNNDeviceTypeCPU.test_EmbeddingBag_per_sample_weights_and_new_offsets_cpu
python test/run_test.py -i test_nn -- TestNNDeviceTypeCPU.test_EmbeddingBag_per_sample_weights_and_no_offsets_cpu
python test/test_nn.py TestNN.test_embeddingbag_from_pretrained
python test/test_nn.py TestNN.test_embeddingbag_from_pretrained_options

Finally run: python test/test_nn.py

Reviewed By: jspark1105

Differential Revision: D21058006

Pulled By: xing-liu

fbshipit-source-id: 65b36a788839e8b722db3e295e58215b5935d6e8
@facebook-github-bot

Copy link
Copy Markdown
Contributor

@xing-liu merged this pull request in 0b693e9.

@facebook-github-bot
facebook-github-bot deleted the gh/xing-liu/2/head branch May 9, 2020 14:17
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

Projects

None yet

Development

Successfully merging this pull request may close these issues.

4 participants