mirror of
https://github.com/luau-lang/luau.git
synced 2025-01-19 01:18:03 +00:00
1212fdacbf
Once again, all of our changes this week are for new type solver and the JIT. In the new type solver, we fixed cyclic type alias handling and multiple stability issues. In the JIT, our main progress was for arm64, where, after lowering 36% of instructions, we start seeing first Luau functions executing natively. For x64, we performed code cleanup and refactoring to allow for future optimizations.
82 lines
2.4 KiB
C++
82 lines
2.4 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
|
|
{
|
|
|
|
// When IrInst operands are used, current instruction index is required to track lifetime
|
|
// In all other calls it is ok to omit the argument
|
|
constexpr uint32_t kInvalidInstIdx = ~0u;
|
|
|
|
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;
|
|
|
|
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
|