c10/core/AutogradState.cpp

#include <c10/core/AutogradState.h>

namespace c10 {

namespace {
// By default, grad mode is enabled and inference mode is disabled
thread_local AutogradState autograd_state_tls = AutogradState(
    /* grad_mode */ true,
    /* inference_mode */ false,
    /* fw_grad_mode */ true);
} // namespace

AutogradState& AutogradState::get_tls_state() {
  return autograd_state_tls;
}

void AutogradState::set_tls_state(AutogradState state) {
  autograd_state_tls = state;
}

} // namespace c10