0

0

深入解析高效计算 CPU 亲和性掩码中活跃核心数的位运算算法

心靈之曲

心靈之曲

发布时间:2026-02-19 22:20:02

|

604人浏览过

|

来源于php中文网

原创

深入解析高效计算 CPU 亲和性掩码中活跃核心数的位运算算法

本文详解 getproccount 函数如何通过系统调用获取 cpu 亲和性掩码,并利用经典位操作(brian kernighan 变体 + 并行位计数)在常数时间内统计掩码中置位比特数,从而准确推导物理/逻辑核心数量。

本文详解 getproccount 函数如何通过系统调用获取 cpu 亲和性掩码,并利用经典位操作(brian kernighan 变体 + 并行位计数)在常数时间内统计掩码中置位比特数,从而准确推导物理/逻辑核心数量。

getproccount 是 Go 语言早期运行时(pre-1.4)中用于探测当前进程可调度 CPU 核心数量的关键函数。其核心逻辑分为三步:获取亲和性掩码 → 解析掩码数组 → 统计所有掩码字中置位(bit set)总数。最终返回值即为操作系统允许该进程使用的逻辑核心数;若系统调用失败或掩码全零,则保守返回 1。

关键步骤解析

1. 获取 CPU 亲和性掩码

int32 r = runtime·sched_getaffinity(0, sizeof(buf), buf);

该系统调用将当前进程的 CPU 亲和性掩码(affinity mask)写入 buf 数组。buf 类型为 uintptr[16],在 64 位系统中每个元素为 8 字节(64 位),因此最多可容纳 16 × 64 = 1024 个 CPU 核心的掩码信息。r 返回实际写入的字节数,故有效数组长度为 r / sizeof(uintptr)。

2. 对每个掩码字执行并行位计数(Population Count)

对 buf[i] 中的每个 uintptr 值 t,采用经典的 SWAR(SIMD Within A Register)位计数算法,仅用 3 行位运算即可在 O(1) 时间内算出其二进制表示中 1 的个数:

t = t - ((t >> 1) & 0x5555555555555555ULL);   // Step 1: 每2位一组,计算组内1的个数(0~2)
t = (t & 0x3333333333333333ULL) + ((t >> 2) & 0x3333333333333333ULL); // Step 2: 每4位一组,累加(0~4)
cnt += (int32)((((t + (t >> 4)) & 0xF0F0F0F0F0F0F0FULL) * 0x101010101010101ULL) >> 56); // Step 3: 全局求和(0~64)

算法原理简述

悟智写作
悟智写作

易开即用的AI写作平台

下载
  • Step 1:0x5555...(二进制 010101...)用于隔离奇偶位。t - (t>>1 & mask) 将每相邻 2 位视为一个“计数单元”,输出值即为该单元中 1 的数量(如 11 → 10₂ = 2)。
  • Step 2:0x3333...(00110011...)提取每 4 位中的低 2 位与高 2 位,相加后得到每 4 位中 1 的总数。
  • Step 3:先用 t + (t>>4) 将每 8 位内的前 4 位与后 4 位相加;& 0xF0F0... 提取每 8 位的高 4 位(即结果);再乘以 0x1010101...(二进制为 1 间隔 7 位),实现“字节内广播求和”——最高字节自动累加所有字节的值;最后 >>56 提取最高字节,即为整个 64 位字中 1 的总数。

✅ 该算法无需分支、无查表、全流水,是编译器内置 __builtin_popcount 的底层实现逻辑之一,在无硬件 POPCNT 指令的老平台(如部分 ARM 或早期 x86)上尤为关键。

3. 边界处理与默认回退

return cnt ? cnt : 1;

若 sched_getaffinity 调用失败(r ≤ 0)或所有掩码字均为 0(cnt == 0),函数返回 1。这是一种安全降级策略:确保程序至少能在单核上运行,避免因探测失败导致不可用。

注意事项与工程启示

  • 可移植性限制:此实现依赖 uintptr 为 64 位且系统支持 sched_getaffinity(Linux/FreeBSD),在 Windows 或旧内核上需适配。
  • 逻辑核 ≠ 物理核:该函数返回的是操作系统可见的逻辑处理器数量(含超线程),非物理核心数。如需区分,需读取 /sys/devices/system/cpu/ 下的拓扑信息。
  • 现代替代方案:Go 1.5+ 已弃用此手写位运算,转而使用 runtime.NumCPU(),其底层可能调用 sysconf(_SC_NPROCESSORS_ONLN) 或直接读取 /proc/sys/kernel/osrelease,更简洁且兼容性更强。
  • 性能提示:尽管位运算极快,但 sched_getaffinity 是系统调用,存在上下文切换开销。生产环境应缓存结果,避免高频调用。

掌握此类底层位操作,不仅有助于理解运行时设计哲学,更是构建高性能系统软件(如调度器、内存分配器)的必备基础能力。

本站声明:本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系admin@php.cn

热门AI工具

更多
DeepSeek
DeepSeek

幻方量化公司旗下的开源大模型平台

豆包大模型
豆包大模型

字节跳动自主研发的一系列大型语言模型

通义千问
通义千问

阿里巴巴推出的全能AI助手

腾讯元宝
腾讯元宝

腾讯混元平台推出的AI助手

文心一言
文心一言

文心一言是百度开发的AI聊天机器人,通过对话可以生成各种形式的内容。

讯飞写作
讯飞写作

基于讯飞星火大模型的AI写作工具,可以快速生成新闻稿件、品宣文案、工作总结、心得体会等各种文文稿

即梦AI
即梦AI

一站式AI创作平台,免费AI图片和视频生成。

ChatGPT
ChatGPT

最最强大的AI聊天机器人程序,ChatGPT不单是聊天机器人,还能进行撰写邮件、视频脚本、文案、翻译、代码等任务。

相关专题

更多
golang如何定义变量
golang如何定义变量

golang定义变量的方法:1、声明变量并赋予初始值“var age int =值”;2、声明变量但不赋初始值“var age int”;3、使用短变量声明“age :=值”等等。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

207

2024.02.23

golang有哪些数据转换方法
golang有哪些数据转换方法

golang数据转换方法:1、类型转换操作符;2、类型断言;3、字符串和数字之间的转换;4、JSON序列化和反序列化;5、使用标准库进行数据转换;6、使用第三方库进行数据转换;7、自定义数据转换函数。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

239

2024.02.23

golang常用库有哪些
golang常用库有哪些

golang常用库有:1、标准库;2、字符串处理库;3、网络库;4、加密库;5、压缩库;6、xml和json解析库;7、日期和时间库;8、数据库操作库;9、文件操作库;10、图像处理库。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

348

2024.02.23

golang和python的区别是什么
golang和python的区别是什么

golang和python的区别是:1、golang是一种编译型语言,而python是一种解释型语言;2、golang天生支持并发编程,而python对并发与并行的支持相对较弱等等。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

212

2024.03.05

golang是免费的吗
golang是免费的吗

golang是免费的。golang是google开发的一种静态强类型、编译型、并发型,并具有垃圾回收功能的开源编程语言,采用bsd开源协议。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

403

2024.05.21

golang结构体相关大全
golang结构体相关大全

本专题整合了golang结构体相关大全,想了解更多内容,请阅读专题下面的文章。

344

2025.06.09

golang相关判断方法
golang相关判断方法

本专题整合了golang相关判断方法,想了解更详细的相关内容,请阅读下面的文章。

197

2025.06.10

golang数组使用方法
golang数组使用方法

本专题整合了golang数组用法,想了解更多的相关内容,请阅读专题下面的文章。

970

2025.06.17

pixiv网页版官网登录与阅读指南_pixiv官网直达入口与在线访问方法
pixiv网页版官网登录与阅读指南_pixiv官网直达入口与在线访问方法

本专题系统整理pixiv网页版官网入口及登录访问方式,涵盖官网登录页面直达路径、在线阅读入口及快速进入方法说明,帮助用户高效找到pixiv官方网站,实现便捷、安全的网页端浏览与账号登录体验。

660

2026.02.13

热门下载

更多
网站特效
/
网站源码
/
网站素材
/
前端模板

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
关于我们 免责申明 举报中心 意见反馈 讲师合作 广告合作 最新更新
php中文网:公益在线php培训,帮助PHP学习者快速成长!
关注服务号 技术交流群
PHP中文网订阅号
每天精选资源文章推送

Copyright 2014-2026 https://www.php.cn/ All Rights Reserved | php.cn | 湘ICP备2023035733号