You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@tvm.apache.org by an...@apache.org on 2022/09/17 00:01:26 UTC
[tvm] 08/18: fix new imports
This is an automated email from the ASF dual-hosted git repository.
andrewzhaoluo pushed a commit to branch aluo/rebase-09162022-autotensorization
in repository https://gitbox.apache.org/repos/asf/tvm.git
commit 581e5c3f5d6d5b1161b56b09dc3b85d1edb13cf4
Author: Andrew Luo <an...@gmail.com>
AuthorDate: Wed Aug 17 21:05:08 2022 -0700
fix new imports
---
python/tvm/meta_schedule/default_config.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/python/tvm/meta_schedule/default_config.py b/python/tvm/meta_schedule/default_config.py
index 4f06b5b15c..eaa026e3b4 100644
--- a/python/tvm/meta_schedule/default_config.py
+++ b/python/tvm/meta_schedule/default_config.py
@@ -24,7 +24,7 @@ from tvm._ffi.registry import register_func
from tvm.contrib import nvcc
from tvm.ir import IRModule
from tvm.target import Target
-from tvm.tir import PrimFunc, tensor_intrin
+from tvm.tir import PrimFunc
from .builder import Builder, LocalBuilder
from .cost_model import CostModel, XGBModel
@@ -460,7 +460,7 @@ class _DefaultCUDATensorCore:
return [
M.MultiLevelTilingTensorCore(
intrin_groups=[
- tensor_intrin.get_wmma_intrin_group(
+ get_wmma_intrin_group(
store_scope="shared",
in_dtype=in_dtype,
out_dtype=out_dtype,