Move resize / zero logic for _thnn_conv_depthwise2d from codegen to native code.#37957
Move resize / zero logic for _thnn_conv_depthwise2d from codegen to native code.#37957gchanan wants to merge 3 commits intogh/gchanan/269/basefrom
Conversation
…ative code. [ghstack-poisoned]
💊 CI failures summary and remediationsAs of commit d376a6a (more details on the Dr. CI page):
🕵️ 1 new failure recognized by patternsThe following CI failures do not appear to be due to upstream breakages:
|
|
|
||
| # grad_weight and grad_bias need to be resized and zeroed | ||
| if arg['name'] == 'grad_weight' and base['name'] != '_thnn_conv2d': | ||
| if arg['name'] == 'grad_weight' and base['name'] != '_thnn_conv2d' and base['name'] != '_thnn_conv_depthwise2d': |
There was a problem hiding this comment.
reupping earlier note - personally I'd rather have a list of names that's getting checked here, but not a blocker if you either like it better this way or just want to blast ahead
There was a problem hiding this comment.
yep, good suggestion. I knew ahead of time (from checking the generated code) that there were only two cases and I was going to remove them in this stack so it didn't seem worth it.
…odegen to native code." Differential Revision: [D21433212](https://our.internmc.facebook.com/intern/diff/D21433212) [ghstack-poisoned]
…odegen to native code." Differential Revision: [D21433212](https://our.internmc.facebook.com/intern/diff/D21433212) [ghstack-poisoned]
…ative code. (pytorch#37957) Summary: Pull Request resolved: pytorch#37957 Test Plan: Imported from OSS Differential Revision: D21433212 Pulled By: gchanan fbshipit-source-id: fb431d5cf06afe2bb87fa2d73e15046f9a8d044d
Stack from ghstack:
Differential Revision: D21433212