pytorch

4143483d - [RPC Framework] Create a separate remote module template when moving CPU tensors to a cuda device is not enabled (#57413)

[RPC Framework] Create a separate remote module template when moving CPU tensors to a cuda device is not enabled (#57413) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/57413 An internal test fails because somehow `Tuple[()]` is not considered compatible with `Tuple[Any]` in TorchScript, even if the code that involves this type of variables is not executed at all. Therefore, create separate templates for instantiation to avoid typing check failure. This can address the FIXME left in https://github.com/pytorch/pytorch/pull/57288 #Closes: https://github.com/pytorch/pytorch/issues/51670 Test Plan: buck test mode/dev-nosan caffe2/test/distributed/rpc:process_group_agent -- RemoteModule -j 1 buck test mode/dev-nosan caffe2/torch/fb/training_toolkit/applications/sparse_nn/batch_distributed_inference/tests:batch_distributed_inference_test -- test_load_di_parts Reviewed By: wanchaol Differential Revision: D28138864 fbshipit-source-id: 39e3e67b0c3979b607ff104d84b4fb1070ffefd6