xref: /aosp_15_r20/external/pytorch/caffe2/utils/threadpool/thread_pool_guard.cpp (revision da0073e96a02ea20f0ac840b70461e3646d07c45) 
1 #include <caffe2/utils/threadpool/thread_pool_guard.h>
2 
3 namespace caffe2 {
4 
5 thread_local bool _NoPThreadPoolGuard_enabled = false;
6 
is_enabled()7 bool _NoPThreadPoolGuard::is_enabled() {
8   return _NoPThreadPoolGuard_enabled;
9 }
10 
set_enabled(bool enabled)11 void _NoPThreadPoolGuard::set_enabled(bool enabled) {
12   _NoPThreadPoolGuard_enabled = enabled;
13 }
14 
15 } // namespace at
16