mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 12:21:27 +01:00
13c975684a
Summary:
`pthread_setname_np` requires Glibc 2.12. The patch reproduces what numactl does: 93867c59b0/syscall.c (L132-L136)
Related to issue https://github.com/pytorch/pytorch/issues/23482 and the `pthread_setname_np.patch` patch that adamjstewart shared.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/55063
Reviewed By: soulitzer
Differential Revision: D28577146
Pulled By: malfet
fbshipit-source-id: 85867b6f04795b1ae7bd46dbbc501cfd0ec9f163
30 lines
593 B
C++
30 lines
593 B
C++
#include <c10/util/thread_name.h>
|
|
|
|
#include <algorithm>
|
|
|
|
#ifndef __GLIBC_PREREQ
|
|
#define __GLIBC_PREREQ(x, y) 0
|
|
#endif
|
|
|
|
#if defined(__GLIBC__) && __GLIBC_PREREQ(2, 12) && !defined(__APPLE__) && \
|
|
!defined(__ANDROID__)
|
|
#define C10_HAS_PTHREAD_SETNAME_NP
|
|
#endif
|
|
|
|
#ifdef C10_HAS_PTHREAD_SETNAME_NP
|
|
#include <pthread.h>
|
|
#endif
|
|
|
|
namespace c10 {
|
|
|
|
void setThreadName(std::string name) {
|
|
#ifdef C10_HAS_PTHREAD_SETNAME_NP
|
|
constexpr size_t kMaxThreadName = 15;
|
|
name.resize(std::min(name.size(), kMaxThreadName));
|
|
|
|
pthread_setname_np(pthread_self(), name.c_str());
|
|
#endif
|
|
}
|
|
|
|
} // namespace c10
|