feat(ir): Add Submit IR node — print task launches as pl.submit(...)

[Hzfengsy]

Summary

Adds Submit as a first-class IR kind sibling to Call, so a pl.submit(self.kernel, ..., deps=[t]) task launch is visually distinct from a plain function call in IR dumps. Mid-pipeline dumps (the user-facing reading point — passes_dump/19_after_InferTileMemorySpace.py and friends) now render task submissions as pl.submit(self.stage1, ..., deps=[a_tid]) instead of the bare self.stage1(...) form that conflated them with synchronous calls.

The migration:

• IR (Submit class) — op_, args_, typed deps_ field (replaces Call::attrs[manual_dep_edges] on the call side), attrs_, kwargs_. ObjectKind::Submit, KindTrait<Submit>, ExprFunctor::VisitExpr_(SubmitPtr). Concrete defaults in IRVisitor / IRMutator walk + rewrite args/deps/attrs.
• Parser — pl.submit(...) now emits ir.Submit directly. Both the unpacked form out, tid = pl.submit(...) and the printer's round-trip-friendly single-LHS form res: pl.Tuple[..., TASK_ID] = pl.submit(...) are accepted.
• Printer — IRPythonPrinter::VisitExpr_(SubmitPtr) emits pl.submit(self.<kernel>, args, deps=[...]). structural_equal / structural_hash get Submit dispatch via the reflection-based comparator using GetFieldDescriptors.
• Lowering — DeriveCallDirections (pass 34) materializes Submit → Call via the new SubmitToCallView adapter: Submit::deps_ is folded back into attrs[manual_dep_edges] on the synthesised Call, the standard direction-derivation runs, and the result Call replaces the Submit in the IR. Codegen and every pass numbered ≥34 are unchanged — they still operate on Call exactly as before. Submit only lives for dumps 0–33, which is the user-relevant window.
• DCE / SSA — Submit-aware: an AssignStmt whose value is a Submit is preserved (task launches are side-effecting); SSA renames Submit's args and typed deps_ field through the IRMutator default plus a small override that also remaps the return type.

A new .claude/rules/pass-submit-awareness.md rule pairs with ir-kind-traits.md and warns pass authors to dispatch on Submit alongside Call.

Files

C++ headers / impls (12): include/pypto/ir/{core.h, expr.h, kind_traits.h}, include/pypto/ir/transforms/base/{functor.h, visitor.h, mutator.h}, src/ir/transforms/{visitor.cpp, mutator.cpp, python_printer.cpp, structural_equal.cpp, structural_hash.cpp, derive_call_directions_pass.cpp, convert_to_ssa_pass.cpp, utils/dead_code_elimination.cpp}, src/ir/arith/{int_set, const_int_bound, modular_set, canonical_simplify, rewrite_simplify}.{cpp,h}, src/codegen/orchestration/orchestration_codegen.cpp

Python bindings / type stubs / parser (3): python/bindings/modules/ir.cpp, python/pypto/pypto_core/ir.pyi, python/pypto/language/parser/ast_parser.py

Tests (5 files, 12 new test cases): tests/ut/ir/printing/test_submit_printer.py, tests/ut/ir/transforms/test_submit_passes.py, tests/ut/ir/transforms/test_submit_end_to_end.py, plus updates to tests/ut/ir/transforms/test_flatten_call_expr_pass.py and tests/ut/language/parser/{test_manual_scope_parsing.py, test_task_id_dsl.py} so the test helpers (_calls_in, _all_calls) collect both Call and Submit RHS values and dep-probing assertions use submit.deps instead of call.attrs[manual_dep_edges].

Docs (2): docs/en/dev/ir/01-hierarchy.md (Submit row + new Submit vs Call section); .claude/rules/pass-submit-awareness.md (new rule).

Test plan

• All 5043 tests/ut/ cases pass (excluding pre-existing simpler-runtime-sync failures in tests/ut/runtime — unrelated, on baseline)
• New per-feature tests:
  • test_submit_printer.py — 6 cases: pl.submit syntax inside Program, deps=[...] kwarg, zero-arg + deps, Submit ≠ Call structurally, deps order matters
  • test_submit_passes.py — 4 cases: SSA preserves Submit kind, SSA renames args and deps_, post-SSA still prints pl.submit, single-LHS round-trip with default VerificationLevel.BASIC
  • test_submit_end_to_end.py — 2 cases: pl.submit visible in mid-pipeline dump (the user's original complaint), deps=[...] kwarg surfaces from typed Submit::deps_
• Default VerificationLevel.BASIC RoundtripInstrument passes — print→re-parse→structural_equal works for Submit-bearing IR
• Existing orchestration codegen tests (TestManualScopeCodegen) still pass — codegen sees Call after the DeriveCallDirections lowering
• clang-format / cpplint / ruff / pyright / markdownlint pre-commit hooks all green on each commit

🤖 Generated with Claude Code

[coderabbitai]

Note

Reviews paused

It looks like this branch is under active development. To avoid overwhelming you with review comments due to an influx of new commits, CodeRabbit has automatically paused this review. You can configure this behavior by changing the reviews.auto_review.auto_pause_after_reviewed_commits setting.

Use the following commands to manage reviews:

• @coderabbitai resume to resume automatic reviews.
• @coderabbitai review to trigger a single review.

Use the checkboxes below for quick actions:

• ▶️ Resume reviews
• 🔍 Trigger review

📝 Walkthrough

Walkthrough

This PR introduces Submit as a first-class IR expression kind representing async task launches from pl.submit(...) within pl.manual_scope. The change adds a new IR node with explicit deps_ field for cross-task dependencies, integrates it across visitor/mutator infrastructure, updates the parser to emit Submit nodes directly, implements pass handlers throughout the compiler pipeline, and provides comprehensive test coverage.

Changes

Submit IR node introduction

Layer / File(s)	Summary
IR node kind definition and core types include/pypto/ir/core.h, include/pypto/ir/expr.h, include/pypto/ir/kind_traits.h	ObjectKind gains Submit; new Submit expression stores op, args, explicit deps_, kwargs/attrs; KindTrait<Submit> added and KindTrait<Expr>::count updated; SubmitToCallView adapter synthesizes a Call view with manual_dep_edges for compatibility.
Visitor and functor dispatch infrastructure include/pypto/ir/transforms/base/functor.h, include/pypto/ir/transforms/base/visitor.h, include/pypto/ir/transforms/base/mutator.h, src/ir/arith/canonical_simplify.h	Added VisitExpr_(const SubmitPtr&) hooks and dispatch entries so ExprFunctor/IRVisitor/IRMutator and related simplifiers dispatch Submit to dedicated handlers.
AST parser Submit handling python/pypto/language/parser/ast_parser.py	Parser now recognizes single-LHS pl.submit(...) forms and constructs ir.Submit directly (augmented return type including TASK_ID, carries deps_list, strips manual_dep_edges from attrs); _parse_submit_single_lhs helper added.
Transformation pass Submit handlers src/ir/transforms/visitor.cpp, src/ir/transforms/mutator.cpp, src/ir/transforms/convert_to_ssa_pass.cpp, src/ir/transforms/derive_call_directions_pass.cpp, src/ir/transforms/utils/dead_code_elimination.cpp, src/codegen/orchestration/orchestration_codegen.cpp, src/ir/transforms/structural_equal.cpp, src/ir/transforms/structural_hash.cpp, src/ir/arith/*	Multiple passes implement VisitExpr_(SubmitPtr): visitor walks args/deps/var-attrs; mutator remaps args/deps/type and arg-direction override vars; SSA substitutes types and preserves node kind; derive-call-directions converts Submit to Call-view for direction derivation; DCE treats Submit as side-effecting and adds CallLikeFinder; arithmetic analyses conservatively return unconstrained results; orchestration codegen funnels Submit through call-shaped codepath; structural equal/hash include Submit.
Python API bindings and type stubs python/bindings/modules/ir.cpp, python/pypto/pypto_core/ir.pyi	Expose ir.Submit to Python (nanobind) with constructors accepting op, args, deps, type/span and overload with kwargs/attrs; read-only properties kwargs, attrs, arg_directions; type stubs updated.
Python printer and function dependency collection src/ir/transforms/python_printer.cpp	Printer prints pl.submit(self.<kernel>, <args>, deps=[...], attrs={"arg_directions": [...]}) when round-trippable; GlobalVarCollector collects submit callees for topological analysis.
Developer guidance for pass authors .claude/rules/pass-submit-awareness.md, docs/en/dev/ir/01-hierarchy.md	Adds documentation explaining Submit vs Call: semantics, return-type TASK_ID augmentation, deps_ as SSA uses, required pass handling (visit both kinds, preserve Submit, normalize TASK_ID when comparing return types), and verifier guidance.
Test coverage for Submit IR node and passes tests/ut/ir/printing/test_submit_printer.py, tests/ut/ir/transforms/test_submit_passes.py, tests/ut/ir/transforms/test_submit_end_to_end.py, tests/ut/ir/transforms/test_flatten_call_expr_pass.py, tests/ut/language/parser/test_manual_scope_parsing.py, tests/ut/language/parser/test_task_id_dsl.py	Adds printer/unit/integration tests validating Submit printing, deps kwarg emission, structural equality/hash behavior, SSA preservation and renaming of args/deps, mid-pipeline visibility, parser round-trips, and updates parser tests to read deps from *.deps instead of attrs["manual_dep_edges"].

Estimated code review effort

🎯 4 (Complex) | ⏱️ ~75 minutes

Possibly related PRs

• hw-native-sys/pypto#1516: Modifies pl.submit/pl.at parsing and deps handling via ir.Submit and deps_ plumbing in the same parser and test files.

Suggested reviewers

• lyfne123

Poem

🐰 Hops through the IR with Submit's new grace,
Async tasks launched in their rightful place,
Deps flow like carrots, structured and true,
Visitors and mutators now know what to do—
A sibling to Call, forever to stay! 🥕✨

🚥 Pre-merge checks | ✅ 4 | ❌ 1

❌ Failed checks (1 warning)

Check name	Status	Explanation	Resolution
Docstring Coverage	⚠️ Warning	Docstring coverage is 38.13% which is insufficient. The required threshold is 80.00%.	Write docstrings for the functions missing them to satisfy the coverage threshold.

✅ Passed checks (4 passed)

Check name	Status	Explanation
Title check	✅ Passed	The title 'feat(ir): Add Submit IR node — print task launches as pl.submit(...)' directly and clearly describes the primary change: introducing a new Submit IR kind and its printing behavior.
Description check	✅ Passed	The description is comprehensive and directly related to the changeset, explaining the Submit IR kind, parser/printer integration, lowering strategy, and test coverage.
Linked Issues check	✅ Passed	Check skipped because no linked issues were found for this pull request.
Out of Scope Changes check	✅ Passed	Check skipped because no linked issues were found for this pull request.

_{✏️ Tip: You can configure your own custom pre-merge checks in the settings.}

---

Thanks for using CodeRabbit! It's free for OSS, and your support helps us grow. If you like it, consider giving us a shout-out.

❤️ Share

• X
• Mastodon
• Reddit
• LinkedIn

_{Comment @coderabbitai help to get the list of available commands and usage tips.}

[gemini-code-assist]

Code Review

This pull request introduces a new first-class IR node kind Submit to represent asynchronous task launches (pl.submit(...) inside pl.manual_scope), separating them structurally from plain synchronous Call nodes. The changes span the C++ IR definitions, AST parser, Python printer, SSA conversion, dead code elimination, Python bindings, and type stubs, alongside comprehensive unit and integration tests. The review feedback highlights two key improvements: addressing a contradiction in the Python printer where a null current_program_ would trigger an abort instead of falling back to a safe printing mode, and replacing a raw assert in the AST parser with a user-friendly ParserSyntaxError when parsing annotated assignments of pl.submit.

[Copilot]

Pull request overview

This PR introduces a first-class Submit IR expression kind to represent pl.submit(...) task launches distinctly from plain synchronous Call, improving IR dumps and enabling submit-aware analyses/passes while preserving the existing late-pipeline/codegen Call-based behavior via a Submit → Call lowering in DeriveCallDirections.

Changes:

• Add ir.Submit across core IR (new ObjectKind, kind traits, reflection descriptors) and wire it through visitors/mutators, structural equality/hash, DCE, SSA, and arithmetic analyzers.
• Update the Python DSL parser and IR Python printer to round-trip pl.submit(...), including the printer’s single-LHS tuple-typed form.
• Add/adjust unit and integration tests and update IR documentation + introduce a .claude rule for pass authors to handle Submit alongside Call.

Reviewed changes

Copilot reviewed 33 out of 33 changed files in this pull request and generated 5 comments.

Show a summary per file

File	Description
tests/ut/language/parser/test_task_id_dsl.py	Update test helpers/assertions to treat Submit as call-like and use submit.deps.
tests/ut/language/parser/test_manual_scope_parsing.py	Broaden call collection to include Submit; add coverage for single-LHS submit binding; switch dep assertions to submit.deps.
tests/ut/ir/transforms/test_submit_passes.py	New tests validating SSA + round-trip behavior on Submit.
tests/ut/ir/transforms/test_submit_end_to_end.py	New integration tests ensuring mid-pipeline dumps show pl.submit(...).
tests/ut/ir/transforms/test_flatten_call_expr_pass.py	Ensure flattening preserves Submit.deps when rewriting submit args.
tests/ut/ir/printing/test_submit_printer.py	New tests for printing and structural equality/hash behavior of Submit.
src/ir/transforms/visitor.cpp	Add IRVisitor traversal of Submit args, deps, and certain var-typed attrs.
src/ir/transforms/utils/dead_code_elimination.cpp	Treat Submit as side-effecting for DCE; ensure expressions containing Submit are preserved.
src/ir/transforms/structural_hash.cpp	Add structural hashing dispatch for Submit.
src/ir/transforms/structural_equal.cpp	Add structural equality dispatch for Submit.
src/ir/transforms/python_printer.cpp	Print Submit as pl.submit(self.<kernel>, ..., deps=[...]) and collect GlobalVars through Submit.
src/ir/transforms/mutator.cpp	Add IRMutator support to rewrite Submit (args/deps/type and selected attrs).
src/ir/transforms/derive_call_directions_pass.cpp	Lower Submit → Call in DeriveCallDirections via SubmitToCallView.
src/ir/transforms/convert_to_ssa_pass.cpp	Ensure SSA conversion also rewrites Submit return type appropriately.
src/ir/arith/rewrite_simplify.h	Add Submit visitor hook to rewrite simplifier.
src/ir/arith/rewrite_simplify.cpp	Treat Submit as a leaf for rewrite simplifier.
src/ir/arith/modular_set.cpp	Treat Submit conservatively in modular set analyzer.
src/ir/arith/int_set.cpp	Treat Submit conservatively in int set analyzer.
src/ir/arith/const_int_bound.cpp	Treat Submit conservatively in const-int bound analyzer.
src/ir/arith/canonical_simplify.h	Add Submit visitor hook to canonical simplifier.
src/ir/arith/canonical_simplify.cpp	Treat Submit as a leaf for canonical simplifier.
src/codegen/orchestration/orchestration_codegen.cpp	Funnel Submit through existing codegen via SubmitToCallView.
python/pypto/pypto_core/ir.pyi	Add Submit stub API (op/args/deps/kwargs/attrs/arg_directions).
python/pypto/language/parser/ast_parser.py	Parse pl.submit(...) into ir.Submit; accept single-LHS printed form; move deps out of attrs into deps_.
python/bindings/modules/ir.cpp	Bind Submit into Python IR module with kwargs/attrs/arg_directions accessors.
include/pypto/ir/transforms/base/visitor.h	Add Submit dispatch in base visitor interface.
include/pypto/ir/transforms/base/mutator.h	Add Submit dispatch in base mutator interface.
include/pypto/ir/transforms/base/functor.h	Add Submit dispatch in ExprFunctor + dispatch table.
include/pypto/ir/kind_traits.h	Register Submit kind trait and include it under Expr kinds.
include/pypto/ir/expr.h	Define Submit IR node and the SubmitToCallView adapter.
include/pypto/ir/core.h	Add ObjectKind::Submit.
docs/en/dev/ir/01-hierarchy.md	Document Submit in hierarchy and add Submit vs Call section.
.claude/rules/pass-submit-awareness.md	New rule documenting how passes must handle Submit vs Call.

[coderabbitai]

Actionable comments posted: 8

🧹 Nitpick comments (1)

tests/ut/ir/transforms/test_submit_passes.py (1)

92-110: ⚡ Quick win

Make the deps-renaming assertion match the test name.

test_ssa_renames_submit_args_and_deps currently verifies arg SSA rewrite only. Either add a deps rewrite assertion (by forcing a deps var rewrite) or rename the test to avoid over-claiming coverage.

🤖 Prompt for AI Agents

Verify each finding against current code. Fix only still-valid issues, skip the
rest with a brief reason, keep changes minimal, and validate.

In `@tests/ut/ir/transforms/test_submit_passes.py` around lines 92 - 110, The test
name claims both args and deps are SSA-renamed but only asserts args; either
update the test to assert deps were renamed or rename the test to match current
behavior. To fix in test_ssa_renames_submit_args_and_deps, either (A) ensure
_build_program_with_submit produces a deps Var that gets rewritten and add an
assertion like checking submit_after.deps[i] is an ir.Var and is not
caller_after.params[j] (use submit_after.deps and caller_after.params to locate
symbols), or (B) rename the test to something like test_ssa_renames_submit_args
to reflect it only verifies args; keep references to _build_program_with_submit
and passes.convert_to_ssa to locate the setup and transformation.

🤖 Prompt for all review comments with AI agents

Verify each finding against current code. Fix only still-valid issues, skip the
rest with a brief reason, keep changes minimal, and validate.

Inline comments:
In `@docs/en/dev/ir/01-hierarchy.md`:
- Around line 156-182: Update the RuntimeScope/dep-encoding paragraph to match
the Submit-first flow: state that the parser emits Submit nodes with a
first-class deps_ field (Submit.deps_) and that DeriveCallDirections lowers
Submit by folding deps_ into Call.attrs["manual_dep_edges"], so later passes and
runtime see Call with attrs["manual_dep_edges"]. Mention Submit, Submit.deps_,
DeriveCallDirections and Call.attrs["manual_dep_edges"] explicitly so the doc
consistently describes Submit-first encoding folded to Call during lowering.

In `@include/pypto/ir/expr.h`:
- Around line 905-923: SubmitToCallView currently copies submit->attrs_
unchanged which preserves any existing kAttrManualDepEdges when submit->deps_ is
empty and it silently drops non-Var-like entries in submit->deps_; update
SubmitToCallView to (1) remove/clear any existing kAttrManualDepEdges from attrs
when submit->deps_ is empty (so synthesized Call never inherits stale manual-dep
metadata) and (2) validate entries in submit->deps_ instead of quietly skipping
non-Var-like objects: iterate submit->deps_ and if an entry is null or its kind
is neither ObjectKind::Var nor ObjectKind::IterArg, surface an error (throw or
LOG and return) rather than ignoring it; continue to convert valid Var/IterArg
entries to VarPtr and call WithManualDepEdgesAttr(attrs, dep_vars) when deps are
present. Ensure you reference and modify the existing symbols SubmitToCallView,
attrs, submit->deps_, kAttrManualDepEdges, WithManualDepEdgesAttr, and Call
construction so the new behavior prevents stale attrs and dropped deps.

In `@python/pypto/language/parser/ast_parser.py`:
- Around line 987-994: The fast-path in the AST parser currently short-circuits
annotated-assignment validation for single-LHS pl.submit calls by calling
_parse_submit_single_lhs and returning; instead, modify _parse_submit_single_lhs
(or create a new helper) to return the constructed ir.Submit node and then let
the normal annotated-assignment binding/override_type validation flow handle it
so annotations are validated; update the call site (the block using _is_pl_call
and _parse_submit_single_lhs) to receive the ir.Submit result and proceed
through the same annotated-assignment/override_type path used for other RHS
expressions (the same logic used elsewhere in the function and mirrored in the
other occurrence around lines ~1363-1395).

In `@src/codegen/orchestration/orchestration_codegen.cpp`:
- Around line 892-899: The synthetic Call created by SubmitToCallView loses the
original Submit* identity so tuple-key lookups (used by
GenerateSubmitReturnAliases) miss entries; preserve the original node identity
by propagating the Submit pointer into the Call view or by recording a mapping
from the synthetic Call to the original Submit. Concretely, update
SubmitToCallView to attach the original Submit* (e.g., a member like
original_submit_ or a side-map from Call* -> Submit*) and change the tuple-key
lookup path (the code that inspects assign->value_ and later lookup logic used
by GenerateSubmitReturnAliases) to consult that original_submit_ mapping when
present so alias bookkeeping uses the Submit* key instead of the synthetic
Call*.

In `@src/ir/transforms/derive_call_directions_pass.cpp`:
- Around line 409-422: The Submit→Call lowering in VisitExpr_(const SubmitPtr&
op) loses original expression identity so first_writer_roots_ (which is keyed by
Call*) can't find Submit-originated first-writer roots, causing Out args to be
misclassified as InOut; fix this by preserving the original Submit identity
before calling VisitExpr_(view_call) — e.g., capture the original submit pointer
(op or submit.get()) and register a temporary mapping from the synthesized Call
pointer (view_call.get()) back to the original submit identity in
first_writer_roots_ (or adjust first_writer_roots_ to accept an alternative key
sourced from Submit) so the direction derivation code (VisitExpr_,
SubmitToCallView, first_writer_roots_) can look up the correct first-writer
context and keep Out arguments as first-writers.

In `@src/ir/transforms/python_printer.cpp`:
- Around line 827-869: The VisitExpr_ for Submit currently hard-fails when
current_program_ or As<GlobalVar> is missing; change it to detect whether the
callee is a GlobalVar inside a Program and only use the "self.<kernel>" form in
that case, otherwise fall back to a generic callee emission so
Print(Function)/Print(Expr) won't crash: in IRPythonPrinter::VisitExpr_(const
SubmitPtr& op) compute gvar = As<GlobalVar>(op->op_) and if (gvar &&
current_program_) emit ".submit(self.<kernel_name>...)" as now, else emit
".submit(" then VisitExpr(op->op_) (or otherwise print the op->op_ generically)
followed by the same args, deps and attrs logic; remove or replace the
INTERNAL_CHECK_SPAN that enforces current_program_ so the fallback path is used
instead of hard-failing.

In `@tests/ut/ir/transforms/test_submit_end_to_end.py`:
- Line 59: Rename the unused unpacked variables from pl.submit(self.stage1, x,
out) — currently named scratch and tid — to underscore-prefixed names (e.g.,
_scratch and _tid or just _) to silence RUF059; locate the call to pl.submit in
test_submit_end_to_end.py (the line invoking pl.submit(self.stage1, x, out)) and
change the left-hand unpacking identifiers accordingly so the test behavior
remains identical while avoiding lint warnings.

In `@tests/ut/ir/transforms/test_submit_passes.py`:
- Around line 12-15: Update the module docstring in the test_submit_passes.py
test module: replace or remove the stale sentence saying “parser does not yet
emit Submit” so it correctly reflects that the parser now emits Submit (e.g.,
state that tests construct Submit-bearing IR to verify passes handle
parser-emitted Submit) and ensure the docstring mentions why the tests exist
(preserve Submit structural shape) rather than claiming the parser can’t emit
it.

---

Nitpick comments:
In `@tests/ut/ir/transforms/test_submit_passes.py`:
- Around line 92-110: The test name claims both args and deps are SSA-renamed
but only asserts args; either update the test to assert deps were renamed or
rename the test to match current behavior. To fix in
test_ssa_renames_submit_args_and_deps, either (A) ensure
_build_program_with_submit produces a deps Var that gets rewritten and add an
assertion like checking submit_after.deps[i] is an ir.Var and is not
caller_after.params[j] (use submit_after.deps and caller_after.params to locate
symbols), or (B) rename the test to something like test_ssa_renames_submit_args
to reflect it only verifies args; keep references to _build_program_with_submit
and passes.convert_to_ssa to locate the setup and transformation.

🪄 Autofix (Beta)

Fix all unresolved CodeRabbit comments on this PR:

• Push a commit to this branch (recommended)
• Create a new PR with the fixes

---

ℹ️ Review info

⚙️ Run configuration

Configuration used: Repository UI

Review profile: CHILL

Plan: Pro

Run ID: 1ae4a749-82b8-42a2-941a-b378f45cbe58

📥 Commits

Reviewing files that changed from the base of the PR and between 1bb34f4 and 0877440.

📒 Files selected for processing (33)

• .claude/rules/pass-submit-awareness.md
• docs/en/dev/ir/01-hierarchy.md
• include/pypto/ir/core.h
• include/pypto/ir/expr.h
• include/pypto/ir/kind_traits.h
• include/pypto/ir/transforms/base/functor.h
• include/pypto/ir/transforms/base/mutator.h
• include/pypto/ir/transforms/base/visitor.h
• python/bindings/modules/ir.cpp
• python/pypto/language/parser/ast_parser.py
• python/pypto/pypto_core/ir.pyi
• src/codegen/orchestration/orchestration_codegen.cpp
• src/ir/arith/canonical_simplify.cpp
• src/ir/arith/canonical_simplify.h
• src/ir/arith/const_int_bound.cpp
• src/ir/arith/int_set.cpp
• src/ir/arith/modular_set.cpp
• src/ir/arith/rewrite_simplify.cpp
• src/ir/arith/rewrite_simplify.h
• src/ir/transforms/convert_to_ssa_pass.cpp
• src/ir/transforms/derive_call_directions_pass.cpp
• src/ir/transforms/mutator.cpp
• src/ir/transforms/python_printer.cpp
• src/ir/transforms/structural_equal.cpp
• src/ir/transforms/structural_hash.cpp
• src/ir/transforms/utils/dead_code_elimination.cpp
• src/ir/transforms/visitor.cpp
• tests/ut/ir/printing/test_submit_printer.py
• tests/ut/ir/transforms/test_flatten_call_expr_pass.py
• tests/ut/ir/transforms/test_submit_end_to_end.py
• tests/ut/ir/transforms/test_submit_passes.py
• tests/ut/language/parser/test_manual_scope_parsing.py
• tests/ut/language/parser/test_task_id_dsl.py