diff --git a/thunder/dynamo/report.py b/thunder/dynamo/report.py index 0a26e02f53..34d55edcea 100644 --- a/thunder/dynamo/report.py +++ b/thunder/dynamo/report.py @@ -396,6 +396,9 @@ def __init__(self, graphs: list[torch.fx.GraphModule], graph_names: list[str] = graph_names = [f"graph{idx}" for idx in range(len(graphs))] self.fx_graph_reports: list[FXGraphReport] = [FXGraphReport(g, name) for name, g in zip(graph_names, graphs)] + def __str__(self): + return f"" + def fx_report(fn: Callable, *args, compile_options: dict = None, **kwargs) -> FXReport: """ diff --git a/thunder/tests/test_dynamo.py b/thunder/tests/test_dynamo.py index d9b5f500f5..b0f9c1d8bf 100644 --- a/thunder/tests/test_dynamo.py +++ b/thunder/tests/test_dynamo.py @@ -1117,7 +1117,7 @@ def check(file_name, cmd): cmd = [sys.executable] if use_benchmark: - cmd = cmd + ["-m", "pytest", "--benchmark-warmup-iterations=5"] + cmd = cmd + ["-m", "pytest"] py_files = list(tmp_path.glob("*.py")) assert len(py_files) == 4