From 2960538c6d145a2bd1efa52c56f2bcaa1ffc45aa Mon Sep 17 00:00:00 2001
From: zjgarvey <47986913+zjgarvey@users.noreply.github.com>
Date: Tue, 3 Sep 2024 11:52:06 -0700
Subject: [PATCH] [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.
---
 python/torch_mlir/extras/fx_importer.py | 5 ++---
 1 file changed, 2 insertions(+), 3 deletions(-)

diff --git a/python/torch_mlir/extras/fx_importer.py b/python/torch_mlir/extras/fx_importer.py
index c984e1e52306..a8d2790e9b00 100644
--- a/python/torch_mlir/extras/fx_importer.py
+++ b/python/torch_mlir/extras/fx_importer.py
@@ -135,7 +135,6 @@
     func as func_dialect,
 )
 
-from .._mlir_libs._torchMlir import get_int64_max, get_int64_min
 
 __all__ = [
     "FxImporter",
@@ -1186,9 +1185,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 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