在 C 中设置 CPU 亲和性的问题

Question

我正在尝试计算 CPUID 指令运行所需的时间。

来源：

#include <stdio.h>
#include <sched.h>

cpu_set_t mask;

CPU_ZERO(&mask);
CPU_SET(0, &mask);
sched_setaffinity(0, sizeof(mask), &mask);

static inline unsigned long long tick()
{
    unsigned long long d;
    asm volatile ("rdtsc" : "=A" (d));
    return d;
}

void cpuid(void)
{
    int i;

    for(i=0; i != 5; i++)
    {
         asm volatile ("cpuid");
    }
}

int main()
{
    long long bef;
    long long aft;
    long long dif;

    bef=tick();
    cpuid();
    aft=tick();
    dif=aft-bef;

    printf("%d\n", bef);
    printf("%d\n", aft);
    printF("%d\n", dif);

    return 0;
}

现在我正在使用以下代码进行编译

gcc -D_GNU_SOURCE -o test test.c

我在不是文件的代码上遇到错误！ 例如：

test.c:6:1: error: expected identifier or '(' before 'do'
test.c:6:1: error: expected identifier or '(' before 'while'
test.c:7:1: error: expected identifier or '(' before '__extension__'
test.c:8:1: warning: data definition has no type or storage class [enable by def...
test.c:8:1: error: intializer element is not constant

“def...”实际上不是输出，因为我的终端窗口很小。我在 ESXi 工作。

任何帮助都会很棒！！！

面向未来的读者

用户@Iwillnotexist Idonotexist 说使用以下函数获得完整的x86 和x64 支持是正确的。

static __inline__ unsigned long long rdtsc(void)
{
  unsigned hi, lo;
  __asm__ __volatile__ ("rdtsc" : "=a"(lo), "=d"(hi));
  return ( (unsigned long long)lo)|( ((unsigned long long)hi)<<32 );
}

Answer 1

您有函数之外的代码。这是不允许的。将以下内容移至main：

CPU_ZERO(&mask);
CPU_SET(0, &mask);
sched_setaffinity(0, sizeof(mask), &mask);

Answer 2

这些指令应该在 main() 中：

cpu_set_t mask;
CPU_ZERO(&mask);
CPU_SET(0, &mask);
sched_setaffinity(0, sizeof(mask), &mask);

这应该可以修复编译错误。

那么，cpuid() 的五次迭代太少，无法给出有意义的结果。

您可以检查这个answer，但使用两个不同长度的序列，仅使用 CPUID 指令。您需要更长的周期，但不会太长以至于内存获取开始发挥作用。

我已经运行了一些测试，TEST 定义在 5 到 1000 之间； CPU 亲和力似乎不会影响四核上的结果：

#include <stdio.h>
#include <sched.h>

static inline unsigned long long tick() {
    unsigned long long d;
    asm volatile ("rdtsc" : "=A" (d));
    return d;
}

static inline void cpuid(void) {
    int i;
    for(i=0; i != TEST; i++) {
         asm volatile ("cpuid");
    }
}

int main()
{
    long long bef, aft, dif;

    bef=tick();
    cpuid();
    aft=tick();
    dif=(aft-bef)/TEST;

    printf("%lld\n", dif);

    return 0;
}

gcc -o0 -DTEST=100 -D_GNU_SOURCE -W -Wall -o time time.c && ./time