Skip to content
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.

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

[TVMScript][Relax] Use tir.SizeVar for shape variables #16949

Open
wants to merge 1 commit into
base: main
Choose a base branch
from

Commits on Apr 28, 2024

  1. [TVMScript][Relax] Use tir.SizeVar for shape variables

    As reported in apache#16877, shape inference performed during a Relax
    transformation may produce different results than shape inference
    performed during TVMScript parsing.  While Relax transformations
    call `Analyzer::MarkGlobalNonNegValue` for each shape expression, this
    is not provided during TVMScript parsing.  As a result, output shapes
    that are conditional on the sign of a variable may produce different
    results when simplifying.
    
    This commit provides a partial resolution for this issue.  Where prior
    to this commit, the TVMScript parser generated a `tir.Var` for each
    symbolic variable, the TVMScript parser now provides a `tir.SizeVar`
    for symbolic variables that are defined in contexts that require a
    non-negative size.
    Lunderberg committed Apr 28, 2024
    Configuration menu
    Copy the full SHA
    b4fff5b View commit details
    Browse the repository at this point in the history