pytorch
daf75c07 - [AOTAutograd] compare with stride hints (#103342)

Commit
1 year ago
[AOTAutograd] compare with stride hints (#103342) We previously compare FakeTensor's strides with real tensor's strides. This cause dynamic dimension of FakeTensor being specialized to static int. This may cause a graph specialized for one shape being used by another shape which is wrong. Use stride hints for the comparison instead. Pull Request resolved: https://github.com/pytorch/pytorch/pull/103342 Approved by: https://github.com/malfet
Author
Committer
Parents
Loading