From 5268df9af1b297095ce3da6788c9a60e7ad63aa7 Mon Sep 17 00:00:00 2001 From: dxtao29 Date: Fri, 20 Jun 2025 14:55:50 +0800 Subject: [PATCH] fix exception when tvm not built with llvm support --- python/tvm/tir/tensor_intrin/__init__.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/python/tvm/tir/tensor_intrin/__init__.py b/python/tvm/tir/tensor_intrin/__init__.py index 7e5a26bdeb43..564655455245 100644 --- a/python/tvm/tir/tensor_intrin/__init__.py +++ b/python/tvm/tir/tensor_intrin/__init__.py @@ -16,4 +16,8 @@ # under the License. # pylint: disable=unused-import """Intrinsics for tensorization.""" -from . import arm_cpu, cuda, rocm, x86, hexagon +from tvm.runtime import enabled +from . import cuda + +if enabled("llvm"): + from . import arm_cpu, x86, rocm, hexagon