Introduce the astcompile pass - #649

Merged

antocuni merged 83 commits into

mainfrom

Aug 10, 2026

Commits on Aug 9, 2026

Rename LoweringStage into LoweringState, move to spy/ast.py
antocuni
committed
9a759e5
Make it possible to specify in which state each AST node is valid by
antocuni
committed
c73e3f3
some notes for when/how to write tests
antocuni
committed
f922ce8
add valid_state annotation to many AST nodes
antocuni
committed
6366d5c
WIP: introduce the astcompile pass.
antocuni
committed
fe3d40b
s/lowering_state/lostate/g
antocuni
committed
eafe451
add Module.lostate and FuncDef.lostate
antocuni
committed
6aa2050
add sanity checks for the expected lostate for modules and functions
antocuni
committed
b25b203
rename LoweringState into LoweringStage AGAIN
antocuni
committed
d8b37fd
make TestBasic::test_NameError[interp] passing
antocuni
committed
bb5b84b
forgot to commit these two files
antocuni
committed
fac4b63
TestBasic::test_binop_right_nested_same_precedence[interp] passes
antocuni
committed
075387c
missing nodes
antocuni
committed
63d3a6b
TestBasic::test_local_variables[interp] passes
antocuni
committed
9c1d80d
move the specialize_Assign/AssignExpr logic from ASTFrame to ASTCompiler. TestBasic::test_assignexpr_basic[interp] passes
antocuni
committed
6817563
TestBasic::test_assignexpr_argument[interp] passes
antocuni
committed
f9f7078
astcompile support for NameOuterDirect and AssignOuterDirect. TestBasic::test_assignexpr_updates_cell[interp] passes
antocuni
committed
865314b
add support for NameLocalCell, the last missing one; TestBasic::test_read_local_cell[interp] passes
antocuni
committed
542cebe
detect and handle assign-to-const errors. TestBasic::test_cannot_assign_to_const_globals[interp] passes
antocuni
committed
b385c9c
TestBasic::test_int_float[interp] passes
antocuni
committed
d64d809
TestBasic::test_BinOp_error[interp] passes
antocuni
committed
fe1a287
TestBasic::test_if_while_assert_cond_type_mismatch[interp] passes
antocuni
committed
1c7ac1c
TestBasic::test_getitem_error_1[interp] passes
antocuni
committed
79dcdf5
TestBasic::test_aug_assign[interp] passes
antocuni
committed
fc29eab
TestBasic::test_redshift_nonglobal_function[interp] passes
antocuni
committed
40e0ab8
TestBasic::test_setattr_error[interp] passes
antocuni
committed
d4888b4
TestBasic::test_automatic_forward_declaration[interp] passes, and fix struct.py
antocuni
committed
2a7e4ab
TestBasic::test_forward_declaration_in_funcdef[interp] passes
antocuni
committed
e2c0a0a
TestBasic::test_call_module_attr[interp] passes
antocuni
committed
b052dfa
TestBasic::test_decorator[interp]
antocuni
committed
62346ab
WIP: compile the For stmt, but we cannot run the test because 'range' is currently broken. Will fix it later
antocuni
committed
e93d237
break and continue
antocuni
committed
013b90d
no longer needed
antocuni
committed
a4d62df

Commits on Aug 10, 2026

TestBasic::test_type_name_attributes[interp]
antocuni
committed
3e79572
test_exception.py::TestException::test_raise[interp]
antocuni
committed
f4b2734
test_itemop.py::TestItemop::test_setitem[interp]
antocuni
committed
241f27a
add support for AssignUnpack and enough to make TestTuple::test_literal_stdlib[interp] passing
antocuni
committed
c25baba
fix; TestBlueGeneric::test_generic_class[interp] passes
antocuni
committed
1947fea
fix print
antocuni
committed
0d01f8d
fix BlockExpr and test_linearize
antocuni
committed
280e83d
these now pass
antocuni
committed
ba3453f
skip spdb tests for now
antocuni
committed
b9cb8d3
run astcompile on the synthetized struct functions
antocuni
committed
d8d733f
we need a slightly different mechanism to catch unsupported stmts in a ClassDef
antocuni
committed
1195594
rename. All interp tests pass 🎉
antocuni
committed
7ca1a48
Shuffle tests around: we want "plain" assign tests to come earlier
antocuni
committed
2750933
start to fix doppler: AssignLocal
antocuni
committed
c6eb2a7
implement reading/writing of Cells in the doppler. See the comments. TestBasic::test_global_variables[doppler] passes
antocuni
committed
2d24d4e
implement this in doppler
antocuni
committed
2b3d5f4
shuffle nodes around
antocuni
committed
eba1754
WIP: break everything again, try to reduce the code duplication between Assign and AssignExpr
antocuni
committed
2444be4
WIP: AssignCell
antocuni
committed
f9f4bd2
unify AssignConstError with it's Expr equivalent
antocuni
committed
60e0814
finally kill compile_assign_common
antocuni
committed
34ec05e
fix
antocuni
committed
6f88e5a
fix AssignUnpack
antocuni
committed
458e35b
fix the spy backend after the Assign refactoring
antocuni
committed
c96cd58
fix test_aug_assign
antocuni
committed
8f46f2d
fix doppler for the new Assign/AssignExpr state; test_assignexpr_* now work out of the box 🎉
antocuni
committed
346dd08
WIP: fix force_inline
antocuni
committed
82413f5
add a .sym to AssignLocalExpr, kill the ugly _set_assignexpr_color, fix TestForceInline::test_outer_var[doppler]
antocuni
committed
d4dbc29
fix
antocuni
committed
4aca5f0
add a sanity check and catch a mistake in w_print; test_builtins now full pass in doppler
antocuni
committed
b44c4bb
fix doppler for UnpackAssign and remove leftover
antocuni
committed
4573ba6
start to fix the C backend
antocuni
committed
74e3134
start to fix test_basic.py -m C
antocuni
committed
2aa4eac
more
antocuni
committed
f0dcf78
more
antocuni
committed
98e9b42
fix AssignUnpack in linearize and C
antocuni
committed
950e999
fix
antocuni
committed
9a34e97
fix test_function
antocuni
committed
deac0c5
astcompile default arguments
antocuni
committed
c74cc0a
improve this diagram
antocuni
committed
cc45f6d
fix spy colorize: after astcompile we have a different AST, and thanks to the new Loc.colorize the astcompile can decide which nodes participate to colorization and which don't
antocuni
committed
f43c1f0
fix the HTML backend tests
antocuni
committed
fc3ac2c
implement spy astcompile --format spy
antocuni
committed
2ef62cf
kill a bunch of leftovers
antocuni
committed
1c03c09
implement spy astcompile -f html
antocuni
committed
79ebfe2
fix SPdb
antocuni
committed
7686b27
kill
antocuni
committed
4c8c3d8
fix spy redshift -x --linearize
antocuni
committed
97255c9
fix mypy
antocuni
committed
025fb95
simplify
antocuni
committed
a4a9915