From e5f52a760f11a86819e2794d282d3de371654406 Mon Sep 17 00:00:00 2001 From: Felix <83807609+felix-ro@users.noreply.github.com> Date: Tue, 4 Jun 2024 13:34:20 +0100 Subject: [PATCH] fixed TensorIntrin not registered bug --- python/tvm/tir/tensor_intrin/__init__.py | 1 + 1 file changed, 1 insertion(+) 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