fix type casting from Tuple to List

KingsleyLiu-NV · KingsleyLiu-NV · commit 30ca1e170e98 · 2023-11-19T21:33:28.000-08:00
diff --git a/torch/distributed/_tensor/tp_conv.py b/torch/distributed/_tensor/tp_conv.py
@@ -135,7 +135,7 @@ def tp_convolution(
         )
 
         # step2 feed local input tensor to op_call
-        local_tensor_args_list = cast(List, local_tensor_args)
+        local_tensor_args_list = list(local_tensor_args)
         local_tensor_args_list[0] = in_tensor
         local_tensor_args = cast(Tuple[object, ...], local_tensor_args_list)
         local_results = op_call(*local_tensor_args, **local_tensor_kwargs)
@@ -205,7 +205,7 @@ def tp_convolution_backward(
             )
 
         # step3 feed local input tensor to op_call
-        local_tensor_args_list = cast(List, local_tensor_args)
+        local_tensor_args_list = list(local_tensor_args)
         local_tensor_args_list[0] = grad_out_tensor
         local_tensor_args_list[1] = in_tensor
         local_tensor_args = cast(Tuple[object, ...], local_tensor_args_list)
