forked from M-Labs/libfringe
Move the Context tests and benches into src/
This concludes preparations for making Context private again.
This commit is contained in:
parent
892a7696ec
commit
b1a6b17d0a
|
@ -1,30 +0,0 @@
|
||||||
// This file is part of libfringe, a low-level green threading library.
|
|
||||||
// Copyright (c) edef <edef@edef.eu>,
|
|
||||||
// whitequark <whitequark@whitequark.org>
|
|
||||||
// See the LICENSE file included in this distribution.
|
|
||||||
#![feature(test)]
|
|
||||||
extern crate test;
|
|
||||||
extern crate fringe;
|
|
||||||
|
|
||||||
use fringe::Context;
|
|
||||||
|
|
||||||
static mut ctx_slot: *mut Context<fringe::OsStack> = 0 as *mut Context<_>;
|
|
||||||
|
|
||||||
#[bench]
|
|
||||||
fn swap(b: &mut test::Bencher) {
|
|
||||||
unsafe extern "C" fn loopback(mut arg: usize) -> ! {
|
|
||||||
// This deliberately does not ignore arg, to measure the time it takes
|
|
||||||
// to move the return value between registers.
|
|
||||||
let ctx_ptr = ctx_slot;
|
|
||||||
loop { arg = Context::swap(ctx_ptr, ctx_ptr, arg) }
|
|
||||||
}
|
|
||||||
|
|
||||||
unsafe {
|
|
||||||
let stack = fringe::OsStack::new(4 << 20).unwrap();
|
|
||||||
let mut ctx = Context::new(stack, loopback);
|
|
||||||
ctx_slot = &mut ctx;
|
|
||||||
|
|
||||||
let ctx_ptr = &mut ctx;
|
|
||||||
b.iter(|| Context::swap(ctx_ptr, ctx_ptr, 0));
|
|
||||||
}
|
|
||||||
}
|
|
115
src/context.rs
115
src/context.rs
|
@ -52,3 +52,118 @@ impl<OldStack> Context<OldStack> where OldStack: stack::Stack {
|
||||||
arch::swap(arg, &mut (*old_ctx).stack_ptr, &(*new_ctx).stack_ptr, &(*new_ctx).stack)
|
arch::swap(arg, &mut (*old_ctx).stack_ptr, &(*new_ctx).stack_ptr, &(*new_ctx).stack)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[cfg(test)]
|
||||||
|
mod test {
|
||||||
|
extern crate test;
|
||||||
|
extern crate simd;
|
||||||
|
|
||||||
|
use std::ptr;
|
||||||
|
use super::Context;
|
||||||
|
use ::OsStack;
|
||||||
|
|
||||||
|
#[thread_local]
|
||||||
|
static mut ctx_slot: *mut Context<OsStack> = ptr::null_mut();
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn context() {
|
||||||
|
unsafe extern "C" fn adder(arg: usize) -> ! {
|
||||||
|
println!("it's alive! arg: {}", arg);
|
||||||
|
let arg = Context::swap(ctx_slot, ctx_slot, arg + 1);
|
||||||
|
println!("still alive! arg: {}", arg);
|
||||||
|
Context::swap(ctx_slot, ctx_slot, arg + 1);
|
||||||
|
panic!("i should be dead");
|
||||||
|
}
|
||||||
|
|
||||||
|
unsafe {
|
||||||
|
let stack = OsStack::new(4 << 20).unwrap();
|
||||||
|
let mut ctx = Context::new(stack, adder);
|
||||||
|
ctx_slot = &mut ctx;
|
||||||
|
|
||||||
|
let ret = Context::swap(ctx_slot, ctx_slot, 10);
|
||||||
|
assert_eq!(ret, 11);
|
||||||
|
let ret = Context::swap(ctx_slot, ctx_slot, 50);
|
||||||
|
assert_eq!(ret, 51);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn context_simd() {
|
||||||
|
unsafe extern "C" fn permuter(arg: usize) -> ! {
|
||||||
|
// This will crash if the stack is not aligned properly.
|
||||||
|
let x = simd::i32x4::splat(arg as i32);
|
||||||
|
let y = x * x;
|
||||||
|
println!("simd result: {:?}", y);
|
||||||
|
Context::swap(ctx_slot, ctx_slot, 0);
|
||||||
|
// And try again after a context switch.
|
||||||
|
let x = simd::i32x4::splat(arg as i32);
|
||||||
|
let y = x * x;
|
||||||
|
println!("simd result: {:?}", y);
|
||||||
|
Context::swap(ctx_slot, ctx_slot, 0);
|
||||||
|
panic!("i should be dead");
|
||||||
|
}
|
||||||
|
|
||||||
|
unsafe {
|
||||||
|
let stack = OsStack::new(4 << 20).unwrap();
|
||||||
|
let mut ctx = Context::new(stack, permuter);
|
||||||
|
ctx_slot = &mut ctx;
|
||||||
|
|
||||||
|
Context::swap(ctx_slot, ctx_slot, 10);
|
||||||
|
Context::swap(ctx_slot, ctx_slot, 20);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
unsafe extern "C" fn do_panic(arg: usize) -> ! {
|
||||||
|
match arg {
|
||||||
|
0 => panic!("arg=0"),
|
||||||
|
1 => {
|
||||||
|
Context::swap(ctx_slot, ctx_slot, 0);
|
||||||
|
panic!("arg=1");
|
||||||
|
}
|
||||||
|
_ => unreachable!()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
#[should_panic="arg=0"]
|
||||||
|
fn panic_after_start() {
|
||||||
|
unsafe {
|
||||||
|
let stack = OsStack::new(4 << 20).unwrap();
|
||||||
|
let mut ctx = Context::new(stack, do_panic);
|
||||||
|
|
||||||
|
Context::swap(&mut ctx, &ctx, 0);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
#[should_panic="arg=1"]
|
||||||
|
fn panic_after_swap() {
|
||||||
|
unsafe {
|
||||||
|
let stack = OsStack::new(4 << 20).unwrap();
|
||||||
|
let mut ctx = Context::new(stack, do_panic);
|
||||||
|
ctx_slot = &mut ctx;
|
||||||
|
|
||||||
|
Context::swap(&mut ctx, &ctx, 1);
|
||||||
|
Context::swap(&mut ctx, &ctx, 0);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[bench]
|
||||||
|
fn swap(b: &mut test::Bencher) {
|
||||||
|
unsafe extern "C" fn loopback(mut arg: usize) -> ! {
|
||||||
|
// This deliberately does not ignore arg, to measure the time it takes
|
||||||
|
// to move the return value between registers.
|
||||||
|
let ctx_ptr = ctx_slot;
|
||||||
|
loop { arg = Context::swap(ctx_ptr, ctx_ptr, arg) }
|
||||||
|
}
|
||||||
|
|
||||||
|
unsafe {
|
||||||
|
let stack = OsStack::new(4 << 20).unwrap();
|
||||||
|
let mut ctx = Context::new(stack, loopback);
|
||||||
|
ctx_slot = &mut ctx;
|
||||||
|
|
||||||
|
let ctx_ptr = &mut ctx;
|
||||||
|
b.iter(|| Context::swap(ctx_ptr, ctx_ptr, 0));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
|
@ -2,6 +2,7 @@
|
||||||
// Copyright (c) edef <edef@edef.eu>
|
// Copyright (c) edef <edef@edef.eu>
|
||||||
// See the LICENSE file included in this distribution.
|
// See the LICENSE file included in this distribution.
|
||||||
#![feature(asm)]
|
#![feature(asm)]
|
||||||
|
#![cfg_attr(test, feature(test, thread_local, const_fn))]
|
||||||
#![cfg_attr(target_arch = "x86", feature(naked_functions, core_intrinsics))]
|
#![cfg_attr(target_arch = "x86", feature(naked_functions, core_intrinsics))]
|
||||||
#![cfg_attr(target_arch = "x86_64", feature(naked_functions, core_intrinsics))]
|
#![cfg_attr(target_arch = "x86_64", feature(naked_functions, core_intrinsics))]
|
||||||
#![no_std]
|
#![no_std]
|
||||||
|
|
|
@ -1,60 +0,0 @@
|
||||||
// This file is part of libfringe, a low-level green threading library.
|
|
||||||
// Copyright (c) edef <edef@edef.eu>,
|
|
||||||
// whitequark <whitequark@whitequark.org>
|
|
||||||
// See the LICENSE file included in this distribution.
|
|
||||||
#![feature(thread_local)]
|
|
||||||
extern crate simd;
|
|
||||||
extern crate fringe;
|
|
||||||
|
|
||||||
use fringe::Context;
|
|
||||||
|
|
||||||
#[thread_local]
|
|
||||||
static mut ctx_slot: *mut Context<fringe::OsStack> = 0 as *mut Context<_>;
|
|
||||||
|
|
||||||
#[test]
|
|
||||||
fn context() {
|
|
||||||
unsafe extern "C" fn adder(arg: usize) -> ! {
|
|
||||||
println!("it's alive! arg: {}", arg);
|
|
||||||
let arg = Context::swap(ctx_slot, ctx_slot, arg + 1);
|
|
||||||
println!("still alive! arg: {}", arg);
|
|
||||||
Context::swap(ctx_slot, ctx_slot, arg + 1);
|
|
||||||
panic!("i should be dead");
|
|
||||||
}
|
|
||||||
|
|
||||||
unsafe {
|
|
||||||
let stack = fringe::OsStack::new(4 << 20).unwrap();
|
|
||||||
let mut ctx = Context::new(stack, adder);
|
|
||||||
ctx_slot = &mut ctx;
|
|
||||||
|
|
||||||
let ret = Context::swap(ctx_slot, ctx_slot, 10);
|
|
||||||
assert_eq!(ret, 11);
|
|
||||||
let ret = Context::swap(ctx_slot, ctx_slot, 50);
|
|
||||||
assert_eq!(ret, 51);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
#[test]
|
|
||||||
fn context_simd() {
|
|
||||||
unsafe extern "C" fn permuter(arg: usize) -> ! {
|
|
||||||
// This will crash if the stack is not aligned properly.
|
|
||||||
let x = simd::i32x4::splat(arg as i32);
|
|
||||||
let y = x * x;
|
|
||||||
println!("simd result: {:?}", y);
|
|
||||||
Context::swap(ctx_slot, ctx_slot, 0);
|
|
||||||
// And try again after a context switch.
|
|
||||||
let x = simd::i32x4::splat(arg as i32);
|
|
||||||
let y = x * x;
|
|
||||||
println!("simd result: {:?}", y);
|
|
||||||
Context::swap(ctx_slot, ctx_slot, 0);
|
|
||||||
panic!("i should be dead");
|
|
||||||
}
|
|
||||||
|
|
||||||
unsafe {
|
|
||||||
let stack = fringe::OsStack::new(4 << 20).unwrap();
|
|
||||||
let mut ctx = Context::new(stack, permuter);
|
|
||||||
ctx_slot = &mut ctx;
|
|
||||||
|
|
||||||
Context::swap(ctx_slot, ctx_slot, 10);
|
|
||||||
Context::swap(ctx_slot, ctx_slot, 20);
|
|
||||||
}
|
|
||||||
}
|
|
|
@ -1,45 +0,0 @@
|
||||||
// This file is part of libfringe, a low-level green threading library.
|
|
||||||
// Copyright (c) whitequark <whitequark@whitequark.org>
|
|
||||||
// See the LICENSE file included in this distribution.
|
|
||||||
#![feature(thread_local)]
|
|
||||||
extern crate fringe;
|
|
||||||
|
|
||||||
use fringe::Context;
|
|
||||||
|
|
||||||
#[thread_local]
|
|
||||||
static mut ctx_slot: *mut Context<fringe::OsStack> = 0 as *mut Context<_>;
|
|
||||||
|
|
||||||
unsafe extern "C" fn do_panic(arg: usize) -> ! {
|
|
||||||
match arg {
|
|
||||||
0 => panic!("arg=0"),
|
|
||||||
1 => {
|
|
||||||
Context::swap(ctx_slot, ctx_slot, 0);
|
|
||||||
panic!("arg=1");
|
|
||||||
}
|
|
||||||
_ => unreachable!()
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
#[test]
|
|
||||||
#[should_panic="arg=0"]
|
|
||||||
fn panic_after_start() {
|
|
||||||
unsafe {
|
|
||||||
let stack = fringe::OsStack::new(4 << 20).unwrap();
|
|
||||||
let mut ctx = Context::new(stack, do_panic);
|
|
||||||
|
|
||||||
Context::swap(&mut ctx, &ctx, 0);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
#[test]
|
|
||||||
#[should_panic="arg=1"]
|
|
||||||
fn panic_after_swap() {
|
|
||||||
unsafe {
|
|
||||||
let stack = fringe::OsStack::new(4 << 20).unwrap();
|
|
||||||
let mut ctx = Context::new(stack, do_panic);
|
|
||||||
ctx_slot = &mut ctx;
|
|
||||||
|
|
||||||
Context::swap(&mut ctx, &ctx, 1);
|
|
||||||
Context::swap(&mut ctx, &ctx, 0);
|
|
||||||
}
|
|
||||||
}
|
|
Loading…
Reference in New Issue