pytorch
2c8dce91 - Show full call stack in TorchScript exception even when calls were inlined.

Commit
5 years ago
Show full call stack in TorchScript exception even when calls were inlined. Summary: This uses newly added InlinedCallStack to print the original call stack even if the real call stack is shallower because of inlining. This change also makes torchscript stacktraces look like python ones. Example: ``` torch.jit.script def baz(c, b): return c + b torch.jit.script def foo(c, b): return baz(c, b) torch.jit.script def bar(c, b): return foo(c, b) bar(torch.rand(10), torch.rand(9)) ``` Output before: ``` Traceback (most recent call last): File "fail.py", line 25, in <module> bar(torch.rand(10), torch.rand(9)) RuntimeError: The size of tensor a (10) must match the size of tensor b (9) at non-singleton dimension 0 The above operation failed in interpreter, with the following stack trace: at fail.py:15:11 torch.jit.script def baz(c, b): return c + b ~~~~~ <--- HERE ``` Output after: ``` Traceback (most recent call last): File "fail.py", line 41, in <module> bar(torch.rand(10), torch.rand(9)) RuntimeError: The size of tensor a (10) must match the size of tensor b (9) at non-singleton dimension 0 The above operation failed in interpreter. Traceback (most recent call last): File "fail.py", line 33 torch.jit.script def bar(c, b): return foo(c, b) ~~~ <--- HERE File "fail.py", line 29, in foo torch.jit.script def foo(c, b): return baz(c, b) ~~~ <--- HERE File "fail.py", line 25, in baz torch.jit.script def baz(c, b): return c + b ~~~~~ <--- HERE ``` Output of non-scripted python code: ``` Traceback (most recent call last): File "fail.py", line 36, in <module> bar(torch.rand(10), torch.rand(9)) File "fail.py", line 21, in bar return foo(c, b) File "fail.py", line 18, in foo return baz(c, b) File "fail.py", line 15, in baz return c + b RuntimeError: The size of tensor a (10) must match the size of tensor b (9) at non-singleton dimension 0 ``` Differential Revision: D18532812 Test Plan: Imported from OSS Pulled By: ZolotukhinM fbshipit-source-id: e7e5ba5e4a8f1c7086406271d0f1685d9db8541a
Author
Mikhail Zolotukhin
Parents
Loading