OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-06 12:20:52 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity

[sym_shapes][perf] Do not calculate hint in advice_is_size (#124472)

Browse Source 
Differential Revision: [D56352412](https://our.internmc.facebook.com/intern/diff/D56352412)
Pull Request resolved: https://github.com/pytorch/pytorch/pull/124472
Approved by: https://github.com/ezyang
This commit is contained in:
IvanKobzarev 2024-04-19 05:01:05 -07:00 committed by PyTorch MergeBot
parent 661fd23640
commit 73f56e1e81
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
torch/fx/experimental/symbolic_shapes.py
View File

@ -556,8 +556,8 @@ def _advise_is_size(a):
if (
isinstance(a, SymInt)
and isinstance(a.node, SymNode)
and not a.node.has_hint()
and isinstance(a.node.expr, sympy.Symbol)
and a.node.shape_env.is_unbacked_symint(a.node.expr)
):
_constrain_range_for_size(a)