summaryrefslogtreecommitdiffstats
path: root/tests/run-make/pgo-use/main.rs
blob: eb9192c87e6f05b213bb62781d253119190e6d6a (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
#[no_mangle]
pub fn cold_function(c: u8) {
    println!("cold {}", c);
}

#[no_mangle]
pub fn hot_function(c: u8) {
    std::env::set_var(format!("var{}", c), format!("hot {}", c));
}

fn main() {
    let arg = std::env::args().skip(1).next().unwrap();

    for i in 0 .. 1000_000 {
        let some_value = arg.as_bytes()[i % arg.len()];
        if some_value == b'!' {
            // This branch is never taken at runtime
            cold_function(some_value);
        } else {
            hot_function(some_value);
        }
    }
}