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
Fix AArch64 folds
  • Loading branch information
@brandtbucher
brandtbucher committed May 1, 2024
commit c40bb341d647f9011acdfd24dff523f3167f35a2
10 changes: 8 additions & 2 deletions Python/jit.c
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -214,6 +214,12 @@ patch_aarch64_12(unsigned char *location, uint64_t value)
set_bits(loc32, 10, value, shift, 12);
}

static inline void
patch_aarch64_12x(unsigned char *location, uint64_t value)
{
patch_aarch64_12(location, value);
}

// 16-bit low part of an absolute address.
static inline void
patch_aarch64_16a(unsigned char *location, uint64_t value)
Expand Down Expand Up @@ -295,7 +301,7 @@ patch_aarch64_26r(unsigned char *location, uint64_t value)
set_bits(loc32, 0, value, 2, 26);
}

// A pair of patch_aarch64_21rx and patch_aarch64_12.
// A pair of patch_aarch64_21rx and patch_aarch64_12x.
static inline void
patch_aarch64_33rx(unsigned char *location, uint64_t value)
{
Expand Down Expand Up @@ -331,7 +337,7 @@ patch_aarch64_33rx(unsigned char *location, uint64_t value)
return;
}
patch_aarch64_21rx(location, value);
patch_aarch64_12(location + 4, value);
patch_aarch64_12x(location + 4, value);
}

// 32-bit relative address.
Expand Down
10 changes: 5 additions & 5 deletions Tools/jit/_stencils.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -52,7 +52,7 @@ class HoleValue(enum.Enum):
# aarch64-apple-darwin:
"ARM64_RELOC_BRANCH26": "patch_aarch64_26r",
"ARM64_RELOC_GOT_LOAD_PAGE21": "patch_aarch64_21rx",
"ARM64_RELOC_GOT_LOAD_PAGEOFF12": "patch_aarch64_12",
"ARM64_RELOC_GOT_LOAD_PAGEOFF12": "patch_aarch64_12x",
"ARM64_RELOC_PAGE21": "patch_aarch64_21r",
"ARM64_RELOC_PAGEOFF12": "patch_aarch64_12",
"ARM64_RELOC_UNSIGNED": "patch_64",
Expand All @@ -62,18 +62,18 @@ class HoleValue(enum.Enum):
"IMAGE_REL_ARM64_BRANCH26": "patch_aarch64_26r",
"IMAGE_REL_ARM64_PAGEBASE_REL21": "patch_aarch64_21rx",
"IMAGE_REL_ARM64_PAGEOFFSET_12A": "patch_aarch64_12",
"IMAGE_REL_ARM64_PAGEOFFSET_12L": "patch_aarch64_12",
"IMAGE_REL_ARM64_PAGEOFFSET_12L": "patch_aarch64_12x",
# i686-pc-windows-msvc:
"IMAGE_REL_I386_DIR32": "patch_32",
"IMAGE_REL_I386_REL32": "patch_x86_64_32rx",
# aarch64-unknown-linux-gnu:
"R_AARCH64_ABS64": "patch_64",
"R_AARCH64_ADD_ABS_LO12_NC": "patch_aarch64_12",
"R_AARCH64_ADR_GOT_PAGE": "patch_aarch64_21rx",
"R_AARCH64_ADR_PREL_PG_HI21": "patch_aarch64_21rx",
"R_AARCH64_ADR_PREL_PG_HI21": "patch_aarch64_21r",
"R_AARCH64_CALL26": "patch_aarch64_26r",
"R_AARCH64_JUMP26": "patch_aarch64_26r",
"R_AARCH64_LD64_GOT_LO12_NC": "patch_aarch64_12",
"R_AARCH64_LD64_GOT_LO12_NC": "patch_aarch64_12x",
"R_AARCH64_MOVW_UABS_G0_NC": "patch_aarch64_16a",
"R_AARCH64_MOVW_UABS_G1_NC": "patch_aarch64_16b",
"R_AARCH64_MOVW_UABS_G2_NC": "patch_aarch64_16c",
Expand Down Expand Up @@ -141,7 +141,7 @@ def fold(self, other: typing.Self) -> typing.Self | None:
and self.symbol == other.symbol
and self.addend == other.addend
and self.func == "patch_aarch64_21rx"
and other.func == "patch_aarch64_12"
and other.func == "patch_aarch64_12x"
):
folded = self.replace()
folded.func = "patch_aarch64_33rx"
Expand Down
0