1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
|
#pragma once
#include <c10/util/flat_hash_map.h>
#include <caffe2/serialize/inline_container.h>
#include <torch/csrc/jit/api/compilation_unit.h>
#include <torch/csrc/jit/ir/scope.h>
#include <torch/csrc/jit/serialization/source_range_serialization.h>
namespace torch::jit {
/*
* MobileDebugTable:
* Deserializes debug_pkl and callstack_map records from PT model's zip archive
* and stores them in a map of debug handles to DebugInfoPair. Debug handles are
* unique per model and runtime, be in lite interpreter or delegate, an
* exception of BackendRuntimeException should raised using debug handles.
* getSourceDebugString method is responsible for translating debug
* handles to correspond debug information.
* This debug informatin includes stack trace of model level source code and
* module hierarchy where the exception occurred.
*/
class MobileDebugTable {
public:
MobileDebugTable() = default;
MobileDebugTable(
std::unique_ptr<caffe2::serialize::PyTorchStreamReader>& reader,
const std::shared_ptr<CompilationUnit>& cu);
template <typename It>
MobileDebugTable(It begin, It end) : callstack_ptr_map_(begin, end) {}
std::string getSourceDebugString(
const int64_t debug_handle,
const std::string& top_module_type_name = "ModuleTypeUnknown") const;
std::string getSourceDebugString(
const std::vector<int64_t>& debug_handles,
const std::string& top_module_type_name = "ModuleTypeUnknown") const;
std::string getModuleHierarchyInfo(
const int64_t debug_handle,
const std::string& top_module_type_name = "ModuleTypeUnknown") const;
std::string getModuleHierarchyInfo(
const std::vector<int64_t>& debug_handles,
const std::string& top_module_type_name = "ModuleTypeUnknown") const;
const ska::flat_hash_map<int64_t, DebugInfoTuple>& getCallStackPtrMap()
const {
return callstack_ptr_map_;
}
private:
std::pair<std::string, std::string> getSourceDebugModuleHierarchyInfo(
const std::vector<int64_t>& debug_handles,
const std::string& top_module_type_name = "ModuleTypeUnknown") const;
ska::flat_hash_map<int64_t, DebugInfoTuple> callstack_ptr_map_;
};
} // namespace torch::jit
|
