Skip to content

Commit

Permalink
pylint
Browse files Browse the repository at this point in the history
  • Loading branch information
masahi committed Apr 18, 2022
1 parent da05e6e commit d6ae848
Showing 1 changed file with 3 additions and 5 deletions.
8 changes: 3 additions & 5 deletions python/tvm/tir/schedule/analysis.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand All @@ -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,
Expand Down Expand Up @@ -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
Expand Down

0 comments on commit d6ae848

Please sign in to comment.