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

Extend 2Dim embedding bag benchmarking to include 3Dim benchmarks #64647

Closed
wants to merge 1 commit into from

Conversation

EdwaRen
Copy link
Contributor

@EdwaRen EdwaRen commented Sep 8, 2021

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

@facebook-github-bot
Copy link
Contributor

facebook-github-bot commented Sep 8, 2021

🔗 Helpful links

💊 CI failures summary and remediations

As 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.

Click here to manually regenerate this comment.

@facebook-github-bot
Copy link
Contributor

This pull request was exported from Phabricator. Differential Revision: D30770085

@codecov
Copy link

codecov bot commented Sep 8, 2021

Codecov Report

Merging #64647 (2c667ff) into master (46c886e) will increase coverage by 12.94%.
The diff coverage is n/a.

@@             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     

Copy link
Member

@jingsh jingsh left a 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

@EdwaRen
Copy link
Contributor Author

EdwaRen commented Sep 8, 2021

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!

@facebook-github-bot
Copy link
Contributor

This pull request was exported from Phabricator. Differential Revision: D30770085

…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
@facebook-github-bot
Copy link
Contributor

This pull request was exported from Phabricator. Differential Revision: D30770085

@facebook-github-bot
Copy link
Contributor

This pull request has been merged in 3fbb49e.

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

Successfully merging this pull request may close these issues.

None yet

3 participants