mirror of
https://github.com/RPCS3/llvm-mirror.git
synced 2024-11-24 03:33:20 +01:00
e01aed9846
But now include a check for CPU_COUNT so we still build on 10 year old versions of glibc. Original message: Use sched_getaffinity instead of std:🧵:hardware_concurrency. The issue with std:🧵:hardware_concurrency is that it forwards to libc and some implementations (like glibc) don't take thread affinity into consideration. With this change a llvm program that can execute in only 2 cores will use 2 threads, even if the machine has 32 cores. This makes benchmarking a lot easier, but should also help if someone doesn't want to use all cores for compilation for example. llvm-svn: 314931
97 lines
2.9 KiB
C++
97 lines
2.9 KiB
C++
//===-- llvm/Support/Threading.cpp- Control multithreading mode --*- C++ -*-==//
|
|
//
|
|
// The LLVM Compiler Infrastructure
|
|
//
|
|
// This file is distributed under the University of Illinois Open Source
|
|
// License. See LICENSE.TXT for details.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
//
|
|
// This file defines helper functions for running LLVM in a multi-threaded
|
|
// environment.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "llvm/Support/Threading.h"
|
|
#include "llvm/Config/config.h"
|
|
#include "llvm/Support/Host.h"
|
|
|
|
#include <cassert>
|
|
#include <errno.h>
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
|
|
using namespace llvm;
|
|
|
|
//===----------------------------------------------------------------------===//
|
|
//=== WARNING: Implementation here must contain only TRULY operating system
|
|
//=== independent code.
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
bool llvm::llvm_is_multithreaded() {
|
|
#if LLVM_ENABLE_THREADS != 0
|
|
return true;
|
|
#else
|
|
return false;
|
|
#endif
|
|
}
|
|
|
|
#if LLVM_ENABLE_THREADS == 0 || \
|
|
(!defined(LLVM_ON_WIN32) && !defined(HAVE_PTHREAD_H))
|
|
// Support for non-Win32, non-pthread implementation.
|
|
void llvm::llvm_execute_on_thread(void (*Fn)(void *), void *UserData,
|
|
unsigned RequestedStackSize) {
|
|
(void)RequestedStackSize;
|
|
Fn(UserData);
|
|
}
|
|
|
|
unsigned llvm::heavyweight_hardware_concurrency() { return 1; }
|
|
|
|
unsigned llvm::hardware_concurrency() { return 1; }
|
|
|
|
uint64_t llvm::get_threadid() { return 0; }
|
|
|
|
uint32_t llvm::get_max_thread_name_length() { return 0; }
|
|
|
|
void llvm::set_thread_name(const Twine &Name) {}
|
|
|
|
void llvm::get_thread_name(SmallVectorImpl<char> &Name) { Name.clear(); }
|
|
|
|
#else
|
|
|
|
#include <thread>
|
|
unsigned llvm::heavyweight_hardware_concurrency() {
|
|
// Since we can't get here unless LLVM_ENABLE_THREADS == 1, it is safe to use
|
|
// `std::thread` directly instead of `llvm::thread` (and indeed, doing so
|
|
// allows us to not define `thread` in the llvm namespace, which conflicts
|
|
// with some platforms such as FreeBSD whose headers also define a struct
|
|
// called `thread` in the global namespace which can cause ambiguity due to
|
|
// ADL.
|
|
int NumPhysical = sys::getHostNumPhysicalCores();
|
|
if (NumPhysical == -1)
|
|
return std::thread::hardware_concurrency();
|
|
return NumPhysical;
|
|
}
|
|
|
|
unsigned llvm::hardware_concurrency() {
|
|
#if defined(HAVE_SCHED_GETAFFINITY) && defined(HAVE_CPU_COUNT)
|
|
cpu_set_t Set;
|
|
if (sched_getaffinity(0, sizeof(Set), &Set))
|
|
return CPU_COUNT(&Set);
|
|
#endif
|
|
// Guard against std::thread::hardware_concurrency() returning 0.
|
|
if (unsigned Val = std::thread::hardware_concurrency())
|
|
return Val;
|
|
return 1;
|
|
}
|
|
|
|
// Include the platform-specific parts of this class.
|
|
#ifdef LLVM_ON_UNIX
|
|
#include "Unix/Threading.inc"
|
|
#endif
|
|
#ifdef LLVM_ON_WIN32
|
|
#include "Windows/Threading.inc"
|
|
#endif
|
|
|
|
#endif
|