pytorch

c06dfd7c - [fx2trt] Check input device in TRTModule (#63893)

[fx2trt] Check input device in TRTModule (#63893) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/63893 Add a check to ensure all the inputs are on cuda device. Test Plan: CI Reviewed By: kflu, houseroad Differential Revision: D30525265 fbshipit-source-id: 6e50b70fd535defc1f802d51e8bb991b2dd73741