pytorch

89d30b29 - [functorch] Added better handling for output device inference (i.e. only use when metatracing)