Skip to content

Commit 360744c

Browse files
committed
Fix pylint error
1 parent f58e4b1 commit 360744c

File tree

1 file changed

+6
-6
lines changed
  • python/tvm/script/ir_builder/tir

1 file changed

+6
-6
lines changed

python/tvm/script/ir_builder/tir/ir.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -98,20 +98,20 @@ def _get_block_name_suffix() -> str:
9898

9999

100100
@contextlib.contextmanager
101-
def block_name_suffix_context(suffix: str):
101+
def block_name_suffix_context(block_suffix: str):
102102
"""Context manager to set block name suffix during macro expansion.
103103
104104
Parameters
105105
----------
106-
suffix : str
106+
block_suffix : str
107107
The suffix to append to block names (e.g., "_1", "_2").
108108
109109
Yields
110110
------
111111
None
112112
"""
113113
old_suffix = getattr(_block_name_suffix, "value", "")
114-
_block_name_suffix.value = suffix
114+
_block_name_suffix.value = block_suffix
115115
try:
116116
yield
117117
finally:
@@ -383,9 +383,9 @@ def block(name: str = "", no_realize: bool = False) -> frame.BlockFrame:
383383
res : frame.BlockFrame
384384
The BlockFrame.
385385
"""
386-
suffix = _get_block_name_suffix()
387-
if suffix and name:
388-
name = name + suffix
386+
block_suffix = _get_block_name_suffix()
387+
if block_suffix and name:
388+
name = name + block_suffix
389389
return _ffi_api.Block(name, no_realize) # type: ignore[attr-defined] # pylint: disable=no-member
390390

391391

0 commit comments

Comments
 (0)