[Bugfix][TIR] Fix version conflict with typing for Python 3.8.0

[Ever-Kid]

I came across this bug under python3.8.0 with error from typing.get_args() while trying to run testcases like tests/python/unittest/test_tir_schedule_set_axis_separator.py::test_set_axis_separator[transform_layout_named]

>           if get_origin(tp) is collections.abc.Callable and res[0] is not Ellipsis:
E           IndexError: tuple index out of range

And the root cause here is a difference between python3.8.0 and later version:

        get_args(Callable[[], T][int]) == ([], int)
    """
-     if isinstance(tp, _GenericAlias):    // python3.8.0
+    if isinstance(tp, _GenericAlias) and not tp._special:    // python3.8.15
        res = tp.__args__
        if get_origin(tp) is collections.abc.Callable and res[0] is not Ellipsis:
}

So I added it back to python/tvm/tir/schedule/_type_checker.py

[tvm-bot]

Thanks for contributing to TVM! Please refer to the contributing guidelines https://tvm.apache.org/docs/contribute/ for useful information and tips. Please request code reviews from Reviewers by @-ing them in a comment.

• cc @Hzfengsy, @junrushao, @quic-sanirudh _{See #10317 for details}

_{Generated by tvm-bot}

[Ever-Kid]

cc. @sunggg

[junrushao]

@sunggg

[sunggg]

LGTM!

[junrushao]

so I’m getting it in!

[junrushao]

for some reason the CI is stuck on the new pipeline added today...let me try to retrigger

[junrushao]

@tvm-bot rerun

[junrushao]

I rebased this branch to latest HEAD so that the CI could run properly

[junrushao]

It's merged! Thanks for the PR!

[Ever-Kid]

Thanks you guys’ kindly help~

[vinx13]

I'm getting some error in Python 3.10

  File "/usr/lib/python3.10/typing.py", line 983, in __getattr__
    return getattr(self.__origin__, attr)
AttributeError: type object 'Callable' has no attribute '_special'

The attribute _special was actually removed since Python 3.9.0 (they introduced SpecialGenericAlias instead).

[junrushao]

Oh no. would you mind having it fixed if it’s not super hard?

[sunggg]

Darn, let me fix this. Shouldn't be hard.

Edit: until fix is merged, please remove ‘ty._special’ as a temporary solution.
PR is under review: #13820 (comment)