Skip to content

Commit 5268df9

Browse files
committed
fix exception when tvm not built with llvm support
1 parent 36f2502 commit 5268df9

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

python/tvm/tir/tensor_intrin/__init__.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -16,4 +16,8 @@
1616
# under the License.
1717
# pylint: disable=unused-import
1818
"""Intrinsics for tensorization."""
19-
from . import arm_cpu, cuda, rocm, x86, hexagon
19+
from tvm.runtime import enabled
20+
from . import cuda
21+
22+
if enabled("llvm"):
23+
from . import arm_cpu, x86, rocm, hexagon

0 commit comments

Comments
 (0)