diff options
| author | Trevor Gross <tmgross@umich.edu> | 2024-06-13 08:11:16 -0500 |
|---|---|---|
| committer | Trevor Gross <tmgross@umich.edu> | 2024-06-14 12:43:48 -0500 |
| commit | c906d2e428b9d20d1d73ec587bff75a6d25efed7 (patch) | |
| tree | d0fb07bd3c1a6b7ae605b70d6b0fe3cfb06f8556 | |
| parent | 921645c737f1d6d107a0a10ca5ee129d364dcd7a (diff) | |
| download | rust-c906d2e428b9d20d1d73ec587bff75a6d25efed7.tar.gz rust-c906d2e428b9d20d1d73ec587bff75a6d25efed7.zip | |
Enable const evaluation for `f16` and `f128`
This excludes casting, which needs more tests.
| -rw-r--r-- | compiler/rustc_const_eval/src/interpret/operator.rs | 10 | ||||
| -rw-r--r-- | compiler/rustc_middle/src/mir/interpret/value.rs | 14 |
2 files changed, 22 insertions, 2 deletions
diff --git a/compiler/rustc_const_eval/src/interpret/operator.rs b/compiler/rustc_const_eval/src/interpret/operator.rs index a6eef9f5662..a6924371632 100644 --- a/compiler/rustc_const_eval/src/interpret/operator.rs +++ b/compiler/rustc_const_eval/src/interpret/operator.rs @@ -362,14 +362,18 @@ impl<'tcx, M: Machine<'tcx>> InterpCx<'tcx, M> { let left = left.to_scalar(); let right = right.to_scalar(); Ok(match fty { - FloatTy::F16 => unimplemented!("f16_f128"), + FloatTy::F16 => { + self.binary_float_op(bin_op, layout, left.to_f16()?, right.to_f16()?) + } FloatTy::F32 => { self.binary_float_op(bin_op, layout, left.to_f32()?, right.to_f32()?) } FloatTy::F64 => { self.binary_float_op(bin_op, layout, left.to_f64()?, right.to_f64()?) } - FloatTy::F128 => unimplemented!("f16_f128"), + FloatTy::F128 => { + self.binary_float_op(bin_op, layout, left.to_f128()?, right.to_f128()?) + } }) } _ if left.layout.ty.is_integral() => { @@ -431,8 +435,10 @@ impl<'tcx, M: Machine<'tcx>> InterpCx<'tcx, M> { let val = val.to_scalar(); // No NaN adjustment here, `-` is a bitwise operation! let res = match (un_op, fty) { + (Neg, FloatTy::F16) => Scalar::from_f16(-val.to_f16()?), (Neg, FloatTy::F32) => Scalar::from_f32(-val.to_f32()?), (Neg, FloatTy::F64) => Scalar::from_f64(-val.to_f64()?), + (Neg, FloatTy::F128) => Scalar::from_f128(-val.to_f128()?), _ => span_bug!(self.cur_span(), "Invalid float op {:?}", un_op), }; Ok(ImmTy::from_scalar(res, layout)) diff --git a/compiler/rustc_middle/src/mir/interpret/value.rs b/compiler/rustc_middle/src/mir/interpret/value.rs index 70e5ad0635b..a84a4c583ed 100644 --- a/compiler/rustc_middle/src/mir/interpret/value.rs +++ b/compiler/rustc_middle/src/mir/interpret/value.rs @@ -69,6 +69,13 @@ impl<Prov: Provenance> fmt::LowerHex for Scalar<Prov> { } } +impl<Prov> From<Half> for Scalar<Prov> { + #[inline(always)] + fn from(f: Half) -> Self { + Scalar::from_f16(f) + } +} + impl<Prov> From<Single> for Scalar<Prov> { #[inline(always)] fn from(f: Single) -> Self { @@ -83,6 +90,13 @@ impl<Prov> From<Double> for Scalar<Prov> { } } +impl<Prov> From<Quad> for Scalar<Prov> { + #[inline(always)] + fn from(f: Quad) -> Self { + Scalar::from_f128(f) + } +} + impl<Prov> From<ScalarInt> for Scalar<Prov> { #[inline(always)] fn from(ptr: ScalarInt) -> Self { |
