shylie/llvm-project
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

[MLIR] [python] Fixed the signature of _OperationBase.get_asm (#136676)

Browse Source 
It claimed to return an `io.StringIO` or an `io.BytesIO`, but it did in
fact return `str` or `bytes`.
This commit is contained in:
Sergei Lebedev 2025-04-23 02:53:35 +01:00 committed by GitHub
parent 85b35a9077
commit 4e679ea89c
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 19 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

31
mlir/python/mlir/_mlir_libs/_mlir/ir.pyi
View File

@ -45,9 +45,8 @@ from __future__ import annotations
import abc import abc
import collections import collections
from collections.abc import Callable, Sequence from collections.abc import Callable, Sequence
import io
from pathlib import Path from pathlib import Path
from typing import Any, BinaryIO, ClassVar, TypeVar, overload from typing import Any, BinaryIO, ClassVar, Literal, TypeVar, overload
__all__ = [ __all__ = [
"AffineAddExpr", "AffineAddExpr",
@ -196,6 +195,19 @@ class _OperationBase:
Detaches the operation from its parent block. Detaches the operation from its parent block.
""" """
def erase(self) -> None: ... def erase(self) -> None: ...
@overload
def get_asm(
binary: Literal[True],
large_elements_limit: int | None = None,
enable_debug_info: bool = False,
pretty_debug_info: bool = False,
print_generic_op_form: bool = False,
use_local_scope: bool = False,
assume_verified: bool = False,
skip_regions: bool = False,
) -> bytes: ...
@overload
def get_asm( def get_asm(
self, self,
binary: bool = False, binary: bool = False,
@ -206,19 +218,14 @@ class _OperationBase:
use_local_scope: bool = False, use_local_scope: bool = False,
assume_verified: bool = False, assume_verified: bool = False,
skip_regions: bool = False, skip_regions: bool = False,
) -> io.BytesIO | io.StringIO: ) -> str:
""" """
Gets the assembly form of the operation with all options available. Returns the assembly form of the operation.
Args: See the print() method for common keyword arguments for configuring
binary: Whether to return a bytes (True) or str (False) object. Defaults to the output.
False.
... others ...: See the print() method for common keyword arguments for
configuring the printout.
Returns:
Either a bytes or str object, depending on the setting of the 'binary'
argument.
""" """
def move_after(self, other: _OperationBase) -> None: def move_after(self, other: _OperationBase) -> None:
""" """
Puts self immediately after the other operation in its parent block. Puts self immediately after the other operation in its parent block.