1 | #include <torch/csrc/autograd/jit_decomp_interface.h> |
---|---|
2 | |
3 | namespace torch { |
4 | namespace autograd { |
5 | namespace impl { |
6 | |
7 | namespace { |
8 | JitDecompInterface* impl = nullptr; |
9 | } |
10 | |
11 | void setJitDecompImpl(JitDecompInterface* impl_) { |
12 | impl = impl_; |
13 | } |
14 | |
15 | JitDecompInterface* getJitDecompImpl() { |
16 | return impl; |
17 | } |
18 | |
19 | } // namespace impl |
20 | } // namespace autograd |
21 | } // namespace torch |
22 |