frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

Add missing docstring for sym_ite (#154201)

Browse Source 
`sym_ite` is listed in [the reference page](https://docs.pytorch.org/docs/stable/torch.html) and has no document.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/154201
Approved by: https://github.com/Skylion007
This commit is contained in:
Yuki Kobayashi
2025-05-26 15:59:19 +00:00
committed by PyTorch MergeBot
parent 02445ec8f0
commit f55f2f42a7
2 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
docs/source/conf.py
View File

@ -999,7 +999,6 @@ coverage_ignore_functions = [
"to_node", "to_node",
"wrap_node", "wrap_node",
"sym_sqrt", "sym_sqrt",
"sym_ite",
# torch.fx.experimental.symbolic_shapes # torch.fx.experimental.symbolic_shapes
"bind_symbols", "bind_symbols",
"cast_symbool_to_symint_guardless", "cast_symbool_to_symint_guardless",

1
torch/__init__.py
View File

@ -978,6 +978,7 @@ __all__.append("sym_sqrt")
def sym_ite(b, t, f): 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)): if overrides.has_torch_function((b, t, f)):
return overrides.handle_torch_function(sym_ite, (b, t, f), 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) assert isinstance(b, (SymBool, builtins.bool)) and type(t) == type(f)