T0b1/wasmtime
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Merge pull request #3363 from dheaton-arm/implement-widening-pairwise-dotprod

Browse Source 
Implement `WideningPairwiseDotProductS` for interpreter
This commit is contained in:
Chris Fallin
2021-09-21 10:05:07 -07:00
committed by GitHub
parent ebe2af6eaa 8abb19cbd8
commit e0bd4bd007
2 changed files with 34 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
cranelift/interpreter/src/step.rs
View File

@@ -900,7 +900,26 @@ where
Opcode::AtomicLoad => unimplemented!("AtomicLoad"),
Opcode::AtomicStore => unimplemented!("AtomicStore"),
Opcode::Fence => unimplemented!("Fence"),
Opcode::WideningPairwiseDotProductS => unimplemented!("WideningPairwiseDotProductS"),
Opcode::WideningPairwiseDotProductS => {
let ctrl_ty = types::I16X8;
let new_type = ctrl_ty.merge_lanes().unwrap();
let arg0 = extractlanes(&arg(0)?, ctrl_ty.lane_type())?;
let arg1 = extractlanes(&arg(1)?, ctrl_ty.lane_type())?;
let new_vec = arg0
.chunks(2)
.into_iter()
.zip(arg1.chunks(2))
.into_iter()
.map(|(x, y)| {
let mut z = 0i128;
for (lhs, rhs) in x.into_iter().zip(y.into_iter()) {
z += lhs.clone().into_int()? * rhs.clone().into_int()?;
}
Value::int(z, new_type.lane_type())
})
.collect::<ValueResult<Vec<_>>>()?;
assign(vectorizelanes(&new_vec, new_type)?)
}
Opcode::SqmulRoundSat => unimplemented!("SqmulRoundSat"),
Opcode::IaddPairwise => assign(binary_pairwise(arg(0)?, arg(1)?, ctrl_ty, Value::add)?),