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 8167a0cb5cd9092a786afbb6d709d18aa9c55a61 Author: Andrew Luo <[email protected]> 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,
