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
|
#pragma once
#include <c10/core/Allocator.h>
#include <torch/csrc/jit/frontend/source_range.h>
#include <unordered_map>
#include <vector>
namespace c10 {
struct IValue;
}
namespace torch {
namespace jit {
class Pickler;
class SourceRangeSerializer;
class SourceRangeDeserializer;
class SourceRangePickler {
public:
SourceRangePickler();
std::vector<char> pickle(const SourceRangeRecords& ranges);
private:
std::shared_ptr<SourceRangeSerializer> srs;
};
class SourceRangeUnpickler {
public:
virtual c10::optional<SourceRange> findSourceRangeThatGenerated(
const SourceRange& range) = 0;
virtual ~SourceRangeUnpickler() {}
};
} // namespace jit
} // namespace torch
|
