[GitHub] [tvm] Hzfengsy opened a new pull request, #12147: [UX][TIR][Schedule] enhance function annotation for tir primitive

[GitBox]

Hzfengsy opened a new pull request, #12147:
URL: https://github.com/apache/tvm/pull/12147

   We do have type annotations for tir primitives
   ```Python
   @type_checked
   def get_block(
       self,
       name: str,
       func_name: Optional[str] = None,
   ) -> BlockRV:
   ```
   
   However, after calling the decorator `@type_checked`, all functions become 
`(func: (...) -> Any) -> ((...) -> Any)`. So there are no code suggestions 
during programming. Please see the following screenshot:
   <img width="615" alt="Screen Shot 2022-07-20 at 20 06 15" 
src="https://user-images.githubusercontent.com/25500082/179981888-6162ab3c-f25a-4d18-a1af-a115e2cce485.png";>
   
   After the fix, it becomes
   <img width="615" alt="Screen Shot 2022-07-20 at 20 05 36" 
src="https://user-images.githubusercontent.com/25500082/179981949-fb946e62-e29d-4bdd-a1ef-5ea146956fa0.png";>
   
   cc @junrushao1994 


-- 
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-unsubscr...@tvm.apache.org

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