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