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
Extend 2Dim embedding bag benchmarking to include 3Dim benchmarks #64647
Conversation
🔗 Helpful links
💊 CI failures summary and remediationsAs of commit 2c667ff (more details on the Dr. CI page): 💚 💚 Looks good so far! There are no failures yet. 💚 💚 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 to the (internal) Dr. CI Users group. |
This pull request was exported from Phabricator. Differential Revision: D30770085 |
Codecov Report
@@ Coverage Diff @@
## master #64647 +/- ##
===========================================
+ Coverage 53.70% 66.64% +12.94%
===========================================
Files 710 710
Lines 92426 92426
===========================================
+ Hits 49638 61601 +11963
+ Misses 42788 30825 -11963 |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
can you fix the failed lint check?
let's go with what we discussed on the diff, do two diffs, first add tests, and second to clean up
Sounds good to me! |
This pull request was exported from Phabricator. Differential Revision: D30770085 |
2eb0a1c
to
d706a12
Compare
…torch#64647) Summary: Pull Request resolved: pytorch#64647 Add support for benchmarking of 8 bit quantizations of N-D batched embeddings. Currently only works for 3Dim embeddings and still requires thought on ramping up from 3Dim to NDim. Test Plan: ```buck run //caffe2/benchmarks/operator_benchmark/pt:qembedding_pack_test``` Differential Revision: D30770085 fbshipit-source-id: e46bf09057347dd9d391d2d4f7468b7a0c46aabc
This pull request was exported from Phabricator. Differential Revision: D30770085 |
d706a12
to
2c667ff
Compare
This pull request has been merged in 3fbb49e. |
Summary: Add support for benchmarking of 8 bit quantizations of N-D batched embeddings. Currently only works for 3Dim embeddings and still requires thought on ramping up from 3Dim to NDim.
Test Plan:
buck run //caffe2/benchmarks/operator_benchmark/pt:qembedding_pack_test
Differential Revision: D30770085