pytorch

958d0cd4 - Adding short tests

Adding short tests Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/29257 Test Plan: Imported from OSS Differential Revision: D18340536 Pulled By: z-a-f fbshipit-source-id: dce470fd0c7ef9c6f639de40f7e0713b335408d1