grad_mode.cpp

#include <torch/csrc/autograd/grad_mode.h>

namespace torch { namespace autograd {

thread_local bool GradMode_enabled = true;

bool GradMode::is_enabled() {
  return GradMode_enabled;
}

void GradMode::set_enabled(bool enabled) {
  GradMode_enabled = enabled;
}
}}