import unittest
from typing import Callable
-from bigint_presentation_code.compiler_ir2 import (GPR_SIZE_IN_BYTES,
- BaseSimState, Fn,
- GenAsmState, OpKind,
- PostRASimState,
- PreRASimState)
-from bigint_presentation_code.register_allocator2 import allocate_registers
+from bigint_presentation_code.compiler_ir import (GPR_SIZE_IN_BYTES,
+ BaseSimState, Fn,
+ GenAsmState, OpKind,
+ PostRASimState,
+ PreRASimState)
+from bigint_presentation_code.register_allocator import allocate_registers
from bigint_presentation_code.toom_cook import ToomCookInstance, simple_mul