Update base for Update on "Create env to test with TORCHLIB_EXPERIMENTAL_PREFER_TRACING on | test(torchlib)"

### Changes

- Create env to test with TORCHLIB_EXPERIMENTAL_PREFER_TRACING on
- Test with Python 3.11 as well

[ghstack-poisoned]

Author: justinchuby

onnxscript/function_libs/torch_lib/_flags.py

@@ -39,7 +39,3 @@ def _load_boolean_flag(
     "TORCHLIB_EXPERIMENTAL_INITIALIZERS_AS_INPUTS",
     this_will="make initializers as inputs to the model graph",
 )
-EXPERIMENTAL_PREFER_TRACING: bool = _load_boolean_flag(
-    "TORCHLIB_EXPERIMENTAL_PREFER_TRACING",
-    this_will="trace all traceable functions to fold if branches and collapse constant expressions",
-)

onnxscript/function_libs/torch_lib/registration.py

@@ -99,7 +99,6 @@ def torch_op(
     trace_only: bool = False,
     private: bool = False,
     complex: bool = False,
-    traceable: bool = False,
 ) -> Callable[[FunctionType], onnxscript.OnnxFunction | onnxscript.values.TracedOnnxFunction]:
     """Register a torch op.
 
@@ -113,7 +112,6 @@ def torch_op(
         private: Whether the function is private (not directly exposed). It should
             be true for all functions with names starting with "_".
         complex: Whether the function expects complex-valued inputs.
-        traceable: Whether the function can be traced.
     """
     if registry is None:
         registry = default_registry
@@ -130,7 +128,6 @@ def wrapper(
         else:
             assert isinstance(func, FunctionType)
             processed_func = onnxscript.script(opset=custom_opset)(func)
-            processed_func.experimental_traceable = traceable
 
         assert registry is not None
         for name_ in _check_and_normalize_names(name):

onnxscript/values.py

@@ -479,9 +479,6 @@ def __init__(
         self._param_schemas: Optional[tuple[ParamSchema, ...]] = None
         self._op_schema: Optional[onnx.defs.OpSchema] = None
 
-        # Experimental fields
-        self.experimental_traceable = False
-
     @property
     @deprecation.deprecated(
         since="0.1",