Repository navigation
[SCRIPT] Support PEP 695 symbolic variables in Relax and TIR - #20107
Merged
Merged
Conversation
tqchen
force-pushed
the
tvm-type-var-parser-syntax
branch
from
August 8, 2026 11:09
5719007 to
af9465b
Compare
spectrometerHBH
approved these changes
Aug 9, 2026
tqchen
added a commit
that referenced
this pull request
Oct 16, 2026
Relax and TIRx dependent type fields need symbolic integer variables to
be declared in a function signature without persisting parser-only
metadata in IR.
- Accept standard module-level `TypeVar` declarations on Python 3.10+
and bare or int-bound PEP 695 parameters on Python 3.12+.
- Treat declarations as syntax sugar for fresh int64 TIR variables used
by dependent signature fields; unused declarations disappear and no
function attrs, type-parameter fields, or runtime parameters are added.
- Parse both versioned forms through shared signature plumbing while
retaining dialect-specific annotation and body handling.
Representative TIRx Python 3.12+ output uses PEP 695 and postponed
annotations, so dependent expressions are unquoted:
```python
from __future__ import annotations
@T.prim_func(private=True)
def main[M](A: T.Buffer((M, M * T.int64(2)), "float32")):
A[0, 0] = T.float32(1.0)
```
Before Python 3.12, the portable TIRx output uses a module-level
`TypeVar` and quotes compound dependent expressions:
```python
M = TypeVar("M")
@T.prim_func(private=True)
def main(A: T.Buffer((M, "M * T.int64(2)"), "float32")):
A[0, 0] = T.float32(1.0)
```
The same declaration and printing rules apply to Relax tensor
signatures.
Validation:
- Latest-main native build completed successfully.
- Consolidated Relax and TIRx TypeVar round-trip tests passed.
- Related Relax/TIRx parser, printer, and well-formedness suites passed
661 tests with 4 expected failures.
- Touched-file pre-commit hooks and diff checks passed.
- The implementation does not change Relax well-formedness analysis.
tqchen
added a commit
that referenced
this pull request
Oct 18, 2026
Relax and TIRx dependent type fields need symbolic integer variables to
be declared in a function signature without persisting parser-only
metadata in IR.
- Accept standard module-level `TypeVar` declarations on Python 3.10+
and bare or int-bound PEP 695 parameters on Python 3.12+.
- Treat declarations as syntax sugar for fresh int64 TIR variables used
by dependent signature fields; unused declarations disappear and no
function attrs, type-parameter fields, or runtime parameters are added.
- Parse both versioned forms through shared signature plumbing while
retaining dialect-specific annotation and body handling.
Representative TIRx Python 3.12+ output uses PEP 695 and postponed
annotations, so dependent expressions are unquoted:
```python
from __future__ import annotations
@T.prim_func(private=True)
def main[M](A: T.Buffer((M, M * T.int64(2)), "float32")):
A[0, 0] = T.float32(1.0)
```
Before Python 3.12, the portable TIRx output uses a module-level
`TypeVar` and quotes compound dependent expressions:
```python
M = TypeVar("M")
@T.prim_func(private=True)
def main(A: T.Buffer((M, "M * T.int64(2)"), "float32")):
A[0, 0] = T.float32(1.0)
```
The same declaration and printing rules apply to Relax tensor
signatures.
Validation:
- Latest-main native build completed successfully.
- Consolidated Relax and TIRx TypeVar round-trip tests passed.
- Related Relax/TIRx parser, printer, and well-formedness suites passed
661 tests with 4 expected failures.
- Touched-file pre-commit hooks and diff checks passed.
- The implementation does not change Relax well-formedness analysis.
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Add this suggestion to a batch that can be applied as a single commit.This suggestion is invalid because no changes were made to the code.Suggestions cannot be applied while the pull request is closed.Suggestions cannot be applied while viewing a subset of changes.Only one suggestion per line can be applied in a batch.Add this suggestion to a batch that can be applied as a single commit.Applying suggestions on deleted lines is not supported.You must change the existing code in this line in order to create a valid suggestion.Outdated suggestions cannot be applied.This suggestion has been applied or marked resolved.Suggestions cannot be applied from pending reviews.Suggestions cannot be applied on multi-line comments.Suggestions cannot be applied while the pull request is queued to merge.Suggestion cannot be applied right now. Please check back later.
Relax and TIRx dependent type fields need symbolic integer variables to be declared in a function signature without persisting parser-only metadata in IR.
TypeVardeclarations on Python 3.10+ and bare or int-bound PEP 695 parameters on Python 3.12+.Representative TIRx Python 3.12+ output uses PEP 695 and postponed annotations, so dependent expressions are unquoted:
Before Python 3.12, the portable TIRx output uses a module-level
TypeVarand quotes compound dependent expressions:The same declaration and printing rules apply to Relax tensor signatures.
Validation: