diff --git a/python/tvm/tir/schedule/analysis.py b/python/tvm/tir/schedule/analysis.py index 7122d8c68813..459931228a77 100644 --- a/python/tvm/tir/schedule/analysis.py +++ b/python/tvm/tir/schedule/analysis.py @@ -17,6 +17,9 @@ """Analysis used in TensorIR scheduling""" from typing import List, Optional +import tvm._ffi +from tvm.runtime import Object + from ..buffer import Buffer from ..stmt import For from ..expr import PrimExpr @@ -25,9 +28,6 @@ from . import _ffi_api from .schedule import Schedule, BlockRV -import tvm._ffi -from tvm.runtime import Object - def suggest_index_map( buffer: Buffer, @@ -66,8 +66,6 @@ def suggest_index_map( class TensorizeInfo(Object): """Necessary information used for tensorization.""" - pass - def get_tensorize_loop_mapping(sch: Schedule, block: BlockRV, desc_func: PrimFunc) -> TensorizeInfo: """Establish a mapping between loops in a target block and an intrinsic description