pytorch/test/cpp
Kimish Patel 5326ec60e6 [Inlined Callstack Fix] Fix inlined callstack for blocks of the node. (#56562)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/56562

Earlier inlined callstack was annotated only nodes. This left out nodes
such as If which have block of nodes. These nodes should also be updated
similarly.

Test Plan:
Added test in test_misc

Imported from OSS

Reviewed By: ZolotukhinM

Differential Revision: D27902516

fbshipit-source-id: 4e65c686fa6b4977e8719db45f71f7d2599d4d8e
2021-05-04 09:21:15 -07:00
..
api s/InplaceOrView/ADInplaceOrView/g (#57372) 2021-05-01 22:56:18 -07:00
common
dist_autograd
jit [Inlined Callstack Fix] Fix inlined callstack for blocks of the node. (#56562) 2021-05-04 09:21:15 -07:00
lite_interpreter_runtime Revert D27958477: [PyTorch][Edge] Add v4 and v5 models and remove unused model 2021-04-23 14:42:01 -07:00
rpc Use Devices instead of DeviceIndexes in TensorPipe agent (#57294) 2021-05-01 16:12:55 -07:00
tensorexpr [TensorExpr] Add methods for inspecting generated code in TensorExprKernel. (#57074) 2021-05-03 20:02:28 -07:00
__init__.py