From ca15255e3a882b89b05bb83079640c929fb63096 Mon Sep 17 00:00:00 2001 From: Masahiro Masuda Date: Thu, 7 Apr 2022 05:41:13 +0900 Subject: [PATCH] add RewriteVNNI --- python/tvm/meta_schedule/postproc/__init__.py | 1 + python/tvm/meta_schedule/tune.py | 1 + 2 files changed, 2 insertions(+) diff --git a/python/tvm/meta_schedule/postproc/__init__.py b/python/tvm/meta_schedule/postproc/__init__.py index 96361e739186..0b95960b313f 100644 --- a/python/tvm/meta_schedule/postproc/__init__.py +++ b/python/tvm/meta_schedule/postproc/__init__.py @@ -22,3 +22,4 @@ from .rewrite_reduction_block import RewriteReductionBlock from .rewrite_unbound_block import RewriteUnboundBlock from .verify_gpu_code import VerifyGPUCode +from .rewrite_vnni import RewriteVNNI diff --git a/python/tvm/meta_schedule/tune.py b/python/tvm/meta_schedule/tune.py index 86157e0fb32e..ecf70053a5d1 100644 --- a/python/tvm/meta_schedule/tune.py +++ b/python/tvm/meta_schedule/tune.py @@ -214,6 +214,7 @@ def _postproc() -> List[Postproc]: M.DisallowDynamicLoop(), M.RewriteParallelVectorizeUnroll(), M.RewriteReductionBlock(), + M.RewriteVNNI(), ] @staticmethod