[GitHub] [tvm] Hzfengsy commented on pull request #11518: [TE] Fix `tvm.te.CreatePrimFunc` for 0-dim computation

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

Hzfengsy commented on PR #11518:
URL: https://github.com/apache/tvm/pull/11518#issuecomment-1143269919

   Thanks @junrushao1994 for pointing it out. However, it's not the correct way to enforce the root block to have empty read/write regions.
   
   The fundamental problem here is that we missed the block "C"
   
   And the expected result is 
   ```python
   def func(a: T.Buffer[(), "int32"], b: T.Buffer[(), "int32"], c: T.Buffer[(), "int32"]) -> None:
       # function attr dict
       T.func_attr({"global_symbol": "main", "tir.noalias": True})
       # body
       # with T.block("root")
       with T.block("C"):    # missed now
           c[()] = a[()] + b[()]
   ```


-- 
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