pytorch

393f7f6a - add layout to slow path (#80429)

add layout to slow path (#80429) Pull Request resolved: https://github.com/pytorch/pytorch/pull/80429 Approved by: https://github.com/ezyang