redstone-ml 0.0.0

High-performance Machine Learning, Auto-Differentiation and Tensor Algebra crate for Rust
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23

use std::env;
use std::hint::black_box;
use std::time::Instant;
// use cpu_time::ProcessTime;

pub fn profile_func(func: impl Fn()) {
    let args: Vec<String> = env::args().collect();

    let trials = args[2].parse::<usize>().unwrap();
    let warmup = args[3].parse::<usize>().unwrap();

    for _ in 0..warmup {
        func();
    }

    for _ in 0..trials {
        let start = Instant::now();
        // let start = ProcessTime::now();
        black_box(func());
        // println!("{}", start.elapsed().as_nanos());
        println!("{}", start.elapsed().as_nanos());
    }
}