File tree Expand file tree Collapse file tree 1 file changed +6
-6
lines changed
python/tvm/script/ir_builder/tir Expand file tree Collapse file tree 1 file changed +6
-6
lines changed Original file line number Diff line number Diff 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
You can’t perform that action at this time.
0 commit comments