frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

[Pytorch Delegated Backend] Save operator name and function name in (#57441)

Browse Source 
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/57441

debug info

Previous diffs did not save operator name in debug info. For delegated
backends that only idenfity op for profiling with debug handle, operator
name should be stores as well.
Furthermore to complete debug informaton also serialize function name.

Test Plan:
Existing lite interpreter and backend tests

Existing lite interpreter and backend tests

Imported from OSS

Differential Revision:
D28144581
D28144581

Reviewed By: raziel

Pulled By: kimishpatel

fbshipit-source-id: 415210f147530a53b444b07f1d6ee699a3570d99
This commit is contained in:
Kimish Patel
2021-05-25 13:16:46 -07:00
committed by Facebook GitHub Bot
parent a7a5992d7d
commit 813adf1076
13 changed files with 95 additions and 69 deletions
Download Patch File Download Diff File Expand all files Collapse all files

BIN
test/cpp/lite_interpreter_runtime/delegated_submodule_with_debug_info.ptl
View File

Binary file not shown.

2
test/cpp/lite_interpreter_runtime/test_lite_interpreter_runtime.cpp
View File

@ -142,7 +142,7 @@ TEST(RunTimeTest, DelegateException) {
inputs.emplace_back(torch::rand({13, 9}));
std::string error_pattern = R"(
Module hierarchy:top(C).A0(backend_with_compiler_demoLoweredModule).AA0(AA)
Module hierarchy:top(C).A0(backend_with_compiler_demoLoweredModule).AA0(AA).aten::add
Traceback of TorchScript (most recent call last):
File "<string>", line 3, in FunctionName_UNKNOWN