numactl和move_pages不匹配

时间:2019-01-18 12:41:32

标签: linux openmp numa numactl

我基于this question开发了一个简单的程序来测试页面在哪个NUMA节点中。

问题在于,将我的程序结果与Xeon E5-2698 v4(两个NUMA节点)上的numactl -H进行比较时会显示不同的输出。 numactl -H节目(已裁剪):

available: 2 nodes (0-1)
node 0 cpus: 0 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59
node 1 cpus: 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79

例如,numactl说CPU 20在节点1上。我有以下代码:

#include <unistd.h>
#include <sys/mman.h>
#include <stdlib.h>
#include <stdio.h>
#include <numaif.h>
#include <omp.h>

int numa_node(void *ptr) {
  int status;
  int ret_code;
  if((ret_code = move_pages(0, 1, &ptr, NULL, &status, 0)) == -1) {
    perror("move_pages");
    return -1;
  }
  return status;
}

int main(int argc, char* argv[]) {
  int pgsize = getpagesize();
  printf("NUMA test(pgsize=%d)\n",pgsize);
  #pragma omp parallel firstprivate(pgsize)
  {
    if(omp_get_thread_num() == 20) {
      char *m = aligned_alloc(pgsize, pgsize);
      m[0] = 'a';
      if(mlock(m, 10) == -1) {
        perror("mlock");
      }
      else {
        int node = numa_node(m);
        printf("thread %d: node %d\n",20,node);
      }
    }
  }
}

我正在使用aligned_alloc尝试分配仅对齐的页面,以便 当该线程“触摸”此页面时,它将被映射到该线程所在的NUMA节点(第一个触摸策略)。然后,我使用mlock,您可以在this question中检入。我想我是第一次触摸,因为我没有对此进行任何修改,但是,我不知道如何检查这一点,以确保。

我正在使用icc -fopenmp -lnuma进行编译,并使用KMP_AFFINITY=granularity=fine,compactOMP_NUM_THREADS=80numactl -m 0,1 ./numa。我使用这种亲缘关系,因为我认为它执行与numactl所看到的系统相同的分配。输出:

NUMA test(pgsize=4096)
thread 20: node 0

因此,该程序说线程20在节点0处,而numactl说线程20在节点1处。为什么?我期待在两者上看到相同的输出。

1 个答案:

答案 0 :(得分:1)

请求的相似性设置compact将连续的线程放置到同一内核上的相邻硬件线程(超线程)中。操作系统cpus的编号稍后会为其他硬件线程编号,因此cpu 0和40在同一内核上。映射如下:

tid -> cpu
0 -> 0
1 -> 40
2 -> 1
3 -> 41
...
20 -> 10

通过将,verbose添加到KMP_AFFINITY可以看到。如果要直接映射,可以使用GOMP_CPU_AFFINITY=0-79代替KMP_AFFINITY设置。这样就可以达到在正确的NUMA节点上获取内存的目的。