max/lean4-htt
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 2

feat(library/vm/vm): add profiler.perf_script_file option

Browse source 
sample usage:
$ lean -Dprofiler.perf_script_file=parser1.script parser1.lean
$ gprof2dot -f perf < parser1.script > parser1.dot
This commit is contained in:
Sebastian Ullrich 2018-10-22 14:58:55 +02:00
parent 89afabae29
commit d9208e2b8b
3 changed files with 39 additions and 12 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
src/frontends/lean/builtin_cmds.cpp
View file

@ -474,6 +474,8 @@ static environment eval_cmd(parser & p) {
if (fn.get_profiler().enabled()) {
if (fn.get_profiler().get_snapshots().display("#eval", p.get_options(), out.get_text_stream().get_stream()))
should_report = true;
if (auto filename = p.get_options().get_string({"profiler", "perf_script_file"}))
fn.get_profiler().save_perf_script(filename);
}
};

47
src/library/vm/vm.cpp
View file

@ -3477,6 +3477,39 @@ vm_state::profiler::~profiler() {
stop();
}
name vm_state::profiler::prettify_decl_name(name decl_name) {
/* Remove unnecessary suffixes. */
while (true) {
if (decl_name.is_atomic()) break;
if (!decl_name.is_string()) break;
char const * str = decl_name.get_string().data(); // <<< HACK: processing Lean string as C string
if (str[0] != '_') break;
if (strncmp(str, "_lambda", 7) == 0) break;
decl_name = decl_name.get_prefix();
}
if (auto prv = hidden_to_user_name(m_state.env(), decl_name))
decl_name = *prv;
return decl_name;
}
void vm_state::profiler::save_perf_script(std::string const & filename) {
stop();
std::ofstream out(filename);
second_duration total_time;
for (snapshot_core const & s : m_snapshots) {
total_time += s.m_duration;
out << "lean 0 " << total_time.count() << " " << s.m_duration.count() << " cycles:uppp:\n";
for (int i = s.m_stack.size() - 1; i >= 0; i--) {
auto const & p = s.m_stack[i];
vm_decl const * decl = m_state.m_decl_map.find(p.first);
lean_assert(decl);
name decl_name = prettify_decl_name(decl->get_name());
out << "\t0 " << decl_name << " ()\n";
}
out << "\n";
}
}
auto vm_state::profiler::get_snapshots() -> snapshots {
stop();
snapshots r;
@ -3493,18 +3526,7 @@ auto vm_state::profiler::get_snapshots() -> snapshots {
auto const & p = s.m_stack[i];
vm_decl const * decl = m_state.m_decl_map.find(p.first);
lean_assert(decl);
name decl_name = decl->get_name();
/* Remove unnecessary suffixes. */
while (true) {
if (decl_name.is_atomic()) break;
if (!decl_name.is_string()) break;
char const * str = decl_name.get_string().data(); // <<< HACK: processing Lean string as C string
if (str[0] != '_') break;
if (strncmp(str, "_lambda", 7) == 0) break;
decl_name = decl_name.get_prefix();
}
if (auto prv = hidden_to_user_name(m_state.env(), decl_name))
decl_name = *prv;
name decl_name = prettify_decl_name(decl->get_name());
if (new_stack.empty() || decl_name != new_stack.back().first)
new_stack.emplace_back(decl_name, p.second);
@ -3719,6 +3741,7 @@ void initialize_vm() {
#if defined(LEAN_MULTI_THREAD)
g_profiler_freq = new name{"profiler", "freq"};
register_unsigned_option(*g_profiler_freq, LEAN_DEFAULT_PROFILER_FREQ, "(profiler) sampling frequency in milliseconds");
register_string_option(name({"profiler", "perf_script_file"}), "", "(profiler) save profile in `perf script` format");
#endif
g_debugger = new name{"debugger"};
register_bool_option(*g_debugger, false, "(debugger) debug code using VM monitors");

2
src/library/vm/vm.h
View file

@ -825,6 +825,7 @@ public:
thread_ptr m_thread_ptr;
std::vector<snapshot_core> m_snapshots;
void stop();
name prettify_decl_name(name decl_name);
public:
profiler(vm_state & s, options const & opts);
~profiler();
@ -849,6 +850,7 @@ public:
};
bool enabled() const { return m_thread_ptr.get() != nullptr; }
snapshots get_snapshots();
void save_perf_script(std::string const & filename);
};
/* performance counters */