apache / tvm

Open deep learning compiler stack for cpu, gpu and specialized accelerators
https://tvm.apache.org/
Apache License 2.0
11.41k stars 3.4k forks source link

[TVMScript] Better Type Annotation for TIR OP #17107

Closed Hzfengsy closed 2 weeks ago

Hzfengsy commented 2 weeks ago

Enable ParamType for TIR op, so that we can have better experience when writing TVMScript in Python with tools.

However, ParamType is introduced in Python 3.10, so we only enable it when Python version is 3.10 or above.