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/16 22:47:31 UTC
[tvm] 07/20: update configs
This is an automated email from the ASF dual-hosted git repository.
andrewzhaoluo pushed a commit to branch aluo/rebase-08312022-autotensorization-fq2i-changes
in repository https://gitbox.apache.org/repos/asf/tvm.git
commit 1bae92e12fd595558fe4abe7b913000378b62d9e
Author: Andrew Luo <an...@gmail.com>
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 c1e886e7f7..a88a9be88d 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,