diff --git a/userbenchmark/dynamo/dynamobench/common.py b/userbenchmark/dynamo/dynamobench/common.py index 9505800360..d3ca51e8f4 100644 --- a/userbenchmark/dynamo/dynamobench/common.py +++ b/userbenchmark/dynamo/dynamobench/common.py @@ -1582,6 +1582,11 @@ def run_n_iterations_onnx(model, inputs, n=2): output_csv( output_error_filename, parsed_error.headers, parsed_error.row ) + if context.onnx_model is not None: + e.onnx_program.save_diagnostics( + f"{context.onnx_model.model_dir}/" + f"{current_onnx_compiler}_{current_name}_{current_device}.sarif" + ) # Check also the raw exception that caused export failure. # Skip if it is already analyzed by diagnostics.