Skip to content

Optimize the JIT's low-level assembly control flow #135904

Description

@brandtbucher

Currently, our JIT backend mostly just takes the machine code that Clang gives us, and emits it without modification. The one exception to this is removing zero-length jumps to the next uop at the end of each blob of machine code, which is both fragile and extremely limited in what it can do.

For example, consider _GUARD_NOS_NULL. This is the (16-byte) sequence that Clang gives us:

cmpq    $0x1, -0x10(%r13)
je      _JIT_CONTINUE
jmp     _JIT_JUMP_TARGET

And this is the (11-byte) sequence we want:

cmpq    $0x1, -0x10(%r13)
jne      _JIT_JUMP_TARGET

We should do a bit more here, and doing more is a lot easier if we're modifying textual assembly at build time. Thankfully, this is pretty straightforward: we just compile to assembly using Clang (-S), modify it, and finish compiling using Clang again.

We should intentionally not take on too much complexity here, especially since we support a few different platforms. There's a lot that we can do by only reasoning about labels, jumps, branches, and returns and treating all other sequences of instructions as black boxes. Once we have the assembly parsed into basic blocks, we can do things like:

  • Inverting the direction of branches (from branch-hot/jump-cold to branch-cold/jump-hot) like in _GUARD_NOS_NULL above is straightforward.
  • Having the assembler encode all _JIT_CONTINUE jumps during this step (by just adding the label at the end of the assembly) instead of doing it at runtime. It will also use more efficient "short" jump encodings most of the time, as an additional benefit.
  • Removing zero length jumps, as we do now, is trivial.
  • Later: Splitting the stencils into "hot" (core uop logic) and "cold" (deopts, error handling, etc) code. The JIT will emit all "hot" code for a trace, followed by all cold code for a trace, keeping the cold code out-of-line.

I have a branch to do all but the last of these, and will open a PR soon.

Linked PRs

Activity

  1. self-assigned this
    on Jun 24, 2025
  2. changed the title [-]JIT Assembly Optimization[/-] [+]JIT assembly optimization[/+] on Jun 24, 2025
  3. changed the title [-]JIT assembly optimization[/-] [+]Optimize the JIT's low-level assembly control flow[/+] on Jun 24, 2025
  4. added
    type-featureA feature request or enhancement
    and removed
    3.15bugs and security fixes
    on Jun 27, 2025
  5. added a commit that references this issue on Jun 27, 2025
  6. added a commit that references this issue on Jul 11, 2025
  7. added a commit that references this issue on Jul 12, 2025
  8. added a commit that references this issue on Jul 14, 2025
  9. added 2 commits that reference this issue on Aug 4, 2025
  10. added 2 commits that reference this issue on Aug 19, 2025
  11. added 2 commits that reference this issue on Oct 17, 2025
  12. added 2 commits that reference this issue on Dec 6, 2025
  13. markshannon commented on Dec 19, 2025

    @markshannon
    Member

    @brandtbucher can we close this? I think it is done.

  14. Fidget-Spinner commented on Dec 25, 2025

    @Fidget-Spinner
    Member

    There's still hot-cold splitting. We can have another issue for that.

  15. Fidget-Spinner commented on Dec 25, 2025

    @Fidget-Spinner
    Member

    Follow-up on hot-cold splitting here #143158

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Metadata

Metadata

Labels

interpreter-core(Objects, Python, Grammar, and Parser dirs)performancePerformance or resource usagetopic-JITtype-featureA feature request or enhancement

Projects

No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions