mirror of
https://github.com/luau-lang/luau.git
synced 2024-12-13 13:30:40 +00:00
ba67fb275e
* `table.sort` was improved further. It now guarentees N*log(N) time complexity in the worst case. * Fix https://github.com/Roblox/luau/issues/880 We are also working on fixing final bugs and crashes in the new type solver. On the CodeGen front we have a few things going on: * We have a smarter register allocator for the x86 JIT * We lower more instructions on arm64 * The vector constructor builtin is now translated to IR --------- Co-authored-by: Arseny Kapoulkine <arseny.kapoulkine@gmail.com> Co-authored-by: Vyacheslav Egorov <vegorov@roblox.com>
79 lines
2.2 KiB
C++
79 lines
2.2 KiB
C++
// This file is part of the Luau programming language and is licensed under MIT License; see LICENSE.txt for details
|
|
#pragma once
|
|
|
|
#include "Luau/AssemblyBuilderX64.h"
|
|
#include "Luau/IrData.h"
|
|
#include "Luau/OperandX64.h"
|
|
#include "Luau/RegisterX64.h"
|
|
|
|
#include <array>
|
|
|
|
// TODO: call wrapper can be used to suggest target registers for ScopedRegX64 to compute data into argument registers directly
|
|
|
|
namespace Luau
|
|
{
|
|
namespace CodeGen
|
|
{
|
|
namespace X64
|
|
{
|
|
|
|
struct IrRegAllocX64;
|
|
struct ScopedRegX64;
|
|
|
|
struct CallArgument
|
|
{
|
|
SizeX64 targetSize = SizeX64::none;
|
|
|
|
OperandX64 source = noreg;
|
|
IrOp sourceOp;
|
|
|
|
OperandX64 target = noreg;
|
|
bool candidate = true;
|
|
};
|
|
|
|
class IrCallWrapperX64
|
|
{
|
|
public:
|
|
IrCallWrapperX64(IrRegAllocX64& regs, AssemblyBuilderX64& build, uint32_t instIdx = kInvalidInstIdx);
|
|
|
|
void addArgument(SizeX64 targetSize, OperandX64 source, IrOp sourceOp = {});
|
|
void addArgument(SizeX64 targetSize, ScopedRegX64& scopedReg);
|
|
|
|
void call(const OperandX64& func);
|
|
|
|
IrRegAllocX64& regs;
|
|
AssemblyBuilderX64& build;
|
|
uint32_t instIdx = ~0u;
|
|
|
|
private:
|
|
void assignTargetRegisters();
|
|
void countRegisterUses();
|
|
CallArgument* findNonInterferingArgument();
|
|
bool interferesWithOperand(const OperandX64& op, RegisterX64 reg) const;
|
|
bool interferesWithActiveSources(const CallArgument& targetArg, int targetArgIndex) const;
|
|
bool interferesWithActiveTarget(RegisterX64 sourceReg) const;
|
|
void moveToTarget(CallArgument& arg);
|
|
void freeSourceRegisters(CallArgument& arg);
|
|
void renameRegister(RegisterX64& target, RegisterX64 reg, RegisterX64 replacement);
|
|
void renameSourceRegisters(RegisterX64 reg, RegisterX64 replacement);
|
|
RegisterX64 findConflictingTarget() const;
|
|
void renameConflictingRegister(RegisterX64 conflict);
|
|
|
|
int getRegisterUses(RegisterX64 reg) const;
|
|
void addRegisterUse(RegisterX64 reg);
|
|
void removeRegisterUse(RegisterX64 reg);
|
|
|
|
static const int kMaxCallArguments = 6;
|
|
std::array<CallArgument, kMaxCallArguments> args;
|
|
int argCount = 0;
|
|
|
|
OperandX64 funcOp;
|
|
|
|
// Internal counters for remaining register use counts
|
|
std::array<uint8_t, 16> gprUses;
|
|
std::array<uint8_t, 16> xmmUses;
|
|
};
|
|
|
|
} // namespace X64
|
|
} // namespace CodeGen
|
|
} // namespace Luau
|