c10/core/GradMode.cpp

#include <c10/core/AutogradState.h>
#include <c10/core/GradMode.h>

#include <stdexcept>

namespace c10 {

bool GradMode::is_enabled() {
  return AutogradState::get_tls_state().get_grad_mode();
}

void GradMode::set_enabled(bool enabled) {
  AutogradState::get_tls_state().set_grad_mode(enabled);
}
} // namespace c10