We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent da99e9d commit 1bbfc02Copy full SHA for 1bbfc02
python/tvm/tir/schedule/_type_checker.py
@@ -98,7 +98,7 @@ def union(type_: Any) -> Optional[List[type]]: # pylint: disable=missing-functi
98
@staticmethod
99
def callable(type_: Any) -> Optional[List[type]]:
100
if _Subtype._origin(type_) is collections.abc.Callable:
101
- if hasattr(typing, "get_args"):
+ if hasattr(typing, "get_args") and not type_._special:
102
subtypes = typing.get_args(type_) # type: ignore
103
else:
104
subtypes = type_.__args__
0 commit comments