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

[dynamo] Fix python_type for UserDefinedClassExceptionVariable (#166251)

Browse Source 
Pull Request resolved: https://github.com/pytorch/pytorch/pull/166251
Approved by: https://github.com/Lucaskabela
This commit is contained in:
Animesh Jain 2025-10-25 23:14:36 -07:00 committed by PyTorch MergeBot
parent 61bad3c1ea
commit 610c09f8f4
2 changed files with 8 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
test/dynamo/test_exceptions.py
View File

@ -889,20 +889,26 @@ class ExceptionTests(torch._dynamo.test_case.TestCase):
assert z == 1
def test_user_defined_exception_variable(self):
@torch.compile(backend="eager", fullgraph=True)
def fn(t):
z = 0
try:
raise CustomException
except ValueError:
z = 1
except CustomException:
except CustomException as e:
# trying to call python_type on the
# UserDefinedExceptionClassVariable
cls = type(e)
if type(cls) is type:
t = t + 1
z = 2
assert z == 2
return t.sin()
t = torch.randn(2)
fn(t)
opt_fn = torch.compile(fn, backend="eager", fullgraph=True)
self.assertEqual(fn(t), opt_fn(t))
def test_user_defined_exception_with_args(self):
@torch.compile(backend="eager", fullgraph=True)

3
torch/_dynamo/variables/user_defined.py
View File

@ -872,9 +872,6 @@ class UserDefinedExceptionClassVariable(UserDefinedClassVariable):
def fn(self):
return self.value
def python_type(self):
return self.value
class NO_SUCH_SUBOBJ:
pass