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