未验证 提交 29a6b8c9 编写于 作者: A Allen Guo 提交者: GitHub

update CompilationProgressLogger (#42665)

上级 000edfd2
...@@ -32,6 +32,20 @@ void RegisterGetter( ...@@ -32,6 +32,20 @@ void RegisterGetter(
options_type[name] = type_str; options_type[name] = type_str;
} }
struct DefaultCompilationProgressLogger {
void operator()(int progress, int total) {
if (progress != progress_ && progress % log_interval_ == 0) {
progress_ = progress;
VLOG(1) << "Graph compile progress: " << progress << "%";
}
}
int log_interval_ = 10;
int progress_ = 0;
// default total progress
int total_ = 100;
};
} // namespace } // namespace
namespace paddle { namespace paddle {
...@@ -417,11 +431,7 @@ IpuStrategy::IpuStrategy() { ...@@ -417,11 +431,7 @@ IpuStrategy::IpuStrategy() {
// Default options // Default options
// Can also be set as a custom logger in python, like using tqdm // Can also be set as a custom logger in python, like using tqdm
popart_options.compilationProgressLogger = [](int progress, int total) { popart_options.compilationProgressLogger = DefaultCompilationProgressLogger();
if (progress % 10 == 0) {
VLOG(1) << "compile progress: " << progress << "%";
}
};
} }
void IpuStrategy::AddBoolOption(const std::string& option, bool value) { void IpuStrategy::AddBoolOption(const std::string& option, bool value) {
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册