MCAI 基准测试

基于配置的基准测试 MCAI 工作的 Crate

lib.rs:

MCAI 基准测试

此 Crate 允许根据配置文件基准测试 MCAI 工作者。

配置

这是一个示例配置文件。

version: 1
output_folder: "./output_folder"
worker_docker_image: "image_name"

my_benchmark_name:
  source_order: "source_order_filename"
  iterations: 5
  hardware:
  cpu:
    - 1
    - 2
  memory:
    - 2048
  envs:
    EXAMPLE_VAR: "my_example_var"
  volumes:
    - host: /path/to/a/volume
      container: /volume
      readonly: true

用法

use mcai_benchmark::{
    benchmark::result::Results,
    configuration::{Configuration, Version1},
    stats::ConsoleWriter,
};
use std::sync::{Arc, Mutex};

#[tokio::main]
async fn main() {
    let config = Version1::read_from_file("./examples/sample_config.yaml").unwrap();
    let mut benchmark = Results::from(Configuration::Version1(config));

    if let Err(error) = benchmark
        .run_benchmark(Some(Arc::new(Mutex::new(ConsoleWriter {}))))
        .await
    {
        println!("Error: {:?}", error);
    }

    println!("{:?}", benchmark);
}