[GitHub] [tvm] junrushao1994 commented on pull request #10843: [TIR] StmtFunctor ReGenerateDef

[GitBox <gi...@apache.org>]

junrushao1994 commented on pull request #10843:
URL: https://github.com/apache/tvm/pull/10843#issuecomment-1086742057


   > Some additional ideas for possible regressions
   > 
   > ## Symbolic shape case that needs remap
   > Symbolic vars needs to be mapped to the same one
   > 
   > ```python
   > def fn(a: T.handle, b: T.handle, n: T.int32)
   >    m = T.var("int32")
   >    A =  match_buffer((n, m), a)
   >    B = match_bufer((n, m*2), b) 
   > ```
   
   Thanks for this interesting case!
   
   As a further step, there might be usecases where buffers are declared as divisible by power-of-2, for example:
   
   ```python
   def fn(a: T.handle):
       m = T.var("int32")
       A = match_buffer(a, (m * 8, ), "float32")
   ```
   
   @tqchen do you think this is a case we want to support?


-- 
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.

To unsubscribe, e-mail: commits-unsubscribe@tvm.apache.org

For queries about this service, please contact Infrastructure at:
users@infra.apache.org