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

Add HAXPY

Browse Source
This commit is contained in:
Derek Christ
2024-02-22 18:00:45 +01:00
parent 86d0a0d635
commit 680933e0e6
13 changed files with 306 additions and 34 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
pim-os/src/bin/vmul.rs
View File

@@ -10,7 +10,7 @@ use half::f16;
use nalgebra::SVector;
use pim_isa::BankMode;
use pim_os::{
kernel::vadd,
kernel::vmul,
pim::{self, vector::F16x1},
uart::Uart0,
};
@@ -19,7 +19,7 @@ const ROWS: usize = 2048;
#[no_mangle]
pub extern "C" fn main() {
pim::state::set_kernel(&vadd::KERNEL);
pim::state::set_kernel(&vmul::KERNEL);
let a = Box::new(pim::continuous_array::Vector(
SVector::<F16x1, ROWS>::from_fn(|_, _| F16x1(f16::from_f32(2 as _))),
@@ -43,7 +43,7 @@ pub extern "C" fn main() {
{
pim::state::set_bank_mode(BankMode::PimAllBank);
vadd::execute(&a.0, &b.0, &mut c.0, dummy.as_ref());
vmul::execute(&a.0, &b.0, &mut c.0, dummy.as_ref());
pim::state::set_bank_mode(BankMode::SingleBank);
}