[TIR][Schedule] Remove `@type_check` decorator for `set_axis_separator`

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

[TIR][Schedule] Remove `@type_check` decorator for `set_axis_separator` #17134

Open abhikran-quic opened 2 days ago

abhikran-quic commented 2 days ago

The decorator is not allowing types like Array to be passed to set_axis_separator directive.
The FFI has a type checking implemented internally making this redundant.
The discussion for this change came up in a review comment in https://github.com/apache/tvm/pull/17115

abhikran-quic commented 2 days ago

@tvm-bot rerun

abhikran-quic commented 1 day ago

cc: @Lunderberg I've tried to address a review comment in https://github.com/apache/tvm/pull/17115 where we discussed about @type_check in set_axis_separator. Please help in reviewing this change. Thank you!