diff --git a/python/tvm/tir/tensor_intrin/__init__.py b/python/tvm/tir/tensor_intrin/__init__.py index d127335e82a6..7e5a26bdeb43 100644 --- a/python/tvm/tir/tensor_intrin/__init__.py +++ b/python/tvm/tir/tensor_intrin/__init__.py @@ -16,3 +16,4 @@ # under the License. # pylint: disable=unused-import """Intrinsics for tensorization.""" +from . import arm_cpu, cuda, rocm, x86, hexagon