pytorch
9b46737f - Add tests for fake tensor striding (#82571)

Commit
3 years ago
Add tests for fake tensor striding (#82571) Add tests for fake tensor striding in OpInfos. I know primtorch is not strictly committing to consistent stride propagation with ATen (see https://github.com/pytorch/pytorch/issues/78050), where as in fake tensor/meta the goal is be completely consistent. This is a little awkward because by default prim refs will register a meta implementation. In any case, I think we can add the tests for fake with a disclaimer in the tests the failure is non-blocking for adding prims. At least as far as OpInfo tests get, the prims seem to do a pretty good job with stride propagation already. Pull Request resolved: https://github.com/pytorch/pytorch/pull/82571 Approved by: https://github.com/ezyang
Author
Committer
Parents
Loading