/* This file is part of the dynarmic project. * Copyright (c) 2016 MerryMage * This software may be used and distributed according to the terms of the GNU * General Public License version 2 or any later version. */ #include #include #include "backend_x64/abi.h" #include "backend_x64/block_of_code.h" #include "backend_x64/emit_x64.h" #include "common/assert.h" #include "common/common_types.h" #include "common/fp/fpcr.h" #include "common/fp/fpsr.h" #include "common/fp/info.h" #include "common/fp/op.h" #include "common/fp/rounding_mode.h" #include "common/fp/util.h" #include "common/mp/cartesian_product.h" #include "common/mp/integer.h" #include "common/mp/list.h" #include "common/mp/lut.h" #include "common/mp/to_tuple.h" #include "common/mp/vlift.h" #include "common/mp/vllift.h" #include "frontend/ir/basic_block.h" #include "frontend/ir/microinstruction.h" #include "frontend/ir/opcodes.h" namespace Dynarmic::BackendX64 { using namespace Xbyak::util; namespace mp = Dynarmic::Common::mp; namespace { constexpr u64 f32_negative_zero = 0x80000000u; constexpr u64 f32_nan = 0x7fc00000u; constexpr u64 f32_non_sign_mask = 0x7fffffffu; constexpr u64 f32_smallest_normal = 0x00800000u; constexpr u64 f64_negative_zero = 0x8000000000000000u; constexpr u64 f64_nan = 0x7ff8000000000000u; constexpr u64 f64_non_sign_mask = 0x7fffffffffffffffu; constexpr u64 f64_smallest_normal = 0x0010000000000000u; constexpr u64 f64_penultimate_positive_denormal = 0x000ffffffffffffeu; constexpr u64 f64_min_s32 = 0xc1e0000000000000u; // -2147483648 as a double constexpr u64 f64_max_s32 = 0x41dfffffffc00000u; // 2147483647 as a double constexpr u64 f64_min_u32 = 0x0000000000000000u; // 0 as a double constexpr u64 f64_max_u32 = 0x41efffffffe00000u; // 4294967295 as a double constexpr u64 f64_min_s64 = 0xc3e0000000000000u; // -2^63 as a double constexpr u64 f64_max_s64_lim = 0x43e0000000000000u; // 2^63 as a double (actual maximum unrepresentable) constexpr u64 f64_min_u64 = 0x0000000000000000u; // 0 as a double constexpr u64 f64_max_u64_lim = 0x43f0000000000000u; // 2^64 as a double (actual maximum unrepresentable) template T ChooseOnFsize([[maybe_unused]] T f32, [[maybe_unused]] T f64) { static_assert(fsize == 32 || fsize == 64, "fsize must be either 32 or 64"); if constexpr (fsize == 32) { return f32; } else { return f64; } } #define FCODE(NAME) (code.*ChooseOnFsize(&Xbyak::CodeGenerator::NAME##s, &Xbyak::CodeGenerator::NAME##d)) template void DenormalsAreZero(BlockOfCode& code, Xbyak::Xmm xmm_value, Xbyak::Reg64 gpr_scratch) { Xbyak::Label end; if constexpr (fsize == 32) { code.movd(gpr_scratch.cvt32(), xmm_value); code.and_(gpr_scratch.cvt32(), u32(0x7FFFFFFF)); code.sub(gpr_scratch.cvt32(), u32(1)); code.cmp(gpr_scratch.cvt32(), u32(0x007FFFFE)); } else { auto mask = code.MConst(xword, f64_non_sign_mask); mask.setBit(64); auto penult_denormal = code.MConst(xword, f64_penultimate_positive_denormal); penult_denormal.setBit(64); code.movq(gpr_scratch, xmm_value); code.and_(gpr_scratch, mask); code.sub(gpr_scratch, u32(1)); code.cmp(gpr_scratch, penult_denormal); } // We need to report back whether we've found a denormal on input. // SSE doesn't do this for us when SSE's DAZ is enabled. code.ja(end); code.andps(xmm_value, code.MConst(xword, fsize == 32 ? f32_negative_zero : f64_negative_zero)); code.mov(dword[r15 + code.GetJitStateInfo().offsetof_FPSCR_IDC], u32(1 << 7)); code.L(end); } template void ZeroIfNaN(BlockOfCode& code, Xbyak::Xmm xmm_value, Xbyak::Xmm xmm_scratch) { code.pxor(xmm_scratch, xmm_scratch); FCODE(cmpords)(xmm_scratch, xmm_value); // true mask when ordered (i.e.: when not an NaN) code.pand(xmm_value, xmm_scratch); } template void PreProcessNaNs(BlockOfCode& code, Xbyak::Xmm a, Xbyak::Xmm b, Xbyak::Label& end) { using FPT = mp::unsigned_integer_of_size; Xbyak::Label nan; FCODE(ucomis)(a, b); code.jp(nan, code.T_NEAR); code.SwitchToFarCode(); code.L(nan); code.sub(rsp, 8); ABI_PushCallerSaveRegistersAndAdjustStackExcept(code, HostLocXmmIdx(a.getIdx())); code.movq(code.ABI_PARAM1, a); code.movq(code.ABI_PARAM2, b); code.CallFunction(static_cast([](FPT a, FPT b) -> FPT { return *FP::ProcessNaNs(a, b); })); code.movq(a, code.ABI_RETURN); ABI_PopCallerSaveRegistersAndAdjustStackExcept(code, HostLocXmmIdx(a.getIdx())); code.add(rsp, 8); code.jmp(end, code.T_NEAR); code.SwitchToNearCode(); } template void PreProcessNaNs(BlockOfCode& code, EmitContext& ctx, Xbyak::Xmm a, Xbyak::Xmm b, Xbyak::Xmm c, Xbyak::Label& end, NaNHandler nan_handler) { using FPT = mp::unsigned_integer_of_size; Xbyak::Label nan; FCODE(ucomis)(a, b); code.jp(nan, code.T_NEAR); FCODE(ucomis)(c, c); code.jp(nan, code.T_NEAR); code.SwitchToFarCode(); code.L(nan); code.sub(rsp, 8); ABI_PushCallerSaveRegistersAndAdjustStackExcept(code, HostLocXmmIdx(a.getIdx())); code.movq(code.ABI_PARAM1, a); code.movq(code.ABI_PARAM2, b); code.movq(code.ABI_PARAM3, c); code.mov(code.ABI_PARAM4, ctx.FPCR()); code.CallFunction(static_cast(nan_handler)); code.movq(a, code.ABI_RETURN); ABI_PopCallerSaveRegistersAndAdjustStackExcept(code, HostLocXmmIdx(a.getIdx())); code.add(rsp, 8); code.jmp(end, code.T_NEAR); code.SwitchToNearCode(); } template void PostProcessNaNs(BlockOfCode& code, Xbyak::Xmm result, Xbyak::Xmm tmp) { if constexpr (fsize == 32) { code.movaps(tmp, result); code.cmpunordps(tmp, tmp); code.pslld(tmp, 31); code.xorps(result, tmp); } else { code.movaps(tmp, result); code.cmpunordpd(tmp, tmp); code.psllq(tmp, 63); code.xorps(result, tmp); } } template void DefaultNaN(BlockOfCode& code, Xbyak::Xmm xmm_value) { Xbyak::Label end; FCODE(ucomis)(xmm_value, xmm_value); code.jnp(end); code.movaps(xmm_value, code.MConst(xword, fsize == 32 ? f32_nan : f64_nan)); code.L(end); } template Xbyak::Label ProcessNaN(BlockOfCode& code, Xbyak::Xmm a) { Xbyak::Label nan, end; FCODE(ucomis)(a, a); code.jp(nan, code.T_NEAR); code.SwitchToFarCode(); code.L(nan); code.orps(a, code.MConst(xword, fsize == 32 ? 0x00400000 : 0x0008'0000'0000'0000)); code.jmp(end, code.T_NEAR); code.SwitchToNearCode(); return end; } template void FPTwoOp(BlockOfCode& code, EmitContext& ctx, IR::Inst* inst, Function fn) { auto args = ctx.reg_alloc.GetArgumentInfo(inst); Xbyak::Label end; Xbyak::Xmm result = ctx.reg_alloc.UseScratchXmm(args[0]); if (ctx.AccurateNaN() && !ctx.FPSCR_DN()) { end = ProcessNaN(code, result); } if constexpr (std::is_member_function_pointer_v) { (code.*fn)(result, result); } else { fn(result); } if (ctx.FPSCR_DN()) { DefaultNaN(code, result); } else if (ctx.AccurateNaN()) { PostProcessNaNs(code, result, ctx.reg_alloc.ScratchXmm()); } code.L(end); ctx.reg_alloc.DefineValue(inst, result); } enum class CallDenormalsAreZero { Yes, No, }; template void FPThreeOp(BlockOfCode& code, EmitContext& ctx, IR::Inst* inst, [[maybe_unused]] PreprocessFunction preprocess, Function fn, CallDenormalsAreZero call_denormals_are_zero = CallDenormalsAreZero::No) { auto args = ctx.reg_alloc.GetArgumentInfo(inst); Xbyak::Label end; Xbyak::Xmm result = ctx.reg_alloc.UseScratchXmm(args[0]); Xbyak::Xmm operand = ctx.reg_alloc.UseScratchXmm(args[1]); Xbyak::Reg64 gpr_scratch = ctx.reg_alloc.ScratchGpr(); if (ctx.FPSCR_FTZ() && call_denormals_are_zero == CallDenormalsAreZero::Yes) { DenormalsAreZero(code, result, gpr_scratch); DenormalsAreZero(code, operand, gpr_scratch); } if constexpr(!std::is_same_v) { preprocess(result, operand, gpr_scratch, end); } if (ctx.AccurateNaN() && !ctx.FPSCR_DN()) { PreProcessNaNs(code, result, operand, end); } if constexpr (std::is_member_function_pointer_v) { (code.*fn)(result, operand); } else { fn(result, operand); } if (ctx.FPSCR_DN()) { DefaultNaN(code, result); } else if (ctx.AccurateNaN()) { PostProcessNaNs(code, result, operand); } code.L(end); ctx.reg_alloc.DefineValue(inst, result); } template void FPThreeOp(BlockOfCode& code, EmitContext& ctx, IR::Inst* inst, Function fn, CallDenormalsAreZero call_denormals_are_zero = CallDenormalsAreZero::No) { FPThreeOp(code, ctx, inst, nullptr, fn, call_denormals_are_zero); } } // anonymous namespace void EmitX64::EmitFPAbs32(EmitContext& ctx, IR::Inst* inst) { auto args = ctx.reg_alloc.GetArgumentInfo(inst); Xbyak::Xmm result = ctx.reg_alloc.UseScratchXmm(args[0]); code.pand(result, code.MConst(xword, f32_non_sign_mask)); ctx.reg_alloc.DefineValue(inst, result); } void EmitX64::EmitFPAbs64(EmitContext& ctx, IR::Inst* inst) { auto args = ctx.reg_alloc.GetArgumentInfo(inst); Xbyak::Xmm result = ctx.reg_alloc.UseScratchXmm(args[0]); code.pand(result, code.MConst(xword, f64_non_sign_mask)); ctx.reg_alloc.DefineValue(inst, result); } void EmitX64::EmitFPNeg32(EmitContext& ctx, IR::Inst* inst) { auto args = ctx.reg_alloc.GetArgumentInfo(inst); Xbyak::Xmm result = ctx.reg_alloc.UseScratchXmm(args[0]); code.pxor(result, code.MConst(xword, f32_negative_zero)); ctx.reg_alloc.DefineValue(inst, result); } void EmitX64::EmitFPNeg64(EmitContext& ctx, IR::Inst* inst) { auto args = ctx.reg_alloc.GetArgumentInfo(inst); Xbyak::Xmm result = ctx.reg_alloc.UseScratchXmm(args[0]); code.pxor(result, code.MConst(xword, f64_negative_zero)); ctx.reg_alloc.DefineValue(inst, result); } void EmitX64::EmitFPAdd32(EmitContext& ctx, IR::Inst* inst) { FPThreeOp<32>(code, ctx, inst, &Xbyak::CodeGenerator::addss); } void EmitX64::EmitFPAdd64(EmitContext& ctx, IR::Inst* inst) { FPThreeOp<64>(code, ctx, inst, &Xbyak::CodeGenerator::addsd); } void EmitX64::EmitFPDiv32(EmitContext& ctx, IR::Inst* inst) { FPThreeOp<32>(code, ctx, inst, &Xbyak::CodeGenerator::divss); } void EmitX64::EmitFPDiv64(EmitContext& ctx, IR::Inst* inst) { FPThreeOp<64>(code, ctx, inst, &Xbyak::CodeGenerator::divsd); } template static void EmitFPMax(BlockOfCode& code, EmitContext& ctx, IR::Inst* inst) { if (ctx.FPSCR_DN()) { auto args = ctx.reg_alloc.GetArgumentInfo(inst); const Xbyak::Xmm result = ctx.reg_alloc.UseScratchXmm(args[0]); const Xbyak::Xmm operand = ctx.reg_alloc.UseScratchXmm(args[1]); if (ctx.FPSCR_FTZ()) { const Xbyak::Reg64 gpr_scratch = ctx.reg_alloc.ScratchGpr(); DenormalsAreZero(code, result, gpr_scratch); DenormalsAreZero(code, operand, gpr_scratch); } Xbyak::Label equal, end, nan; FCODE(ucomis)(result, operand); code.jz(equal, code.T_NEAR); FCODE(maxs)(result, operand); code.L(end); code.SwitchToFarCode(); code.L(equal); code.jp(nan); code.andps(result, operand); code.jmp(end); code.L(nan); code.movaps(result, code.MConst(xword, fsize == 32 ? f32_nan : f64_nan)); code.jmp(end); code.SwitchToNearCode(); ctx.reg_alloc.DefineValue(inst, result); return; } FPThreeOp(code, ctx, inst, [&](Xbyak::Xmm result, Xbyak::Xmm operand){ Xbyak::Label equal, end; FCODE(ucomis)(result, operand); code.jz(equal); FCODE(maxs)(result, operand); code.jmp(end); code.L(equal); code.andps(result, operand); code.L(end); }, CallDenormalsAreZero::Yes); } void EmitX64::EmitFPMax32(EmitContext& ctx, IR::Inst* inst) { EmitFPMax<32>(code, ctx, inst); } void EmitX64::EmitFPMax64(EmitContext& ctx, IR::Inst* inst) { EmitFPMax<64>(code, ctx, inst); } void EmitX64::EmitFPMaxNumeric32(EmitContext& ctx, IR::Inst* inst) { FPThreeOp<32>(code, ctx, inst, [&](Xbyak::Xmm result, Xbyak::Xmm operand, Xbyak::Reg64 scratch, Xbyak::Label& end){ Xbyak::Label normal, normal_or_equal, result_is_result; code.ucomiss(result, operand); code.jnp(normal_or_equal); // If operand == QNaN, result = result. code.movd(scratch.cvt32(), operand); code.shl(scratch.cvt32(), 1); code.cmp(scratch.cvt32(), 0xff800000u); code.jae(result_is_result); // If operand == SNaN, let usual NaN code handle it. code.cmp(scratch.cvt32(), 0xff000000u); code.ja(normal); // If result == SNaN, && operand != NaN, result = result. code.movd(scratch.cvt32(), result); code.shl(scratch.cvt32(), 1); code.cmp(scratch.cvt32(), 0xff800000u); code.jnae(result_is_result); // If result == QNaN && operand != NaN, result = operand. code.movaps(result, operand); code.jmp(end, code.T_NEAR); code.L(result_is_result); code.movaps(operand, result); code.jmp(normal); code.L(normal_or_equal); code.jnz(normal); code.andps(operand, result); code.L(normal); }, &Xbyak::CodeGenerator::maxss, CallDenormalsAreZero::Yes); } void EmitX64::EmitFPMaxNumeric64(EmitContext& ctx, IR::Inst* inst) { FPThreeOp<64>(code, ctx, inst, [&](Xbyak::Xmm result, Xbyak::Xmm operand, Xbyak::Reg64 scratch, Xbyak::Label& end){ Xbyak::Label normal, normal_or_equal, result_is_result; code.ucomisd(result, operand); code.jnp(normal_or_equal); // If operand == QNaN, result = result. code.movq(scratch, operand); code.shl(scratch, 1); code.cmp(scratch, code.MConst(qword, 0xfff0'0000'0000'0000u)); code.jae(result_is_result); // If operand == SNaN, let usual NaN code handle it. code.cmp(scratch, code.MConst(qword, 0xffe0'0000'0000'0000u)); code.ja(normal); // If result == SNaN, && operand != NaN, result = result. code.movq(scratch, result); code.shl(scratch, 1); code.cmp(scratch, code.MConst(qword, 0xfff0'0000'0000'0000u)); code.jnae(result_is_result); // If result == QNaN && operand != NaN, result = operand. code.movaps(result, operand); code.jmp(end, code.T_NEAR); code.L(result_is_result); code.movaps(operand, result); code.jmp(normal); code.L(normal_or_equal); code.jnz(normal); code.andps(operand, result); code.L(normal); }, &Xbyak::CodeGenerator::maxsd, CallDenormalsAreZero::Yes); } template static void EmitFPMin(BlockOfCode& code, EmitContext& ctx, IR::Inst* inst) { if (ctx.FPSCR_DN()) { auto args = ctx.reg_alloc.GetArgumentInfo(inst); const Xbyak::Xmm result = ctx.reg_alloc.UseScratchXmm(args[0]); const Xbyak::Xmm operand = ctx.reg_alloc.UseScratchXmm(args[1]); if (ctx.FPSCR_FTZ()) { const Xbyak::Reg64 gpr_scratch = ctx.reg_alloc.ScratchGpr(); DenormalsAreZero(code, result, gpr_scratch); DenormalsAreZero(code, operand, gpr_scratch); } Xbyak::Label equal, end, nan; FCODE(ucomis)(result, operand); code.jz(equal, code.T_NEAR); FCODE(mins)(result, operand); code.L(end); code.SwitchToFarCode(); code.L(equal); code.jp(nan); code.orps(result, operand); code.jmp(end); code.L(nan); code.movaps(result, code.MConst(xword, fsize == 32 ? f32_nan : f64_nan)); code.jmp(end); code.SwitchToNearCode(); ctx.reg_alloc.DefineValue(inst, result); return; } FPThreeOp(code, ctx, inst, [&](Xbyak::Xmm result, Xbyak::Xmm operand){ Xbyak::Label equal, end; FCODE(ucomis)(result, operand); code.jz(equal); FCODE(mins)(result, operand); code.jmp(end); code.L(equal); code.orps(result, operand); code.L(end); }, CallDenormalsAreZero::Yes); } void EmitX64::EmitFPMin32(EmitContext& ctx, IR::Inst* inst) { EmitFPMin<32>(code, ctx, inst); } void EmitX64::EmitFPMin64(EmitContext& ctx, IR::Inst* inst) { EmitFPMin<64>(code, ctx, inst); } void EmitX64::EmitFPMinNumeric32(EmitContext& ctx, IR::Inst* inst) { FPThreeOp<32>(code, ctx, inst, [&](Xbyak::Xmm result, Xbyak::Xmm operand, Xbyak::Reg64 scratch, Xbyak::Label& end){ Xbyak::Label normal, normal_or_equal, result_is_result; code.ucomiss(result, operand); code.jnp(normal_or_equal); // If operand == QNaN, result = result. code.movd(scratch.cvt32(), operand); code.shl(scratch.cvt32(), 1); code.cmp(scratch.cvt32(), 0xff800000u); code.jae(result_is_result); // If operand == SNaN, let usual NaN code handle it. code.cmp(scratch.cvt32(), 0xff000000u); code.ja(normal); // If result == SNaN, && operand != NaN, result = result. code.movd(scratch.cvt32(), result); code.shl(scratch.cvt32(), 1); code.cmp(scratch.cvt32(), 0xff800000u); code.jnae(result_is_result); // If result == QNaN && operand != NaN, result = operand. code.movaps(result, operand); code.jmp(end, code.T_NEAR); code.L(result_is_result); code.movaps(operand, result); code.jmp(normal); code.L(normal_or_equal); code.jnz(normal); code.orps(operand, result); code.L(normal); }, &Xbyak::CodeGenerator::minss, CallDenormalsAreZero::Yes); } void EmitX64::EmitFPMinNumeric64(EmitContext& ctx, IR::Inst* inst) { FPThreeOp<64>(code, ctx, inst, [&](Xbyak::Xmm result, Xbyak::Xmm operand, Xbyak::Reg64 scratch, Xbyak::Label& end){ Xbyak::Label normal, normal_or_equal, result_is_result; code.ucomisd(result, operand); code.jnp(normal_or_equal); // If operand == QNaN, result = result. code.movq(scratch, operand); code.shl(scratch, 1); code.cmp(scratch, code.MConst(qword, 0xfff0'0000'0000'0000u)); code.jae(result_is_result); // If operand == SNaN, let usual NaN code handle it. code.cmp(scratch, code.MConst(qword, 0xffe0'0000'0000'0000u)); code.ja(normal); // If result == SNaN, && operand != NaN, result = result. code.movq(scratch, result); code.shl(scratch, 1); code.cmp(scratch, code.MConst(qword, 0xfff0'0000'0000'0000u)); code.jnae(result_is_result); // If result == QNaN && operand != NaN, result = operand. code.movaps(result, operand); code.jmp(end, code.T_NEAR); code.L(result_is_result); code.movaps(operand, result); code.jmp(normal); code.L(normal_or_equal); code.jnz(normal); code.orps(operand, result); code.L(normal); }, &Xbyak::CodeGenerator::minsd, CallDenormalsAreZero::Yes); } void EmitX64::EmitFPMul32(EmitContext& ctx, IR::Inst* inst) { FPThreeOp<32>(code, ctx, inst, &Xbyak::CodeGenerator::mulss); } void EmitX64::EmitFPMul64(EmitContext& ctx, IR::Inst* inst) { FPThreeOp<64>(code, ctx, inst, &Xbyak::CodeGenerator::mulsd); } template static void EmitFPMulAdd(BlockOfCode& code, EmitContext& ctx, IR::Inst* inst) { using FPT = mp::unsigned_integer_of_size; if (code.DoesCpuSupport(Xbyak::util::Cpu::tFMA)) { auto args = ctx.reg_alloc.GetArgumentInfo(inst); Xbyak::Label end, fallback; const Xbyak::Xmm operand1 = ctx.reg_alloc.UseXmm(args[0]); const Xbyak::Xmm operand2 = ctx.reg_alloc.UseXmm(args[1]); const Xbyak::Xmm operand3 = ctx.reg_alloc.UseXmm(args[2]); const Xbyak::Xmm result = ctx.reg_alloc.ScratchXmm(); const Xbyak::Xmm tmp = ctx.reg_alloc.ScratchXmm(); code.movaps(result, operand1); FCODE(vfmadd231s)(result, operand2, operand3); code.movaps(tmp, code.MConst(xword, fsize == 32 ? f32_non_sign_mask : f64_non_sign_mask)); code.andps(tmp, result); FCODE(ucomis)(result, code.MConst(xword, fsize == 32 ? f32_smallest_normal : f64_smallest_normal)); code.jz(fallback, code.T_NEAR); code.L(end); code.SwitchToFarCode(); code.L(fallback); code.sub(rsp, 8); ABI_PushCallerSaveRegistersAndAdjustStackExcept(code, HostLocXmmIdx(result.getIdx())); code.movq(code.ABI_PARAM1, operand1); code.movq(code.ABI_PARAM2, operand2); code.movq(code.ABI_PARAM3, operand3); code.mov(code.ABI_PARAM4.cvt32(), ctx.FPCR()); #ifdef _WIN32 code.sub(rsp, 16 + ABI_SHADOW_SPACE); code.lea(rax, code.ptr[code.r15 + code.GetJitStateInfo().offsetof_fpsr_exc]); code.mov(qword[rsp + ABI_SHADOW_SPACE], rax); code.CallFunction(&FP::FPMulAdd); code.add(rsp, 16 + ABI_SHADOW_SPACE); #else code.lea(code.ABI_PARAM5, code.ptr[code.r15 + code.GetJitStateInfo().offsetof_fpsr_exc]); code.CallFunction(&FP::FPMulAdd); #endif code.movq(result, code.ABI_RETURN); ABI_PopCallerSaveRegistersAndAdjustStackExcept(code, HostLocXmmIdx(result.getIdx())); code.add(rsp, 8); code.jmp(end, code.T_NEAR); code.SwitchToNearCode(); ctx.reg_alloc.DefineValue(inst, result); return; } auto args = ctx.reg_alloc.GetArgumentInfo(inst); ctx.reg_alloc.HostCall(inst, args[0], args[1], args[2]); code.mov(code.ABI_PARAM4.cvt32(), ctx.FPCR()); #ifdef _WIN32 code.sub(rsp, 16 + ABI_SHADOW_SPACE); code.lea(rax, code.ptr[code.r15 + code.GetJitStateInfo().offsetof_fpsr_exc]); code.mov(qword[rsp + ABI_SHADOW_SPACE], rax); code.CallFunction(&FP::FPMulAdd); code.add(rsp, 16 + ABI_SHADOW_SPACE); #else code.lea(code.ABI_PARAM5, code.ptr[code.r15 + code.GetJitStateInfo().offsetof_fpsr_exc]); code.CallFunction(&FP::FPMulAdd); #endif } void EmitX64::EmitFPMulAdd32(EmitContext& ctx, IR::Inst* inst) { EmitFPMulAdd<32>(code, ctx, inst); } void EmitX64::EmitFPMulAdd64(EmitContext& ctx, IR::Inst* inst) { EmitFPMulAdd<64>(code, ctx, inst); } template static void EmitFPRecipEstimate(BlockOfCode& code, EmitContext& ctx, IR::Inst* inst) { auto args = ctx.reg_alloc.GetArgumentInfo(inst); ctx.reg_alloc.HostCall(inst, args[0]); code.mov(code.ABI_PARAM2.cvt32(), ctx.FPCR()); code.lea(code.ABI_PARAM3, code.ptr[code.r15 + code.GetJitStateInfo().offsetof_fpsr_exc]); code.CallFunction(&FP::FPRecipEstimate); } void EmitX64::EmitFPRecipEstimate32(EmitContext& ctx, IR::Inst* inst) { EmitFPRecipEstimate(code, ctx, inst); } void EmitX64::EmitFPRecipEstimate64(EmitContext& ctx, IR::Inst* inst) { EmitFPRecipEstimate(code, ctx, inst); } template static void EmitFPRecipStepFused(BlockOfCode& code, EmitContext& ctx, IR::Inst* inst) { auto args = ctx.reg_alloc.GetArgumentInfo(inst); ctx.reg_alloc.HostCall(inst, args[0], args[1]); code.mov(code.ABI_PARAM3.cvt32(), ctx.FPCR()); code.lea(code.ABI_PARAM4, code.ptr[code.r15 + code.GetJitStateInfo().offsetof_fpsr_exc]); code.CallFunction(&FP::FPRecipStepFused); } void EmitX64::EmitFPRecipStepFused32(EmitContext& ctx, IR::Inst* inst) { EmitFPRecipStepFused(code, ctx, inst); } void EmitX64::EmitFPRecipStepFused64(EmitContext& ctx, IR::Inst* inst) { EmitFPRecipStepFused(code, ctx, inst); } static void EmitFPRound(BlockOfCode& code, EmitContext& ctx, IR::Inst* inst, size_t fsize) { const auto rounding = static_cast(inst->GetArg(1).GetU8()); const bool exact = inst->GetArg(2).GetU1(); if (code.DoesCpuSupport(Xbyak::util::Cpu::tSSE41) && rounding != FP::RoundingMode::ToNearest_TieAwayFromZero && !exact) { const int round_imm = [&]{ switch (rounding) { case FP::RoundingMode::ToNearest_TieEven: return 0b00; case FP::RoundingMode::TowardsPlusInfinity: return 0b10; case FP::RoundingMode::TowardsMinusInfinity: return 0b01; case FP::RoundingMode::TowardsZero: return 0b11; default: UNREACHABLE(); } return 0; }(); if (fsize == 64) { FPTwoOp<64>(code, ctx, inst, [&](Xbyak::Xmm result) { code.roundsd(result, result, round_imm); }); } else { FPTwoOp<32>(code, ctx, inst, [&](Xbyak::Xmm result) { code.roundss(result, result, round_imm); }); } return; } using fsize_list = mp::list, mp::vlift>; using rounding_list = mp::list< std::integral_constant, std::integral_constant, std::integral_constant, std::integral_constant, std::integral_constant >; using exact_list = mp::list, mp::vlift>; using key_type = std::tuple; using value_type = u64(*)(u64, FP::FPSR&, FP::FPCR); static const auto lut = mp::GenerateLookupTableFromList( [](auto args) { return std::pair{ mp::to_tuple, static_cast( [](u64 input, FP::FPSR& fpsr, FP::FPCR fpcr) { constexpr auto t = mp::to_tuple; constexpr size_t fsize = std::get<0>(t); constexpr FP::RoundingMode rounding_mode = std::get<1>(t); constexpr bool exact = std::get<2>(t); using InputSize = mp::unsigned_integer_of_size; return FP::FPRoundInt(static_cast(input), fpcr, rounding_mode, exact, fpsr); } ) }; }, mp::cartesian_product{} ); auto args = ctx.reg_alloc.GetArgumentInfo(inst); ctx.reg_alloc.HostCall(inst, args[0]); code.lea(code.ABI_PARAM2, code.ptr[code.r15 + code.GetJitStateInfo().offsetof_fpsr_exc]); code.mov(code.ABI_PARAM3.cvt32(), ctx.FPCR()); code.CallFunction(lut.at(std::make_tuple(fsize, rounding, exact))); } void EmitX64::EmitFPRoundInt32(EmitContext& ctx, IR::Inst* inst) { EmitFPRound(code, ctx, inst, 32); } void EmitX64::EmitFPRoundInt64(EmitContext& ctx, IR::Inst* inst) { EmitFPRound(code, ctx, inst, 64); } template static void EmitFPRSqrtEstimate(BlockOfCode& code, EmitContext& ctx, IR::Inst* inst) { auto args = ctx.reg_alloc.GetArgumentInfo(inst); ctx.reg_alloc.HostCall(inst, args[0]); code.mov(code.ABI_PARAM2.cvt32(), ctx.FPCR()); code.lea(code.ABI_PARAM3, code.ptr[code.r15 + code.GetJitStateInfo().offsetof_fpsr_exc]); code.CallFunction(&FP::FPRSqrtEstimate); } void EmitX64::EmitFPRSqrtEstimate32(EmitContext& ctx, IR::Inst* inst) { EmitFPRSqrtEstimate(code, ctx, inst); } void EmitX64::EmitFPRSqrtEstimate64(EmitContext& ctx, IR::Inst* inst) { EmitFPRSqrtEstimate(code, ctx, inst); } template static void EmitFPRSqrtStepFused(BlockOfCode& code, EmitContext& ctx, IR::Inst* inst) { auto args = ctx.reg_alloc.GetArgumentInfo(inst); ctx.reg_alloc.HostCall(inst, args[0], args[1]); code.mov(code.ABI_PARAM3.cvt32(), ctx.FPCR()); code.lea(code.ABI_PARAM4, code.ptr[code.r15 + code.GetJitStateInfo().offsetof_fpsr_exc]); code.CallFunction(&FP::FPRSqrtStepFused); } void EmitX64::EmitFPRSqrtStepFused32(EmitContext& ctx, IR::Inst* inst) { EmitFPRSqrtStepFused(code, ctx, inst); } void EmitX64::EmitFPRSqrtStepFused64(EmitContext& ctx, IR::Inst* inst) { EmitFPRSqrtStepFused(code, ctx, inst); } void EmitX64::EmitFPSqrt32(EmitContext& ctx, IR::Inst* inst) { FPTwoOp<32>(code, ctx, inst, &Xbyak::CodeGenerator::sqrtss); } void EmitX64::EmitFPSqrt64(EmitContext& ctx, IR::Inst* inst) { FPTwoOp<64>(code, ctx, inst, &Xbyak::CodeGenerator::sqrtsd); } void EmitX64::EmitFPSub32(EmitContext& ctx, IR::Inst* inst) { FPThreeOp<32>(code, ctx, inst, &Xbyak::CodeGenerator::subss); } void EmitX64::EmitFPSub64(EmitContext& ctx, IR::Inst* inst) { FPThreeOp<64>(code, ctx, inst, &Xbyak::CodeGenerator::subsd); } static Xbyak::Reg64 SetFpscrNzcvFromFlags(BlockOfCode& code, EmitContext& ctx) { ctx.reg_alloc.ScratchGpr({HostLoc::RCX}); // shifting requires use of cl Xbyak::Reg64 nzcv = ctx.reg_alloc.ScratchGpr(); // x64 flags ARM flags // ZF PF CF NZCV // Unordered 1 1 1 0011 // Greater than 0 0 0 0010 // Less than 0 0 1 1000 // Equal 1 0 0 0110 // // Thus we can take use ZF:CF as an index into an array like so: // x64 ARM ARM as x64 // ZF:CF NZCV NZ-----C-------V // 0 0010 0000000100000000 = 0x0100 // 1 1000 1000000000000000 = 0x8000 // 2 0110 0100000100000000 = 0x4100 // 3 0011 0000000100000001 = 0x0101 code.mov(nzcv, 0x0101'4100'8000'0100); code.sete(cl); code.rcl(cl, 5); // cl = ZF:CF:0000 code.shr(nzcv, cl); return nzcv; } void EmitX64::EmitFPCompare32(EmitContext& ctx, IR::Inst* inst) { auto args = ctx.reg_alloc.GetArgumentInfo(inst); Xbyak::Xmm reg_a = ctx.reg_alloc.UseXmm(args[0]); Xbyak::Xmm reg_b = ctx.reg_alloc.UseXmm(args[1]); bool exc_on_qnan = args[2].GetImmediateU1(); if (exc_on_qnan) { code.comiss(reg_a, reg_b); } else { code.ucomiss(reg_a, reg_b); } Xbyak::Reg64 nzcv = SetFpscrNzcvFromFlags(code, ctx); ctx.reg_alloc.DefineValue(inst, nzcv); } void EmitX64::EmitFPCompare64(EmitContext& ctx, IR::Inst* inst) { auto args = ctx.reg_alloc.GetArgumentInfo(inst); Xbyak::Xmm reg_a = ctx.reg_alloc.UseXmm(args[0]); Xbyak::Xmm reg_b = ctx.reg_alloc.UseXmm(args[1]); bool exc_on_qnan = args[2].GetImmediateU1(); if (exc_on_qnan) { code.comisd(reg_a, reg_b); } else { code.ucomisd(reg_a, reg_b); } Xbyak::Reg64 nzcv = SetFpscrNzcvFromFlags(code, ctx); ctx.reg_alloc.DefineValue(inst, nzcv); } void EmitX64::EmitFPSingleToDouble(EmitContext& ctx, IR::Inst* inst) { auto args = ctx.reg_alloc.GetArgumentInfo(inst); Xbyak::Xmm result = ctx.reg_alloc.UseScratchXmm(args[0]); Xbyak::Reg64 gpr_scratch = ctx.reg_alloc.ScratchGpr(); if (ctx.FPSCR_FTZ()) { DenormalsAreZero<32>(code, result, gpr_scratch); } code.cvtss2sd(result, result); if (ctx.FPSCR_DN()) { DefaultNaN<64>(code, result); } ctx.reg_alloc.DefineValue(inst, result); } void EmitX64::EmitFPDoubleToSingle(EmitContext& ctx, IR::Inst* inst) { auto args = ctx.reg_alloc.GetArgumentInfo(inst); Xbyak::Xmm result = ctx.reg_alloc.UseScratchXmm(args[0]); Xbyak::Reg64 gpr_scratch = ctx.reg_alloc.ScratchGpr(); if (ctx.FPSCR_FTZ()) { DenormalsAreZero<64>(code, result, gpr_scratch); } code.cvtsd2ss(result, result); if (ctx.FPSCR_DN()) { DefaultNaN<32>(code, result); } ctx.reg_alloc.DefineValue(inst, result); } static void EmitFPToFixed(BlockOfCode& code, EmitContext& ctx, IR::Inst* inst, size_t fsize, bool unsigned_, size_t isize) { auto args = ctx.reg_alloc.GetArgumentInfo(inst); const size_t fbits = args[1].GetImmediateU8(); const auto rounding = static_cast(args[2].GetImmediateU8()); if (code.DoesCpuSupport(Xbyak::util::Cpu::tSSE41) && rounding != FP::RoundingMode::ToNearest_TieAwayFromZero){ const Xbyak::Xmm src = ctx.reg_alloc.UseScratchXmm(args[0]); const int round_imm = [&]{ switch (rounding) { case FP::RoundingMode::ToNearest_TieEven: default: return 0b00; case FP::RoundingMode::TowardsPlusInfinity: return 0b10; case FP::RoundingMode::TowardsMinusInfinity: return 0b01; case FP::RoundingMode::TowardsZero: return 0b11; } }(); const Xbyak::Xmm scratch = ctx.reg_alloc.ScratchXmm(); const Xbyak::Reg64 result = ctx.reg_alloc.ScratchGpr().cvt64(); if (fsize == 64) { if (fbits != 0) { const u64 scale_factor = static_cast((fbits + 1023) << 52); code.mulsd(src, code.MConst(xword, scale_factor)); } code.roundsd(src, src, round_imm); } else { if (fbits != 0) { const u32 scale_factor = static_cast((fbits + 127) << 23); code.mulss(src, code.MConst(xword, scale_factor)); } code.roundss(src, src, round_imm); code.cvtss2sd(src, src); } ZeroIfNaN<64>(code, src, scratch); if (isize == 64) { Xbyak::Label saturate_max, end; code.maxsd(src, code.MConst(xword, unsigned_ ? f64_min_u64 : f64_min_s64)); code.movsd(scratch, code.MConst(xword, unsigned_ ? f64_max_u64_lim : f64_max_s64_lim)); code.comisd(scratch, src); code.jna(saturate_max, code.T_NEAR); if (unsigned_) { Xbyak::Label below_max; code.movsd(scratch, code.MConst(xword, f64_max_s64_lim)); code.comisd(src, scratch); code.jb(below_max); code.subsd(src, scratch); code.cvttsd2si(result, src); code.btc(result, 63); code.jmp(end); code.L(below_max); } code.cvttsd2si(result, src); // 64 bit gpr code.L(end); code.SwitchToFarCode(); code.L(saturate_max); code.mov(result, unsigned_ ? 0xFFFF'FFFF'FFFF'FFFF : 0x7FFF'FFFF'FFFF'FFFF); code.jmp(end, code.T_NEAR); code.SwitchToNearCode(); } else { code.minsd(src, code.MConst(xword, unsigned_ ? f64_max_u32 : f64_max_s32)); code.maxsd(src, code.MConst(xword, unsigned_ ? f64_min_u32 : f64_min_s32)); code.cvttsd2si(result, src); // 64 bit gpr } ctx.reg_alloc.DefineValue(inst, result); return; } using fsize_list = mp::list, mp::vlift>; using unsigned_list = mp::list, mp::vlift>; using isize_list = mp::list, mp::vlift>; using rounding_list = mp::list< std::integral_constant, std::integral_constant, std::integral_constant, std::integral_constant, std::integral_constant >; using key_type = std::tuple; using value_type = u64(*)(u64, u8, FP::FPSR&, FP::FPCR); static const auto lut = mp::GenerateLookupTableFromList( [](auto args) { return std::pair{ mp::to_tuple, static_cast( [](u64 input, u8 fbits, FP::FPSR& fpsr, FP::FPCR fpcr) { constexpr auto t = mp::to_tuple; constexpr size_t fsize = std::get<0>(t); constexpr bool unsigned_ = std::get<1>(t); constexpr size_t isize = std::get<2>(t); constexpr FP::RoundingMode rounding_mode = std::get<3>(t); using InputSize = mp::unsigned_integer_of_size; return FP::FPToFixed(isize, static_cast(input), fbits, unsigned_, fpcr, rounding_mode, fpsr); } ) }; }, mp::cartesian_product{} ); ctx.reg_alloc.HostCall(inst, args[0], args[1]); code.lea(code.ABI_PARAM3, code.ptr[code.r15 + code.GetJitStateInfo().offsetof_fpsr_exc]); code.mov(code.ABI_PARAM4.cvt32(), ctx.FPCR()); code.CallFunction(lut.at(std::make_tuple(fsize, unsigned_, isize, rounding))); } void EmitX64::EmitFPDoubleToFixedS32(EmitContext& ctx, IR::Inst* inst) { EmitFPToFixed(code, ctx, inst, 64, false, 32); } void EmitX64::EmitFPDoubleToFixedS64(EmitContext& ctx, IR::Inst* inst) { EmitFPToFixed(code, ctx, inst, 64, false, 64); } void EmitX64::EmitFPDoubleToFixedU32(EmitContext& ctx, IR::Inst* inst) { EmitFPToFixed(code, ctx, inst, 64, true, 32); } void EmitX64::EmitFPDoubleToFixedU64(EmitContext& ctx, IR::Inst* inst) { EmitFPToFixed(code, ctx, inst, 64, true, 64); } void EmitX64::EmitFPSingleToFixedS32(EmitContext& ctx, IR::Inst* inst) { EmitFPToFixed(code, ctx, inst, 32, false, 32); } void EmitX64::EmitFPSingleToFixedS64(EmitContext& ctx, IR::Inst* inst) { EmitFPToFixed(code, ctx, inst, 32, false, 64); } void EmitX64::EmitFPSingleToFixedU32(EmitContext& ctx, IR::Inst* inst) { EmitFPToFixed(code, ctx, inst, 32, true, 32); } void EmitX64::EmitFPSingleToFixedU64(EmitContext& ctx, IR::Inst* inst) { EmitFPToFixed(code, ctx, inst, 32, true, 64); } void EmitX64::EmitFPS32ToSingle(EmitContext& ctx, IR::Inst* inst) { auto args = ctx.reg_alloc.GetArgumentInfo(inst); Xbyak::Reg32 from = ctx.reg_alloc.UseGpr(args[0]).cvt32(); Xbyak::Xmm to = ctx.reg_alloc.ScratchXmm(); bool round_to_nearest = args[1].GetImmediateU1(); ASSERT_MSG(!round_to_nearest, "round_to_nearest unimplemented"); code.cvtsi2ss(to, from); ctx.reg_alloc.DefineValue(inst, to); } void EmitX64::EmitFPU32ToSingle(EmitContext& ctx, IR::Inst* inst) { auto args = ctx.reg_alloc.GetArgumentInfo(inst); const Xbyak::Xmm to = ctx.reg_alloc.ScratchXmm(); const bool round_to_nearest = args[1].GetImmediateU1(); ASSERT_MSG(!round_to_nearest, "round_to_nearest unimplemented"); if (code.DoesCpuSupport(Xbyak::util::Cpu::tAVX512F)) { const Xbyak::Reg64 from = ctx.reg_alloc.UseGpr(args[0]); code.vcvtusi2ss(to, to, from.cvt32()); } else { // We are using a 64-bit GPR register to ensure we don't end up treating the input as signed const Xbyak::Reg64 from = ctx.reg_alloc.UseScratchGpr(args[0]); code.mov(from.cvt32(), from.cvt32()); // TODO: Verify if this is necessary code.cvtsi2ss(to, from); } ctx.reg_alloc.DefineValue(inst, to); } void EmitX64::EmitFPS32ToDouble(EmitContext& ctx, IR::Inst* inst) { auto args = ctx.reg_alloc.GetArgumentInfo(inst); Xbyak::Reg32 from = ctx.reg_alloc.UseGpr(args[0]).cvt32(); Xbyak::Xmm to = ctx.reg_alloc.ScratchXmm(); bool round_to_nearest = args[1].GetImmediateU1(); ASSERT_MSG(!round_to_nearest, "round_to_nearest unimplemented"); code.cvtsi2sd(to, from); ctx.reg_alloc.DefineValue(inst, to); } void EmitX64::EmitFPS64ToDouble(EmitContext& ctx, IR::Inst* inst) { auto args = ctx.reg_alloc.GetArgumentInfo(inst); const Xbyak::Reg64 from = ctx.reg_alloc.UseGpr(args[0]); const Xbyak::Xmm result = ctx.reg_alloc.ScratchXmm(); const bool round_to_nearest = args[1].GetImmediateU1(); ASSERT_MSG(!round_to_nearest, "round_to_nearest unimplemented"); code.cvtsi2sd(result, from); ctx.reg_alloc.DefineValue(inst, result); } void EmitX64::EmitFPS64ToSingle(EmitContext& ctx, IR::Inst* inst) { auto args = ctx.reg_alloc.GetArgumentInfo(inst); const Xbyak::Reg64 from = ctx.reg_alloc.UseGpr(args[0]); const Xbyak::Xmm result = ctx.reg_alloc.ScratchXmm(); const bool round_to_nearest = args[1].GetImmediateU1(); ASSERT_MSG(!round_to_nearest, "round_to_nearest unimplemented"); code.cvtsi2ss(result, from); ctx.reg_alloc.DefineValue(inst, result); } void EmitX64::EmitFPU32ToDouble(EmitContext& ctx, IR::Inst* inst) { auto args = ctx.reg_alloc.GetArgumentInfo(inst); const Xbyak::Xmm to = ctx.reg_alloc.ScratchXmm(); const bool round_to_nearest = args[1].GetImmediateU1(); ASSERT_MSG(!round_to_nearest, "round_to_nearest unimplemented"); if (code.DoesCpuSupport(Xbyak::util::Cpu::tAVX512F)) { const Xbyak::Reg64 from = ctx.reg_alloc.UseGpr(args[0]); code.vcvtusi2sd(to, to, from.cvt32()); } else { // We are using a 64-bit GPR register to ensure we don't end up treating the input as signed const Xbyak::Reg64 from = ctx.reg_alloc.UseScratchGpr(args[0]); code.mov(from.cvt32(), from.cvt32()); // TODO: Verify if this is necessary code.cvtsi2sd(to, from); } ctx.reg_alloc.DefineValue(inst, to); } void EmitX64::EmitFPU64ToDouble(EmitContext& ctx, IR::Inst* inst) { auto args = ctx.reg_alloc.GetArgumentInfo(inst); const Xbyak::Reg64 from = ctx.reg_alloc.UseGpr(args[0]); const Xbyak::Xmm result = ctx.reg_alloc.ScratchXmm(); const bool round_to_nearest = args[1].GetImmediateU1(); ASSERT_MSG(!round_to_nearest, "round_to_nearest unimplemented"); if (code.DoesCpuSupport(Xbyak::util::Cpu::tAVX512F)) { code.vcvtusi2sd(result, result, from); } else { const Xbyak::Xmm tmp = ctx.reg_alloc.ScratchXmm(); code.movq(tmp, from); code.punpckldq(tmp, code.MConst(xword, 0x4530000043300000, 0)); code.subpd(tmp, code.MConst(xword, 0x4330000000000000, 0x4530000000000000)); code.pshufd(result, tmp, 0b01001110); code.addpd(result, tmp); if (ctx.FPSCR_RMode() == FP::RoundingMode::TowardsMinusInfinity) { code.pand(result, code.MConst(xword, f64_non_sign_mask)); } } ctx.reg_alloc.DefineValue(inst, result); } void EmitX64::EmitFPU64ToSingle(EmitContext& ctx, IR::Inst* inst) { auto args = ctx.reg_alloc.GetArgumentInfo(inst); const Xbyak::Xmm result = ctx.reg_alloc.ScratchXmm(); const bool round_to_nearest = args[1].GetImmediateU1(); ASSERT_MSG(!round_to_nearest, "round_to_nearest unimplemented"); if (code.DoesCpuSupport(Xbyak::util::Cpu::tAVX512F)) { const Xbyak::Reg64 from = ctx.reg_alloc.UseGpr(args[0]); code.vcvtusi2ss(result, result, from); } else { const Xbyak::Reg64 from = ctx.reg_alloc.UseScratchGpr(args[0]); code.pxor(result, result); Xbyak::Label negative; Xbyak::Label end; code.test(from, from); code.js(negative); code.cvtsi2ss(result, from); code.jmp(end); code.L(negative); const Xbyak::Reg64 tmp = ctx.reg_alloc.ScratchGpr(); code.mov(tmp, from); code.shr(tmp, 1); code.and_(from.cvt32(), 1); code.or_(from, tmp); code.cvtsi2ss(result, from); code.addss(result, result); code.L(end); } ctx.reg_alloc.DefineValue(inst, result); } } // namespace Dynarmic::BackendX64