pytorch
2a20a1f3 - [Quant][core][gpu][improvement] Refactored implementation for conv2d_cudnn to use packed parameters (Reland PR#73510) (#74318)

Commit
2 years ago
[Quant][core][gpu][improvement] Refactored implementation for conv2d_cudnn to use packed parameters (Reland PR#73510) (#74318) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/74318 This a reland of https://github.com/pytorch/pytorch/pull/73510 -- please look at this PR directly for a summary and test plan. The only change in this PR is we add the ops to check_forward_backward_compatibility.py to get around the backwards compatibility issue introduced in the previous PR that changes the name of the cudnn conv operations. Differential Revision: D34934630 D34934630 Test Plan: Imported from OSS Reviewed By: jerryzh168 Pulled By: dzdang fbshipit-source-id: 04bdc7c4123dc60e82504b032dbde484427dda93 (cherry picked from commit 7bf672994309468862d7266292ce5000a36d0f5a)
Author
Committer
Parents
Loading