mirror of
https://github.com/FEX-Emu/FEX.git
synced 2026-10-14 11:01:01 +02:00
Ryan Houdek
199cfd76d8
I had to change how blocks are represented to make it easier to parse This required a fairly substantial refactor that makes it so blocks are represented differently and we can walk them sequentially. This will make future analysis easier to deal with. Had to rewrite the passes and core's parsing of the IR afterwards. Moved RA in to a optimization pass to be shared between the JIT backends This works because x86-64 and AArch64 RA can be identical. Still doesn't support PHI nodes or spilling correctly, this is the first step in the process of getting there.
367 lines
14 KiB
Python
367 lines
14 KiB
Python
import json
|
|
import sys
|
|
|
|
# Print out enum values
|
|
def print_enums(ops, defines):
|
|
output_file.write("#ifdef IROP_ENUM\n")
|
|
output_file.write("enum IROps : uint8_t {\n")
|
|
|
|
for op_key, op_vals in ops.items():
|
|
output_file.write("\t\tOP_%s,\n" % op_key.upper())
|
|
|
|
output_file.write("};\n")
|
|
|
|
output_file.write("#undef IROP_ENUM\n")
|
|
output_file.write("#endif\n\n")
|
|
|
|
# Print out struct definitions
|
|
def print_ir_structs(ops, defines):
|
|
output_file.write("#ifdef IROP_STRUCTS\n")
|
|
|
|
# Print out defines here
|
|
for op_val in defines:
|
|
output_file.write("\t%s;\n" % op_val)
|
|
|
|
output_file.write("// Default structs\n")
|
|
output_file.write("struct __attribute__((packed)) IROp_Header {\n")
|
|
output_file.write("\tvoid* Data[0];\n")
|
|
output_file.write("\tIROps Op;\n\n")
|
|
output_file.write("\tuint8_t Size;\n")
|
|
output_file.write("\tuint8_t NumArgs;\n")
|
|
output_file.write("\tuint8_t Elements : 7;\n")
|
|
output_file.write("\tbool HasDest : 1;\n")
|
|
|
|
output_file.write("\ttemplate<typename T>\n")
|
|
output_file.write("\tT const* C() const { return reinterpret_cast<T const*>(Data); }\n")
|
|
output_file.write("\ttemplate<typename T>\n")
|
|
output_file.write("\tT* CW() { return reinterpret_cast<T*>(Data); }\n")
|
|
|
|
output_file.write("\tOrderedNodeWrapper Args[0];\n")
|
|
|
|
output_file.write("};\n\n");
|
|
|
|
output_file.write("struct __attribute__((packed)) IROp_Empty {\n")
|
|
output_file.write("\tIROp_Header Header;\n")
|
|
output_file.write("};\n\n")
|
|
|
|
output_file.write("// User defined IR Op structs\n")
|
|
for op_key, op_vals in ops.items():
|
|
SSAArgs = 0
|
|
HasArgs = False
|
|
HasSSANames = False
|
|
|
|
if ("SSAArgs" in op_vals):
|
|
SSAArgs = int(op_vals["SSAArgs"])
|
|
|
|
if ("Args" in op_vals and len(op_vals["Args"]) != 0):
|
|
HasArgs = True
|
|
|
|
if ("SSANames" in op_vals and len(op_vals["SSANames"]) != 0):
|
|
HasSSANames = True
|
|
|
|
if (HasArgs or SSAArgs != 0):
|
|
output_file.write("struct __attribute__((packed)) IROp_%s {\n" % op_key)
|
|
output_file.write("\tIROp_Header Header;\n\n")
|
|
|
|
# SSA arguments have a hard requirement to appear after the header
|
|
if (SSAArgs != 0):
|
|
if (HasSSANames):
|
|
for i in range(0, SSAArgs):
|
|
output_file.write("\tOrderedNodeWrapper %s;\n" % (op_vals["SSANames"][i]));
|
|
|
|
else:
|
|
output_file.write("private:\n")
|
|
for i in range(0, SSAArgs):
|
|
output_file.write("\tuint64_t : (sizeof(OrderedNodeWrapper) * 8);\n");
|
|
output_file.write("public:\n")
|
|
|
|
if (HasArgs):
|
|
output_file.write("\t// User defined data\n")
|
|
|
|
# Print out arguments in IR Op
|
|
for i in range(0, len(op_vals["Args"]), 2):
|
|
data_type = op_vals["Args"][i]
|
|
data_name = op_vals["Args"][i+1]
|
|
output_file.write("\t%s %s;\n" % (data_type, data_name))
|
|
|
|
output_file.write("};\n")
|
|
else:
|
|
output_file.write("using IROp_%s = IROp_Empty;\n" % op_key)
|
|
|
|
# Add a static assert that the IR ops must be pod
|
|
output_file.write("static_assert(std::is_pod<IROp_%s>::value);\n\n" % op_key)
|
|
|
|
output_file.write("#undef IROP_STRUCTS\n")
|
|
output_file.write("#endif\n\n")
|
|
|
|
# Print out const expression to calculate IR Op sizes
|
|
def print_ir_sizes(ops, defines):
|
|
output_file.write("#ifdef IROP_SIZES\n")
|
|
|
|
output_file.write("constexpr std::array<size_t, IROps::OP_LAST + 1> IRSizes = {\n")
|
|
for op_key, op_vals in ops.items():
|
|
if ("Last" in op_vals):
|
|
output_file.write("\t-1ULL,\n")
|
|
else:
|
|
output_file.write("\tsizeof(IROp_%s),\n" % op_key)
|
|
|
|
output_file.write("};\n\n")
|
|
|
|
output_file.write("// Make sure our array maps directly to the IROps enum\n")
|
|
output_file.write("static_assert(IRSizes[IROps::OP_LAST] == -1ULL);\n\n")
|
|
|
|
output_file.write("[[maybe_unused]] static size_t GetSize(IROps Op) { return IRSizes[Op]; }\n\n")
|
|
|
|
output_file.write("std::string_view const& GetName(IROps Op);\n")
|
|
|
|
output_file.write("#undef IROP_SIZES\n")
|
|
output_file.write("#endif\n\n")
|
|
|
|
# Print out the name printer implementation
|
|
def print_ir_getname(ops, defines):
|
|
output_file.write("#ifdef IROP_GETNAME_IMPL\n")
|
|
output_file.write("constexpr std::array<std::string_view const, OP_LAST + 1> IRNames = {\n")
|
|
for op_key, op_vals in ops.items():
|
|
output_file.write("\t\"%s\",\n" % op_key)
|
|
|
|
output_file.write("};\n\n")
|
|
|
|
output_file.write("static_assert(IRNames[OP_LAST] == \"Last\");\n\n")
|
|
|
|
output_file.write("std::string_view const& GetName(IROps Op) {\n")
|
|
output_file.write(" return IRNames[Op];\n")
|
|
output_file.write("}\n")
|
|
|
|
output_file.write("#undef IROP_GETNAME_IMPL\n")
|
|
output_file.write("#endif\n\n")
|
|
|
|
# Print out IR argument printing
|
|
def print_ir_arg_printer(ops, defines):
|
|
output_file.write("#ifdef IROP_ARGPRINTER_HELPER\n")
|
|
output_file.write("switch (IROp->Op) {\n")
|
|
for op_key, op_vals in ops.items():
|
|
if not ("Last" in op_vals):
|
|
SSAArgs = 0
|
|
HasArgs = False
|
|
|
|
# Does this not want a printer?
|
|
if ("ArgPrinter" in op_vals and op_vals["ArgPrinter"] == False):
|
|
continue
|
|
|
|
if ("SSAArgs" in op_vals):
|
|
SSAArgs = int(op_vals["SSAArgs"])
|
|
|
|
if ("Args" in op_vals and len(op_vals["Args"]) != 0):
|
|
HasArgs = True
|
|
|
|
output_file.write("case IROps::OP_%s: {\n" % op_key.upper())
|
|
if (HasArgs or SSAArgs != 0):
|
|
output_file.write("\tauto Op = IROp->C<IR::IROp_%s>();\n" % op_key)
|
|
output_file.write("\t*out << \" \";\n")
|
|
|
|
# Print SSA args first
|
|
if (SSAArgs != 0):
|
|
for i in range(0, SSAArgs):
|
|
LastArg = (SSAArgs - i - 1) == 0 and not HasArgs
|
|
output_file.write("\tPrintArg(out, IR, Op->Header.Args[%d]);\n" % i)
|
|
if not (LastArg):
|
|
output_file.write("\t*out << \", \";\n")
|
|
|
|
# Now print user defined arguments
|
|
if (HasArgs):
|
|
ArgCount = len(op_vals["Args"])
|
|
|
|
for i in range(0, len(op_vals["Args"]), 2):
|
|
data_name = op_vals["Args"][i+1]
|
|
LastArg = (ArgCount - i - 2) == 0
|
|
CondArg2 = (", ", "")
|
|
output_file.write("\tPrintArg(out, IR, Op->%s);\n" % data_name)
|
|
if not (LastArg):
|
|
output_file.write("\t*out << \", \";\n")
|
|
|
|
output_file.write("break;\n")
|
|
output_file.write("}\n")
|
|
|
|
|
|
output_file.write("#undef IROP_ARGPRINTER_HELPER\n")
|
|
output_file.write("#endif\n")
|
|
|
|
# Print out IR allocator helpers
|
|
def print_ir_allocator_helpers(ops, defines):
|
|
output_file.write("#ifdef IROP_ALLOCATE_HELPERS\n")
|
|
|
|
output_file.write("\ttemplate <class T>\n")
|
|
output_file.write("\tstruct Wrapper final {\n")
|
|
output_file.write("\t\tT *first;\n")
|
|
output_file.write("\t\tOrderedNode *Node; ///< Actual offset of this IR in ths list\n")
|
|
output_file.write("\t\t\n")
|
|
output_file.write("\t\toperator Wrapper<IROp_Header>() const { return Wrapper<IROp_Header> {reinterpret_cast<IROp_Header*>(first), Node}; }\n")
|
|
output_file.write("\t\toperator OrderedNode *() { return Node; }\n")
|
|
output_file.write("\t\toperator OpNodeWrapper () { return Node->Header.Value; }\n")
|
|
output_file.write("\t};\n")
|
|
|
|
output_file.write("\ttemplate <class T>\n")
|
|
output_file.write("\tusing IRPair = Wrapper<T>;\n\n")
|
|
|
|
output_file.write("\tIRPair<IROp_Header> AllocateRawOp(size_t HeaderSize) {\n")
|
|
output_file.write("\t\tauto Op = reinterpret_cast<IROp_Header*>(Data.Allocate(HeaderSize));\n")
|
|
output_file.write("\t\tmemset(Op, 0, HeaderSize);\n")
|
|
output_file.write("\t\tOp->Op = IROps::OP_DUMMY;\n")
|
|
output_file.write("\t\treturn IRPair<IROp_Header>{Op, CreateNode(Op)};\n")
|
|
output_file.write("\t}\n\n")
|
|
|
|
output_file.write("\ttemplate<class T, IROps T2>\n")
|
|
output_file.write("\tT *AllocateOrphanOp() {\n")
|
|
output_file.write("\t\tsize_t Size = FEXCore::IR::GetSize(T2);\n")
|
|
output_file.write("\t\tauto Op = reinterpret_cast<T*>(Data.Allocate(Size));\n")
|
|
output_file.write("\t\tmemset(Op, 0, Size);\n")
|
|
output_file.write("\t\tOp->Header.Op = T2;\n")
|
|
output_file.write("\t\treturn Op;\n")
|
|
output_file.write("\t}\n\n")
|
|
|
|
output_file.write("\ttemplate<class T, IROps T2>\n")
|
|
output_file.write("\tIRPair<T> AllocateOp() {\n")
|
|
output_file.write("\t\tsize_t Size = FEXCore::IR::GetSize(T2);\n")
|
|
output_file.write("\t\tauto Op = reinterpret_cast<T*>(Data.Allocate(Size));\n")
|
|
output_file.write("\t\tmemset(Op, 0, Size);\n")
|
|
output_file.write("\t\tOp->Header.Op = T2;\n")
|
|
output_file.write("\t\treturn IRPair<T>{Op, CreateNode(&Op->Header)};\n")
|
|
output_file.write("\t}\n\n")
|
|
|
|
output_file.write("\tuint8_t GetOpSize(OrderedNode *Op) const {\n")
|
|
output_file.write("\t\tauto HeaderOp = Op->Header.Value.GetNode(Data.Begin());\n")
|
|
output_file.write("\t\tLogMan::Throw::A(HeaderOp->HasDest, \"Op %s has no dest\\n\", GetName(HeaderOp->Op));\n")
|
|
output_file.write("\t\treturn HeaderOp->Size;\n")
|
|
output_file.write("\t}\n\n")
|
|
|
|
output_file.write("\tuint8_t GetOpElements(OrderedNode *Op) const {\n")
|
|
output_file.write("\t\tauto HeaderOp = Op->Header.Value.GetNode(Data.Begin());\n")
|
|
output_file.write("\t\tLogMan::Throw::A(HeaderOp->HasDest, \"Op %s has no dest\\n\", GetName(HeaderOp->Op));\n")
|
|
output_file.write("\t\treturn HeaderOp->Elements;\n")
|
|
output_file.write("\t}\n\n")
|
|
|
|
output_file.write("\tbool OpHasDest(OrderedNode *Op) const {\n")
|
|
output_file.write("\t\tauto HeaderOp = Op->Header.Value.GetNode(Data.Begin());\n")
|
|
output_file.write("\t\treturn HeaderOp->HasDest;\n")
|
|
output_file.write("\t}\n\n")
|
|
|
|
# Generate helpers with operands
|
|
for op_key, op_vals in ops.items():
|
|
if not ("Last" in op_vals):
|
|
SSAArgs = 0
|
|
HasArgs = False
|
|
HasDest = False
|
|
HasFixedDestSize = False
|
|
FixedDestSize = 0
|
|
HasDestSize = False;
|
|
NumElements = 1
|
|
DestSize = ""
|
|
|
|
if ("SSAArgs" in op_vals):
|
|
SSAArgs = int(op_vals["SSAArgs"])
|
|
|
|
if ("Args" in op_vals and len(op_vals["Args"]) != 0):
|
|
HasArgs = True
|
|
|
|
if ("HelperGen" in op_vals and op_vals["HelperGen"] == False):
|
|
continue;
|
|
|
|
if ("HasDest" in op_vals and op_vals["HasDest"] == True):
|
|
HasDest = True
|
|
|
|
if ("FixedDestSize" in op_vals):
|
|
HasFixedDestSize = True
|
|
FixedDestSize = int(op_vals["FixedDestSize"])
|
|
|
|
if ("DestSize" in op_vals):
|
|
HasDestSize = True
|
|
DestSize = op_vals["DestSize"]
|
|
|
|
if ("NumElements" in op_vals):
|
|
NumElements = int(op_vals["NumElements"])
|
|
|
|
|
|
output_file.write("\tIRPair<IROp_%s> _%s(" % (op_key, op_key))
|
|
|
|
# Output SSA args first
|
|
if (SSAArgs != 0):
|
|
for i in range(0, SSAArgs):
|
|
LastArg = (SSAArgs - i - 1) == 0 and not HasArgs
|
|
CondArg2 = (", ", "")
|
|
output_file.write("OrderedNode *ssa%d%s" % (i, CondArg2[LastArg]))
|
|
|
|
if (HasArgs):
|
|
ArgCount = len(op_vals["Args"])
|
|
for i in range(0, len(op_vals["Args"]), 2):
|
|
data_type = op_vals["Args"][i]
|
|
data_name = op_vals["Args"][i+1]
|
|
LastArg = (ArgCount - i - 2) == 0
|
|
CondArg2 = (", ", "")
|
|
|
|
output_file.write("%s %s%s" % (data_type, data_name, CondArg2[LastArg]))
|
|
|
|
output_file.write(") {\n")
|
|
|
|
output_file.write("\t\tauto Op = AllocateOp<IROp_%s, IROps::OP_%s>();\n" % (op_key, op_key.upper()))
|
|
output_file.write("\t\tOp.first->Header.NumArgs = %d;\n" % (SSAArgs))
|
|
|
|
if (SSAArgs != 0):
|
|
for i in range(0, SSAArgs):
|
|
output_file.write("\t\tOp.first->Header.Args[%d] = ssa%d->Wrapped(ListData.Begin());\n" % (i, i))
|
|
output_file.write("\t\tssa%d->AddUse();\n" % (i))
|
|
|
|
if (HasArgs):
|
|
for i in range(1, len(op_vals["Args"]), 2):
|
|
data_name = op_vals["Args"][i]
|
|
output_file.write("\t\tOp.first->%s = %s;\n" % (data_name, data_name))
|
|
|
|
if (HasDest):
|
|
if (HasFixedDestSize):
|
|
output_file.write("\t\tOp.first->Header.Size = %d;\n" % FixedDestSize)
|
|
if (HasDestSize):
|
|
output_file.write("\t\tOp.first->Header.Size = %s;\n" % DestSize)
|
|
output_file.write("\t\tOp.first->Header.Elements = %s;\n" % NumElements)
|
|
|
|
if not (HasFixedDestSize or HasDestSize):
|
|
# We need to infer destination size
|
|
output_file.write("\t\tuint8_t InferSize = 0;\n")
|
|
if (SSAArgs != 0):
|
|
for i in range(0, SSAArgs):
|
|
output_file.write("\t\tuint8_t Size%d = GetOpSize(ssa%s);\n" % (i, i))
|
|
output_file.write("\t\tInferSize = std::max(InferSize, Size%d);\n" % (i))
|
|
|
|
output_file.write("\t\tOp.first->Header.Size = InferSize;\n")
|
|
|
|
output_file.write("\t\tOp.first->Header.HasDest = true;\n")
|
|
|
|
output_file.write("\t\treturn Op;\n")
|
|
output_file.write("\t}\n\n")
|
|
|
|
output_file.write("#undef IROP_ALLOCATE_HELPERS\n")
|
|
output_file.write("#endif\n")
|
|
|
|
if (len(sys.argv) < 3):
|
|
sys.exit()
|
|
|
|
output_filename = sys.argv[2]
|
|
json_file = open(sys.argv[1], "r")
|
|
json_text = json_file.read()
|
|
json_file.close()
|
|
|
|
json_object = json.loads(json_text)
|
|
json_object = {k.upper(): v for k, v in json_object.items()}
|
|
|
|
ops = json_object["OPS"]
|
|
defines = json_object["DEFINES"]
|
|
|
|
output_file = open(output_filename, "w")
|
|
|
|
print_enums(ops, defines)
|
|
print_ir_structs(ops, defines)
|
|
print_ir_sizes(ops, defines)
|
|
print_ir_getname(ops, defines)
|
|
print_ir_arg_printer(ops, defines)
|
|
print_ir_allocator_helpers(ops, defines)
|
|
|
|
output_file.close()
|