Skip to content

Commit

Permalink
Revert "[fximporter] Avoid importing from _torchMlir (#3685)"
Browse files Browse the repository at this point in the history
This reverts commit 2960538.
  • Loading branch information
patel-vimal committed Sep 10, 2024
1 parent b35675a commit 35be39e
Showing 1 changed file with 3 additions and 2 deletions.
5 changes: 3 additions & 2 deletions python/torch_mlir/extras/fx_importer.py
Original file line number Diff line number Diff line change
Expand Up @@ -135,6 +135,7 @@
func as func_dialect,
)

from .._mlir_libs._torchMlir import get_int64_max, get_int64_min

__all__ = [
"FxImporter",
Expand Down Expand Up @@ -1185,9 +1186,9 @@ def set_symbolic_guards(
def _sympy_int_to_int(val: sympy.Expr, adjust_func: Callable):
# Convert simple sympy Integers into concrete int
if val in infs:
return torch.iinfo(torch.int64).max
return get_int64_max()
if val in tuple(-inf for inf in infs):
return torch.iinfo(torch.int64).min
return get_int64_min()
if isinstance(val, sympy.Integer):
return int(val)
# TODO: Remove this adjustment when fractional ranges are removed
Expand Down

0 comments on commit 35be39e

Please sign in to comment.