| 1234567891011121314151617181920212223242526272829303132333435363738394041 |
- #if !defined(TORCH_STABLE_ONLY) && !defined(TORCH_TARGET_VERSION)
- /*
- * Copyright (c) Meta Platforms, Inc. and affiliates.
- * All rights reserved.
- *
- * This source code is licensed under the BSD-style license found in the
- * LICENSE file in the root directory of this source tree.
- */
- #pragma once
- #include <sys/stat.h>
- #include <cstdint>
- #include <string>
- #include <utility>
- #include <vector>
- namespace libkineto {
- int32_t systemThreadId(bool cache = true);
- int32_t threadId();
- bool setThreadName(const std::string& name);
- std::string getThreadName();
- int32_t pidNamespace(ino_t& ns);
- int32_t processId(bool cache = true);
- std::string processName(int32_t pid);
- // Return a list of pids and process names for the current process
- // and its parents.
- std::vector<std::pair<int32_t, std::string>> pidCommandPairsOfAncestors();
- // Resets all cached Thread local state, this must be done on
- // forks to prevent stale values from being retained.
- void resetTLS();
- } // namespace libkineto
- #else
- #error "This file should not be included when either TORCH_STABLE_ONLY or TORCH_TARGET_VERSION is defined."
- #endif // !defined(TORCH_STABLE_ONLY) && !defined(TORCH_TARGET_VERSION)
|
