xref: /aosp_15_r20/external/pytorch/torch/csrc/autograd/jit_decomp_interface.cpp (revision da0073e96a02ea20f0ac840b70461e3646d07c45) 
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_)9 void setJitDecompImpl(JitDecompInterface* impl_) {
10   impl = impl_;
11 }
12 
getJitDecompImpl()13 JitDecompInterface* getJitDecompImpl() {
14   return impl;
15 }
16 
17 } // namespace torch::autograd::impl
18