use rustc::ty::subst::SubstsRef;
-macro_rules! intrinsic_pat {
+macro intrinsic_pat {
(_) => {
_
- };
+ },
($name:ident) => {
stringify!($name)
+ },
+ ($name:literal) => {
+ stringify!($name)
+ },
+ ($x:ident . $($xs:tt).*) => {
+ concat!(stringify!($x), ".", intrinsic_pat!($($xs).*))
}
}
-macro_rules! intrinsic_arg {
+macro intrinsic_arg {
(o $fx:expr, $arg:ident) => {
$arg
- };
+ },
(c $fx:expr, $arg:ident) => {
trans_operand($fx, $arg)
- };
+ },
(v $fx:expr, $arg:ident) => {
trans_operand($fx, $arg).load_scalar($fx)
- };
+ }
}
-macro_rules! intrinsic_substs {
- ($substs:expr, $index:expr,) => {};
+macro intrinsic_substs {
+ ($substs:expr, $index:expr,) => {},
($substs:expr, $index:expr, $first:ident $(,$rest:ident)*) => {
let $first = $substs.type_at($index);
intrinsic_substs!($substs, $index+1, $($rest),*);
- };
+ }
}
-macro_rules! intrinsic_match {
- ($fx:expr, $intrinsic:expr, $substs:expr, $args:expr, $(
- $($name:tt)|+ $(if $cond:expr)?, $(<$($subst:ident),*>)? ($($a:ident $arg:ident),*) $content:block;
+pub macro intrinsic_match {
+ ($fx:expr, $intrinsic:expr, $substs:expr, $args:expr,
+ _ => $unknown:block;
+ $(
+ $($($name:tt).*)|+ $(if $cond:expr)?, $(<$($subst:ident),*>)? ($($a:ident $arg:ident),*) $content:block;
)*) => {
match $intrinsic {
$(
- $(intrinsic_pat!($name))|* $(if $cond)? => {
+ $(intrinsic_pat!($($name).*))|* $(if $cond)? => {
#[allow(unused_parens, non_snake_case)]
{
$(
}
}
)*
- _ => unimpl!("unsupported intrinsic {}", $intrinsic),
+ _ => $unknown,
}
- };
+ }
}
macro_rules! call_intrinsic_match {
};
}
-fn lane_type_and_count<'tcx>(
+pub fn lane_type_and_count<'tcx>(
fx: &FunctionCx<'_, 'tcx, impl Backend>,
layout: TyLayout<'tcx>,
intrinsic: &str,
(lane_layout, lane_count)
}
-fn simd_for_each_lane<'tcx, B: Backend>(
+pub fn simd_for_each_lane<'tcx, B: Backend>(
fx: &mut FunctionCx<'_, 'tcx, B>,
intrinsic: &str,
x: CValue<'tcx>,
}
}
-fn bool_to_zero_or_max_uint<'tcx>(
+pub fn bool_to_zero_or_max_uint<'tcx>(
fx: &mut FunctionCx<'_, 'tcx, impl Backend>,
layout: TyLayout<'tcx>,
val: Value,
) -> CValue<'tcx> {
let ty = fx.clif_type(layout.ty).unwrap();
- let zero = fx.bcx.ins().iconst(ty, 0);
- let max = fx.bcx.ins().iconst(ty, (u64::max_value() >> (64 - ty.bits())) as i64);
- let res = crate::common::codegen_select(&mut fx.bcx, val, max, zero);
+ let int_ty = match ty {
+ types::F32 => types::I32,
+ types::F64 => types::I64,
+ ty => ty,
+ };
+
+ let zero = fx.bcx.ins().iconst(int_ty, 0);
+ let max = fx.bcx.ins().iconst(int_ty, (u64::max_value() >> (64 - int_ty.bits())) as i64);
+ let mut res = crate::common::codegen_select(&mut fx.bcx, val, max, zero);
+
+ if ty.is_float() {
+ res = fx.bcx.ins().bitcast(ty, res);
+ }
+
CValue::by_val(res, layout)
}
macro_rules! simd_cmp {
($fx:expr, $intrinsic:expr, $cc:ident($x:ident, $y:ident) -> $ret:ident) => {
- simd_for_each_lane($fx, $intrinsic, $x, $y, $ret, |fx, _lane_layout, res_lane_layout, x_lane, y_lane| {
- let res_lane = fx.bcx.ins().icmp(IntCC::$cc, x_lane, y_lane);
+ simd_for_each_lane($fx, $intrinsic, $x, $y, $ret, |fx, lane_layout, res_lane_layout, x_lane, y_lane| {
+ let res_lane = match lane_layout.ty.sty {
+ ty::Uint(_) | ty::Int(_) => fx.bcx.ins().icmp(IntCC::$cc, x_lane, y_lane),
+ _ => unreachable!("{:?}", lane_layout.ty),
+ };
bool_to_zero_or_max_uint(fx, res_lane_layout, res_lane)
});
};
}
-macro_rules! simd_binop {
+macro_rules! simd_int_binop {
($fx:expr, $intrinsic:expr, $op:ident($x:ident, $y:ident) -> $ret:ident) => {
- simd_for_each_lane($fx, $intrinsic, $x, $y, $ret, |fx, _lane_layout, ret_lane_layout, x_lane, y_lane| {
- let res_lane = fx.bcx.ins().$op(x_lane, y_lane);
+ simd_for_each_lane($fx, $intrinsic, $x, $y, $ret, |fx, lane_layout, ret_lane_layout, x_lane, y_lane| {
+ let res_lane = match lane_layout.ty.sty {
+ ty::Uint(_) | ty::Int(_) => fx.bcx.ins().$op(x_lane, y_lane),
+ _ => unreachable!("{:?}", lane_layout.ty),
+ };
CValue::by_val(res_lane, ret_lane_layout)
});
};
};
}
+macro_rules! simd_int_flt_binop {
+ ($fx:expr, $intrinsic:expr, $op:ident|$op_f:ident($x:ident, $y:ident) -> $ret:ident) => {
+ simd_for_each_lane($fx, $intrinsic, $x, $y, $ret, |fx, lane_layout, ret_lane_layout, x_lane, y_lane| {
+ let res_lane = match lane_layout.ty.sty {
+ ty::Uint(_) | ty::Int(_) => fx.bcx.ins().$op(x_lane, y_lane),
+ ty::Float(_) => fx.bcx.ins().$op_f(x_lane, y_lane),
+ _ => unreachable!("{:?}", lane_layout.ty),
+ };
+ CValue::by_val(res_lane, ret_lane_layout)
+ });
+ };
+ ($fx:expr, $intrinsic:expr, $op_u:ident|$op_s:ident|$op_f:ident($x:ident, $y:ident) -> $ret:ident) => {
+ simd_for_each_lane($fx, $intrinsic, $x, $y, $ret, |fx, lane_layout, ret_lane_layout, x_lane, y_lane| {
+ let res_lane = match lane_layout.ty.sty {
+ ty::Uint(_) => fx.bcx.ins().$op_u(x_lane, y_lane),
+ ty::Int(_) => fx.bcx.ins().$op_s(x_lane, y_lane),
+ ty::Float(_) => fx.bcx.ins().$op_f(x_lane, y_lane),
+ _ => unreachable!("{:?}", lane_layout.ty),
+ };
+ CValue::by_val(res_lane, ret_lane_layout)
+ });
+ };
+}
+
+macro_rules! simd_flt_binop {
+ ($fx:expr, $intrinsic:expr, $op:ident($x:ident, $y:ident) -> $ret:ident) => {
+ simd_for_each_lane($fx, $intrinsic, $x, $y, $ret, |fx, lane_layout, ret_lane_layout, x_lane, y_lane| {
+ let res_lane = match lane_layout.ty.sty {
+ ty::Float(_) => fx.bcx.ins().$op(x_lane, y_lane),
+ _ => unreachable!("{:?}", lane_layout.ty),
+ };
+ CValue::by_val(res_lane, ret_lane_layout)
+ });
+ }
+}
+
pub fn codegen_intrinsic_call<'a, 'tcx: 'a>(
fx: &mut FunctionCx<'a, 'tcx, impl Backend>,
def_id: DefId,
intrinsic_match! {
fx, intrinsic, substs, args,
+ _ => {
+ unimpl!("unsupported intrinsic {}", intrinsic)
+ };
assume, (c _a) {};
likely | unlikely, (c a) {
let indexes = {
use rustc::mir::interpret::*;
- let idx_place = match idx {
- Operand::Copy(idx_place) => {
- idx_place
- }
- _ => panic!("simd_shuffle* idx is not Operand::Copy, but {:?}", idx),
- };
-
- assert!(idx_place.projection.is_none());
- let static_ = match &idx_place.base {
- PlaceBase::Static(static_) => {
- static_
- }
- PlaceBase::Local(_) => panic!("simd_shuffle* idx is not constant, but a local"),
- };
-
- let idx_const = match &static_.kind {
- StaticKind::Static(_) => unimplemented!(),
- StaticKind::Promoted(promoted) => {
- fx.tcx.const_eval(ParamEnv::reveal_all().and(GlobalId {
- instance: fx.instance,
- promoted: Some(*promoted),
- })).unwrap()
- }
- };
+ let idx_const = crate::constant::mir_operand_get_const_val(fx, idx).expect("simd_shuffle* idx not const");
let idx_bytes = match idx_const.val {
ConstValue::ByRef { align: _, offset, alloc } => {
assert!(idx < total_len, "idx {} out of range 0..{}", idx, total_len);
}
-
-
- println!("{:?}", indexes);
- unimplemented!();
+ for (out_idx, in_idx) in indexes.into_iter().enumerate() {
+ let in_lane = if in_idx < lane_count {
+ x.value_field(fx, mir::Field::new(in_idx.try_into().unwrap()))
+ } else {
+ y.value_field(fx, mir::Field::new((in_idx - lane_count).try_into().unwrap()))
+ };
+ let out_lane = ret.place_field(fx, mir::Field::new(out_idx));
+ out_lane.write_cvalue(fx, in_lane);
+ }
};
simd_add, (c x, c y) {
- simd_binop!(fx, intrinsic, iadd(x, y) -> ret);
+ simd_int_flt_binop!(fx, intrinsic, iadd|fadd(x, y) -> ret);
};
simd_sub, (c x, c y) {
- simd_binop!(fx, intrinsic, isub(x, y) -> ret);
+ simd_int_flt_binop!(fx, intrinsic, isub|fsub(x, y) -> ret);
};
simd_mul, (c x, c y) {
- simd_binop!(fx, intrinsic, imul(x, y) -> ret);
+ simd_int_flt_binop!(fx, intrinsic, imul|fmul(x, y) -> ret);
};
simd_div, (c x, c y) {
- simd_binop!(fx, intrinsic, udiv|sdiv(x, y) -> ret);
- };
- simd_rem, (c x, c y) {
- simd_binop!(fx, intrinsic, urem|srem(x, y) -> ret);
+ simd_int_flt_binop!(fx, intrinsic, udiv|sdiv|fdiv(x, y) -> ret);
};
simd_shl, (c x, c y) {
- simd_binop!(fx, intrinsic, ishl(x, y) -> ret);
+ simd_int_binop!(fx, intrinsic, ishl(x, y) -> ret);
};
simd_shr, (c x, c y) {
- simd_binop!(fx, intrinsic, ushr|sshr(x, y) -> ret);
+ simd_int_binop!(fx, intrinsic, ushr|sshr(x, y) -> ret);
};
simd_and, (c x, c y) {
- simd_binop!(fx, intrinsic, band(x, y) -> ret);
+ simd_int_binop!(fx, intrinsic, band(x, y) -> ret);
};
simd_or, (c x, c y) {
- simd_binop!(fx, intrinsic, bor(x, y) -> ret);
+ simd_int_binop!(fx, intrinsic, bor(x, y) -> ret);
};
simd_xor, (c x, c y) {
- simd_binop!(fx, intrinsic, bxor(x, y) -> ret);
+ simd_int_binop!(fx, intrinsic, bxor(x, y) -> ret);
};
simd_fmin, (c x, c y) {
- simd_binop!(fx, intrinsic, fmin(x, y) -> ret);
+ simd_flt_binop!(fx, intrinsic, fmin(x, y) -> ret);
};
simd_fmax, (c x, c y) {
- simd_binop!(fx, intrinsic, fmax(x, y) -> ret);
+ simd_flt_binop!(fx, intrinsic, fmax(x, y) -> ret);
};
}