Convert arith to x86

@dataclass
class ArithConstantToX86(RewritePattern):
    arch: X86Arch

    @op_type_rewrite_pattern
    def match_and_rewrite(self, op: arith.ConstantOp, rewriter: PatternRewriter):
        if not isa(op.value, IntegerAttr):
            raise DiagnosticException(
                "Lowering of arith.constant is only implemented for integers"
            )
        reg_type = self.arch.register_type_for_type(op.result.type)
        assert issubclass(reg_type, GeneralRegisterType)
        mov_op = x86.DI_MovOp(
            immediate=op.value.value.data, destination=reg_type.unallocated()
        )
        cast_op = asm.FromRegOp.get(mov_op.destination, op.result.type)
        rewriter.replace(op, [mov_op, cast_op])

@dataclass
class ArithBinaryToX86(RewritePattern):
    arch: X86Arch

    def match_and_rewrite(self, op: Operation, rewriter: PatternRewriter):
        new_type = X86_OP_BY_ARITH_BINARY_OP.get(type(op))  # pyright: ignore
        if new_type is None:
            return

        lhs = op.operands[0]

        if isinstance(lhs.type, builtin.ShapedType):
            raise DiagnosticException(
                f"Lowering of {op.name} not implemented for ShapedType"
            )
        rewriter.name_hint = op.results[0].name_hint

        lhs_x86, rhs_x86 = self.arch.cast_to_regs(op.operands, rewriter)
        moved_rhs = self.arch.move_value_to_unallocated(
            rhs_x86, rewriter, value_type=op.operands[1].type
        )
        add_op = new_type(source=lhs_x86, register_in=moved_rhs)
        result_cast_op = asm.FromRegOp.get(add_op.register_out, lhs.type)
        rewriter.replace(op, [add_op, result_cast_op])

@dataclass(frozen=True)
class ConvertArithToX86Pass(ModulePass):
    name = "convert-arith-to-x86"

    def apply(self, ctx: Context, op: builtin.ModuleOp) -> None:
        arch = X86Arch.arch_for_name(None)
        PatternRewriteWalker(
            GreedyRewritePatternApplier(
                [
                    ArithBinaryToX86(arch),
                    ArithConstantToX86(arch),
                ],
            ),
            apply_recursively=False,
        ).rewrite_module(op)