derek/pim-rs
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Use gem5 ops ISA method instead of ADDR method

Browse Source
This commit is contained in:
Derek Christ
2024-01-07 21:46:37 +01:00
parent 44f38fb443
commit ec8ae87aaa
7 changed files with 160 additions and 26 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
pim-os/src/main.rs
View File

@@ -10,6 +10,7 @@ use half::f16;
use nalgebra::Matrix;
use pim_isa::BankMode;
use pim_os::{
m5op,
pim::{
self,
array::{DummyArray, PimMatrixArena, PimStorage, NUMBER_OF_BANKS},
@@ -65,12 +66,18 @@ pub extern "C" fn main() {
pim::state::set_bank_mode(BankMode::PimAllBank);
matrix_matrix_mul::execute(
pim_matrix_arena0,
pim_matrix_arena1,
pim_matrix_arena2,
dummy_array.as_ref(),
);
m5op::reset_stats();
for _ in 0..100 {
matrix_matrix_mul::execute(
pim_matrix_arena0,
pim_matrix_arena1,
pim_matrix_arena2,
dummy_array.as_ref(),
);
}
m5op::dump_stats();
pim::state::set_bank_mode(BankMode::SingleBank);
}