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