Skip to content

Commit feacb59

Browse files
committed
auto merge of #11841 : huonw/rust/noinline-fail, r=alexcrichton
In two ways: - for a plain `fail!(a)` we make the generic part of `begin_unwind` as small as possible (makes `fn main() { fail!() }` compile 2-3x faster, due to less monomorphisation bloat) - for `fail!("format {}", "string")`, we avoid touching the generics completely by doing the formatting in a specialised function, which (with optimisations) saves a function call at the call-site of `fail!`. (This one has significantly less benefit than the first.)
2 parents d6d7812 + b4bb8c0 commit feacb59

File tree

3 files changed

+43
-6
lines changed

3 files changed

+43
-6
lines changed

src/libstd/macros.rs

+12-2
Original file line numberDiff line numberDiff line change
@@ -48,8 +48,18 @@ macro_rules! fail(
4848
::std::rt::begin_unwind($msg, file!(), line!())
4949
);
5050
($fmt:expr, $($arg:tt)*) => (
51-
::std::rt::begin_unwind(format!($fmt, $($arg)*), file!(), line!())
52-
)
51+
{
52+
// a closure can't have return type !, so we need a full
53+
// function to pass to format_args!, *and* we need the
54+
// file and line numbers right here; so an inner bare fn
55+
// is our only choice.
56+
#[inline]
57+
fn run_fmt(fmt: &::std::fmt::Arguments) -> ! {
58+
::std::rt::begin_unwind_fmt(fmt, file!(), line!())
59+
}
60+
format_args!(run_fmt, $fmt, $($arg)*)
61+
}
62+
)
5363
)
5464

5565
#[macro_export]

src/libstd/rt/mod.rs

+1-1
Original file line numberDiff line numberDiff line change
@@ -69,7 +69,7 @@ use self::task::{Task, BlockedTask};
6969
pub use self::util::default_sched_threads;
7070

7171
// Export unwinding facilities used by the failure macros
72-
pub use self::unwind::{begin_unwind, begin_unwind_raw};
72+
pub use self::unwind::{begin_unwind, begin_unwind_raw, begin_unwind_fmt};
7373

7474
// FIXME: these probably shouldn't be public...
7575
#[doc(hidden)]

src/libstd/rt/unwind.rs

+30-3
Original file line numberDiff line numberDiff line change
@@ -58,6 +58,7 @@
5858
use any::{Any, AnyRefExt};
5959
use c_str::CString;
6060
use cast;
61+
use fmt;
6162
use kinds::Send;
6263
use option::{Some, None, Option};
6364
use prelude::drop;
@@ -382,17 +383,43 @@ pub fn begin_unwind_raw(msg: *u8, file: *u8, line: uint) -> ! {
382383
begin_unwind(msg, file, line as uint)
383384
}
384385

386+
/// The entry point for unwinding with a formatted message.
387+
///
388+
/// This is designed to reduce the amount of code required at the call
389+
/// site as much as possible (so that `fail!()` has as low an implact
390+
/// on (e.g.) the inlining of other functions as possible), by moving
391+
/// the actual formatting into this shared place.
392+
#[inline(never)] #[cold]
393+
pub fn begin_unwind_fmt(msg: &fmt::Arguments, file: &'static str, line: uint) -> ! {
394+
begin_unwind_inner(~fmt::format(msg), file, line)
395+
}
396+
385397
/// This is the entry point of unwinding for fail!() and assert!().
386-
#[inline(never)] #[cold] // this is the slow path, please never inline this
398+
#[inline(never)] #[cold] // avoid code bloat at the call sites as much as possible
387399
pub fn begin_unwind<M: Any + Send>(msg: M, file: &'static str, line: uint) -> ! {
388-
// Note that this should be the only allocation performed in this block.
400+
// Note that this should be the only allocation performed in this code path.
389401
// Currently this means that fail!() on OOM will invoke this code path,
390402
// but then again we're not really ready for failing on OOM anyway. If
391403
// we do start doing this, then we should propagate this allocation to
392404
// be performed in the parent of this task instead of the task that's
393405
// failing.
394-
let msg = ~msg as ~Any;
395406

407+
// see below for why we do the `Any` coercion here.
408+
begin_unwind_inner(~msg, file, line)
409+
}
410+
411+
412+
/// The core of the unwinding.
413+
///
414+
/// This is non-generic to avoid instantiation bloat in other crates
415+
/// (which makes compilation of small crates noticably slower). (Note:
416+
/// we need the `Any` object anyway, we're not just creating it to
417+
/// avoid being generic.)
418+
///
419+
/// Do this split took the LLVM IR line counts of `fn main() { fail!()
420+
/// }` from ~1900/3700 (-O/no opts) to 180/590.
421+
#[inline(never)] #[cold] // this is the slow path, please never inline this
422+
fn begin_unwind_inner(msg: ~Any, file: &'static str, line: uint) -> ! {
396423
let mut task;
397424
{
398425
let msg_s = match msg.as_ref::<&'static str>() {

0 commit comments

Comments
 (0)