magnus167/rustframe
1
0
Fork 0
mirror of https://github.com/Magnus167/rustframe.git synced 2025-08-21 03:49:59 +00:00
Code Issues Packages Projects Releases Wiki Activity

Compare commits

base: magnus167:f30b51b4b894eee31c45beea5f9688951c2ec266
Branches Tags
magnus167:main
magnus167:csv
magnus167:ci-update
magnus167:dataframe
magnus167:v0.0.1-a.20250805
magnus167:v0.0.1-a.20250716
..
compare: magnus167:8484661fc427f0288a686813005d4c7b1f6a8cfa
Branches Tags
magnus167:csv
magnus167:ci-update
magnus167:main
magnus167:dataframe
magnus167:v0.0.1-a.20250805
magnus167:v0.0.1-a.20250716

1 Commits
f30b51b4b8 ... 8484661fc4

Author SHA1 Message Date
Palash Tyagi
8484661fc4
Merge e61494edc18712fde6a0b25e3ef3ddc3cd6fa7ca into aeaf93638050524c8251e0a6d3a2a6913753d1c8 2025-05-03 01:31:39 +01:00
1 changed files with 31 additions and 74 deletions
Show Stats Expand all files Collapse all files

103
benches/benchmarks.rs
View File

@ -1,15 +1,14 @@
// Combined benchmarks for rustframe // Combined benchmarks for rustframe
use chrono::NaiveDate;
use criterion::{criterion_group, criterion_main, Criterion}; use criterion::{criterion_group, criterion_main, Criterion};
use rustframe::{ use rustframe::{
frame::{Frame, RowIndex}, frame::{Frame, RowIndex},
matrix::{BoolMatrix, Matrix}, matrix::{BoolMatrix, Matrix},
utils::{BDateFreq, BDatesList}, utils::{BDateFreq, BDatesList},
}; };
use chrono::NaiveDate;
fn bool_matrix_operations_benchmark(c: &mut Criterion) { fn bool_matrix_operations_benchmark(c: &mut Criterion) {
// let sizes = [1, 100, 1000]; let sizes = [1, 100, 1000];
let sizes = [1000];
for &size in &sizes { for &size in &sizes {
let data1: Vec<bool> = (0..size * size).map(|x| x % 2 == 0).collect(); let data1: Vec<bool> = (0..size * size).map(|x| x % 2 == 0).collect();
@ -44,8 +43,7 @@ fn bool_matrix_operations_benchmark(c: &mut Criterion) {
} }
fn matrix_boolean_operations_benchmark(c: &mut Criterion) { fn matrix_boolean_operations_benchmark(c: &mut Criterion) {
// let sizes = [1, 100, 1000]; let sizes = [1, 100, 1000];
let sizes = [1000];
for &size in &sizes { for &size in &sizes {
let data1: Vec<bool> = (0..size * size).map(|x| x % 2 == 0).collect(); let data1: Vec<bool> = (0..size * size).map(|x| x % 2 == 0).collect();
@ -80,8 +78,32 @@ fn matrix_boolean_operations_benchmark(c: &mut Criterion) {
} }
fn matrix_operations_benchmark(c: &mut Criterion) { fn matrix_operations_benchmark(c: &mut Criterion) {
// let sizes = [1, 100, 1000]; let n_periods = 4;
let sizes = [1000]; let dates: Vec<NaiveDate> =
BDatesList::from_n_periods("2024-01-02".to_string(), BDateFreq::Daily, n_periods)
.unwrap()
.list()
.unwrap();
let col_names: Vec<String> = vec!["a".to_string(), "b".to_string()];
let ma = Matrix::from_cols(vec![vec![1.0, 2.0, 3.0, 4.0], vec![5.0, 6.0, 7.0, 8.0]]);
let mb = Matrix::from_cols(vec![vec![4.0, 3.0, 2.0, 1.0], vec![8.0, 7.0, 6.0, 5.0]]);
let fa = Frame::new(
ma.clone(),
col_names.clone(),
Some(RowIndex::Date(dates.clone())),
);
let fb = Frame::new(mb, col_names, Some(RowIndex::Date(dates)));
c.bench_function("element-wise multiply", |b| {
b.iter(|| {
let _result = &fa * &fb;
});
});
let sizes = [1, 100, 1000];
for &size in &sizes { for &size in &sizes {
let data: Vec<f64> = (0..size * size).map(|x| x as f64).collect(); let data: Vec<f64> = (0..size * size).map(|x| x as f64).collect();
@ -111,72 +133,7 @@ fn matrix_operations_benchmark(c: &mut Criterion) {
}); });
}); });
} }
// Benchmarking matrix addition
for &size in &sizes {
let data1: Vec<f64> = (0..size * size).map(|x| x as f64).collect();
let data2: Vec<f64> = (0..size * size).map(|x| (x + 1) as f64).collect();
let ma = Matrix::from_vec(data1.clone(), size, size);
let mb = Matrix::from_vec(data2.clone(), size, size);
c.bench_function(&format!("matrix add ({}x{})", size, size), |b| {
b.iter(|| {
let _result = &ma + &mb;
});
});
c.bench_function(&format!("matrix subtract ({}x{})", size, size), |b| {
b.iter(|| {
let _result = &ma - &mb;
});
});
c.bench_function(&format!("matrix multiply ({}x{})", size, size), |b| {
b.iter(|| {
let _result = &ma * &mb;
});
});
c.bench_function(&format!("matrix divide ({}x{})", size, size), |b| {
b.iter(|| {
let _result = &ma / &mb;
});
});
}
} }
fn benchmark_frame_operations(c: &mut Criterion) { criterion_group!(combined_benches, bool_matrix_operations_benchmark, matrix_boolean_operations_benchmark, matrix_operations_benchmark);
let n_periods = 4;
let dates: Vec<NaiveDate> =
BDatesList::from_n_periods("2024-01-02".to_string(), BDateFreq::Daily, n_periods)
.unwrap()
.list()
.unwrap();
let col_names: Vec<String> = vec!["a".to_string(), "b".to_string()];
let ma = Matrix::from_cols(vec![vec![1.0, 2.0, 3.0, 4.0], vec![5.0, 6.0, 7.0, 8.0]]);
let mb = Matrix::from_cols(vec![vec![4.0, 3.0, 2.0, 1.0], vec![8.0, 7.0, 6.0, 5.0]]);
let fa = Frame::new(
ma.clone(),
col_names.clone(),
Some(RowIndex::Date(dates.clone())),
);
let fb = Frame::new(mb, col_names, Some(RowIndex::Date(dates)));
c.bench_function("frame element-wise multiply", |b| {
b.iter(|| {
let _result = &fa * &fb;
});
});
}
criterion_group!(
combined_benches,
bool_matrix_operations_benchmark,
matrix_boolean_operations_benchmark,
matrix_operations_benchmark,
benchmark_frame_operations
);
criterion_main!(combined_benches); criterion_main!(combined_benches);