Skip to content

Commit f21d222

Browse files
Update specialize.py
1 parent c19ad9f commit f21d222

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

mypyc/irbuild/specialize.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -520,7 +520,7 @@ def translate_sum_call(builder: IRBuilder, expr: CallExpr, callee: RefExpr) -> V
520520
return None
521521

522522
arg = expr.args[0]
523-
if not isinstance(arg, GeneratorExpr) and not _is_supported_forloop_iter(builder, arg):
523+
if not isinstance(arg, GeneratorExpr) and not expr_has_specialized_for_helper(builder, arg):
524524
return None
525525

526526
# handle 'start' argument, if given
@@ -572,7 +572,7 @@ def body_insts() -> None:
572572
builder.assign(retval, total, expr.line)
573573

574574
index_type = builder._analyze_iterable_item_type(arg)
575-
index = _create_iterable_lexpr(index_name, index_type)
575+
index = create_synthetic_nameexpr(index_name, index_type)
576576
index_reg = builder.add_local_reg(index.node, builder.type_to_rtype(index_type)) # type: ignore [arg-type]
577577

578578
for_loop_helper(builder, index, arg, body_insts, None, is_async=False, line=expr.line)

0 commit comments

Comments
 (0)