File: fibonacci.rs

package info (click to toggle)
rustc 1.85.0%2Bdfsg3-1
  • links: PTS, VCS
  • area: main
  • in suites: experimental, sid, trixie
  • size: 893,396 kB
  • sloc: xml: 158,127; python: 35,830; javascript: 19,497; cpp: 19,002; sh: 17,245; ansic: 13,127; asm: 4,376; makefile: 1,051; perl: 29; lisp: 29; ruby: 19; sql: 11
file content (57 lines) | stat: -rw-r--r-- 1,675 bytes parent folder | download | duplicates (7)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
use tracing::{info, instrument};
use tracing_subscriber::prelude::*;

#[instrument]
fn fibonacci(n: usize) -> usize {
    if n < 2 {
        n
    } else {
        fibonacci(n - 1) + fibonacci(n - 2)
    }
}

#[instrument]
fn fibonacci_parallel(n: usize) -> usize {
    if n < 20 {
        fibonacci(n - 1) + fibonacci(n - 2)
    } else {
        let (a, b) = rayon::join(|| fibonacci_parallel(n - 1), || fibonacci_parallel(n - 2));
        if (a + b) > 1000 {
            info!("Big number!");
        }
        a + b
    }
}

fn main() {
    rayon::ThreadPoolBuilder::new()
        .thread_name(|idx| format!("Worker {idx}"))
        .build_global()
        .expect("Failed to build thread pool.");

    let guard = if std::env::args().any(|arg| arg == "--no-trace") {
        None
    } else {
        let (chrome_layer, guard) = tracing_chrome::ChromeLayerBuilder::new()
            .include_args(true)
            .build();
        tracing_subscriber::registry().with(chrome_layer).init();
        Some(guard)
    };

    let before = std::time::Instant::now();
    println!("fibonacci_serial(24) -> {}", fibonacci(24));
    println!("took {} s", before.elapsed().as_secs_f32());
    if let Some(guard) = &guard {
        guard.start_new(None);
    }
    let before = std::time::Instant::now();
    println!("fibonacci_parallel(24) -> {}", fibonacci_parallel(24));
    println!("took {} s", before.elapsed().as_secs_f32());
    if let Some(guard) = &guard {
        guard.start_new(None);
    }
    let before = std::time::Instant::now();
    println!("fibonacci_parallel(20) -> {}", fibonacci_parallel(20));
    println!("took {} s", before.elapsed().as_secs_f32());
}