Skip to content

[mypyc] Call generator helper method directly in await expression #19376

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
JukkaL merged 22 commits into from
Jul 7, 2025
Merged

[mypyc] Call generator helper method directly in await expression #19376

Changes from 1 commit
Commits
Show all changes
22 commits
Select commit Hold shift + click to select a range
d92c437
WIP populate generator classes before main irbuild phase
JukkaL Jun 9, 2025
4c44b4d
Omit CPy_GetCoro call when calling native coro
JukkaL Jun 9, 2025
ab44852
WIP better
JukkaL Jun 9, 2025
74db611
WIP attempt to fix naming
JukkaL Jun 9, 2025
9959e24
Fix segfault
JukkaL Jun 10, 2025
baca816
Simplify
JukkaL Jun 10, 2025
e1282a7
Call __next__ directly if possible in await
JukkaL Jun 10, 2025
5e6b575
Call helper method directly
JukkaL Jun 10, 2025
e3d3578
Generator class is final (fix after rebase)
JukkaL Jun 19, 2025
7bdc1f5
Fix helper method return type -- it should be object
JukkaL Jun 20, 2025
a925378
Don't apply to decorated functions, since we don't know concrete retu…
JukkaL Jun 20, 2025
3328d18
Fix nested generators
JukkaL Jul 2, 2025
f293150
Add test case
JukkaL Jul 3, 2025
3048d03
Support method calls
JukkaL Jul 3, 2025
2c184dc
Add comments and docstrings
JukkaL Jul 4, 2025
bcf9bbc
Fix generator class detection
JukkaL Jul 4, 2025
ce74b2f
Lint
JukkaL Jul 4, 2025
0ab4b19
Test calling async method
JukkaL Jul 4, 2025
11a1e75
Add comment
JukkaL Jul 4, 2025
fff29df
Add some type annotations
JukkaL Jul 4, 2025
c33a683
Fix self check
JukkaL Jul 4, 2025
1fa32b2
Add comments based on review feedback
JukkaL Jul 7, 2025
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
Call __next__ directly if possible in await
  • Loading branch information
@JukkaL
JukkaL committed Jul 4, 2025
commit e1282a75ff655fc6c96e3b6ce1c4b20fe9da2466
10 changes: 9 additions & 1 deletion mypyc/irbuild/statement.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -48,6 +48,7 @@
)
from mypyc.common import TEMP_ATTR_NAME
from mypyc.ir.ops import (
ERR_NEVER,
NAMESPACE_MODULE,
NO_TRACEBACK_LINE_NO,
Assign,
Expand Down Expand Up @@ -944,7 +945,14 @@ def emit_yield_from_or_await(
iter_reg = builder.maybe_spill_assignable(iter_val)

stop_block, main_block, done_block = BasicBlock(), BasicBlock(), BasicBlock()
_y_init = builder.call_c(next_raw_op, [builder.read(iter_reg)], line)

if isinstance(iter_reg.type, RInstance) and iter_reg.type.class_ir.has_method("__next__"):
m = MethodCall(builder.read(iter_reg), "__next__", [], line)
m.error_kind = ERR_NEVER
_y_init = builder.add(m)
else:
_y_init = builder.call_c(next_raw_op, [builder.read(iter_reg)], line)

builder.add(Branch(_y_init, stop_block, main_block, Branch.IS_ERROR))

# Try extracting a return value from a StopIteration and return it.
Expand Down