pytorch
defde3bb - [NNC] Use index for stride mapping in kernel.cpp (#72266)

Commit
4 years ago
[NNC] Use index for stride mapping in kernel.cpp (#72266) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/72266 Within the kernel, we may manipulate `Value *` in `OptimizeCat`, which would invalidate the input `Value *` -> Stride mapping. Fix for https://github.com/pytorch/pytorch/issues/72173 Test Plan: Imported from OSS Reviewed By: dagitses, davidberard98 Differential Revision: D33986306 Pulled By: eellison fbshipit-source-id: dc33cd2b545e49e90d1e46b9fcf1e6dbb4b829db (cherry picked from commit 5e4555968a0d7b9e42ab6368575137b1c1db814f)
Author
Elias Ellison
Committer
Parents
Loading