Home
last modified time | relevance | path

Searched refs:currentThreadLocalKey_ (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/distributed/rpc/profiler/
H A Dremote_profiler_manager.cpp11 RemoteProfilerManager::currentThreadLocalKey_ = std::nullopt; member in torch::distributed::rpc::RemoteProfilerManager
20 if (RemoteProfilerManager::currentThreadLocalKey_) { in setCurrentKey()
25 currentThreadLocalKey_ = std::move(key); in setCurrentKey()
29 return currentThreadLocalKey_.has_value(); in isCurrentKeySet()
33 currentThreadLocalKey_ = std::nullopt; in unsetCurrentKey()
66 RemoteProfilerManager::currentThreadLocalKey_, in getCurrentProfilingKey()
68 return *currentThreadLocalKey_; in getCurrentProfilingKey()
H A Dremote_profiler_manager.h51 static thread_local std::optional<std::string> currentThreadLocalKey_; variable