fix rebase conflict

Author: kiya00

thunder/dynamo/compiler.py

@@ -87,8 +87,8 @@ def __call__(self, gm: torch.fx.GraphModule, sample_args: list[torch.SymInt, tor
     def save_reproducer_to_folder(self, reproducer_folder: str | PathLike):
         """Save the reproducer script for the GraphModule executed by Thunder to the specified `reproducer_folder`.
         Each saved script is named as "g[graph_id]_thunder_[module_id]", where:
-          - `graph_id` indexes the graph generated by Dynamo, which is then passed to Thunder.
-          - `module_id` indexes the submodule split by the :func:`thunder.dynamo.utils._splitter`.
+            - `graph_id` indexes the graph generated by Dynamo, which is then passed to Thunder.
+            - `module_id` indexes the submodule split by the :func:`thunder.dynamo.utils._splitter`.
         Both `graph_id` and `module_id` start from 1.
         """
         if not self.subgraph_infos:

thunder/dynamo/utils.py

@@ -564,30 +564,6 @@ def checkpoint_converter(gm: torch.fx.GraphModule, sub_gm: torch.fx.GraphModule)
                 _checkpoint_function_converter(function_module)
 
 
-from torch.nn.modules.module import _addindent
-import typing
-
-
-@dataclasses.dataclass
-class ExampleInputMetaData:
-    """
-    Describes the metadata of a tensor, used to generate a random tensor with matching properties
-    """
-
-    requires_grad: bool
-    layout: torch.layout
-    device: str | torch.device
-    dtype: torch.dtype
-    shape: list[int]
-    storage_shape: list[int]
-    strides: list[int]
-    min_val: int | None = None
-    max_val: int | None = None
-
-    def stride(self) -> list[int]:
-        return self.strides
-
-
 def arg_like_tensor_integer(arg: torch.Tensor | ExampleInputMetaData, f: TextIO):
     """Creates a new argument like the given tensor, which must be an integer
     type. This is separated out because randn() does not work for integer

thunder/tests/test_dynamo.py

@@ -692,6 +692,7 @@ def find_target_module(model, target_module_name):
             assert isinstance(n.target, Symbol)
 
 
+@requiresCUDA
 def test_dynamo_reproducer_2graph(tmp_path):
     backend = ThunderCompiler()
     # Test non-contiguous input tensor