讲师中心 微信公众号
AI工具推荐 视频效率加速

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

酷雪大大_1633

酷雪大大_1633

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

|

644人浏览过

|

来源于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)

算法原理简述:

  • 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大模型工具。

Seko
Seko Hot

一款AI视频创作工具,主要用于商汤科技推出的创编一体的AI短视频创作Agent,适合需要提升相关任务效率的用户。

豆包大模型

豆包大模型是一款由字节跳动推出的企业级大语言模型服务平台。

PixPix
PixPix Hot

PixPix是一款面向电商视觉生产的AI商品图生成工具。

UP简历
UP简历 Hot

一款AI办公效率工具,主要用于基于AI技术的免费在线简历制作工具,适合需要提升相关任务效率的用户。

VibeKnow
VibeKnow Hot

一款AI视频创作工具,主要用于全球首个AI知识视频创作平台,文档、文章、网页,一键生成视频,适合需要提升相关任务效率的用户。

讯飞智作

讯飞智作是一款AI视频创作工具,AI文本配音工具,数字人课程、营销视频制作。

二狗PPT
二狗PPT Hot

一款AI演示文稿工具,主要用于专为中式职场打造的AI PPT生成工具,适合需要提升相关任务效率的用户。

WorkBuddy

一款AI办公效率工具,主要用于腾讯云推出的AI原生桌面智能体工作台,适合需要提升相关任务效率的用户。

相关专题

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

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

499

2024.02.23

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

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

596

2024.02.23

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

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

1056

2024.02.23

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

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

771

2024.03.05

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

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

1546

2024.05.21

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

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

4314

2025.06.09

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

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

1954

2025.06.10

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

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

3866

2025.06.17

LLVM自定义Pass怎么写
LLVM自定义Pass怎么写

本专题聚焦LLVM自定义Pass开发,整理Pass类结构、run()方法、PreservedAnalyses、CMake构建、插件注册、-load-pass-plugin加载和测试用例编写流程。

80

2026.09.30

热门下载

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

精品课程

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

Copyright 2014-2026 https://www.php.cn/ All Rights Reserved | php.cn