src/test/bench/rt-parfib.rs

   1 // Copyright 2013 The Rust Project Developers. See the COPYRIGHT
   2 // file at the top-level directory of this distribution and at
   3 // http://rust-lang.org/COPYRIGHT.
   4 //
   5 // Licensed under the Apache License, Version 2.0 <LICENSE-APACHE or
   6 // http://www.apache.org/licenses/LICENSE-2.0> or the MIT license
   7 // <LICENSE-MIT or http://opensource.org/licenses/MIT>, at your
   8 // option. This file may not be copied, modified, or distributed
   9 // except according to those terms.
  10
  11 use std::os;
  12 use std::uint;
  13
  14 // A simple implementation of parfib. One subtree is found in a new
  15 // task and communicated over a oneshot pipe, the other is found
  16 // locally. There is no sequential-mode threshold.
  17
  18 fn parfib(n: uint) -> uint {
  19     if(n == 0 || n == 1) {
  20         return 1;
  21     }
  22
  23     let (tx, rx) = channel();
  24     spawn(proc() {
  25         tx.send(parfib(n-1));
  26     });
  27     let m2 = parfib(n-2);
  28     return (rx.recv() + m2);
  29 }
  30
  31 fn main() {
  32
  33     let args = os::args();
  34     let args = args.as_slice();
  35     let n = if args.len() == 2 {
  36         from_str::<uint>(args[1].as_slice()).unwrap()
  37     } else {
  38         10
  39     };
  40
  41     parfib(n);
  42
  43 }