Method _fixup_constants

Tools/jit/_optimizers.py:498–547

(self)

        return Instruction(InstructionKind.OTHER, "", marker, None)

    def _fixup_constants(self) -> None:
        if not self.supports_small_constants:
            return
        index = 0
        for block in self._blocks():
            fixed: list[Instruction] = []
            small_const_index = -1
            for inst in block.instructions:
                if inst.kind == InstructionKind.SMALL_CONST_1:
                    marker = f"jit_pending_{inst.target}{index}:"
                    fixed.append(self._make_temp_label(index))
                    index += 1
                    small_const_index = len(fixed)
                    fixed.append(inst)
                elif inst.kind == InstructionKind.SMALL_CONST_2:
                    if small_const_index < 0:
                        fixed.append(inst)
                        continue
                    small_const_1 = fixed[small_const_index]
                    if not self._small_consts_match(small_const_1, inst):
                        small_const_index = -1
                        fixed.append(inst)
                        continue
                    assert small_const_1.target is not None
                    if small_const_1.target.endswith("16"):
                        fixed[small_const_index] = self._make_temp_label(index)
                        index += 1
                    else:
                        assert small_const_1.target.endswith("32")
                        patch_kind, replacement = self._small_const_1(small_const_1)
                        if replacement is not None:
                            label = f"{self.const_reloc}{patch_kind}_JIT_RELOCATION_CONST{small_const_1.target[:-3]}_JIT_RELOCATION_{index}:"
                            index += 1
                            fixed[small_const_index - 1] = Instruction(
                                InstructionKind.OTHER, "", label, None
                            )
                            fixed[small_const_index] = replacement
                    patch_kind, replacement = self._small_const_2(inst)
                    if replacement is not None:
                        assert inst.target is not None
                        label = f"{self.const_reloc}{patch_kind}_JIT_RELOCATION_CONST{inst.target[:-3]}_JIT_RELOCATION_{index}:"
                        index += 1
                        fixed.append(
                            Instruction(InstructionKind.OTHER, "", label, None)
                        )
                        fixed.append(replacement)
                    small_const_index = -1
                else:
                    fixed.append(inst)
            block.instructions = fixed

    def _small_const_1(self, inst: Instruction) -> tuple[str, Instruction | None]:
        raise NotImplementedError()