test within ir structure

Author: titaiwangms

onnxscript/function_libs/torch_lib/_flags.py

@@ -51,8 +51,3 @@ def _load_boolean_flag(
     this_will="trace all traceable functions to fold if branches and collapse constant expressions",
     default=True,
 )
-EXPERIMENTAL_USE_IR: bool = _load_boolean_flag(
-    "TORCHLIB_EXPERIMENTAL_USE_IR",
-    this_will="use the ONNX IR instead of the PyTorch Graph for graph building",
-    deprecated=True,
-)

onnxscript/function_libs/torch_lib/graph_building/__init__.py

@@ -40,17 +40,9 @@
     "TorchScriptTracingEvaluator",
 ]
 
-from onnxscript.function_libs.torch_lib import _flags
-
-if _flags.EXPERIMENTAL_USE_IR:
-    from ._graph_building_ir import (
-        TorchScriptGraph,
-        TorchScriptTensor,
-        TorchScriptTracingEvaluator,
-    )
-else:
-    from ._graph_building_torch import (  # type: ignore[assignment]
-        TorchScriptGraph,
-        TorchScriptTensor,
-        TorchScriptTracingEvaluator,
-    )
+
+from ._graph_building_ir import (
+    TorchScriptGraph,
+    TorchScriptTensor,
+    TorchScriptTracingEvaluator,
+)