Skip to content

Commit

Permalink
fix
Browse files Browse the repository at this point in the history
  • Loading branch information
kiya00 committed Jan 10, 2025
1 parent ac3c6cb commit df71313
Showing 1 changed file with 2 additions and 3 deletions.
5 changes: 2 additions & 3 deletions thunder/dynamo/report.py
Original file line number Diff line number Diff line change
Expand Up @@ -35,11 +35,10 @@ def thunderfx_save_report(
try:
compiled = thunderfx(fn, **compile_kwargs) if compile_kwargs is not None else thunderfx(fn)
compiled(*args, **kwargs)
backend = compiled._backend
except Exception as e:
print(f"Failed to run the function using ThunderFX with exception: {e}")
try:
backend.save_reproducer_to_folder(folder_path)
compiled._backend.save_reproducer_to_folder(folder_path)
except Exception as repro_e:
print(f"Failed to save reproducer due to {repro_e}")
return
Expand All @@ -49,7 +48,7 @@ def thunderfx_save_report(
if not check_benchmark and not check_consistency:
return

thunder_graph_names = get_thunder_graph_names(backend.subgraph_infos)
thunder_graph_names = get_thunder_graph_names(compiled._backend.subgraph_infos)
EXECUTOR_NAMES = ("eager", "thunder", "torch_inductor")

report_result: dict[str, list] = {}
Expand Down

0 comments on commit df71313

Please sign in to comment.