Fix missing import in bifrost schedule (#6479)

diff --git a/python/tvm/topi/bifrost/gemm.py b/python/tvm/topi/bifrost/gemm.py
index 6f147d9..c06f623 100644 (file)
--- 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