Added missing example file

This commit is contained in:
Saurabh 2022-02-26 18:24:24 -07:00
parent 0d2437b5c8
commit 46a757fc42
1 changed files with 25 additions and 0 deletions

View File

@ -0,0 +1,25 @@
extern crate nalgebra_sparse;
use nalgebra_sparse::CsrMatrix;
use std::time::{Duration, Instant};
#[cfg(feature = "io")]
use nalgebra_sparse::io::load_coo_from_matrix_market_file;
fn main() {
#[cfg(feature = "io")]
{
let sparse_input_matrix =
load_coo_from_matrix_market_file::<f64, _>("./data/crankseg_1.mtx").unwrap();
let sparse_input_matrix = CsrMatrix::from(&sparse_input_matrix);
let spmm_result = &sparse_input_matrix * &sparse_input_matrix;
let now = Instant::now();
let spmm_result = &sparse_input_matrix * &sparse_input_matrix;
let spmm_time = now.elapsed().as_millis();
println!("SGEMM time was {}", spmm_time);
let sum: f64 = spmm_result.triplet_iter().map(|(_, _, v)| v).sum();
println!("sum of product is {}", sum);
}
#[cfg(not(feature = "io"))]
{
panic!("Run with IO feature only");
}
}