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

Remove dead get_shape_groups (#120813)

Browse Source 
Signed-off-by: Edward Z. Yang <ezyang@meta.com>
Pull Request resolved: https://github.com/pytorch/pytorch/pull/120813
Approved by: https://github.com/albanD
This commit is contained in:
Edward Z. Yang
2024-03-06 18:36:45 +00:00
committed by PyTorch MergeBot
parent 18d574a07a
commit 6490441d8f
1 changed files with 0 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -3403,13 +3403,6 @@ class ShapeEnv:
return '\n'.join(f" - {guard.expr}{format_tb(guard.stack)}" for guard in self.guards)
def get_shape_groups(self):
"""Returns lists of symbols grouped by the expression they are equivalent to"""
shape_groups = collections.defaultdict(list)
for k, v in self.replacements.items():
shape_groups[v].append(k)
return shape_groups
def bound_sympy(self, expr: sympy.Expr, size_oblivious: bool = False) -> ValueRanges:
"""Given a sympy expression, computes a ValueRanges bound for what values it can be"""
var_to_range = {x: self.var_to_range.get(x, None) for x in expr.free_symbols}