File tree Expand file tree Collapse file tree 1 file changed +8
-4
lines changed
userbenchmark/dynamo/dynamobench Expand file tree Collapse file tree 1 file changed +8
-4
lines changed Original file line number Diff line number Diff line change @@ -2284,9 +2284,11 @@ def record_status(accuracy_status, dynamo_start_stats):
22842284 )
22852285 ):
22862286 is_same = False
2287- except Exception :
2287+ except Exception as e :
22882288 # Sometimes torch.allclose may throw RuntimeError
2289- is_same = False
2289+ exception_string = str (e )
2290+ accuracy_status = f"fail_exception: { exception_string } "
2291+ return record_status (accuracy_status , dynamo_start_stats = start_stats )
22902292
22912293 if not is_same :
22922294 accuracy_status = "eager_two_runs_differ"
@@ -2403,9 +2405,11 @@ def record_status(accuracy_status, dynamo_start_stats):
24032405 force_max_multiplier = force_max_multiplier ,
24042406 ):
24052407 is_same = False
2406- except Exception :
2408+ except Exception as e :
24072409 # Sometimes torch.allclose may throw RuntimeError
2408- is_same = False
2410+ exception_string = str (e )
2411+ accuracy_status = f"fail_exception: { exception_string } "
2412+ return record_status (accuracy_status , dynamo_start_stats = start_stats )
24092413
24102414 if not is_same :
24112415 if self .args .skip_accuracy_check :
You can’t perform that action at this time.
0 commit comments