mirror of
https://github.com/luau-lang/luau.git
synced 2025-05-04 10:33:46 +01:00
e25de95445
* Fixed indexing table intersections using `x["prop"]` syntax: https://github.com/Roblox/luau/pull/971 * Add console output codepage for Windows: https://github.com/Roblox/luau/pull/967 * Added `Frontend::parse` for a fast source graph preparation * luau_load should check GC * Work toward a type-diff system for nicer error messages New Solver * Correctly suppress errors in more cases * Further improvements to typechecking of function calls and return statements * Crash fixes * Propagate refinements drawn from the condition of a while loop into the loop body JIT * Fix accidental bailout for math.frexp/modf/sign in A64 * Work toward bringing type annotation info in * Do not propagate Luau IR constants of wrong type into load instructions * CHECK_SAFEENV exits to VM on failure * Implement error handling in A64 reg allocator * Inline the string.len builtin * Do not enter native code of a function if arguments don’t match --------- Co-authored-by: Arseny Kapoulkine <arseny.kapoulkine@gmail.com> Co-authored-by: Vyacheslav Egorov <vegorov@roblox.com>
86 lines
2.3 KiB
C++
86 lines
2.3 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/IrData.h"
|
|
#include "Luau/RegisterA64.h"
|
|
|
|
#include <initializer_list>
|
|
#include <utility>
|
|
#include <vector>
|
|
|
|
namespace Luau
|
|
{
|
|
namespace CodeGen
|
|
{
|
|
namespace A64
|
|
{
|
|
|
|
class AssemblyBuilderA64;
|
|
|
|
struct IrRegAllocA64
|
|
{
|
|
IrRegAllocA64(IrFunction& function, std::initializer_list<std::pair<RegisterA64, RegisterA64>> regs);
|
|
|
|
RegisterA64 allocReg(KindA64 kind, uint32_t index);
|
|
RegisterA64 allocTemp(KindA64 kind);
|
|
RegisterA64 allocReuse(KindA64 kind, uint32_t index, std::initializer_list<IrOp> oprefs);
|
|
|
|
RegisterA64 takeReg(RegisterA64 reg, uint32_t index);
|
|
|
|
void freeReg(RegisterA64 reg);
|
|
|
|
void freeLastUseReg(IrInst& target, uint32_t index);
|
|
void freeLastUseRegs(const IrInst& inst, uint32_t index);
|
|
|
|
void freeTempRegs();
|
|
|
|
// Spills all live registers that outlive current instruction; all allocated registers are assumed to be undefined
|
|
size_t spill(AssemblyBuilderA64& build, uint32_t index, std::initializer_list<RegisterA64> live = {});
|
|
|
|
// Restores registers starting from the offset returned by spill(); all spills will be restored to the original registers
|
|
void restore(AssemblyBuilderA64& build, size_t start);
|
|
|
|
// Restores register for a single instruction; may not assign the previously used register!
|
|
void restoreReg(AssemblyBuilderA64& build, IrInst& inst);
|
|
|
|
void assertNoSpills() const;
|
|
|
|
struct Set
|
|
{
|
|
// which registers are in the set that the allocator manages (initialized at construction)
|
|
uint32_t base = 0;
|
|
|
|
// which subset of initial set is free
|
|
uint32_t free = 0;
|
|
|
|
// which subset of initial set is allocated as temporary
|
|
uint32_t temp = 0;
|
|
|
|
// which instruction is defining which register (for spilling); only valid if not free and not temp
|
|
uint32_t defs[32];
|
|
};
|
|
|
|
struct Spill
|
|
{
|
|
uint32_t inst;
|
|
|
|
RegisterA64 origin;
|
|
int8_t slot;
|
|
};
|
|
|
|
Set& getSet(KindA64 kind);
|
|
|
|
IrFunction& function;
|
|
Set gpr, simd;
|
|
|
|
std::vector<Spill> spills;
|
|
|
|
// which 8-byte slots are free
|
|
uint32_t freeSpillSlots = 0;
|
|
|
|
bool error = false;
|
|
};
|
|
|
|
} // namespace A64
|
|
} // namespace CodeGen
|
|
} // namespace Luau
|