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

Hack SymInt.__iadd__ to be working. (#94136)

Browse Source 
Signed-off-by: Edward Z. Yang <ezyang@meta.com>
Pull Request resolved: https://github.com/pytorch/pytorch/pull/94136
Approved by: https://github.com/Skylion007
This commit is contained in:
Edward Z. Yang 2023-02-04 08:00:40 -08:00 committed by PyTorch MergeBot
parent c1da35af5e
commit 834e8f0464
1 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
torch/__init__.py
View File

@ -243,6 +243,11 @@ class SymInt:
def __int__(self):
return self.node.int_()
# This is a hack, shouldn't be necessary. Helps
# pyhpc_turbulent_kinetic_energy and vision_maskrcnn
def __iadd__(self, other):
return self + other
# Magic methods installed by torch.fx.experimental.symbolic_shapes
def __eq__(self, other: object) -> builtins.bool: