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

[TIR] Fix RenewDef for symbolic input shapes #15163

Merged

Commits on Jun 27, 2023

  1. [TIR] Fix RenewDef for symbolic input shapes

    There are cases where the shapes of input buffers are symbolic, but the
    first symbol is a composite PrimExpr rather than a TIR Var, which the
    original implementation does not take this into account.
    
    Example:
    
    ```python
    @T.prim_func
    def main(a: T.handle, b: T.handle):
        m = T.int64()
        A = T.match_buffer(a, (m * 2,))  // `m` first appears as composite
        B = T.match_buffer(b, (m, 2))
    ```
    junrushao committed Jun 27, 2023
    Configuration menu
    Copy the full SHA
    ac2b079 View commit details
    Browse the repository at this point in the history