CPUFeature

CPUFeature 是一个用于检测 x86/AMD64 处理器上 CPU 功能的工具。特别设计为在导入时具有非常低的开销（约 1 毫秒）并检测在混合使用多线程和多进程的高性能计算中经常重要的功能。检测到的 CPU 功能包括

• SIMD 指令支持
• 核心数量（虚拟和物理核心）
• 主板上的 CPU 数量。
• 缓存大小

它使用 C99 编写。它支持 Windows 上的 MSVC 和 Linux 上的 GCC/LLVM 编译器。MacOSX 应该也支持 GCC/LLVM。CPUFeature 在 Creative Commons Zero 下授权，因此可以自由地将其作为子模块纳入您的项目。

CPUFeature 被认为是 Alpha 状态，因为在 GitHub Actions 上对各种平台进行了测试，但我们预计会出现边缘情况。

示例

该模块在导入时生成一个名为 CPUFeature 的字典，其中包含有关宿主机的优化和编译的有用信息。对于我的 Intel i7-7820X Skylake CPU，它生成以下信息

    > python -c "import cpufeature; cpufeature.print_features()"
  === CPU FEATURES ===
      VendorId                : GenuineIntel
      num_virtual_cores       : 16
      num_physical_cores      : 8
      num_threads_per_core    : 2
      num_cpus                : 1
      cache_line_size         : 64
      cache_L1_size           : 32768
      cache_L2_size           : 1048576
      cache_L3_size           : 11534336
      OS_x64                  : True
      OS_AVX                  : True
      OS_AVX512               : True
      MMX                     : True
      x64                     : True
      ABM                     : True
      RDRAND                  : True
      BMI1                    : True
      BMI2                    : True
      ADX                     : True
      PREFETCHWT1             : False
      MPX                     : True
      SSE                     : True
      SSE2                    : True
      SSE3                    : True
      SSSE3                   : True
      SSE4.1                  : True
      SSE4.2                  : True
      SSE4.a                  : False
      AES                     : True
      SHA                     : False
      AVX                     : True
      XOP                     : False
      FMA3                    : True
      FMA4                    : False
      AVX2                    : True
      AVX512f                 : True
      AVX512pf                : False
      AVX512er                : False
      AVX512cd                : True
      AVX512vl                : True
      AVX512bw                : True
      AVX512dq                : True
      AVX512ifma              : False
      AVX512vbmi              : False
      AVX512vbmi2             : False
      AVX512vnni              : False

字段说明

• 如果 num_threads_per_core > 1，则可能存在 Intel Hyperthreading（或等效架构）。在虚拟环境中，物理核心数量可能不可靠。
• 缓存大小以字节为单位。
• 对于 AVX2，请检查 AVX2 和 OS_AVX2。
• 对于AVX512，请检查AVX512f和OS_AVX512。在ICC编译之外，Python对AVX512指令的支持非常有限。直到MSVC2017，MSVC都没有AVX512支持。

作者

• 基于Alexander Yee的FeatureDetector（https://github.com/Mysticial/FeatureDetector）。
• 由Robert A. McLeod移植到Python，并增加了核心和缓存大小的检测。
• drfinkus为AMD支持做出了贡献。
• WidgetA为使用Clang编译提供了支持。

发行说明

0.2.1

• 支持的Python版本现在为3.8 - 3.11。
• 添加了AVX512vbmi2和AVX512vnni的测试。

0.2.0

• 通过cibuildwheels和GitHub Actions添加了预构建的轮子，适用于Linux/Windows/MacOSX上的Python 3.6-3.9。
• 添加了对AMD处理器的支持。
• 添加了一个用于valgrind的脚本，以在模块崩溃时更容易地定位问题。

0.1.1

• 针对clang编译的修复。

0.1.0

• 添加了对处理器、核心和缓存大小的检测
• 创建了FeatureDetector的Python包装器
• 将FeatureDetector转换为C99，以获得更好的跨平台兼容性