mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-06 12:20:52 +01:00
911d527b87
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/70339 When a python program is translated to TorchScript, the python exception type is dropped. This makes users's life hard when they need to categorize errors based more than only exception message. Here we make the change so when we raise a python exception, we record the fully qualified class name for the exception. Later on when the TorchScript is interpreted, a special exception CustomJITException is thrown. User can get the python class name from CustomJITException::getPythonClassName . Note that, this diff does not customize the mapping from C++ exception to Python exception. It's left to the users to do whatever mapping they want. Code under scripts/shunting are just my own experimental code. I can split them out if requested. ghstack-source-id: 146221879 Test Plan: buck test mode/opt //caffe2/test:jit Reviewed By: gmagogsfm Differential Revision: D33282878 fbshipit-source-id: 910f67a764519f1053a48589d1a34df69001525d
9 lines
291 B
Python
9 lines
291 B
Python
r"""
|
|
Define exceptions used in test_exception.py. We define them in a
|
|
separate file on purpose to make sure the fully qualified exception class name
|
|
is captured correctly in suce cases.
|
|
"""
|
|
class MyKeyError(KeyError):
|
|
def __init__(self, msg):
|
|
super(KeyError, self).__init__(msg)
|