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 <[email protected]>
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
