Move monomorphize code to its own crate.
This commit is contained in:
parent
bba4be681d
commit
81a600b6b7
13 changed files with 72 additions and 23 deletions
73
compiler/rustc_monomorphize/src/util.rs
Normal file
73
compiler/rustc_monomorphize/src/util.rs
Normal file
|
@ -0,0 +1,73 @@
|
|||
use rustc_middle::ty::{self, ClosureSizeProfileData, Instance, TyCtxt};
|
||||
use std::fs::OpenOptions;
|
||||
use std::io::prelude::*;
|
||||
|
||||
/// For a given closure, writes out the data for the profiling the impact of RFC 2229 on
|
||||
/// closure size into a CSV.
|
||||
///
|
||||
/// During the same compile all closures dump the information in the same file
|
||||
/// "closure_profile_XXXXX.csv", which is created in the directory where the compiler is invoked.
|
||||
crate fn dump_closure_profile(tcx: TyCtxt<'tcx>, closure_instance: Instance<'tcx>) {
|
||||
let mut file = if let Ok(file) = OpenOptions::new()
|
||||
.create(true)
|
||||
.append(true)
|
||||
.open(&format!("closure_profile_{}.csv", std::process::id()))
|
||||
{
|
||||
file
|
||||
} else {
|
||||
eprintln!("Cound't open file for writing closure profile");
|
||||
return;
|
||||
};
|
||||
|
||||
let closure_def_id = closure_instance.def_id();
|
||||
let typeck_results = tcx.typeck(closure_def_id.expect_local());
|
||||
|
||||
if typeck_results.closure_size_eval.contains_key(&closure_def_id) {
|
||||
let param_env = ty::ParamEnv::reveal_all();
|
||||
|
||||
let ClosureSizeProfileData { before_feature_tys, after_feature_tys } =
|
||||
typeck_results.closure_size_eval[&closure_def_id];
|
||||
|
||||
let before_feature_tys = tcx.subst_and_normalize_erasing_regions(
|
||||
closure_instance.substs,
|
||||
param_env,
|
||||
before_feature_tys,
|
||||
);
|
||||
let after_feature_tys = tcx.subst_and_normalize_erasing_regions(
|
||||
closure_instance.substs,
|
||||
param_env,
|
||||
after_feature_tys,
|
||||
);
|
||||
|
||||
let new_size = tcx
|
||||
.layout_of(param_env.and(after_feature_tys))
|
||||
.map(|l| format!("{:?}", l.size.bytes()))
|
||||
.unwrap_or_else(|e| format!("Failed {:?}", e));
|
||||
|
||||
let old_size = tcx
|
||||
.layout_of(param_env.and(before_feature_tys))
|
||||
.map(|l| format!("{:?}", l.size.bytes()))
|
||||
.unwrap_or_else(|e| format!("Failed {:?}", e));
|
||||
|
||||
let closure_hir_id = tcx.hir().local_def_id_to_hir_id(closure_def_id.expect_local());
|
||||
let closure_span = tcx.hir().span(closure_hir_id);
|
||||
let src_file = tcx.sess.source_map().span_to_filename(closure_span);
|
||||
let line_nos = tcx
|
||||
.sess
|
||||
.source_map()
|
||||
.span_to_lines(closure_span)
|
||||
.map(|l| format!("{:?} {:?}", l.lines.first(), l.lines.last()))
|
||||
.unwrap_or_else(|e| format!("{:?}", e));
|
||||
|
||||
if let Err(e) = writeln!(
|
||||
file,
|
||||
"{}, {}, {}, {:?}",
|
||||
old_size,
|
||||
new_size,
|
||||
src_file.prefer_local(),
|
||||
line_nos
|
||||
) {
|
||||
eprintln!("Error writting to file {}", e.to_string())
|
||||
}
|
||||
}
|
||||
}
|
Loading…
Add table
Add a link
Reference in a new issue