[discuss] Better hinting and auto completion for TIR script

[yzh119]

Current TIR script is a sub-language embedded in python frontend, user cannot use type hinting and auto completion tools because the code is not parsed by python.

We can create a .pyi stub for these keywords which only annotates type and leave the implementations empty (actually we don't need them). The stub could also help us generate language reference for user to grasp the syntax and semantics of TIR scripts.

For example, if I write a __init__.pyi under tvm.tir (from simplicity, it only cover a few keywords):

from typing import Tuple, Union, List
from .schedule import LoopRV, BlockRV
from ..ir import PrimExpr

class serial:
    def __init__(self, start: PrimExpr, end: PrimExpr) -> None: ...
    
    def __enter__(self) -> LoopRV: ...

class grid:
    def __init__(self, *args: PrimExpr) -> None: ...

    def __enter__(self) -> LoopRV: ...

class block:
    def __init__(self, *args: PrimExpr, name: str = '') -> None: ...

    def __enter__(self) -> List[LoopRV]: ...

it could help type hinting and auto completion when user write their TIR scripts.

However, there might be some conflict between python functions under tvm.tir and tir script keywords. @junrushao1994 's suggestion in to create another namespace (e.g. tvm.tir.T), I wonder do you have any elegant proposal for this?

[MasterJH5574]

However, there might be some conflict between python functions under tvm.tir and tir script keywords.

I'm not clear that how the conflict looks like? Could you give an example 😳?

[yzh119]

For example, functions defined under tir.op always have an argument span, which is not required in script mode.

[junrushao]

Moved to #471