diff --git a/docs/source/conf.py b/docs/source/conf.py index 66043c65133c..a326c8cbd202 100644 --- a/docs/source/conf.py +++ b/docs/source/conf.py @@ -999,7 +999,6 @@ coverage_ignore_functions = [ "to_node", "wrap_node", "sym_sqrt", - "sym_ite", # torch.fx.experimental.symbolic_shapes "bind_symbols", "cast_symbool_to_symint_guardless", diff --git a/torch/__init__.py b/torch/__init__.py index f2bf77527c3d..719eb121f880 100644 --- a/torch/__init__.py +++ b/torch/__init__.py @@ -978,6 +978,7 @@ __all__.append("sym_sqrt") def sym_ite(b, t, f): + """SymInt-aware utility for ternary operator (``t if b else f``.)""" if overrides.has_torch_function((b, t, f)): return overrides.handle_torch_function(sym_ite, (b, t, f), b, t, f) assert isinstance(b, (SymBool, builtins.bool)) and type(t) == type(f)