frontend/ir_emitter: Add half-precision variant of FPAbs
This commit is contained in:
parent
10abc77fad
commit
8309ec7a9f
4 changed files with 23 additions and 6 deletions
|
@ -39,6 +39,7 @@ namespace {
|
||||||
const Xbyak::Reg64 INVALID_REG = Xbyak::Reg64(-1);
|
const Xbyak::Reg64 INVALID_REG = Xbyak::Reg64(-1);
|
||||||
|
|
||||||
constexpr u64 f16_negative_zero = 0x8000;
|
constexpr u64 f16_negative_zero = 0x8000;
|
||||||
|
constexpr u64 f16_non_sign_mask = 0x7fff;
|
||||||
|
|
||||||
constexpr u64 f32_negative_zero = 0x80000000u;
|
constexpr u64 f32_negative_zero = 0x80000000u;
|
||||||
constexpr u64 f32_nan = 0x7fc00000u;
|
constexpr u64 f32_nan = 0x7fc00000u;
|
||||||
|
@ -325,9 +326,18 @@ void FPThreeOp(BlockOfCode& code, EmitContext& ctx, IR::Inst* inst, Function fn)
|
||||||
|
|
||||||
} // anonymous namespace
|
} // anonymous namespace
|
||||||
|
|
||||||
|
void EmitX64::EmitFPAbs16(EmitContext& ctx, IR::Inst* inst) {
|
||||||
|
auto args = ctx.reg_alloc.GetArgumentInfo(inst);
|
||||||
|
const Xbyak::Xmm result = ctx.reg_alloc.UseScratchXmm(args[0]);
|
||||||
|
|
||||||
|
code.pand(result, code.MConst(xword, f16_non_sign_mask));
|
||||||
|
|
||||||
|
ctx.reg_alloc.DefineValue(inst, result);
|
||||||
|
}
|
||||||
|
|
||||||
void EmitX64::EmitFPAbs32(EmitContext& ctx, IR::Inst* inst) {
|
void EmitX64::EmitFPAbs32(EmitContext& ctx, IR::Inst* inst) {
|
||||||
auto args = ctx.reg_alloc.GetArgumentInfo(inst);
|
auto args = ctx.reg_alloc.GetArgumentInfo(inst);
|
||||||
Xbyak::Xmm result = ctx.reg_alloc.UseScratchXmm(args[0]);
|
const Xbyak::Xmm result = ctx.reg_alloc.UseScratchXmm(args[0]);
|
||||||
|
|
||||||
code.pand(result, code.MConst(xword, f32_non_sign_mask));
|
code.pand(result, code.MConst(xword, f32_non_sign_mask));
|
||||||
|
|
||||||
|
@ -336,7 +346,7 @@ void EmitX64::EmitFPAbs32(EmitContext& ctx, IR::Inst* inst) {
|
||||||
|
|
||||||
void EmitX64::EmitFPAbs64(EmitContext& ctx, IR::Inst* inst) {
|
void EmitX64::EmitFPAbs64(EmitContext& ctx, IR::Inst* inst) {
|
||||||
auto args = ctx.reg_alloc.GetArgumentInfo(inst);
|
auto args = ctx.reg_alloc.GetArgumentInfo(inst);
|
||||||
Xbyak::Xmm result = ctx.reg_alloc.UseScratchXmm(args[0]);
|
const Xbyak::Xmm result = ctx.reg_alloc.UseScratchXmm(args[0]);
|
||||||
|
|
||||||
code.pand(result, code.MConst(xword, f64_non_sign_mask));
|
code.pand(result, code.MConst(xword, f64_non_sign_mask));
|
||||||
|
|
||||||
|
|
|
@ -1773,11 +1773,17 @@ U128 IREmitter::ZeroVector() {
|
||||||
return Inst<U128>(Opcode::ZeroVector);
|
return Inst<U128>(Opcode::ZeroVector);
|
||||||
}
|
}
|
||||||
|
|
||||||
U32U64 IREmitter::FPAbs(const U32U64& a) {
|
U16U32U64 IREmitter::FPAbs(const U16U32U64& a) {
|
||||||
if (a.GetType() == Type::U32) {
|
switch (a.GetType()) {
|
||||||
|
case Type::U16:
|
||||||
|
return Inst<U16>(Opcode::FPAbs16, a);
|
||||||
|
case Type::U32:
|
||||||
return Inst<U32>(Opcode::FPAbs32, a);
|
return Inst<U32>(Opcode::FPAbs32, a);
|
||||||
} else {
|
case Type::U64:
|
||||||
return Inst<U64>(Opcode::FPAbs64, a);
|
return Inst<U64>(Opcode::FPAbs64, a);
|
||||||
|
default:
|
||||||
|
UNREACHABLE();
|
||||||
|
return U16U32U64{};
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
|
@ -292,7 +292,7 @@ public:
|
||||||
U128 VectorZeroUpper(const U128& a);
|
U128 VectorZeroUpper(const U128& a);
|
||||||
U128 ZeroVector();
|
U128 ZeroVector();
|
||||||
|
|
||||||
U32U64 FPAbs(const U32U64& a);
|
U16U32U64 FPAbs(const U16U32U64& a);
|
||||||
U32U64 FPAdd(const U32U64& a, const U32U64& b, bool fpcr_controlled);
|
U32U64 FPAdd(const U32U64& a, const U32U64& b, bool fpcr_controlled);
|
||||||
NZCV FPCompare(const U32U64& a, const U32U64& b, bool exc_on_qnan, bool fpcr_controlled);
|
NZCV FPCompare(const U32U64& a, const U32U64& b, bool exc_on_qnan, bool fpcr_controlled);
|
||||||
U32U64 FPDiv(const U32U64& a, const U32U64& b, bool fpcr_controlled);
|
U32U64 FPDiv(const U32U64& a, const U32U64& b, bool fpcr_controlled);
|
||||||
|
|
|
@ -460,6 +460,7 @@ OPCODE(VectorZeroUpper, U128, U128
|
||||||
OPCODE(ZeroVector, U128, )
|
OPCODE(ZeroVector, U128, )
|
||||||
|
|
||||||
// Floating-point operations
|
// Floating-point operations
|
||||||
|
OPCODE(FPAbs16, U16, U16 )
|
||||||
OPCODE(FPAbs32, U32, U32 )
|
OPCODE(FPAbs32, U32, U32 )
|
||||||
OPCODE(FPAbs64, U64, U64 )
|
OPCODE(FPAbs64, U64, U64 )
|
||||||
OPCODE(FPAdd32, U32, U32, U32 )
|
OPCODE(FPAdd32, U32, U32, U32 )
|
||||||
|
|
Loading…
Reference in a new issue