lint

Author: mengshyu

python/tvm/dlight/cpu/gemv.py

@@ -145,10 +145,10 @@ class GEMV(CPUScheduleRule):
     """A rule for GEMV and DecodeGEMV."""
 
     def apply(  # pylint: disable=too-many-locals,too-many-branches,too-many-return-statements
-            self,
-            func: tir.PrimFunc,
-            target: Target,
-            _: bool,
+        self,
+        func: tir.PrimFunc,
+        target: Target,
+        _: bool,
     ) -> Union[None, tir.Schedule, List[tir.Schedule]]:
         if not isinstance(func, tir.PrimFunc) or not self.is_target_available(target):
             return None
@@ -211,7 +211,7 @@ def apply(
             gemv,
             vector_width: int = 8,
             parallel_threads: int = 8,
-            unroll_factor: int = 256
+            unroll_factor: int = 256,
         ):
             batch, s, r, c = sch.get_loops(block)
             len_batch, len_s, len_r, len_c = (

python/tvm/dlight/cpu/utils.py

@@ -40,4 +40,3 @@ def auto_vectorize(sch: tir.Schedule, loop: tir.schedule.LoopRV, max_vec: int):
         return
     v = loop if extent <= max_vec else sch.split(loop, factors=[None, max_vec])[-1]
     sch.vectorize(v)
-