Remove use_c10_dispatcher option#54969
Conversation
With all use cases to hacky wrapper removed, all kernels will be dispatched with c10 full dispatcher. Differential Revision: [D27436596](https://our.internmc.facebook.com/intern/diff/D27436596/) [ghstack-poisoned]
💊 CI failures summary and remediationsAs of commit eddd659 (more details on the Dr. CI page):
🕵️ 3 new failures recognized by patternsThe following CI failures do not appear to be due to upstream breakages:
|
| Job | Step | Action |
|---|---|---|
| Unknown | 🔁 rerun |
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.
With all use cases to hacky wrapper removed, all kernels will be dispatched with c10 full dispatcher. Differential Revision: [D27436596](https://our.internmc.facebook.com/intern/diff/D27436596/) [ghstack-poisoned]
Pull Request resolved: #54969 With all use cases to hacky wrapper removed, all kernels will be dispatched with c10 full dispatcher. ghstack-source-id: 125269241 Differential Revision: [D27436596](https://our.internmc.facebook.com/intern/diff/D27436596/)
With all use cases to hacky wrapper removed, all kernels will be dispatched with c10 full dispatcher. Differential Revision: [D27436596](https://our.internmc.facebook.com/intern/diff/D27436596/) [ghstack-poisoned]
Pull Request resolved: #54969 With all use cases to hacky wrapper removed, all kernels will be dispatched with c10 full dispatcher. ghstack-source-id: 125278882 Differential Revision: [D27436596](https://our.internmc.facebook.com/intern/diff/D27436596/)
With all use cases to hacky wrapper removed, all kernels will be dispatched with c10 full dispatcher. Differential Revision: [D27436596](https://our.internmc.facebook.com/intern/diff/D27436596/) [ghstack-poisoned]
Pull Request resolved: #54969 With all use cases to hacky wrapper removed, all kernels will be dispatched with c10 full dispatcher. ghstack-source-id: 125341486 Differential Revision: [D27436596](https://our.internmc.facebook.com/intern/diff/D27436596/)
| namespace impl { | ||
|
|
||
| inline c10::optional<MemoryFormat> | ||
| check_tensor_options_and_extract_memory_format( |
There was a problem hiding this comment.
check_tensor_options_and_extract_memory_format is still required. But not sure how to rename hacky_wrapper_for_legacy_signatures.h. Use adaption.h in this PR but let me know if there is any better suggestions. cc @smessmer
There was a problem hiding this comment.
check_tensor_options_and_extract_memory_format is actually a good example of another thing we need to get rid of; it's due to MemoryFormat being a separate argument in some JIT schema but also being specified in TensorOptions.
|
|
||
| class Locals(threading.local): | ||
| use_c10_dispatcher: Optional[UseC10Dispatcher] = None | ||
| _locals = Locals() |
There was a problem hiding this comment.
This looks like we can remove the whole Locals class, cc @ezyang did you want to use it for something else?
There was a problem hiding this comment.
@wenleix and I chatted about it, and it might make sense to keep this around for a little bit, in case there's some other incremental refactoring we need to do (I don't think we're done, btw)
There was a problem hiding this comment.
Discussed this offline with @ezyang . Supporting dynamic scoping during codegen seems a useful feature (e.g. other refactor) , plus is not very much code.
smessmer
left a comment
There was a problem hiding this comment.
Yay :) Stoked to finally get rid of hacky_wrapper
With all use cases to hacky wrapper removed, all kernels will be dispatched with c10 full dispatcher. Differential Revision: [D27436596](https://our.internmc.facebook.com/intern/diff/D27436596/) [ghstack-poisoned]
Pull Request resolved: #54969 With all use cases to hacky wrapper removed, all kernels will be dispatched with c10 full dispatcher. ghstack-source-id: 125401584 Differential Revision: [D27436596](https://our.internmc.facebook.com/intern/diff/D27436596/)
With all use cases to hacky wrapper removed, all kernels will be dispatched with c10 full dispatcher. Differential Revision: [D27436596](https://our.internmc.facebook.com/intern/diff/D27436596/) [ghstack-poisoned]
Pull Request resolved: #54969 With all use cases to hacky wrapper removed, all kernels will be dispatched with c10 full dispatcher. ghstack-source-id: 125434790 Differential Revision: [D27436596](https://our.internmc.facebook.com/intern/diff/D27436596/)
|
This pull request has been merged in 70af5db. |
Stack from ghstack:
With all use cases to hacky wrapper removed, all kernels will be
dispatched with c10 full dispatcher.
Differential Revision: D27436596