8000 GH-113464: Generate a more efficient JIT by brandtbucher · Pull Request #118512 · python/cpython · GitHub
[go: up one dir, main page]
Skip to content

GH-113464: Generate a more efficient JIT #118512

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 18 commits into from
May 3, 2024
Merged

GH-113464: Generate a more efficient JIT #118512

Show file tree
Hide file tree
Changes from 1 commit
Commits
Show all changes
18 commits
Select commit Hold shift + click to select a range
2901caf
Replace stencils with dedicated writer functions
brandtbucher Apr 19, 2024
f30fa64
Generate patching logic
brandtbucher Apr 19, 2024
23e211c
Cleanup
brandtbucher Apr 19, 2024
431fbed
uint64_t -> uintptr_t
brandtbucher Apr 19, 2024
3e6b25c
uintptr_t -> uint64_t
brandtbucher Apr 20, 2024
82030c8
Linting
brandtbucher Apr 20, 2024
236af82
Restore AArch64 pair folding
brandtbucher Apr 28, 2024
3b7e693
Cleanup
brandtbucher Apr 29, 2024
fbb97fc
Add missing relocations
brandtbucher Apr 29, 2024
c40bb34
Fix AArch64 folds
brandtbucher Apr 30, 2024
bd570b5
Dedent
brandtbucher May 1, 2024
b2fd9d2
Use a single array of structs
brandtbucher May 1, 2024
7aa12a2
Move C initializer formation to StencilGroup
brandtbucher May 1, 2024
9ec64ac
Add comment on why data is first
brandtbucher May 1, 2024
eb0826f
Silence warnings
brandtbucher May 2, 2024
a04d7f8
Add missing space
brandtbucher May 2, 2024
b919fcc
Clarify which patch functions are relaxing (and what that means)
brandtbucher May 3, 2024
46adf09
Exaplain why GOT is commented out
brandtbucher May 3, 2024
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
Move C initializer formation to StencilGroup
  • Loading branch information
@brandtbucher
brandtbucher committed May 1, 2024
commit 7aa12a2c3e3d3e11cad07f59cecb35b9e45e63c2
4 changes: 4 additions & 0 deletions Tools/jit/_stencils.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -350,6 +350,10 @@ def _emit_global_offset_table(self) -> None:
)
self.data.body.extend([0] * 8)

def as_c(self, opname: str) -> str:
"""Dump this hole as a StencilGroup initializer."""
return f"{{emit_{opname}, {len(self.code.body)}, {len(self.data.body)}}}"


def symbol_to_value(symbol: str) -> tuple[HoleValue, str | None]:
"""
Expand Down
7 changes: 2 additions & 5 deletions Tools/jit/_writer.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -5,8 +5,6 @@

import _stencils

def _initialize_stencil_group(opname: str, group: _stencils.StencilGroup) -> str:
return f"{{emit_{opname}, {len(group.code.body)}, {len(group.data.body)}}}"

def _dump_footer(groups: dict[str, _stencils.StencilGroup]) -> typing.Iterator[str]:
yield "typedef struct {"
Expand All @@ -17,14 +15,13 @@ def _dump_footer(groups: dict[str, _stencils.StencilGroup]) -> typing.Iterator[s
yield " size_t data_size;"
yield "} StencilGroup;"
yield ""
initializer = _initialize_stencil_group('trampoline', groups['trampoline'])
yield f"static const StencilGroup trampoline = {initializer};"
yield f"static const StencilGroup trampoline = {groups['trampoline'].as_c('trampoline')};"
yield ""
yield "static const StencilGroup stencil_groups[MAX_UOP_ID + 1] = {"
for opname, group in sorted(groups.items()):
if opname == "trampoline":
continue
yield f" [{opname}] = {_initialize_stencil_group(opname, group)},"
yield f" [{opname}] = {group.as_c(opname)},"
yield "};"


Expand Down
0