d7755701ad
This reduces the peak RSS for a cut-down version of the program in #36799 by 10%, from 951MB to 856MB.
73 lines
1.9 KiB
Rust
73 lines
1.9 KiB
Rust
// Copyright 2015 The Rust Project Developers. See the COPYRIGHT
|
|
// file at the top-level directory of this distribution and at
|
|
// http://rust-lang.org/COPYRIGHT.
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 <LICENSE-APACHE or
|
|
// http://www.apache.org/licenses/LICENSE-2.0> or the MIT license
|
|
// <LICENSE-MIT or http://opensource.org/licenses/MIT>, at your
|
|
// option. This file may not be copied, modified, or distributed
|
|
// except according to those terms.
|
|
|
|
//! This pass just dumps MIR at a specified point.
|
|
|
|
use std::fmt;
|
|
|
|
use rustc::ty::TyCtxt;
|
|
use rustc::mir::*;
|
|
use rustc::mir::transform::{Pass, MirPass, MirPassHook, MirSource};
|
|
use pretty;
|
|
|
|
pub struct Marker<'a>(pub &'a str);
|
|
|
|
impl<'b, 'tcx> MirPass<'tcx> for Marker<'b> {
|
|
fn run_pass<'a>(&mut self, _tcx: TyCtxt<'a, 'tcx, 'tcx>,
|
|
_src: MirSource, _mir: &mut Mir<'tcx>)
|
|
{}
|
|
}
|
|
|
|
impl<'b> Pass for Marker<'b> {
|
|
fn name(&self) -> ::std::borrow::Cow<'static, str> { String::from(self.0).into() }
|
|
}
|
|
|
|
pub struct Disambiguator<'a> {
|
|
pass: &'a Pass,
|
|
is_after: bool
|
|
}
|
|
|
|
impl<'a> fmt::Display for Disambiguator<'a> {
|
|
fn fmt(&self, formatter: &mut fmt::Formatter) -> fmt::Result {
|
|
let title = if self.is_after { "after" } else { "before" };
|
|
if let Some(fmt) = self.pass.disambiguator() {
|
|
write!(formatter, "{}-{}", fmt, title)
|
|
} else {
|
|
write!(formatter, "{}", title)
|
|
}
|
|
}
|
|
}
|
|
|
|
pub struct DumpMir;
|
|
|
|
impl<'tcx> MirPassHook<'tcx> for DumpMir {
|
|
fn on_mir_pass<'a>(
|
|
&mut self,
|
|
tcx: TyCtxt<'a, 'tcx, 'tcx>,
|
|
src: MirSource,
|
|
mir: &Mir<'tcx>,
|
|
pass: &Pass,
|
|
is_after: bool)
|
|
{
|
|
pretty::dump_mir(
|
|
tcx,
|
|
&*pass.name(),
|
|
&Disambiguator {
|
|
pass: pass,
|
|
is_after: is_after
|
|
},
|
|
src,
|
|
mir
|
|
);
|
|
}
|
|
}
|
|
|
|
impl<'b> Pass for DumpMir {}
|