mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-21 05:34:18 +08:00
e0a5536cc9
Follows #133180 Pull Request resolved: https://github.com/pytorch/pytorch/pull/133295 Approved by: https://github.com/Skylion007
18 lines
307 B
C++
18 lines
307 B
C++
#include <torch/csrc/autograd/jit_decomp_interface.h>
|
|
|
|
namespace torch::autograd::impl {
|
|
|
|
namespace {
|
|
JitDecompInterface* impl = nullptr;
|
|
}
|
|
|
|
void setJitDecompImpl(JitDecompInterface* impl_) {
|
|
impl = impl_;
|
|
}
|
|
|
|
JitDecompInterface* getJitDecompImpl() {
|
|
return impl;
|
|
}
|
|
|
|
} // namespace torch::autograd::impl
|