-
Notifications
You must be signed in to change notification settings - Fork 26.3k
2 Bit Embedding Conversion Operator support. #43077
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
Conversation
|
This pull request was exported from Phabricator. Differential Revision: D23143262 |
💊 CI failures summary and remediationsAs of commit 9e97a5d (more details on the Dr. CI page): ✅ None of the CI failures appear to be your fault 💚
🚧 1 ongoing upstream failure:These were probably caused by upstream breakages that are not fixed 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 on the GitHub issue tracker or post in the (internal) Dr. CI Users group. This comment has been revised 10 times. |
supriyar
left a comment
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.
Looks good, left some minor comments.
|
This pull request was exported from Phabricator. Differential Revision: D23143262 |
e4abb76 to
4081495
Compare
Summary: Pull Request resolved: pytorch#43077 2 Bit Embedding weight conversion operation is quite similar to 4 bit embedding weight conversion. The diff contains both the 1. 2bit packing op `embedding_bag_2bit_prepack`. 2. 2bit unpacking op `embedding_bag_2bit_unpack`. Comments about the op are inline with the op definition. Test Plan: buck test caffe2/test:quantization -- test_embedding_bag_2bit_unpack Reviewed By: supriyar Differential Revision: D23143262 fbshipit-source-id: 97cd6da969abafb5d302b3150ba5349117f22b8a
|
This pull request was exported from Phabricator. Differential Revision: D23143262 |
4081495 to
9e97a5d
Compare
|
This pull request has been merged in fa6b34b. |
Summary:
2 Bit Embedding weight conversion operation is quite similar to
4 bit embedding weight conversion.
The diff contains both the
embedding_bag_2bit_prepack.embedding_bag_2bit_unpack.Comments about the op are inline with the op definition.
Test Plan: buck test caffe2/test:quantization -- test_embedding_bag_2bit_unpack
Differential Revision: D23143262