Skip to content

Commit dcbc15e

Browse files
authored
Verbose log transform passes (#294)
For debugging with environment variable like `LOG_THRESHOLD=-1` Signed-off-by: take-cheeze <takechi101010@gmail.com>
1 parent 7942e19 commit dcbc15e

1 file changed

Lines changed: 7 additions & 0 deletions

File tree

onnxoptimizer/pass_manager.cc

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -3,6 +3,7 @@
33
// SPDX-License-Identifier: Apache-2.0
44

55
#include "onnxoptimizer/pass_manager.h"
6+
#include "onnxoptimizer/passes/logging.h"
67

78
namespace ONNX_NAMESPACE {
89
namespace optimization {
@@ -36,10 +37,16 @@ std::shared_ptr<PassManagerAnalysis> FixedPointPassManager::run(Graph& graph) {
3637
}
3738
std::shared_ptr<CountBasedPassAnalysis> count_analysis =
3839
std::static_pointer_cast<CountBasedPassAnalysis>(analysis);
40+
if (count_analysis->num_positive_transforms != 0) {
41+
VLOG(1) << "Pass " << pass->getPassName() << " transformed " << count_analysis->num_positive_transforms;
42+
}
3943

4044
while (count_analysis->fixedPointOptimizationNeeded()) {
4145
count_analysis = std::static_pointer_cast<CountBasedPassAnalysis>(
4246
pass->runPass(graph));
47+
if (count_analysis->num_positive_transforms != 0) {
48+
VLOG(1) << "Pass " << pass->getPassName() << " transformed " << count_analysis->num_positive_transforms;
49+
}
4350
fixed_point_optimization_done = true;
4451
}
4552
}

0 commit comments

Comments
 (0)