Riscv scf

@register_impls
@dataclass
class RiscvScfFunctions(InterpreterFunctions):
    bitwidth: int = field(default=32)

    @impl(riscv_scf.ForOp)
    def run_for(
        self,
        interpreter: Interpreter,
        op: riscv_scf.ForOp,
        args: tuple[Any, ...],
    ):
        args = RiscvFunctions.get_reg_values(interpreter, op.operands, args)
        match op.step:
            case IntegerAttr(value=IntAttr(data=step)):
                lb, ub, *loop_args = args
            case _:
                lb, ub, step, *loop_args = args
                step = step.value.data
        loop_args = tuple(loop_args)

        for i in range(lb.value.data, ub.value.data, step):
            induction = RiscvFunctions.set_reg_value(
                interpreter,
                op.body.block.args[0].type,
                IntegerAttr(i, interpreter.index_bitwidth, truncate_bits=True),
            )
            loop_args = interpreter.run_ssacfg_region(
                op.body, (induction, *loop_args), "for_loop"
            )

        return loop_args

    @impl_terminator(riscv_scf.YieldOp)
    def run_yield(
        self, interpreter: Interpreter, op: riscv_scf.YieldOp, args: tuple[Any, ...]
    ):
        return ReturnedValues(args), ()

    @impl(riscv_scf.WhileOp)
    def run_while(
        self,
        interpreter: Interpreter,
        op: riscv_scf.WhileOp,
        args: tuple[Any, ...],
    ):
        cond, *results = interpreter.run_ssacfg_region(
            op.before_region, args, "while_head"
        )

        while cond:
            args = interpreter.run_ssacfg_region(
                op.after_region, tuple(results), "while_body"
            )
            cond, *results = interpreter.run_ssacfg_region(
                op.before_region, args, "while_head"
            )

        return tuple(results)

    @impl_terminator(riscv_scf.ConditionOp)
    def run_condition(
        self, interpreter: Interpreter, op: riscv_scf.ConditionOp, args: tuple[Any, ...]
    ):
        return ReturnedValues(args), ()