Replace loops with iterator methods.

Copyright (c) 2021, Arm Limited
This commit is contained in:
dheaton-arm
2021-09-09 10:34:29 +01:00
parent d2cbe4fc30
commit a595bd22e3

View File

@@ -842,24 +842,20 @@ where
)?), )?),
Opcode::SwidenLow => { Opcode::SwidenLow => {
let new_type = ctrl_ty.merge_lanes().unwrap(); let new_type = ctrl_ty.merge_lanes().unwrap();
let mut new_vec = SimdVec::new(); let new_vec = extractlanes(&arg(0)?, ctrl_ty.lane_type())?
let mut arg0 = extractlanes(&arg(0)?, ctrl_ty.lane_type())?; .into_iter()
arg0.truncate(new_type.lane_count() as usize); .take(new_type.lane_count() as usize)
for lane in arg0 { .map(|lane| lane.convert(ValueConversionKind::SignExtend(new_type.lane_type())))
let lane = lane.convert(ValueConversionKind::SignExtend(new_type.lane_type()))?; .collect::<ValueResult<Vec<_>>>()?;
new_vec.push(lane);
}
assign(vectorizelanes(&new_vec, new_type)?) assign(vectorizelanes(&new_vec, new_type)?)
} }
Opcode::SwidenHigh => { Opcode::SwidenHigh => {
let new_type = ctrl_ty.merge_lanes().unwrap(); let new_type = ctrl_ty.merge_lanes().unwrap();
let mut new_vec = SimdVec::new(); let new_vec = extractlanes(&arg(0)?, ctrl_ty.lane_type())?
let mut arg0 = extractlanes(&arg(0)?, ctrl_ty.lane_type())?; .into_iter()
arg0.drain(0..new_type.lane_count() as usize); .skip(new_type.lane_count() as usize)
for lane in arg0 { .map(|lane| lane.convert(ValueConversionKind::SignExtend(new_type.lane_type())))
let lane = lane.convert(ValueConversionKind::SignExtend(new_type.lane_type()))?; .collect::<ValueResult<Vec<_>>>()?;
new_vec.push(lane);
}
assign(vectorizelanes(&new_vec, new_type)?) assign(vectorizelanes(&new_vec, new_type)?)
} }
Opcode::UwidenLow => { Opcode::UwidenLow => {