mirror of
https://github.com/saymrwulf/onnxruntime.git
synced 2026-07-28 20:11:22 +00:00
891b50cc68
### Description Adds logging to `GraphTransformer::Apply` whether modification has taken place or not. ### Motivation and Context A general high level info logging to track which optimization occurred for a given model. To help improve dynamo exported model performance by monitoring the difference of triggered transformations between that of torchscript exported model.
29 lines
1,012 B
C++
29 lines
1,012 B
C++
// Copyright (c) Microsoft Corporation. All rights reserved.
|
|
// Licensed under the MIT License.
|
|
|
|
#include "core/optimizer/graph_transformer.h"
|
|
|
|
using namespace ::onnxruntime::common;
|
|
|
|
namespace onnxruntime {
|
|
|
|
Status GraphTransformer::Apply(Graph& graph, bool& modified, const logging::Logger& logger) const {
|
|
// the Graph should be in a good state prior this being called, so there should be no need to call Resolve here
|
|
// ORT_RETURN_IF_ERROR(graph.Resolve());
|
|
|
|
auto status = ApplyImpl(graph, modified, 0, logger);
|
|
LOGS(logger, INFO) << "GraphTransformer " << Name() << " modified: " << modified << " with status: " << status;
|
|
ORT_RETURN_IF_ERROR(status);
|
|
|
|
#if !defined(ORT_MINIMAL_BUILD)
|
|
// At least currently, some transformers (InsertCastTransformer and MemcpyTransformer) need this to be called
|
|
// after they complete to put the graph back into a valid state for the next transformer.
|
|
if (modified) {
|
|
status = graph.Resolve();
|
|
}
|
|
#endif
|
|
|
|
return status;
|
|
}
|
|
|
|
} // namespace onnxruntime
|