From 3dbcab38d5f9a95d85dc25f406730a8760d78a35 Mon Sep 17 00:00:00 2001 From: Krzysztof Parzyszek Date: Sat, 27 May 2023 16:41:35 -0700 Subject: [PATCH] Rearrange pylint directives for better formatting --- python/tvm/tir/schedule/schedule.py | 75 ++++++++++++----------------- 1 file changed, 30 insertions(+), 45 deletions(-) diff --git a/python/tvm/tir/schedule/schedule.py b/python/tvm/tir/schedule/schedule.py index e6dac80824e28..8996ad49a68d7 100644 --- a/python/tvm/tir/schedule/schedule.py +++ b/python/tvm/tir/schedule/schedule.py @@ -305,9 +305,8 @@ def get( """ if isinstance(rand_var_or_sref, StmtSRef): return rand_var_or_sref.stmt - result = _ffi_api.ScheduleGet( - self, rand_var_or_sref - ) # type: ignore # pylint: disable=no-member + # pylint: disable-next=no-member + result = _ffi_api.ScheduleGet(self, rand_var_or_sref) # type: ignore if isinstance(result, IntImm): result = result.value return result @@ -468,9 +467,8 @@ def get_loops(self, block: Union[BlockRV, str]) -> List[LoopRV]: A list of loops above the given block in its scope, from outer to inner """ block = self._normalize_block_arg(block) - return list( - _ffi_api.ScheduleGetLoops(self, block) - ) # type: ignore # pylint: disable=no-member + # pylint: disable-next=no-member + return list(_ffi_api.ScheduleGetLoops(self, block)) # type: ignore @type_checked def get_child_blocks(self, block_or_loop: Union[BlockRV, LoopRV]) -> List[BlockRV]: @@ -486,9 +484,8 @@ def get_child_blocks(self, block_or_loop: Union[BlockRV, LoopRV]) -> List[BlockR blocks : List[LoopRV] A list of leaf blocks inside a specific block/loop """ - return list( - _ffi_api.ScheduleGetChildBlocks(self, block_or_loop) - ) # type: ignore # pylint: disable=no-member + # pylint: disable-next=no-member + return list(_ffi_api.ScheduleGetChildBlocks(self, block_or_loop)) # type: ignore @type_checked def get_producers(self, block: Union[BlockRV, str]) -> List[BlockRV]: @@ -505,9 +502,8 @@ def get_producers(self, block: Union[BlockRV, str]) -> List[BlockRV]: A list of producers of the given block """ block = self._normalize_block_arg(block) - return list( - _ffi_api.ScheduleGetProducers(self, block) - ) # type: ignore # pylint: disable=no-member + # pylint: disable-next=no-member + return list(_ffi_api.ScheduleGetProducers(self, block)) # type: ignore @type_checked def get_consumers(self, block: Union[BlockRV, str]) -> List[BlockRV]: @@ -524,9 +520,8 @@ def get_consumers(self, block: Union[BlockRV, str]) -> List[BlockRV]: A list of consumers of the given block """ block = self._normalize_block_arg(block) - return list( - _ffi_api.ScheduleGetConsumers(self, block) - ) # type: ignore # pylint: disable=no-member + # pylint: disable-next=no-member + return list(_ffi_api.ScheduleGetConsumers(self, block)) # type: ignore @type_checked def get_output_blocks(self, scope_block: Union[BlockRV, str]) -> List[BlockRV]: @@ -546,9 +541,8 @@ def get_output_blocks(self, scope_block: Union[BlockRV, str]) -> List[BlockRV]: """ scope_block = self._normalize_block_arg(scope_block) - return list( - _ffi_api.ScheduleGetOutputBlocks(self, scope_block) - ) # type: ignore # pylint: disable=no-member + # pylint: disable-next=no-member + return list(_ffi_api.ScheduleGetOutputBlocks(self, scope_block)) # type: ignore ########## Schedule: Transform loops ########## @type_checked @@ -685,9 +679,8 @@ def after_fuse(a: T.handle, b: T.handle) -> None: B[vi, vj] = A[vi, vj] * 2.0 """ - return _ffi_api.ScheduleFuse( - self, loops, preserve_unit_iters - ) # type: ignore # pylint: disable=no-member + # pylint: disable-next=no-member + return _ffi_api.ScheduleFuse(self, loops, preserve_unit_iters) # type: ignore @type_checked def split( @@ -894,9 +887,8 @@ def matmul_after_reorder_block_iter_var( -------- reorder """ - _ffi_api.ScheduleReorderBlockIterVar( - self, block, new_order - ) # type: ignore # pylint: disable=no-member + # pylint: disable-next=no-member + _ffi_api.ScheduleReorderBlockIterVar(self, block, new_order) # type: ignore @type_checked def add_unit_loop(self, block_or_loop: Union[LoopRV, BlockRV]) -> LoopRV: @@ -952,9 +944,8 @@ def after_add_unit_loop( vi = T.axis.spatial(1, 0) C[()] = A[()] + B[()] """ - return _ffi_api.ScheduleAddUnitLoop( - self, block_or_loop - ) # type: ignore # pylint: disable=no-member + # pylint: disable-next=no-member + return _ffi_api.ScheduleAddUnitLoop(self, block_or_loop) # type: ignore ########## Schedule: Manipulate ForKind ########## @@ -2172,9 +2163,8 @@ def after_inline(a: T.handle, c: T.handle) -> None: """ block = self._normalize_block_arg(block) - _ffi_api.ScheduleReverseComputeInline( - self, block - ) # type: ignore # pylint: disable=no-member + # pylint: disable-next=no-member + _ffi_api.ScheduleReverseComputeInline(self, block) # type: ignore ########## Schedule: Reduction ########## @@ -2254,9 +2244,8 @@ def after_decompose(a: ty.handle, c: ty.handle) -> None: """ block = self._normalize_block_arg(block) - return _ffi_api.ScheduleDecomposeReduction( - self, block, loop - ) # type: ignore # pylint: disable=no-member + # pylint: disable-next=no-member + return _ffi_api.ScheduleDecomposeReduction(self, block, loop) # type: ignore @type_checked def rfactor(self, loop: LoopRV, factor_axis: int) -> BlockRV: @@ -2405,9 +2394,8 @@ def after_rfactor(a: T.handle, b: T.handle) -> None: where `B` is the buffer that the reduction block writes to. Negative indexing is normalized according to numpy convention. """ - return _ffi_api.ScheduleRFactor( - self, loop, factor_axis - ) # type: ignore # pylint: disable=no-member + # pylint: disable-next=no-member + return _ffi_api.ScheduleRFactor(self, loop, factor_axis) # type: ignore ######## Schedule: Block annotation ######## @@ -2724,9 +2712,8 @@ def after_blockize( block are divisible by the subspace represented by the loops starting at the given loop. """ - return _ffi_api.ScheduleBlockize( - self, target, preserve_unit_iters - ) # type: ignore # pylint: disable=no-member + # pylint: disable-next=no-member + return _ffi_api.ScheduleBlockize(self, target, preserve_unit_iters) # type: ignore @type_checked def tensorize( @@ -3489,9 +3476,8 @@ def after_decompose(x: T.Buffer(128, "int32"), y: T.Buffer(140, "int32")): @type_checked def can_decompose_padding(self, block: Union[BlockRV, str], loop: LoopRV) -> bool: """Check whether the block match padding pattern and can be decomposed.""" - return _ffi_api.CanDecomposePadding( - self, block, loop - ) # type: ignore # pylint: disable=no-member + # pylint: disable-next=no-member + return _ffi_api.CanDecomposePadding(self, block, loop) # type: ignore @type_checked def pad_einsum(self, block: Union[BlockRV, str], padding: List[int]) -> None: @@ -3717,9 +3703,8 @@ def after_rolling_buffer( The region_cover property of the consumer block of the target buffer will become false. """ block = self._normalize_block_arg(block) - return _ffi_api.ScheduleRollingBuffer( - self, block, write_buffer_index - ) # type: ignore # pylint: disable=no-member + # pylint: disable-next=no-member + return _ffi_api.ScheduleRollingBuffer(self, block, write_buffer_index) # type: ignore ########## Schedule: Misc ##########