[BugFix][MetaSchedule] Fix TensorIntrin ‘dot_4x4_i8i8s32_sdot’ is not registered (#17066)

felix-ro · web-flow · commit 0e622e140c2d · 2024-06-06T09:38:32.000+01:00
fixed TensorIntrin not registered bug
diff --git 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
