zyl910

优化技巧、硬件体系、图像处理、图形学、游戏编程、国际化与文本信息处理。

  博客园 :: 首页 :: 博问 :: 闪存 :: 新随笔 :: 联系 :: 订阅 订阅 :: 管理 ::

作者:zyl910

  之前的ccpuid V1.02的mmx/sse指令可用性检查存在缺陷。现在的V1.03版改进了mmx/sse指令可用性检查,使用signal、setjmp,能够支持纯C程序。修正了AVX检查Bug。增加多文件链接ccpuid的测试例程。


一、更新说明

1.1 改进mmx/sse指令可用性检查

1.1.1 问题背景

  以前是使用结构化异常处理来确认当前环境是否能运行mmx/sse指令的。该方法存在两个问题。
  首先,仅有C++支持结构化异常处理,而纯C是不支持的。所以在V1.02版中,是根据__cplusplus宏来做条件编译的。造成仅有C++版支持指令可用性检查,而纯C版没有该检查能力。
  更重大的问题是——结构化异常处理并不能捕获mmx/sse指令错误,程序会崩溃。
  这是因为——当运行了当前环境不支持指令时,不会触发结构化异常处理,而是触发SIG_ILL(非法指令)信号,若没有该信号的处理函数,程序就崩溃了。

  本来我以为“硬件支持MMX/SSE,但当前运行环境不支持”这种情况是很罕见的,因为现在的主流操作系统(Windows、Linux、Mac OS X等)都是支持mmx和sse。
  但后来发现,这种情况是很容易出现的。这是因为AVX指令集的影响。
  当gcc用上“-mavx”编译参数时,gcc会将所有的SSE代码编译为“VEX前缀编码的SSE指令”,而不是“传统SSE编码的指令”。“VEX前缀编码的SSE指令”实际上是128位的AVX指令,只有CPU硬件支持AVX,并且操作系统支持AVX时,才能运行“VEX前缀编码的SSE指令”。
  就算你的代码中没有使用SSE,但是由于编译器的自动矢量化编译优化,也可能会自动生成VEX SSE指令,导致程序崩溃。

  而VC在这方面好一点,依然是生成传统编码的SSE指令,程序能够在非AVX环境下运行。


1.1.2 解决办法

  怎么捕获SIG_ILL(非法指令)信号呢?C标准库提供了<signal.h>,用于处理信号。
  从SIG_ILL信号中怎么恢复呢?C标准库提供了<setjmp.h>,可以实现非局部转移。

  但是还存在一个问题——在信号处理函数中,怎么知道应该回到哪一个位置呢?这时只有靠一个全局变量来存放返回位置。
  而使用全局变量,可能会引起线程安全问题。暂时不管了,目前的跨平台多线程处理很麻烦。还是等C11普及后再说吧。


1.1.3 具体实现1(修改ccpuid.h)

  在具体实现时,会遇到这个问题——纯C版的ccpuid只是一个头文件,按照惯例只能含有声明性内容,不能含有定义性内容。而信号处理函数必须是一个实际定义的函数,这样才能获取它的函数指针进行注册。
  如果再添加一个ccpuid.c源文件,在它里面编写信号处理函数。虽然这样更符合C语言惯例,但这样用起来很麻烦,而且可能还需要修改代码才能与新版的ccpuid兼容。
  所以我决定还是将信号处理函数及相关的全局变量放在头文件中。为了避免外部暴露,可以加上static关键字,使它们仅在文件内可见、不参与链接。

  由于现在需要检查MMX与SSE这两类指令,所以可以考虑将这种检查封装成为一个函数——

// 收到SIGILL信号时的跳回地址.
static jmp_buf *volatile simd_pjmpback_sigill = NULL;

// 处理SIGILL信号.
static void simd_catch_sigill(int sig)
{
    jmp_buf * pjmp = simd_pjmpback_sigill;
    // 能够跳回.
    if (NULL!=pjmp)
    {
        longjmp(*pjmp, 1);    // 跳回.
    }
    // 不能跳回.
    //fprintf(stderr, "!SIGILL!");
    abort();
}

// 尝试调用一个可能会发生SIGILL信号的函数.
//
// result: 若正常运行,就返回pfunc的返回值. 否则返回0.
// pfunc: 欲测试的函数.
// puserdata: 用户自定参数, 在调用pfunc时会传递该参数.
static int simd_try_sigill(int (*pfunc)(void*), void* puserdata)
{
    int rt = 0;
    jmp_buf myjmp;
    jmp_buf* poldjmp=NULL;    // 以前的跳回地址.
    void (*old_signal)(int) = SIG_DFL;    // 以前的信号处理函数.

    // 注册跳转.
    if (0==setjmp(myjmp))
    {
        // 登记跳回.
        poldjmp = simd_pjmpback_sigill;
        simd_pjmpback_sigill = &myjmp;

        // 注册信号处理函数.
        old_signal = signal(SIGILL, simd_catch_sigill);

        // [try]
        rt = pfunc(puserdata);
    }
    else
    {
        // [catch]
    }

    // 恢复信号处理函数.
    simd_pjmpback_sigill = poldjmp;
    signal(SIGILL, old_signal);

    return rt;
}

 

 

  验证MMX指令是否能运行——

// 验证mmx指令是否能运行_实际指令测试.
static int    simd_try_mmx_pfunc(void* puserdata)
{
    #if defined(_M_X64) && defined(_MSC_VER) && !defined(__INTEL_COMPILER)
        // VC编译器不支持64位下的MMX.
        return 0;
    #else
        _mm_empty();    // MMX instruction: emms
        return 1;
    #endif    // #if defined(_M_X64) && defined(_MSC_VER)
}

// 验证mmx指令是否能运行.
static int    simd_try_mmx()
{
    int rt = 0;
    #if defined(_M_X64) && defined(_MSC_VER) && !defined(__INTEL_COMPILER)
        // VC编译器不支持64位下的MMX.
    #else
        rt = simd_try_sigill(simd_try_mmx_pfunc, 0);
    #endif    // #if defined(_M_X64) && defined(_MSC_VER)
    return rt;
}

 

  验证SSE指令是否能运行——

// 验证sse指令是否能运行_实际指令测试.
static int    simd_try_sse_pfunc(void* puserdata)
{
    int rt = 0;
    volatile __m128 xmm1 = _mm_setzero_ps();    // SSE instruction: xorps
    int* pxmm1 = (int*)&xmm1;    // 避免GCC的 -Wstrict-aliasing 警告.
    if (0==*pxmm1)    rt = 1;    // 避免Release模式编译优化时剔除_mm_setzero_ps.
    return rt;
}

// 验证sse指令是否能运行.
static int    simd_try_sse()
{
    int rt = simd_try_sigill(simd_try_sse_pfunc, 0);
    return rt;
}

 

  然后再修改原来的simd_mmx、simd_sse_level函数,在最后调用simd_try_mmx、simd_try_sse验证当前环境——

// 是否支持MMX指令集.
//
// result: 返回当前运行环境是否支持MMX指令集. 非0表示支持, 0表示不支持.
// phwmmx: 返回硬件是否支持MMX指令集. 非0表示支持, 0表示不支持.
INLINE int    simd_mmx(int* phwmmx)
{
    int    rt = 0;    // result
    #ifdef CCPUID_X86
        const uint32_t    BIT_D_MMX = 0x00800000;    // bit 23
        uint32_t dwBuf[4];

        // check processor support
        getcpuid(dwBuf, 1);    // Function 1: Feature Information
        if ( dwBuf[3] & BIT_D_MMX )    rt=1;
        if (NULL!=phwmmx)    *phwmmx=rt;

        // check OS support
        if ( rt>0 )
        {
            if (!simd_try_mmx()) rt=0;
        }
    #else    // #ifdef CCPUID_X86
        if (NULL!=phwmmx)    *phwmmx=rt;
    #endif    // #ifdef CCPUID_X86
    return rt;
}

// 检测SSE系列指令集的支持级别.
//
// result: 返回当前运行环境的SSE系列指令集支持级别. 详见SIMD_SSE_常数.
// phwmmx: 返回硬件的SSE系列指令集支持级别. 详见SIMD_SSE_常数.
INLINE int    simd_sse_level(int* phwsse)
{
    int    rt = SIMD_SSE_NONE;    // result
    #ifdef CCPUID_X86
        const uint32_t    BIT_D_SSE = 0x02000000;    // bit 25
        const uint32_t    BIT_D_SSE2 = 0x04000000;    // bit 26
        const uint32_t    BIT_C_SSE3 = 0x00000001;    // bit 0
        const uint32_t    BIT_C_SSSE3 = 0x00000100;    // bit 9
        const uint32_t    BIT_C_SSE41 = 0x00080000;    // bit 19
        const uint32_t    BIT_C_SSE42 = 0x00100000;    // bit 20
        uint32_t dwBuf[4];

        // check processor support
        getcpuid(dwBuf, 1);    // Function 1: Feature Information
        if ( dwBuf[3] & BIT_D_SSE )
        {
            rt = SIMD_SSE_1;
            if ( dwBuf[3] & BIT_D_SSE2 )
            {
                rt = SIMD_SSE_2;
                if ( dwBuf[2] & BIT_C_SSE3 )
                {
                    rt = SIMD_SSE_3;
                    if ( dwBuf[2] & BIT_C_SSSE3 )
                    {
                        rt = SIMD_SSE_3S;
                        if ( dwBuf[2] & BIT_C_SSE41 )
                        {
                            rt = SIMD_SSE_41;
                            if ( dwBuf[2] & BIT_C_SSE42 )
                            {
                                rt = SIMD_SSE_42;
                            }
                        }
                    }
                }
            }
        }
        if (NULL!=phwsse)    *phwsse=rt;

        // check OS support
        if ( rt>0 )
        {
            if (!simd_try_sse()) rt=SIMD_SSE_NONE;
        }
    #else    // #ifdef CCPUID_X86
        if (NULL!=phwmmx)    *phwmmx=rt;
    #endif    // #ifdef CCPUID_X86
    return rt;
}

 


1.1.4 具体实现2(修改ccpuid.hpp、ccpuid.cpp)

  再考虑一下如何修改CCPUID类。原先是在调用RefreshProperty函数时,就一起更新_mmx、_sse等变量。而现在simd_try_mmx、simd_try_sse等验证函数不是线程安全的,如果还是一起更新的话,恐怕会影响稳定性。
  所以设计为在需要时才验证的模式更好。具体做法是,simd_mmx、simd_sse_level返回-1,由mmx、sse负责检查。

  mmx、sse原先设计为 const成员函数,而现在需要保存验证结果而修改_mmx、_sse变量,该怎么办呢?有两种方案——
1. 去掉该函数的const关键字,使其成为普通成员函数。
2. 将_mmx、_sse变量加上mutable关键字,使其能在 const成员函数 中修改。

  考虑到旧代码的兼容性,我选择了方案2。

  修改ccpuid.hpp:给_mmx、_sse变量加上mutable关键字,将mmx、sse函数挪至cpp文件中——

    int mmx() const;    // 系统支持MMX.
    int sse() const;    // 系统支持SSE.
...
    mutable int _mmx;    // 系统支持MMX.
    mutable int _sse;    // 系统支持SSE.

 



  修改ccpuid.cpp:simd_mmx、simd_sse_level返回-1,由mmx、sse负责检查——

int    CCPUID::simd_mmx(int* phwmmx) const
{
    int    rt = 0;    // result
    #ifdef CCPUID_X86
        const uint32_t    BIT_D_MMX = 0x00800000;    // bit 23
        uint32_t dwBuf[4];

        // check processor support
        GetData(dwBuf, 1);    // Function 1: Feature Information
        if ( dwBuf[3] & BIT_D_MMX )    rt=1;
        if (NULL!=phwmmx)    *phwmmx=rt;

        // check OS support
        rt = -1;    // 需要时才检查, 见mmx().
    #else    // #ifdef CCPUID_X86
        if (NULL!=phwmmx)    *phwmmx=rt;
    #endif    // #ifdef CCPUID_X86
    return rt;
}

int    CCPUID::simd_sse_level(int* phwsse) const
{
    int    rt = SIMD_SSE_NONE;    // result
    #ifdef CCPUID_X86
        const uint32_t    BIT_D_SSE = 0x02000000;    // bit 25
        const uint32_t    BIT_D_SSE2 = 0x04000000;    // bit 26
        const uint32_t    BIT_C_SSE3 = 0x00000001;    // bit 0
        const uint32_t    BIT_C_SSSE3 = 0x00000100;    // bit 9
        const uint32_t    BIT_C_SSE41 = 0x00080000;    // bit 19
        const uint32_t    BIT_C_SSE42 = 0x00100000;    // bit 20
        uint32_t dwBuf[4];

        // check processor support
        GetData(dwBuf, 1);    // Function 1: Feature Information
        if ( dwBuf[3] & BIT_D_SSE )
        {
            rt = SIMD_SSE_1;
            if ( dwBuf[3] & BIT_D_SSE2 )
            {
                rt = SIMD_SSE_2;
                if ( dwBuf[2] & BIT_C_SSE3 )
                {
                    rt = SIMD_SSE_3;
                    if ( dwBuf[2] & BIT_C_SSSE3 )
                    {
                        rt = SIMD_SSE_3S;
                        if ( dwBuf[2] & BIT_C_SSE41 )
                        {
                            rt = SIMD_SSE_41;
                            if ( dwBuf[2] & BIT_C_SSE42 )
                            {
                                rt = SIMD_SSE_42;
                            }
                        }
                    }
                }
            }
        }
        if (NULL!=phwsse)    *phwsse=rt;

        // check OS support
        rt = -1;    // 需要时才检查, 见sse().
    #else    // #ifdef CCPUID_X86
        if (NULL!=phwsse)    *phwsse=rt;
    #endif    // #ifdef CCPUID_X86
    return rt;
}

...


int CCPUID::mmx() const
{
    if (_mmx<0)
    {
        // 发现未检查, 进行检查.
        _mmx = _hwmmx;
        if (_mmx>0)
        {
            if (!simd_try_mmx()) _mmx=0;
        }
    }
    return _mmx;
}

int CCPUID::sse() const
{
    if (_sse<0)
    {
        // 发现未检查, 进行检查.
        _sse = _hwsse;
        if (_sse>0)
        {
            if (!simd_try_sse()) _sse=SIMD_SSE_NONE;
        }
    }
    return _sse;
}

 

 

1.2 修正AVX检查Bug

  V1.02版因X86平台判断功能改动了AVX检查代码,不小心留下了一个Bug——当操作系统不支持OSXSAVE时,simd_avx_level函数却返回一个正数,报告当前环境支持AVX。而按照规定,它应该返回0。

  经过检查,发现是条件判断覆盖问题。于是修改了 simd_avx_level、CCPUID::simd_avx_level,修正了此Bug。


1.3 增加多文件链接ccpuid的测试例程

  因SIGILL信号处理,现在ccpuid.h这个头文件中不得不含有定义性代码。一般来说,当头文件中含有定义性代码时,有时会造成多文件链接时报错。
  虽然我们已经加上了static关键字使那部分代码变为仅文件内可见、不参与链接。但是为了验证,还是编写一个测试例程比较好。

  于是在C++测试例程项目中 增加两个文件——test2.h和test2.cpp。
  test2.h用于声明test2函数——
void test2(void);

  在test2.cpp中也引用了ccpuid.hpp,并写了简单的测试代码——

#include <stdio.h>

#include "ccpuid.hpp"

#include "test2.h"

// 测试调用 const对象 的 const成员函数.
void test2_const(const CCPUID& ccid)
{
    printf("test2_SSE:\t%d\n",ccid.sse());
}

// 测试多文件链接使用ccpuid.
void test2(void)
{
    test2_const(CCPUID::cur());
}

 

  最后在testccpuid.cpp中添加对test2函数的调用——

#include "test2.h"
...
test2();

 

 

二、全部代码

(略)

2.1 ccpuid模块

2.1.1 ccpuid.h: CPUID信息(纯C版)

  全部代码——


2.1.2 ccpuid.hpp: CPUID信息(C++版头文件)

  全部代码——


2.1.3 ccpuid.cpp: CPUID信息(C++版源文件)

  全部代码——


2.2 纯C版测试例程

2.2.1 testccpuidc.c : [C] 测试ccpuid.h, 显示CPUID信息

  全部代码——


2.3 C++版测试例程

2.3.1 testccpuid.cpp : [C++] 测试ccpuid.hpp, 显示所有的CPUID信息

  全部代码——


2.3.2 test2.h: 多文件链接ccpuid的测试(头文件)

  全部代码——


2.3.3 test2.cpp: 多文件链接ccpuid的测试(源文件)

  全部代码——


2.4 makefile

  makefile——


三、编译测试

  在以下编译器中成功编译——
VC6:x86版。
VC2003:x86版。
VC2005:x86版。
VC2010:x86版、x64版。
GCC 4.7.0(Fedora 17 x64):x86版、x64版。
GCC 4.6.2(MinGW (20120426)):x86版。
GCC 4.7.1(TDM-GCC (MinGW-w64)):x86版、x64版。
llvm-gcc-4.2(Mac OS X Lion 10.7.4, Xcode 4.4.1):x86版、x64版。

 

 

参考文献——
《Intel® 64 and IA-32 Architectures Software Developer’s Manual Combined Volumes:1, 2A, 2B, 2C, 3A, 3B, and 3C》044US. August 2012. http://www.intel.com/content/www/us/en/processors/architectures-software-developer-manuals.html
《Intel® Architecture Instruction Set Extensions Programming Reference》014. AUGUST 2012. http://software.intel.com/en-us/avx/
《Intel® Processor Identification and the CPUID Instruction》. May 2012. http://developer.intel.com/content/www/us/en/processors/processor-identification-cpuid-instruction-note.html
《AMD64 Architecture Programmer's Manual Volume 3: General Purpose and System Instructions》. December 2011. http://support.amd.com/us/Processor_TechDocs/24594_APM_v3.pdf
《AMD CPUID Specification》. September 2010. http://support.amd.com/us/Embedded_TechDocs/25481.pdf
《x86 architecture CPUID》. http://www.sandpile.org/x86/cpuid.htm
《Haswell New Instruction Descriptions Now Available! 》. Mark Buxton. http://software.intel.com/en-us/blogs/2011/06/13/haswell-new-instruction-descriptions-now-available/
[IDF2012]ARCS002《Introduction to the upcoming Intel® Advanced Vector Extensions 2 (Intel® AVX2)》. 王有伟, Henry Ou. 2012-4.
[IDF2012]ARCS002《即将推出的英特尔® 高级矢量扩展指令集2(英特尔® AVX2)介绍》. 王有伟, Henry Ou. 2012-4.
《x86/x64 指令系统》. mik(邓志). http://www.mouseos.com/x64/default.html
《[x86]SIMD指令集发展历程表(MMX、SSE、AVX等)》. http://www.cnblogs.com/zyl910/archive/2012/02/26/x86_simd_table.html
《如何在各个版本的VC及64位下使用CPUID指令》. http://www.cnblogs.com/zyl910/archive/2012/05/21/vcgetcpuid.html
《[VC兼容32位和64位] 检查MMX和SSE系列指令集的支持级别》. http://www.cnblogs.com/zyl910/archive/2012/05/25/checksimd64.html
《[VC] CPUIDFIELD:CPUID字段的统一编号、读取方案。范例:检查SSE4A、AES、PCLMULQDQ指令》. http://www.cnblogs.com/zyl910/archive/2012/06/29/getcpuidfield.html
《[VC] 检测AVX系列指令集的支持级别(AVX、AVX2、F16C、FMA、FMA4、XOP)》. http://www.cnblogs.com/zyl910/archive/2012/07/04/checkavx.html
《[C#] cmdarg_ui:“简单参数命令行程序”的通用图形界面》.  http://www.cnblogs.com/zyl910/archive/2012/06/19/cmdarg_ui.html
《[C/C++] 显示各种C/C++编译器的预定义宏(C11标准、C++11标准、VC、BCB、Intel、GCC)》. http://www.cnblogs.com/zyl910/archive/2012/08/02/printmacro.html
《[C] 让VC、BCB支持C99的整数类型(stdint.h、inttypes.h)(兼容GCC)》. http://www.cnblogs.com/zyl910/archive/2012/08/08/c99int.html
《GCC 64位程序的makefile条件编译心得——32位版与64位版、debug版与release版(兼容MinGW、TDM-GCC)》. http://www.cnblogs.com/zyl910/archive/2012/08/14/gcc64_make.html
《[C] 在GCC中获取CPUID信息(兼容VC)》. http://www.cnblogs.com/zyl910/archive/2012/08/06/getcpuid_gcc.html
《ccpuid:CPUID信息模块。范例:显示所有的CPUID信息》. http://www.cnblogs.com/zyl910/archive/2012/07/11/ccpuid.html
《ccpuid:CPUID信息模块 V1.01版,支持GCC(兼容32位或64位的Windows/Linux)》. http://www.cnblogs.com/zyl910/archive/2012/08/22/ccpuid_v101.html
《ccpuid:CPUID信息模块 V1.02版,支持Mac OS X,支持纯C,增加CPUF常数》. http://www.cnblogs.com/zyl910/archive/2012/09/29/ccpuid_v102.html


源码下载——
https://files.cnblogs.com/zyl910/ccpuid_v103.rar

posted on 2012-10-13 00:01  zyl910  阅读(4905)  评论(1编辑  收藏  举报