Add preprocessing of list inputs for op by op execution

tenstorrent · Feb 11, 2025 · 8fe87e0 · 8fe87e0
1 parent c8b40f3
commit 8fe87e0
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/tt_torch/dynamo/backend.py b/tt_torch/dynamo/backend.py
@@ -336,6 +336,16 @@ def pre_process_inputs(self, *inputs):
                 if not inp.is_contiguous():
                     inp = inp.contiguous()
                 processed_inputs.append(inp)
+            elif isinstance(inp, list):
+                for ele in inp:
+                    if isinstance(ele, torch.nn.Parameter):
+                        if not ele.data.is_contiguous():
+                            ele.data = ele.data.contiguous()
+                        processed_inputs.append(ele.data)
+                    elif isinstance(ele, torch.Tensor):
+                        if not ele.is_contiguous():
+                            ele = ele.contiguous()
+                        processed_inputs.append(ele)
 
         # Typecast the unsupported data types to hardware supported types.
         supported_inputs = ()