You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
42 lines
1.2 KiB
42 lines
1.2 KiB
#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)
|