You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@tvm.apache.org by tq...@apache.org on 2020/09/16 15:18:58 UTC
[incubator-tvm] branch master updated: Fix missing import in
bifrost schedule (#6479)
This is an automated email from the ASF dual-hosted git repository.
tqchen pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/incubator-tvm.git
The following commit(s) were added to refs/heads/master by this push:
new 8f4e361 Fix missing import in bifrost schedule (#6479)
8f4e361 is described below
commit 8f4e361cbe063a5d512ceb187b7d43ae83e1d656
Author: Shizhi Tang <rd...@gmail.com>
AuthorDate: Wed Sep 16 23:18:49 2020 +0800
Fix missing import in bifrost schedule (#6479)
---
python/tvm/topi/bifrost/gemm.py | 2 ++
1 file changed, 2 insertions(+)
diff --git a/python/tvm/topi/bifrost/gemm.py b/python/tvm/topi/bifrost/gemm.py
index 6f147d9..c06f623 100644
--- a/python/tvm/topi/bifrost/gemm.py
+++ b/python/tvm/topi/bifrost/gemm.py
@@ -16,6 +16,8 @@
# under the License.
# pylint: disable=invalid-name,unused-variable,unused-argument
"""GEMM schedules for Mali Bifrost"""
+from tvm import te
+
from .transforms import tile_and_bind, tile_and_bind3d, interleave_transpose, transpose_interleave
from .. import util