1#include <c10/core/AutogradState.h>
2
3namespace c10 {
4
5namespace {
6// By default, grad mode and mulithreading are enabled, inference mode is
7// disabled,
8thread_local AutogradState autograd_state_tls = AutogradState(
9 /* grad_mode */ true,
10 /* inference_mode */ false,
11 /* fw_grad_mode */ true,
12 /* multithreading_enabled */ true);
13} // namespace
14
15AutogradState& AutogradState::get_tls_state() {
16 return autograd_state_tls;
17}
18
19void AutogradState::set_tls_state(AutogradState state) {
20 autograd_state_tls = state;
21}
22
23} // namespace c10
24