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 9a86b32917c2e0ab89ea77c79e23e55292691698 Author: Andrew Luo <[email protected]> AuthorDate: Wed Aug 17 11:01:56 2022 -0700 update configs --- 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 eaa026e3b4..4f06b5b15c 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 +from tvm.tir import PrimFunc, tensor_intrin from .builder import Builder, LocalBuilder from .cost_model import CostModel, XGBModel @@ -460,7 +460,7 @@ class _DefaultCUDATensorCore: return [ M.MultiLevelTilingTensorCore( intrin_groups=[ - get_wmma_intrin_group( + tensor_intrin.get_wmma_intrin_group( store_scope="shared", in_dtype=in_dtype, out_dtype=out_dtype,
