pytorch
23815480 - add stride constraints to fallbacks (#88534)

Commit
3 years ago
add stride constraints to fallbacks (#88534) Add stride/contiguity constraints to fallbacks so that inputs will be in the right stride permutation for the fallback kernel. Improves perf of coat_lite_mini from 1.48415536054865 -> 2.010956856330101. Pull Request resolved: https://github.com/pytorch/pytorch/pull/88534 Approved by: https://github.com/ngimel
Author
Committer
Parents
Loading