[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

View File

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