-
Notifications
You must be signed in to change notification settings - Fork 24.4k
[quant] Expose supportedQEngines to python #26474
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
Summary: att Test Plan: python test/test_torch.py Reviewers: pt1quant Subscribers: Tasks: Tags: [ghstack-poisoned]
Summary: att Test Plan: python test/test_torch.py Reviewers: pt1quant Subscribers: Tasks: Tags: [ghstack-poisoned]
qengines = torch._C._supported_qengines() | ||
# [TODO] Enable after the interface change | ||
# original_qe = torch._C._get_qengine() | ||
# for qe in qengines: |
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.
qe
is an int but set_qengine expects at::QEngine
. Will this work?
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.
it doens't work right now that's why I commented them out. it should work after the interface change. we should be returning/pass int
in _get_qengine
and _set_qengine
as well
Summary: att Test Plan: python test/test_torch.py Reviewers: pt1quant Subscribers: Tasks: Tags: [ghstack-poisoned]
Summary: att Test Plan: python test/test_torch.py Reviewers: pt1quant Subscribers: Tasks: Tags: [ghstack-poisoned]
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.
The plan after this is to change qengine to string (user facing) and return an int from supported_qengines
.
plan is to change |
Summary: att Test Plan: python test/test_torch.py Reviewers: pt1quant Subscribers: Tasks: Tags: [ghstack-poisoned]
Summary: att Test Plan: python test/test_torch.py Reviewers: pt1quant Subscribers: Tasks: Tags: [ghstack-poisoned]
This pull request has been merged in 2667493. |
Summary: Pull Request resolved: pytorch#26474 att Test Plan: python test/test_torch.py Imported from OSS Differential Revision: D17517373 fbshipit-source-id: af931761d6ee31a88808d05f686002a83b6b25af
Summary: att Test Plan: python test/test_torch.py Reviewers: pt1quant Subscribers: Tasks: Tags: ghstack-source-id: 6688d99 Pull Request resolved: pytorch/pytorch#26474
Stack from ghstack:
Summary:
att
Test Plan:
python test/test_torch.py
Reviewers:
pt1quant
Subscribers:
Tasks:
Tags:
Differential Revision: D17517373