llvm
/
torch-mlir
mirror of https://github.com/llvm/torch-mlir
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

[fximporter] Avoid importing from `_torchMlir` (#3685) 

Downstream projects don't necessarily register this C++ module. This
package removes the dependency and uses `torch.iinfo` to access the max
and min values instead.
pull/3686/head
zjgarvey 2024-09-03 11:52:06 -07:00 committed by GitHub
parent b3942ff984
commit 2960538c6d
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
python/torch_mlir/extras/fx_importer.py
View File

@ -135,7 +135,6 @@ from ..dialects import (
func as func_dialect, func as func_dialect,
) )
from .._mlir_libs._torchMlir import get_int64_max, get_int64_min
__all__ = [ __all__ = [
"FxImporter", "FxImporter",
@ -1186,9 +1185,9 @@ class ContextCache:
def _sympy_int_to_int(val: sympy.Expr, adjust_func: Callable): def _sympy_int_to_int(val: sympy.Expr, adjust_func: Callable):
# Convert simple sympy Integers into concrete int # Convert simple sympy Integers into concrete int
if val in infs: if val in infs:
return get_int64_max() return torch.iinfo(torch.int64).max
if val in tuple(-inf for inf in infs): if val in tuple(-inf for inf in infs):
return get_int64_min() return torch.iinfo(torch.int64).min
if isinstance(val, sympy.Integer): if isinstance(val, sympy.Integer):
return int(val) return int(val)
# TODO: Remove this adjustment when fractional ranges are removed # TODO: Remove this adjustment when fractional ranges are removed