1a8c51b3fSopenharmony_ci// Copyright 2015 Google Inc. All rights reserved. 2a8c51b3fSopenharmony_ci// 3a8c51b3fSopenharmony_ci// Licensed under the Apache License, Version 2.0 (the "License"); 4a8c51b3fSopenharmony_ci// you may not use this file except in compliance with the License. 5a8c51b3fSopenharmony_ci// You may obtain a copy of the License at 6a8c51b3fSopenharmony_ci// 7a8c51b3fSopenharmony_ci// http://www.apache.org/licenses/LICENSE-2.0 8a8c51b3fSopenharmony_ci// 9a8c51b3fSopenharmony_ci// Unless required by applicable law or agreed to in writing, software 10a8c51b3fSopenharmony_ci// distributed under the License is distributed on an "AS IS" BASIS, 11a8c51b3fSopenharmony_ci// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 12a8c51b3fSopenharmony_ci// See the License for the specific language governing permissions and 13a8c51b3fSopenharmony_ci// limitations under the License. 14a8c51b3fSopenharmony_ci 15a8c51b3fSopenharmony_ci#include "internal_macros.h" 16a8c51b3fSopenharmony_ci 17a8c51b3fSopenharmony_ci#ifdef BENCHMARK_OS_WINDOWS 18a8c51b3fSopenharmony_ci#include <shlwapi.h> 19a8c51b3fSopenharmony_ci#undef StrCat // Don't let StrCat in string_util.h be renamed to lstrcatA 20a8c51b3fSopenharmony_ci#include <versionhelpers.h> 21a8c51b3fSopenharmony_ci#include <windows.h> 22a8c51b3fSopenharmony_ci 23a8c51b3fSopenharmony_ci#include <codecvt> 24a8c51b3fSopenharmony_ci#else 25a8c51b3fSopenharmony_ci#include <fcntl.h> 26a8c51b3fSopenharmony_ci#if !defined(BENCHMARK_OS_FUCHSIA) && !defined(BENCHMARK_OS_QURT) 27a8c51b3fSopenharmony_ci#include <sys/resource.h> 28a8c51b3fSopenharmony_ci#endif 29a8c51b3fSopenharmony_ci#include <sys/time.h> 30a8c51b3fSopenharmony_ci#include <sys/types.h> // this header must be included before 'sys/sysctl.h' to avoid compilation error on FreeBSD 31a8c51b3fSopenharmony_ci#include <unistd.h> 32a8c51b3fSopenharmony_ci#if defined BENCHMARK_OS_FREEBSD || defined BENCHMARK_OS_MACOSX || \ 33a8c51b3fSopenharmony_ci defined BENCHMARK_OS_NETBSD || defined BENCHMARK_OS_OPENBSD || \ 34a8c51b3fSopenharmony_ci defined BENCHMARK_OS_DRAGONFLY 35a8c51b3fSopenharmony_ci#define BENCHMARK_HAS_SYSCTL 36a8c51b3fSopenharmony_ci#include <sys/sysctl.h> 37a8c51b3fSopenharmony_ci#endif 38a8c51b3fSopenharmony_ci#endif 39a8c51b3fSopenharmony_ci#if defined(BENCHMARK_OS_SOLARIS) 40a8c51b3fSopenharmony_ci#include <kstat.h> 41a8c51b3fSopenharmony_ci#include <netdb.h> 42a8c51b3fSopenharmony_ci#endif 43a8c51b3fSopenharmony_ci#if defined(BENCHMARK_OS_QNX) 44a8c51b3fSopenharmony_ci#include <sys/syspage.h> 45a8c51b3fSopenharmony_ci#endif 46a8c51b3fSopenharmony_ci#if defined(BENCHMARK_OS_QURT) 47a8c51b3fSopenharmony_ci#include <qurt.h> 48a8c51b3fSopenharmony_ci#endif 49a8c51b3fSopenharmony_ci#if defined(BENCHMARK_HAS_PTHREAD_AFFINITY) 50a8c51b3fSopenharmony_ci#include <pthread.h> 51a8c51b3fSopenharmony_ci#endif 52a8c51b3fSopenharmony_ci 53a8c51b3fSopenharmony_ci#include <algorithm> 54a8c51b3fSopenharmony_ci#include <array> 55a8c51b3fSopenharmony_ci#include <bitset> 56a8c51b3fSopenharmony_ci#include <cerrno> 57a8c51b3fSopenharmony_ci#include <climits> 58a8c51b3fSopenharmony_ci#include <cstdint> 59a8c51b3fSopenharmony_ci#include <cstdio> 60a8c51b3fSopenharmony_ci#include <cstdlib> 61a8c51b3fSopenharmony_ci#include <cstring> 62a8c51b3fSopenharmony_ci#include <fstream> 63a8c51b3fSopenharmony_ci#include <iostream> 64a8c51b3fSopenharmony_ci#include <iterator> 65a8c51b3fSopenharmony_ci#include <limits> 66a8c51b3fSopenharmony_ci#include <locale> 67a8c51b3fSopenharmony_ci#include <memory> 68a8c51b3fSopenharmony_ci#include <random> 69a8c51b3fSopenharmony_ci#include <sstream> 70a8c51b3fSopenharmony_ci#include <utility> 71a8c51b3fSopenharmony_ci 72a8c51b3fSopenharmony_ci#include "benchmark/benchmark.h" 73a8c51b3fSopenharmony_ci#include "check.h" 74a8c51b3fSopenharmony_ci#include "cycleclock.h" 75a8c51b3fSopenharmony_ci#include "internal_macros.h" 76a8c51b3fSopenharmony_ci#include "log.h" 77a8c51b3fSopenharmony_ci#include "string_util.h" 78a8c51b3fSopenharmony_ci#include "timers.h" 79a8c51b3fSopenharmony_ci 80a8c51b3fSopenharmony_cinamespace benchmark { 81a8c51b3fSopenharmony_cinamespace { 82a8c51b3fSopenharmony_ci 83a8c51b3fSopenharmony_civoid PrintImp(std::ostream& out) { out << std::endl; } 84a8c51b3fSopenharmony_ci 85a8c51b3fSopenharmony_citemplate <class First, class... Rest> 86a8c51b3fSopenharmony_civoid PrintImp(std::ostream& out, First&& f, Rest&&... rest) { 87a8c51b3fSopenharmony_ci out << std::forward<First>(f); 88a8c51b3fSopenharmony_ci PrintImp(out, std::forward<Rest>(rest)...); 89a8c51b3fSopenharmony_ci} 90a8c51b3fSopenharmony_ci 91a8c51b3fSopenharmony_citemplate <class... Args> 92a8c51b3fSopenharmony_ciBENCHMARK_NORETURN void PrintErrorAndDie(Args&&... args) { 93a8c51b3fSopenharmony_ci PrintImp(std::cerr, std::forward<Args>(args)...); 94a8c51b3fSopenharmony_ci std::exit(EXIT_FAILURE); 95a8c51b3fSopenharmony_ci} 96a8c51b3fSopenharmony_ci 97a8c51b3fSopenharmony_ci#ifdef BENCHMARK_HAS_SYSCTL 98a8c51b3fSopenharmony_ci 99a8c51b3fSopenharmony_ci/// ValueUnion - A type used to correctly alias the byte-for-byte output of 100a8c51b3fSopenharmony_ci/// `sysctl` with the result type it's to be interpreted as. 101a8c51b3fSopenharmony_cistruct ValueUnion { 102a8c51b3fSopenharmony_ci union DataT { 103a8c51b3fSopenharmony_ci int32_t int32_value; 104a8c51b3fSopenharmony_ci int64_t int64_value; 105a8c51b3fSopenharmony_ci // For correct aliasing of union members from bytes. 106a8c51b3fSopenharmony_ci char bytes[8]; 107a8c51b3fSopenharmony_ci }; 108a8c51b3fSopenharmony_ci using DataPtr = std::unique_ptr<DataT, decltype(&std::free)>; 109a8c51b3fSopenharmony_ci 110a8c51b3fSopenharmony_ci // The size of the data union member + its trailing array size. 111a8c51b3fSopenharmony_ci std::size_t size; 112a8c51b3fSopenharmony_ci DataPtr buff; 113a8c51b3fSopenharmony_ci 114a8c51b3fSopenharmony_ci public: 115a8c51b3fSopenharmony_ci ValueUnion() : size(0), buff(nullptr, &std::free) {} 116a8c51b3fSopenharmony_ci 117a8c51b3fSopenharmony_ci explicit ValueUnion(std::size_t buff_size) 118a8c51b3fSopenharmony_ci : size(sizeof(DataT) + buff_size), 119a8c51b3fSopenharmony_ci buff(::new (std::malloc(size)) DataT(), &std::free) {} 120a8c51b3fSopenharmony_ci 121a8c51b3fSopenharmony_ci ValueUnion(ValueUnion&& other) = default; 122a8c51b3fSopenharmony_ci 123a8c51b3fSopenharmony_ci explicit operator bool() const { return bool(buff); } 124a8c51b3fSopenharmony_ci 125a8c51b3fSopenharmony_ci char* data() const { return buff->bytes; } 126a8c51b3fSopenharmony_ci 127a8c51b3fSopenharmony_ci std::string GetAsString() const { return std::string(data()); } 128a8c51b3fSopenharmony_ci 129a8c51b3fSopenharmony_ci int64_t GetAsInteger() const { 130a8c51b3fSopenharmony_ci if (size == sizeof(buff->int32_value)) 131a8c51b3fSopenharmony_ci return buff->int32_value; 132a8c51b3fSopenharmony_ci else if (size == sizeof(buff->int64_value)) 133a8c51b3fSopenharmony_ci return buff->int64_value; 134a8c51b3fSopenharmony_ci BENCHMARK_UNREACHABLE(); 135a8c51b3fSopenharmony_ci } 136a8c51b3fSopenharmony_ci 137a8c51b3fSopenharmony_ci template <class T, int N> 138a8c51b3fSopenharmony_ci std::array<T, N> GetAsArray() { 139a8c51b3fSopenharmony_ci const int arr_size = sizeof(T) * N; 140a8c51b3fSopenharmony_ci BM_CHECK_LE(arr_size, size); 141a8c51b3fSopenharmony_ci std::array<T, N> arr; 142a8c51b3fSopenharmony_ci std::memcpy(arr.data(), data(), arr_size); 143a8c51b3fSopenharmony_ci return arr; 144a8c51b3fSopenharmony_ci } 145a8c51b3fSopenharmony_ci}; 146a8c51b3fSopenharmony_ci 147a8c51b3fSopenharmony_ciValueUnion GetSysctlImp(std::string const& name) { 148a8c51b3fSopenharmony_ci#if defined BENCHMARK_OS_OPENBSD 149a8c51b3fSopenharmony_ci int mib[2]; 150a8c51b3fSopenharmony_ci 151a8c51b3fSopenharmony_ci mib[0] = CTL_HW; 152a8c51b3fSopenharmony_ci if ((name == "hw.ncpu") || (name == "hw.cpuspeed")) { 153a8c51b3fSopenharmony_ci ValueUnion buff(sizeof(int)); 154a8c51b3fSopenharmony_ci 155a8c51b3fSopenharmony_ci if (name == "hw.ncpu") { 156a8c51b3fSopenharmony_ci mib[1] = HW_NCPU; 157a8c51b3fSopenharmony_ci } else { 158a8c51b3fSopenharmony_ci mib[1] = HW_CPUSPEED; 159a8c51b3fSopenharmony_ci } 160a8c51b3fSopenharmony_ci 161a8c51b3fSopenharmony_ci if (sysctl(mib, 2, buff.data(), &buff.Size, nullptr, 0) == -1) { 162a8c51b3fSopenharmony_ci return ValueUnion(); 163a8c51b3fSopenharmony_ci } 164a8c51b3fSopenharmony_ci return buff; 165a8c51b3fSopenharmony_ci } 166a8c51b3fSopenharmony_ci return ValueUnion(); 167a8c51b3fSopenharmony_ci#else 168a8c51b3fSopenharmony_ci std::size_t cur_buff_size = 0; 169a8c51b3fSopenharmony_ci if (sysctlbyname(name.c_str(), nullptr, &cur_buff_size, nullptr, 0) == -1) 170a8c51b3fSopenharmony_ci return ValueUnion(); 171a8c51b3fSopenharmony_ci 172a8c51b3fSopenharmony_ci ValueUnion buff(cur_buff_size); 173a8c51b3fSopenharmony_ci if (sysctlbyname(name.c_str(), buff.data(), &buff.size, nullptr, 0) == 0) 174a8c51b3fSopenharmony_ci return buff; 175a8c51b3fSopenharmony_ci return ValueUnion(); 176a8c51b3fSopenharmony_ci#endif 177a8c51b3fSopenharmony_ci} 178a8c51b3fSopenharmony_ci 179a8c51b3fSopenharmony_ciBENCHMARK_MAYBE_UNUSED 180a8c51b3fSopenharmony_cibool GetSysctl(std::string const& name, std::string* out) { 181a8c51b3fSopenharmony_ci out->clear(); 182a8c51b3fSopenharmony_ci auto buff = GetSysctlImp(name); 183a8c51b3fSopenharmony_ci if (!buff) return false; 184a8c51b3fSopenharmony_ci out->assign(buff.data()); 185a8c51b3fSopenharmony_ci return true; 186a8c51b3fSopenharmony_ci} 187a8c51b3fSopenharmony_ci 188a8c51b3fSopenharmony_citemplate <class Tp, 189a8c51b3fSopenharmony_ci class = typename std::enable_if<std::is_integral<Tp>::value>::type> 190a8c51b3fSopenharmony_cibool GetSysctl(std::string const& name, Tp* out) { 191a8c51b3fSopenharmony_ci *out = 0; 192a8c51b3fSopenharmony_ci auto buff = GetSysctlImp(name); 193a8c51b3fSopenharmony_ci if (!buff) return false; 194a8c51b3fSopenharmony_ci *out = static_cast<Tp>(buff.GetAsInteger()); 195a8c51b3fSopenharmony_ci return true; 196a8c51b3fSopenharmony_ci} 197a8c51b3fSopenharmony_ci 198a8c51b3fSopenharmony_citemplate <class Tp, size_t N> 199a8c51b3fSopenharmony_cibool GetSysctl(std::string const& name, std::array<Tp, N>* out) { 200a8c51b3fSopenharmony_ci auto buff = GetSysctlImp(name); 201a8c51b3fSopenharmony_ci if (!buff) return false; 202a8c51b3fSopenharmony_ci *out = buff.GetAsArray<Tp, N>(); 203a8c51b3fSopenharmony_ci return true; 204a8c51b3fSopenharmony_ci} 205a8c51b3fSopenharmony_ci#endif 206a8c51b3fSopenharmony_ci 207a8c51b3fSopenharmony_citemplate <class ArgT> 208a8c51b3fSopenharmony_cibool ReadFromFile(std::string const& fname, ArgT* arg) { 209a8c51b3fSopenharmony_ci *arg = ArgT(); 210a8c51b3fSopenharmony_ci std::ifstream f(fname.c_str()); 211a8c51b3fSopenharmony_ci if (!f.is_open()) return false; 212a8c51b3fSopenharmony_ci f >> *arg; 213a8c51b3fSopenharmony_ci return f.good(); 214a8c51b3fSopenharmony_ci} 215a8c51b3fSopenharmony_ci 216a8c51b3fSopenharmony_ciCPUInfo::Scaling CpuScaling(int num_cpus) { 217a8c51b3fSopenharmony_ci // We don't have a valid CPU count, so don't even bother. 218a8c51b3fSopenharmony_ci if (num_cpus <= 0) return CPUInfo::Scaling::UNKNOWN; 219a8c51b3fSopenharmony_ci#if defined(BENCHMARK_OS_QNX) 220a8c51b3fSopenharmony_ci return CPUInfo::Scaling::UNKNOWN; 221a8c51b3fSopenharmony_ci#elif !defined(BENCHMARK_OS_WINDOWS) 222a8c51b3fSopenharmony_ci // On Linux, the CPUfreq subsystem exposes CPU information as files on the 223a8c51b3fSopenharmony_ci // local file system. If reading the exported files fails, then we may not be 224a8c51b3fSopenharmony_ci // running on Linux, so we silently ignore all the read errors. 225a8c51b3fSopenharmony_ci std::string res; 226a8c51b3fSopenharmony_ci for (int cpu = 0; cpu < num_cpus; ++cpu) { 227a8c51b3fSopenharmony_ci std::string governor_file = 228a8c51b3fSopenharmony_ci StrCat("/sys/devices/system/cpu/cpu", cpu, "/cpufreq/scaling_governor"); 229a8c51b3fSopenharmony_ci if (ReadFromFile(governor_file, &res) && res != "performance") 230a8c51b3fSopenharmony_ci return CPUInfo::Scaling::ENABLED; 231a8c51b3fSopenharmony_ci } 232a8c51b3fSopenharmony_ci return CPUInfo::Scaling::DISABLED; 233a8c51b3fSopenharmony_ci#else 234a8c51b3fSopenharmony_ci return CPUInfo::Scaling::UNKNOWN; 235a8c51b3fSopenharmony_ci#endif 236a8c51b3fSopenharmony_ci} 237a8c51b3fSopenharmony_ci 238a8c51b3fSopenharmony_ciint CountSetBitsInCPUMap(std::string val) { 239a8c51b3fSopenharmony_ci auto CountBits = [](std::string part) { 240a8c51b3fSopenharmony_ci using CPUMask = std::bitset<sizeof(std::uintptr_t) * CHAR_BIT>; 241a8c51b3fSopenharmony_ci part = "0x" + part; 242a8c51b3fSopenharmony_ci CPUMask mask(benchmark::stoul(part, nullptr, 16)); 243a8c51b3fSopenharmony_ci return static_cast<int>(mask.count()); 244a8c51b3fSopenharmony_ci }; 245a8c51b3fSopenharmony_ci std::size_t pos; 246a8c51b3fSopenharmony_ci int total = 0; 247a8c51b3fSopenharmony_ci while ((pos = val.find(',')) != std::string::npos) { 248a8c51b3fSopenharmony_ci total += CountBits(val.substr(0, pos)); 249a8c51b3fSopenharmony_ci val = val.substr(pos + 1); 250a8c51b3fSopenharmony_ci } 251a8c51b3fSopenharmony_ci if (!val.empty()) { 252a8c51b3fSopenharmony_ci total += CountBits(val); 253a8c51b3fSopenharmony_ci } 254a8c51b3fSopenharmony_ci return total; 255a8c51b3fSopenharmony_ci} 256a8c51b3fSopenharmony_ci 257a8c51b3fSopenharmony_ciBENCHMARK_MAYBE_UNUSED 258a8c51b3fSopenharmony_cistd::vector<CPUInfo::CacheInfo> GetCacheSizesFromKVFS() { 259a8c51b3fSopenharmony_ci std::vector<CPUInfo::CacheInfo> res; 260a8c51b3fSopenharmony_ci std::string dir = "/sys/devices/system/cpu/cpu0/cache/"; 261a8c51b3fSopenharmony_ci int idx = 0; 262a8c51b3fSopenharmony_ci while (true) { 263a8c51b3fSopenharmony_ci CPUInfo::CacheInfo info; 264a8c51b3fSopenharmony_ci std::string fpath = StrCat(dir, "index", idx++, "/"); 265a8c51b3fSopenharmony_ci std::ifstream f(StrCat(fpath, "size").c_str()); 266a8c51b3fSopenharmony_ci if (!f.is_open()) break; 267a8c51b3fSopenharmony_ci std::string suffix; 268a8c51b3fSopenharmony_ci f >> info.size; 269a8c51b3fSopenharmony_ci if (f.fail()) 270a8c51b3fSopenharmony_ci PrintErrorAndDie("Failed while reading file '", fpath, "size'"); 271a8c51b3fSopenharmony_ci if (f.good()) { 272a8c51b3fSopenharmony_ci f >> suffix; 273a8c51b3fSopenharmony_ci if (f.bad()) 274a8c51b3fSopenharmony_ci PrintErrorAndDie( 275a8c51b3fSopenharmony_ci "Invalid cache size format: failed to read size suffix"); 276a8c51b3fSopenharmony_ci else if (f && suffix != "K") 277a8c51b3fSopenharmony_ci PrintErrorAndDie("Invalid cache size format: Expected bytes ", suffix); 278a8c51b3fSopenharmony_ci else if (suffix == "K") 279a8c51b3fSopenharmony_ci info.size *= 1024; 280a8c51b3fSopenharmony_ci } 281a8c51b3fSopenharmony_ci if (!ReadFromFile(StrCat(fpath, "type"), &info.type)) 282a8c51b3fSopenharmony_ci PrintErrorAndDie("Failed to read from file ", fpath, "type"); 283a8c51b3fSopenharmony_ci if (!ReadFromFile(StrCat(fpath, "level"), &info.level)) 284a8c51b3fSopenharmony_ci PrintErrorAndDie("Failed to read from file ", fpath, "level"); 285a8c51b3fSopenharmony_ci std::string map_str; 286a8c51b3fSopenharmony_ci if (!ReadFromFile(StrCat(fpath, "shared_cpu_map"), &map_str)) 287a8c51b3fSopenharmony_ci PrintErrorAndDie("Failed to read from file ", fpath, "shared_cpu_map"); 288a8c51b3fSopenharmony_ci info.num_sharing = CountSetBitsInCPUMap(map_str); 289a8c51b3fSopenharmony_ci res.push_back(info); 290a8c51b3fSopenharmony_ci } 291a8c51b3fSopenharmony_ci 292a8c51b3fSopenharmony_ci return res; 293a8c51b3fSopenharmony_ci} 294a8c51b3fSopenharmony_ci 295a8c51b3fSopenharmony_ci#ifdef BENCHMARK_OS_MACOSX 296a8c51b3fSopenharmony_cistd::vector<CPUInfo::CacheInfo> GetCacheSizesMacOSX() { 297a8c51b3fSopenharmony_ci std::vector<CPUInfo::CacheInfo> res; 298a8c51b3fSopenharmony_ci std::array<int, 4> cache_counts{{0, 0, 0, 0}}; 299a8c51b3fSopenharmony_ci GetSysctl("hw.cacheconfig", &cache_counts); 300a8c51b3fSopenharmony_ci 301a8c51b3fSopenharmony_ci struct { 302a8c51b3fSopenharmony_ci std::string name; 303a8c51b3fSopenharmony_ci std::string type; 304a8c51b3fSopenharmony_ci int level; 305a8c51b3fSopenharmony_ci int num_sharing; 306a8c51b3fSopenharmony_ci } cases[] = {{"hw.l1dcachesize", "Data", 1, cache_counts[1]}, 307a8c51b3fSopenharmony_ci {"hw.l1icachesize", "Instruction", 1, cache_counts[1]}, 308a8c51b3fSopenharmony_ci {"hw.l2cachesize", "Unified", 2, cache_counts[2]}, 309a8c51b3fSopenharmony_ci {"hw.l3cachesize", "Unified", 3, cache_counts[3]}}; 310a8c51b3fSopenharmony_ci for (auto& c : cases) { 311a8c51b3fSopenharmony_ci int val; 312a8c51b3fSopenharmony_ci if (!GetSysctl(c.name, &val)) continue; 313a8c51b3fSopenharmony_ci CPUInfo::CacheInfo info; 314a8c51b3fSopenharmony_ci info.type = c.type; 315a8c51b3fSopenharmony_ci info.level = c.level; 316a8c51b3fSopenharmony_ci info.size = val; 317a8c51b3fSopenharmony_ci info.num_sharing = c.num_sharing; 318a8c51b3fSopenharmony_ci res.push_back(std::move(info)); 319a8c51b3fSopenharmony_ci } 320a8c51b3fSopenharmony_ci return res; 321a8c51b3fSopenharmony_ci} 322a8c51b3fSopenharmony_ci#elif defined(BENCHMARK_OS_WINDOWS) 323a8c51b3fSopenharmony_cistd::vector<CPUInfo::CacheInfo> GetCacheSizesWindows() { 324a8c51b3fSopenharmony_ci std::vector<CPUInfo::CacheInfo> res; 325a8c51b3fSopenharmony_ci DWORD buffer_size = 0; 326a8c51b3fSopenharmony_ci using PInfo = SYSTEM_LOGICAL_PROCESSOR_INFORMATION; 327a8c51b3fSopenharmony_ci using CInfo = CACHE_DESCRIPTOR; 328a8c51b3fSopenharmony_ci 329a8c51b3fSopenharmony_ci using UPtr = std::unique_ptr<PInfo, decltype(&std::free)>; 330a8c51b3fSopenharmony_ci GetLogicalProcessorInformation(nullptr, &buffer_size); 331a8c51b3fSopenharmony_ci UPtr buff(static_cast<PInfo*>(std::malloc(buffer_size)), &std::free); 332a8c51b3fSopenharmony_ci if (!GetLogicalProcessorInformation(buff.get(), &buffer_size)) 333a8c51b3fSopenharmony_ci PrintErrorAndDie("Failed during call to GetLogicalProcessorInformation: ", 334a8c51b3fSopenharmony_ci GetLastError()); 335a8c51b3fSopenharmony_ci 336a8c51b3fSopenharmony_ci PInfo* it = buff.get(); 337a8c51b3fSopenharmony_ci PInfo* end = buff.get() + (buffer_size / sizeof(PInfo)); 338a8c51b3fSopenharmony_ci 339a8c51b3fSopenharmony_ci for (; it != end; ++it) { 340a8c51b3fSopenharmony_ci if (it->Relationship != RelationCache) continue; 341a8c51b3fSopenharmony_ci using BitSet = std::bitset<sizeof(ULONG_PTR) * CHAR_BIT>; 342a8c51b3fSopenharmony_ci BitSet b(it->ProcessorMask); 343a8c51b3fSopenharmony_ci // To prevent duplicates, only consider caches where CPU 0 is specified 344a8c51b3fSopenharmony_ci if (!b.test(0)) continue; 345a8c51b3fSopenharmony_ci const CInfo& cache = it->Cache; 346a8c51b3fSopenharmony_ci CPUInfo::CacheInfo C; 347a8c51b3fSopenharmony_ci C.num_sharing = static_cast<int>(b.count()); 348a8c51b3fSopenharmony_ci C.level = cache.Level; 349a8c51b3fSopenharmony_ci C.size = cache.Size; 350a8c51b3fSopenharmony_ci C.type = "Unknown"; 351a8c51b3fSopenharmony_ci switch (cache.Type) { 352a8c51b3fSopenharmony_ci case CacheUnified: 353a8c51b3fSopenharmony_ci C.type = "Unified"; 354a8c51b3fSopenharmony_ci break; 355a8c51b3fSopenharmony_ci case CacheInstruction: 356a8c51b3fSopenharmony_ci C.type = "Instruction"; 357a8c51b3fSopenharmony_ci break; 358a8c51b3fSopenharmony_ci case CacheData: 359a8c51b3fSopenharmony_ci C.type = "Data"; 360a8c51b3fSopenharmony_ci break; 361a8c51b3fSopenharmony_ci case CacheTrace: 362a8c51b3fSopenharmony_ci C.type = "Trace"; 363a8c51b3fSopenharmony_ci break; 364a8c51b3fSopenharmony_ci } 365a8c51b3fSopenharmony_ci res.push_back(C); 366a8c51b3fSopenharmony_ci } 367a8c51b3fSopenharmony_ci return res; 368a8c51b3fSopenharmony_ci} 369a8c51b3fSopenharmony_ci#elif BENCHMARK_OS_QNX 370a8c51b3fSopenharmony_cistd::vector<CPUInfo::CacheInfo> GetCacheSizesQNX() { 371a8c51b3fSopenharmony_ci std::vector<CPUInfo::CacheInfo> res; 372a8c51b3fSopenharmony_ci struct cacheattr_entry* cache = SYSPAGE_ENTRY(cacheattr); 373a8c51b3fSopenharmony_ci uint32_t const elsize = SYSPAGE_ELEMENT_SIZE(cacheattr); 374a8c51b3fSopenharmony_ci int num = SYSPAGE_ENTRY_SIZE(cacheattr) / elsize; 375a8c51b3fSopenharmony_ci for (int i = 0; i < num; ++i) { 376a8c51b3fSopenharmony_ci CPUInfo::CacheInfo info; 377a8c51b3fSopenharmony_ci switch (cache->flags) { 378a8c51b3fSopenharmony_ci case CACHE_FLAG_INSTR: 379a8c51b3fSopenharmony_ci info.type = "Instruction"; 380a8c51b3fSopenharmony_ci info.level = 1; 381a8c51b3fSopenharmony_ci break; 382a8c51b3fSopenharmony_ci case CACHE_FLAG_DATA: 383a8c51b3fSopenharmony_ci info.type = "Data"; 384a8c51b3fSopenharmony_ci info.level = 1; 385a8c51b3fSopenharmony_ci break; 386a8c51b3fSopenharmony_ci case CACHE_FLAG_UNIFIED: 387a8c51b3fSopenharmony_ci info.type = "Unified"; 388a8c51b3fSopenharmony_ci info.level = 2; 389a8c51b3fSopenharmony_ci break; 390a8c51b3fSopenharmony_ci case CACHE_FLAG_SHARED: 391a8c51b3fSopenharmony_ci info.type = "Shared"; 392a8c51b3fSopenharmony_ci info.level = 3; 393a8c51b3fSopenharmony_ci break; 394a8c51b3fSopenharmony_ci default: 395a8c51b3fSopenharmony_ci continue; 396a8c51b3fSopenharmony_ci break; 397a8c51b3fSopenharmony_ci } 398a8c51b3fSopenharmony_ci info.size = cache->line_size * cache->num_lines; 399a8c51b3fSopenharmony_ci info.num_sharing = 0; 400a8c51b3fSopenharmony_ci res.push_back(std::move(info)); 401a8c51b3fSopenharmony_ci cache = SYSPAGE_ARRAY_ADJ_OFFSET(cacheattr, cache, elsize); 402a8c51b3fSopenharmony_ci } 403a8c51b3fSopenharmony_ci return res; 404a8c51b3fSopenharmony_ci} 405a8c51b3fSopenharmony_ci#endif 406a8c51b3fSopenharmony_ci 407a8c51b3fSopenharmony_cistd::vector<CPUInfo::CacheInfo> GetCacheSizes() { 408a8c51b3fSopenharmony_ci#ifdef BENCHMARK_OS_MACOSX 409a8c51b3fSopenharmony_ci return GetCacheSizesMacOSX(); 410a8c51b3fSopenharmony_ci#elif defined(BENCHMARK_OS_WINDOWS) 411a8c51b3fSopenharmony_ci return GetCacheSizesWindows(); 412a8c51b3fSopenharmony_ci#elif defined(BENCHMARK_OS_QNX) 413a8c51b3fSopenharmony_ci return GetCacheSizesQNX(); 414a8c51b3fSopenharmony_ci#elif defined(BENCHMARK_OS_QURT) 415a8c51b3fSopenharmony_ci return std::vector<CPUInfo::CacheInfo>(); 416a8c51b3fSopenharmony_ci#else 417a8c51b3fSopenharmony_ci return GetCacheSizesFromKVFS(); 418a8c51b3fSopenharmony_ci#endif 419a8c51b3fSopenharmony_ci} 420a8c51b3fSopenharmony_ci 421a8c51b3fSopenharmony_cistd::string GetSystemName() { 422a8c51b3fSopenharmony_ci#if defined(BENCHMARK_OS_WINDOWS) 423a8c51b3fSopenharmony_ci std::string str; 424a8c51b3fSopenharmony_ci static constexpr int COUNT = MAX_COMPUTERNAME_LENGTH + 1; 425a8c51b3fSopenharmony_ci TCHAR hostname[COUNT] = {'\0'}; 426a8c51b3fSopenharmony_ci DWORD DWCOUNT = COUNT; 427a8c51b3fSopenharmony_ci if (!GetComputerName(hostname, &DWCOUNT)) return std::string(""); 428a8c51b3fSopenharmony_ci#ifndef UNICODE 429a8c51b3fSopenharmony_ci str = std::string(hostname, DWCOUNT); 430a8c51b3fSopenharmony_ci#else 431a8c51b3fSopenharmony_ci // `WideCharToMultiByte` returns `0` when conversion fails. 432a8c51b3fSopenharmony_ci int len = WideCharToMultiByte(CP_UTF8, WC_ERR_INVALID_CHARS, hostname, 433a8c51b3fSopenharmony_ci DWCOUNT, NULL, 0, NULL, NULL); 434a8c51b3fSopenharmony_ci str.resize(len); 435a8c51b3fSopenharmony_ci WideCharToMultiByte(CP_UTF8, WC_ERR_INVALID_CHARS, hostname, DWCOUNT, &str[0], 436a8c51b3fSopenharmony_ci str.size(), NULL, NULL); 437a8c51b3fSopenharmony_ci#endif 438a8c51b3fSopenharmony_ci return str; 439a8c51b3fSopenharmony_ci#elif defined(BENCHMARK_OS_QURT) 440a8c51b3fSopenharmony_ci std::string str = "Hexagon DSP"; 441a8c51b3fSopenharmony_ci qurt_arch_version_t arch_version_struct; 442a8c51b3fSopenharmony_ci if (qurt_sysenv_get_arch_version(&arch_version_struct) == QURT_EOK) { 443a8c51b3fSopenharmony_ci str += " v"; 444a8c51b3fSopenharmony_ci str += std::to_string(arch_version_struct.arch_version); 445a8c51b3fSopenharmony_ci } 446a8c51b3fSopenharmony_ci return str; 447a8c51b3fSopenharmony_ci#else 448a8c51b3fSopenharmony_ci#ifndef HOST_NAME_MAX 449a8c51b3fSopenharmony_ci#ifdef BENCHMARK_HAS_SYSCTL // BSD/Mac doesn't have HOST_NAME_MAX defined 450a8c51b3fSopenharmony_ci#define HOST_NAME_MAX 64 451a8c51b3fSopenharmony_ci#elif defined(BENCHMARK_OS_NACL) 452a8c51b3fSopenharmony_ci#define HOST_NAME_MAX 64 453a8c51b3fSopenharmony_ci#elif defined(BENCHMARK_OS_QNX) 454a8c51b3fSopenharmony_ci#define HOST_NAME_MAX 154 455a8c51b3fSopenharmony_ci#elif defined(BENCHMARK_OS_RTEMS) 456a8c51b3fSopenharmony_ci#define HOST_NAME_MAX 256 457a8c51b3fSopenharmony_ci#elif defined(BENCHMARK_OS_SOLARIS) 458a8c51b3fSopenharmony_ci#define HOST_NAME_MAX MAXHOSTNAMELEN 459a8c51b3fSopenharmony_ci#else 460a8c51b3fSopenharmony_ci#pragma message("HOST_NAME_MAX not defined. using 64") 461a8c51b3fSopenharmony_ci#define HOST_NAME_MAX 64 462a8c51b3fSopenharmony_ci#endif 463a8c51b3fSopenharmony_ci#endif // def HOST_NAME_MAX 464a8c51b3fSopenharmony_ci char hostname[HOST_NAME_MAX]; 465a8c51b3fSopenharmony_ci int retVal = gethostname(hostname, HOST_NAME_MAX); 466a8c51b3fSopenharmony_ci if (retVal != 0) return std::string(""); 467a8c51b3fSopenharmony_ci return std::string(hostname); 468a8c51b3fSopenharmony_ci#endif // Catch-all POSIX block. 469a8c51b3fSopenharmony_ci} 470a8c51b3fSopenharmony_ci 471a8c51b3fSopenharmony_ciint GetNumCPUs() { 472a8c51b3fSopenharmony_ci#ifdef BENCHMARK_HAS_SYSCTL 473a8c51b3fSopenharmony_ci int num_cpu = -1; 474a8c51b3fSopenharmony_ci if (GetSysctl("hw.ncpu", &num_cpu)) return num_cpu; 475a8c51b3fSopenharmony_ci fprintf(stderr, "Err: %s\n", strerror(errno)); 476a8c51b3fSopenharmony_ci std::exit(EXIT_FAILURE); 477a8c51b3fSopenharmony_ci#elif defined(BENCHMARK_OS_WINDOWS) 478a8c51b3fSopenharmony_ci SYSTEM_INFO sysinfo; 479a8c51b3fSopenharmony_ci // Use memset as opposed to = {} to avoid GCC missing initializer false 480a8c51b3fSopenharmony_ci // positives. 481a8c51b3fSopenharmony_ci std::memset(&sysinfo, 0, sizeof(SYSTEM_INFO)); 482a8c51b3fSopenharmony_ci GetSystemInfo(&sysinfo); 483a8c51b3fSopenharmony_ci return sysinfo.dwNumberOfProcessors; // number of logical 484a8c51b3fSopenharmony_ci // processors in the current 485a8c51b3fSopenharmony_ci // group 486a8c51b3fSopenharmony_ci#elif defined(BENCHMARK_OS_SOLARIS) 487a8c51b3fSopenharmony_ci // Returns -1 in case of a failure. 488a8c51b3fSopenharmony_ci long num_cpu = sysconf(_SC_NPROCESSORS_ONLN); 489a8c51b3fSopenharmony_ci if (num_cpu < 0) { 490a8c51b3fSopenharmony_ci fprintf(stderr, "sysconf(_SC_NPROCESSORS_ONLN) failed with error: %s\n", 491a8c51b3fSopenharmony_ci strerror(errno)); 492a8c51b3fSopenharmony_ci } 493a8c51b3fSopenharmony_ci return (int)num_cpu; 494a8c51b3fSopenharmony_ci#elif defined(BENCHMARK_OS_QNX) 495a8c51b3fSopenharmony_ci return static_cast<int>(_syspage_ptr->num_cpu); 496a8c51b3fSopenharmony_ci#elif defined(BENCHMARK_OS_QURT) 497a8c51b3fSopenharmony_ci qurt_sysenv_max_hthreads_t hardware_threads; 498a8c51b3fSopenharmony_ci if (qurt_sysenv_get_max_hw_threads(&hardware_threads) != QURT_EOK) { 499a8c51b3fSopenharmony_ci hardware_threads.max_hthreads = 1; 500a8c51b3fSopenharmony_ci } 501a8c51b3fSopenharmony_ci return hardware_threads.max_hthreads; 502a8c51b3fSopenharmony_ci#else 503a8c51b3fSopenharmony_ci int num_cpus = 0; 504a8c51b3fSopenharmony_ci int max_id = -1; 505a8c51b3fSopenharmony_ci std::ifstream f("/proc/cpuinfo"); 506a8c51b3fSopenharmony_ci if (!f.is_open()) { 507a8c51b3fSopenharmony_ci std::cerr << "failed to open /proc/cpuinfo\n"; 508a8c51b3fSopenharmony_ci return -1; 509a8c51b3fSopenharmony_ci } 510a8c51b3fSopenharmony_ci const std::string Key = "processor"; 511a8c51b3fSopenharmony_ci std::string ln; 512a8c51b3fSopenharmony_ci while (std::getline(f, ln)) { 513a8c51b3fSopenharmony_ci if (ln.empty()) continue; 514a8c51b3fSopenharmony_ci std::size_t split_idx = ln.find(':'); 515a8c51b3fSopenharmony_ci std::string value; 516a8c51b3fSopenharmony_ci#if defined(__s390__) 517a8c51b3fSopenharmony_ci // s390 has another format in /proc/cpuinfo 518a8c51b3fSopenharmony_ci // it needs to be parsed differently 519a8c51b3fSopenharmony_ci if (split_idx != std::string::npos) 520a8c51b3fSopenharmony_ci value = ln.substr(Key.size() + 1, split_idx - Key.size() - 1); 521a8c51b3fSopenharmony_ci#else 522a8c51b3fSopenharmony_ci if (split_idx != std::string::npos) value = ln.substr(split_idx + 1); 523a8c51b3fSopenharmony_ci#endif 524a8c51b3fSopenharmony_ci if (ln.size() >= Key.size() && ln.compare(0, Key.size(), Key) == 0) { 525a8c51b3fSopenharmony_ci num_cpus++; 526a8c51b3fSopenharmony_ci if (!value.empty()) { 527a8c51b3fSopenharmony_ci const int cur_id = benchmark::stoi(value); 528a8c51b3fSopenharmony_ci max_id = std::max(cur_id, max_id); 529a8c51b3fSopenharmony_ci } 530a8c51b3fSopenharmony_ci } 531a8c51b3fSopenharmony_ci } 532a8c51b3fSopenharmony_ci if (f.bad()) { 533a8c51b3fSopenharmony_ci std::cerr << "Failure reading /proc/cpuinfo\n"; 534a8c51b3fSopenharmony_ci return -1; 535a8c51b3fSopenharmony_ci } 536a8c51b3fSopenharmony_ci if (!f.eof()) { 537a8c51b3fSopenharmony_ci std::cerr << "Failed to read to end of /proc/cpuinfo\n"; 538a8c51b3fSopenharmony_ci return -1; 539a8c51b3fSopenharmony_ci } 540a8c51b3fSopenharmony_ci f.close(); 541a8c51b3fSopenharmony_ci 542a8c51b3fSopenharmony_ci if ((max_id + 1) != num_cpus) { 543a8c51b3fSopenharmony_ci fprintf(stderr, 544a8c51b3fSopenharmony_ci "CPU ID assignments in /proc/cpuinfo seem messed up." 545a8c51b3fSopenharmony_ci " This is usually caused by a bad BIOS.\n"); 546a8c51b3fSopenharmony_ci } 547a8c51b3fSopenharmony_ci return num_cpus; 548a8c51b3fSopenharmony_ci#endif 549a8c51b3fSopenharmony_ci BENCHMARK_UNREACHABLE(); 550a8c51b3fSopenharmony_ci} 551a8c51b3fSopenharmony_ci 552a8c51b3fSopenharmony_ciclass ThreadAffinityGuard final { 553a8c51b3fSopenharmony_ci public: 554a8c51b3fSopenharmony_ci ThreadAffinityGuard() : reset_affinity(SetAffinity()) { 555a8c51b3fSopenharmony_ci if (!reset_affinity) 556a8c51b3fSopenharmony_ci std::cerr << "***WARNING*** Failed to set thread affinity. Estimated CPU " 557a8c51b3fSopenharmony_ci "frequency may be incorrect." 558a8c51b3fSopenharmony_ci << std::endl; 559a8c51b3fSopenharmony_ci } 560a8c51b3fSopenharmony_ci 561a8c51b3fSopenharmony_ci ~ThreadAffinityGuard() { 562a8c51b3fSopenharmony_ci if (!reset_affinity) return; 563a8c51b3fSopenharmony_ci 564a8c51b3fSopenharmony_ci#if defined(BENCHMARK_HAS_PTHREAD_AFFINITY) 565a8c51b3fSopenharmony_ci int ret = pthread_setaffinity_np(self, sizeof(previous_affinity), 566a8c51b3fSopenharmony_ci &previous_affinity); 567a8c51b3fSopenharmony_ci if (ret == 0) return; 568a8c51b3fSopenharmony_ci#elif defined(BENCHMARK_OS_WINDOWS_WIN32) 569a8c51b3fSopenharmony_ci DWORD_PTR ret = SetThreadAffinityMask(self, previous_affinity); 570a8c51b3fSopenharmony_ci if (ret != 0) return; 571a8c51b3fSopenharmony_ci#endif // def BENCHMARK_HAS_PTHREAD_AFFINITY 572a8c51b3fSopenharmony_ci PrintErrorAndDie("Failed to reset thread affinity"); 573a8c51b3fSopenharmony_ci } 574a8c51b3fSopenharmony_ci 575a8c51b3fSopenharmony_ci ThreadAffinityGuard(ThreadAffinityGuard&&) = delete; 576a8c51b3fSopenharmony_ci ThreadAffinityGuard(const ThreadAffinityGuard&) = delete; 577a8c51b3fSopenharmony_ci ThreadAffinityGuard& operator=(ThreadAffinityGuard&&) = delete; 578a8c51b3fSopenharmony_ci ThreadAffinityGuard& operator=(const ThreadAffinityGuard&) = delete; 579a8c51b3fSopenharmony_ci 580a8c51b3fSopenharmony_ci private: 581a8c51b3fSopenharmony_ci bool SetAffinity() { 582a8c51b3fSopenharmony_ci#if defined(BENCHMARK_HAS_PTHREAD_AFFINITY) 583a8c51b3fSopenharmony_ci int ret; 584a8c51b3fSopenharmony_ci self = pthread_self(); 585a8c51b3fSopenharmony_ci ret = pthread_getaffinity_np(self, sizeof(previous_affinity), 586a8c51b3fSopenharmony_ci &previous_affinity); 587a8c51b3fSopenharmony_ci if (ret != 0) return false; 588a8c51b3fSopenharmony_ci 589a8c51b3fSopenharmony_ci cpu_set_t affinity; 590a8c51b3fSopenharmony_ci memcpy(&affinity, &previous_affinity, sizeof(affinity)); 591a8c51b3fSopenharmony_ci 592a8c51b3fSopenharmony_ci bool is_first_cpu = true; 593a8c51b3fSopenharmony_ci 594a8c51b3fSopenharmony_ci for (int i = 0; i < CPU_SETSIZE; ++i) 595a8c51b3fSopenharmony_ci if (CPU_ISSET(i, &affinity)) { 596a8c51b3fSopenharmony_ci if (is_first_cpu) 597a8c51b3fSopenharmony_ci is_first_cpu = false; 598a8c51b3fSopenharmony_ci else 599a8c51b3fSopenharmony_ci CPU_CLR(i, &affinity); 600a8c51b3fSopenharmony_ci } 601a8c51b3fSopenharmony_ci 602a8c51b3fSopenharmony_ci if (is_first_cpu) return false; 603a8c51b3fSopenharmony_ci 604a8c51b3fSopenharmony_ci ret = pthread_setaffinity_np(self, sizeof(affinity), &affinity); 605a8c51b3fSopenharmony_ci return ret == 0; 606a8c51b3fSopenharmony_ci#elif defined(BENCHMARK_OS_WINDOWS_WIN32) 607a8c51b3fSopenharmony_ci self = GetCurrentThread(); 608a8c51b3fSopenharmony_ci DWORD_PTR mask = static_cast<DWORD_PTR>(1) << GetCurrentProcessorNumber(); 609a8c51b3fSopenharmony_ci previous_affinity = SetThreadAffinityMask(self, mask); 610a8c51b3fSopenharmony_ci return previous_affinity != 0; 611a8c51b3fSopenharmony_ci#else 612a8c51b3fSopenharmony_ci return false; 613a8c51b3fSopenharmony_ci#endif // def BENCHMARK_HAS_PTHREAD_AFFINITY 614a8c51b3fSopenharmony_ci } 615a8c51b3fSopenharmony_ci 616a8c51b3fSopenharmony_ci#if defined(BENCHMARK_HAS_PTHREAD_AFFINITY) 617a8c51b3fSopenharmony_ci pthread_t self; 618a8c51b3fSopenharmony_ci cpu_set_t previous_affinity; 619a8c51b3fSopenharmony_ci#elif defined(BENCHMARK_OS_WINDOWS_WIN32) 620a8c51b3fSopenharmony_ci HANDLE self; 621a8c51b3fSopenharmony_ci DWORD_PTR previous_affinity; 622a8c51b3fSopenharmony_ci#endif // def BENCHMARK_HAS_PTHREAD_AFFINITY 623a8c51b3fSopenharmony_ci bool reset_affinity; 624a8c51b3fSopenharmony_ci}; 625a8c51b3fSopenharmony_ci 626a8c51b3fSopenharmony_cidouble GetCPUCyclesPerSecond(CPUInfo::Scaling scaling) { 627a8c51b3fSopenharmony_ci // Currently, scaling is only used on linux path here, 628a8c51b3fSopenharmony_ci // suppress diagnostics about it being unused on other paths. 629a8c51b3fSopenharmony_ci (void)scaling; 630a8c51b3fSopenharmony_ci 631a8c51b3fSopenharmony_ci#if defined BENCHMARK_OS_LINUX || defined BENCHMARK_OS_CYGWIN 632a8c51b3fSopenharmony_ci long freq; 633a8c51b3fSopenharmony_ci 634a8c51b3fSopenharmony_ci // If the kernel is exporting the tsc frequency use that. There are issues 635a8c51b3fSopenharmony_ci // where cpuinfo_max_freq cannot be relied on because the BIOS may be 636a8c51b3fSopenharmony_ci // exporintg an invalid p-state (on x86) or p-states may be used to put the 637a8c51b3fSopenharmony_ci // processor in a new mode (turbo mode). Essentially, those frequencies 638a8c51b3fSopenharmony_ci // cannot always be relied upon. The same reasons apply to /proc/cpuinfo as 639a8c51b3fSopenharmony_ci // well. 640a8c51b3fSopenharmony_ci if (ReadFromFile("/sys/devices/system/cpu/cpu0/tsc_freq_khz", &freq) 641a8c51b3fSopenharmony_ci // If CPU scaling is disabled, use the *current* frequency. 642a8c51b3fSopenharmony_ci // Note that we specifically don't want to read cpuinfo_cur_freq, 643a8c51b3fSopenharmony_ci // because it is only readable by root. 644a8c51b3fSopenharmony_ci || (scaling == CPUInfo::Scaling::DISABLED && 645a8c51b3fSopenharmony_ci ReadFromFile("/sys/devices/system/cpu/cpu0/cpufreq/scaling_cur_freq", 646a8c51b3fSopenharmony_ci &freq)) 647a8c51b3fSopenharmony_ci // Otherwise, if CPU scaling may be in effect, we want to use 648a8c51b3fSopenharmony_ci // the *maximum* frequency, not whatever CPU speed some random processor 649a8c51b3fSopenharmony_ci // happens to be using now. 650a8c51b3fSopenharmony_ci || ReadFromFile("/sys/devices/system/cpu/cpu0/cpufreq/cpuinfo_max_freq", 651a8c51b3fSopenharmony_ci &freq)) { 652a8c51b3fSopenharmony_ci // The value is in kHz (as the file name suggests). For example, on a 653a8c51b3fSopenharmony_ci // 2GHz warpstation, the file contains the value "2000000". 654a8c51b3fSopenharmony_ci return freq * 1000.0; 655a8c51b3fSopenharmony_ci } 656a8c51b3fSopenharmony_ci 657a8c51b3fSopenharmony_ci const double error_value = -1; 658a8c51b3fSopenharmony_ci double bogo_clock = error_value; 659a8c51b3fSopenharmony_ci 660a8c51b3fSopenharmony_ci std::ifstream f("/proc/cpuinfo"); 661a8c51b3fSopenharmony_ci if (!f.is_open()) { 662a8c51b3fSopenharmony_ci std::cerr << "failed to open /proc/cpuinfo\n"; 663a8c51b3fSopenharmony_ci return error_value; 664a8c51b3fSopenharmony_ci } 665a8c51b3fSopenharmony_ci 666a8c51b3fSopenharmony_ci auto StartsWithKey = [](std::string const& Value, std::string const& Key) { 667a8c51b3fSopenharmony_ci if (Key.size() > Value.size()) return false; 668a8c51b3fSopenharmony_ci auto Cmp = [&](char X, char Y) { 669a8c51b3fSopenharmony_ci return std::tolower(X) == std::tolower(Y); 670a8c51b3fSopenharmony_ci }; 671a8c51b3fSopenharmony_ci return std::equal(Key.begin(), Key.end(), Value.begin(), Cmp); 672a8c51b3fSopenharmony_ci }; 673a8c51b3fSopenharmony_ci 674a8c51b3fSopenharmony_ci std::string ln; 675a8c51b3fSopenharmony_ci while (std::getline(f, ln)) { 676a8c51b3fSopenharmony_ci if (ln.empty()) continue; 677a8c51b3fSopenharmony_ci std::size_t split_idx = ln.find(':'); 678a8c51b3fSopenharmony_ci std::string value; 679a8c51b3fSopenharmony_ci if (split_idx != std::string::npos) value = ln.substr(split_idx + 1); 680a8c51b3fSopenharmony_ci // When parsing the "cpu MHz" and "bogomips" (fallback) entries, we only 681a8c51b3fSopenharmony_ci // accept positive values. Some environments (virtual machines) report zero, 682a8c51b3fSopenharmony_ci // which would cause infinite looping in WallTime_Init. 683a8c51b3fSopenharmony_ci if (StartsWithKey(ln, "cpu MHz")) { 684a8c51b3fSopenharmony_ci if (!value.empty()) { 685a8c51b3fSopenharmony_ci double cycles_per_second = benchmark::stod(value) * 1000000.0; 686a8c51b3fSopenharmony_ci if (cycles_per_second > 0) return cycles_per_second; 687a8c51b3fSopenharmony_ci } 688a8c51b3fSopenharmony_ci } else if (StartsWithKey(ln, "bogomips")) { 689a8c51b3fSopenharmony_ci if (!value.empty()) { 690a8c51b3fSopenharmony_ci bogo_clock = benchmark::stod(value) * 1000000.0; 691a8c51b3fSopenharmony_ci if (bogo_clock < 0.0) bogo_clock = error_value; 692a8c51b3fSopenharmony_ci } 693a8c51b3fSopenharmony_ci } 694a8c51b3fSopenharmony_ci } 695a8c51b3fSopenharmony_ci if (f.bad()) { 696a8c51b3fSopenharmony_ci std::cerr << "Failure reading /proc/cpuinfo\n"; 697a8c51b3fSopenharmony_ci return error_value; 698a8c51b3fSopenharmony_ci } 699a8c51b3fSopenharmony_ci if (!f.eof()) { 700a8c51b3fSopenharmony_ci std::cerr << "Failed to read to end of /proc/cpuinfo\n"; 701a8c51b3fSopenharmony_ci return error_value; 702a8c51b3fSopenharmony_ci } 703a8c51b3fSopenharmony_ci f.close(); 704a8c51b3fSopenharmony_ci // If we found the bogomips clock, but nothing better, we'll use it (but 705a8c51b3fSopenharmony_ci // we're not happy about it); otherwise, fallback to the rough estimation 706a8c51b3fSopenharmony_ci // below. 707a8c51b3fSopenharmony_ci if (bogo_clock >= 0.0) return bogo_clock; 708a8c51b3fSopenharmony_ci 709a8c51b3fSopenharmony_ci#elif defined BENCHMARK_HAS_SYSCTL 710a8c51b3fSopenharmony_ci constexpr auto* freqStr = 711a8c51b3fSopenharmony_ci#if defined(BENCHMARK_OS_FREEBSD) || defined(BENCHMARK_OS_NETBSD) 712a8c51b3fSopenharmony_ci "machdep.tsc_freq"; 713a8c51b3fSopenharmony_ci#elif defined BENCHMARK_OS_OPENBSD 714a8c51b3fSopenharmony_ci "hw.cpuspeed"; 715a8c51b3fSopenharmony_ci#elif defined BENCHMARK_OS_DRAGONFLY 716a8c51b3fSopenharmony_ci "hw.tsc_frequency"; 717a8c51b3fSopenharmony_ci#else 718a8c51b3fSopenharmony_ci "hw.cpufrequency"; 719a8c51b3fSopenharmony_ci#endif 720a8c51b3fSopenharmony_ci unsigned long long hz = 0; 721a8c51b3fSopenharmony_ci#if defined BENCHMARK_OS_OPENBSD 722a8c51b3fSopenharmony_ci if (GetSysctl(freqStr, &hz)) return hz * 1000000; 723a8c51b3fSopenharmony_ci#else 724a8c51b3fSopenharmony_ci if (GetSysctl(freqStr, &hz)) return hz; 725a8c51b3fSopenharmony_ci#endif 726a8c51b3fSopenharmony_ci fprintf(stderr, "Unable to determine clock rate from sysctl: %s: %s\n", 727a8c51b3fSopenharmony_ci freqStr, strerror(errno)); 728a8c51b3fSopenharmony_ci fprintf(stderr, 729a8c51b3fSopenharmony_ci "This does not affect benchmark measurements, only the " 730a8c51b3fSopenharmony_ci "metadata output.\n"); 731a8c51b3fSopenharmony_ci 732a8c51b3fSopenharmony_ci#elif defined BENCHMARK_OS_WINDOWS_WIN32 733a8c51b3fSopenharmony_ci // In NT, read MHz from the registry. If we fail to do so or we're in win9x 734a8c51b3fSopenharmony_ci // then make a crude estimate. 735a8c51b3fSopenharmony_ci DWORD data, data_size = sizeof(data); 736a8c51b3fSopenharmony_ci if (IsWindowsXPOrGreater() && 737a8c51b3fSopenharmony_ci SUCCEEDED( 738a8c51b3fSopenharmony_ci SHGetValueA(HKEY_LOCAL_MACHINE, 739a8c51b3fSopenharmony_ci "HARDWARE\\DESCRIPTION\\System\\CentralProcessor\\0", 740a8c51b3fSopenharmony_ci "~MHz", nullptr, &data, &data_size))) 741a8c51b3fSopenharmony_ci return static_cast<double>(static_cast<int64_t>(data) * 742a8c51b3fSopenharmony_ci static_cast<int64_t>(1000 * 1000)); // was mhz 743a8c51b3fSopenharmony_ci#elif defined(BENCHMARK_OS_SOLARIS) 744a8c51b3fSopenharmony_ci kstat_ctl_t* kc = kstat_open(); 745a8c51b3fSopenharmony_ci if (!kc) { 746a8c51b3fSopenharmony_ci std::cerr << "failed to open /dev/kstat\n"; 747a8c51b3fSopenharmony_ci return -1; 748a8c51b3fSopenharmony_ci } 749a8c51b3fSopenharmony_ci kstat_t* ksp = kstat_lookup(kc, const_cast<char*>("cpu_info"), -1, 750a8c51b3fSopenharmony_ci const_cast<char*>("cpu_info0")); 751a8c51b3fSopenharmony_ci if (!ksp) { 752a8c51b3fSopenharmony_ci std::cerr << "failed to lookup in /dev/kstat\n"; 753a8c51b3fSopenharmony_ci return -1; 754a8c51b3fSopenharmony_ci } 755a8c51b3fSopenharmony_ci if (kstat_read(kc, ksp, NULL) < 0) { 756a8c51b3fSopenharmony_ci std::cerr << "failed to read from /dev/kstat\n"; 757a8c51b3fSopenharmony_ci return -1; 758a8c51b3fSopenharmony_ci } 759a8c51b3fSopenharmony_ci kstat_named_t* knp = (kstat_named_t*)kstat_data_lookup( 760a8c51b3fSopenharmony_ci ksp, const_cast<char*>("current_clock_Hz")); 761a8c51b3fSopenharmony_ci if (!knp) { 762a8c51b3fSopenharmony_ci std::cerr << "failed to lookup data in /dev/kstat\n"; 763a8c51b3fSopenharmony_ci return -1; 764a8c51b3fSopenharmony_ci } 765a8c51b3fSopenharmony_ci if (knp->data_type != KSTAT_DATA_UINT64) { 766a8c51b3fSopenharmony_ci std::cerr << "current_clock_Hz is of unexpected data type: " 767a8c51b3fSopenharmony_ci << knp->data_type << "\n"; 768a8c51b3fSopenharmony_ci return -1; 769a8c51b3fSopenharmony_ci } 770a8c51b3fSopenharmony_ci double clock_hz = knp->value.ui64; 771a8c51b3fSopenharmony_ci kstat_close(kc); 772a8c51b3fSopenharmony_ci return clock_hz; 773a8c51b3fSopenharmony_ci#elif defined(BENCHMARK_OS_QNX) 774a8c51b3fSopenharmony_ci return static_cast<double>((int64_t)(SYSPAGE_ENTRY(cpuinfo)->speed) * 775a8c51b3fSopenharmony_ci (int64_t)(1000 * 1000)); 776a8c51b3fSopenharmony_ci#elif defined(BENCHMARK_OS_QURT) 777a8c51b3fSopenharmony_ci // QuRT doesn't provide any API to query Hexagon frequency. 778a8c51b3fSopenharmony_ci return 1000000000; 779a8c51b3fSopenharmony_ci#endif 780a8c51b3fSopenharmony_ci // If we've fallen through, attempt to roughly estimate the CPU clock rate. 781a8c51b3fSopenharmony_ci 782a8c51b3fSopenharmony_ci // Make sure to use the same cycle counter when starting and stopping the 783a8c51b3fSopenharmony_ci // cycle timer. We just pin the current thread to a cpu in the previous 784a8c51b3fSopenharmony_ci // affinity set. 785a8c51b3fSopenharmony_ci ThreadAffinityGuard affinity_guard; 786a8c51b3fSopenharmony_ci 787a8c51b3fSopenharmony_ci static constexpr double estimate_time_s = 1.0; 788a8c51b3fSopenharmony_ci const double start_time = ChronoClockNow(); 789a8c51b3fSopenharmony_ci const auto start_ticks = cycleclock::Now(); 790a8c51b3fSopenharmony_ci 791a8c51b3fSopenharmony_ci // Impose load instead of calling sleep() to make sure the cycle counter 792a8c51b3fSopenharmony_ci // works. 793a8c51b3fSopenharmony_ci using PRNG = std::minstd_rand; 794a8c51b3fSopenharmony_ci using Result = PRNG::result_type; 795a8c51b3fSopenharmony_ci PRNG rng(static_cast<Result>(start_ticks)); 796a8c51b3fSopenharmony_ci 797a8c51b3fSopenharmony_ci Result state = 0; 798a8c51b3fSopenharmony_ci 799a8c51b3fSopenharmony_ci do { 800a8c51b3fSopenharmony_ci static constexpr size_t batch_size = 10000; 801a8c51b3fSopenharmony_ci rng.discard(batch_size); 802a8c51b3fSopenharmony_ci state += rng(); 803a8c51b3fSopenharmony_ci 804a8c51b3fSopenharmony_ci } while (ChronoClockNow() - start_time < estimate_time_s); 805a8c51b3fSopenharmony_ci 806a8c51b3fSopenharmony_ci DoNotOptimize(state); 807a8c51b3fSopenharmony_ci 808a8c51b3fSopenharmony_ci const auto end_ticks = cycleclock::Now(); 809a8c51b3fSopenharmony_ci const double end_time = ChronoClockNow(); 810a8c51b3fSopenharmony_ci 811a8c51b3fSopenharmony_ci return static_cast<double>(end_ticks - start_ticks) / (end_time - start_time); 812a8c51b3fSopenharmony_ci // Reset the affinity of current thread when the lifetime of affinity_guard 813a8c51b3fSopenharmony_ci // ends. 814a8c51b3fSopenharmony_ci} 815a8c51b3fSopenharmony_ci 816a8c51b3fSopenharmony_cistd::vector<double> GetLoadAvg() { 817a8c51b3fSopenharmony_ci#if (defined BENCHMARK_OS_FREEBSD || defined(BENCHMARK_OS_LINUX) || \ 818a8c51b3fSopenharmony_ci defined BENCHMARK_OS_MACOSX || defined BENCHMARK_OS_NETBSD || \ 819a8c51b3fSopenharmony_ci defined BENCHMARK_OS_OPENBSD || defined BENCHMARK_OS_DRAGONFLY) && \ 820a8c51b3fSopenharmony_ci !(defined(__ANDROID__) && __ANDROID_API__ < 29) 821a8c51b3fSopenharmony_ci static constexpr int kMaxSamples = 3; 822a8c51b3fSopenharmony_ci std::vector<double> res(kMaxSamples, 0.0); 823a8c51b3fSopenharmony_ci const int nelem = getloadavg(res.data(), kMaxSamples); 824a8c51b3fSopenharmony_ci if (nelem < 1) { 825a8c51b3fSopenharmony_ci res.clear(); 826a8c51b3fSopenharmony_ci } else { 827a8c51b3fSopenharmony_ci res.resize(nelem); 828a8c51b3fSopenharmony_ci } 829a8c51b3fSopenharmony_ci return res; 830a8c51b3fSopenharmony_ci#else 831a8c51b3fSopenharmony_ci return {}; 832a8c51b3fSopenharmony_ci#endif 833a8c51b3fSopenharmony_ci} 834a8c51b3fSopenharmony_ci 835a8c51b3fSopenharmony_ci} // end namespace 836a8c51b3fSopenharmony_ci 837a8c51b3fSopenharmony_ciconst CPUInfo& CPUInfo::Get() { 838a8c51b3fSopenharmony_ci static const CPUInfo* info = new CPUInfo(); 839a8c51b3fSopenharmony_ci return *info; 840a8c51b3fSopenharmony_ci} 841a8c51b3fSopenharmony_ci 842a8c51b3fSopenharmony_ciCPUInfo::CPUInfo() 843a8c51b3fSopenharmony_ci : num_cpus(GetNumCPUs()), 844a8c51b3fSopenharmony_ci scaling(CpuScaling(num_cpus)), 845a8c51b3fSopenharmony_ci cycles_per_second(GetCPUCyclesPerSecond(scaling)), 846a8c51b3fSopenharmony_ci caches(GetCacheSizes()), 847a8c51b3fSopenharmony_ci load_avg(GetLoadAvg()) {} 848a8c51b3fSopenharmony_ci 849a8c51b3fSopenharmony_ciconst SystemInfo& SystemInfo::Get() { 850a8c51b3fSopenharmony_ci static const SystemInfo* info = new SystemInfo(); 851a8c51b3fSopenharmony_ci return *info; 852a8c51b3fSopenharmony_ci} 853a8c51b3fSopenharmony_ci 854a8c51b3fSopenharmony_ciSystemInfo::SystemInfo() : name(GetSystemName()) {} 855a8c51b3fSopenharmony_ci} // end namespace benchmark 856