saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-07-27 20:02:15 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

add SAVE_TEST_GRAPH macro (#18696)

Browse source 
### Description
<!-- Describe your changes. -->

Add a macro `SAVE_TEST_GRAPH ` in `graph_transform_test_builder.cc`. 


### Motivation and Context
<!-- - Why is this change required? What problem does it solve?
- If it fixes an open issue, please link to the issue here. -->

This will help us debug the graph and Unitest.

Co-authored-by: ruiren <ruiren@microsoft.com>
This commit is contained in:
rui-ren 2023-12-05 07:46:08 -08:00 committed by GitHub
parent 2b3050bb0c
commit c14fae9461
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 11 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

13
onnxruntime/test/optimizer/graph_transform_test_builder.cc
View file

@ -14,6 +14,9 @@
#include "test/util/include/asserts.h"
#include "test/util/include/inference_session_wrapper.h"
// enable to dump model for debugging
#define SAVE_TEST_GRAPH 0
namespace onnxruntime {
namespace test {
@ -73,7 +76,7 @@ void TransformerTester(const std::function<void(ModelTestBuilder& helper)>& buil
std::unique_ptr<GraphTransformer> transformer = nullptr) {
SessionOptions session_options;
session_options.graph_optimization_level = transformer ? baseline_level : level;
#if 0 // enable to dump model for debugging
#if SAVE_TEST_GRAPH
session_options.optimized_model_filepath =
ToPathString("model" + std::to_string(static_cast<int>(level)) + ".onnx");
#endif
@ -156,11 +159,17 @@ Status TestGraphTransformer(const std::function<void(ModelTestBuilder& helper)>&
if (pre_graph_checker) {
ORT_RETURN_IF_ERROR(pre_graph_checker(graph));
}
#if SAVE_TEST_GRAPH
ORT_RETURN_IF_ERROR(Model::Save(model, "model_original.onnx"));
#endif
ORT_RETURN_IF_ERROR(graph_transformation_mgr.ApplyTransformers(graph, level, logger));
if (post_graph_checker) {
ORT_RETURN_IF_ERROR(post_graph_checker(graph));
}
}
#if SAVE_TEST_GRAPH
ORT_RETURN_IF_ERROR(Model::Save(model, "model_optimized.onnx"));
#endif
};
return Status::OK();
}