1#include <c10/core/GradMode.h>
2
3namespace c10 {
4
5bool GradMode::is_enabled() {
6 return AutogradState::get_tls_state().get_grad_mode();
7}
8
9void GradMode::set_enabled(bool enabled) {
10 AutogradState::get_tls_state().set_grad_mode(enabled);
11}
12} // namespace c10
13