技术头条 - 一个快速在微博传播文章的方式     搜索本站
您现在的位置首页 --> 系统架构 --> 多核环境下编写程序需注意Cache

多核环境下编写程序需注意Cache

浏览:2086次  出处信息

前 阵子接触到一道关于数组内部链表(多用于内存池技术)的数据结构的题, 这种数据结构能够比普通链表在cache中更容易命中, 理由很简单, 就是因为其在地址上是连续的(=.=!), 借这个机会, 就对cpu cache进行了一个研究, 今天做一个简单的分享, 首先先来普及一下cpu cache的知识, 这里的cache是指cpu的高速缓存. 在我们程序员看来, 缓存是一个透明部件. 因此, 程序员通常无法直接干预对缓存的操作. 但是, 确实可以根据缓存的特点对程序代码实施特定优化, 从而更好地利用高速缓存. 

高速缓存的置换策略会尽可能地将访问频繁的数据放入cache中, 这是一个动态的过程, 所以cache中的数据不会一直不变. 目前一般的机器的cpu cache可分为一级缓存和二级缓存. 一级缓存更靠近cpu, 速度比二级缓存更快. 二级缓存比一级缓存速度更慢, 容量更大, 主要就是做一级缓存和内存之间数据临时交换的地方用.
这两者和RAM在空间和效率上的关系如下:
L1 Cache ―> L2 Cache ―> RAM
――――> 容量递增 ――――>
――――> 速度递减 ――――>
―-> CPU访问优先级递减 ―->

在linux系统中, 我们可以使用cat /proc/cpuinfo 来获知机器的cpu和核数.
而cpu cache的信息, 我们通过dmesg | grep cache来获知.
例如:
CPU: L1 I Cache: 64K (64 bytes/line), D cache 64K (64 bytes/line)
CPU: L1 I Cache: 64K (64 bytes/line), D cache 64K (64 bytes/line)
说明我这台机器有两个处理器, 并只有一级缓存, 大小为 64K, 缓存行/快 大小为64 bytes.
由 于不同的处理器之间都具有自己的高速缓存, 所以当两个cpu的cache中都存有数据a, 那么就有可能需要进行同步数据, 而cache之间同步数据的最小单元为cache行大小, 可以把一个cache想象成一张表, 表的每一行都是64bytes(假设), 当cpu被告知cache第一行的第一个byte为脏数据时, cpu会将第一行都进行同步.
例如以下场景:

CPU1读取了数据a(假设a小于cache行大小),并存入CPU1的高速缓存.

CPU2也读取了数据a,并存入CPU2的高速缓存.

CPU1修改了数据a, a被放回CPU1的高速缓存行. 但是该信息并没有被写入RAM.

CPU2访问a, 但由于CPU1并未将数据写入RAM, 导致了数据不同步.

为 了解决这个问题, 芯片设计者制定了一个规则. 当一个CPU修改高速缓存行中的字节时, 计算机中的其它CPU会被通知, 它们的高速缓存将视为无效. 于是, 在上面的情况下, CPU2发现自己的高速缓存中数据已无效, CPU1将立即把自己的数据写回RAM, 然后CPU2重新读取该数据. 这样就完成了一次两个cpu之间cache的同步.

为了测试上述场景, 我编写了如下程序进行测试:
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
#include<sys/types.h>
#include<sys/sysinfo.h>
#include <sys/time.h>
#include<unistd.h>
#include <stdlib.h>
#include <stdio.h>
#include <pthread.h>
 
#include <iostream>
using namespace std;
 
#define ENABLE_WHCIH_CPU
#define ENABLE_SET_CPU
 
#define EXEC_COUNT (100 * 1000 * 1000)
 
struct bits_t
{
 int a;
    char placeholder[64];
 int b;
};
 
struct bits_t bits;
 
int which_cpu(const char* prefix_)
{
    #ifdef ENABLE_WHCIH_CPU
 cpu_set_t cur_cpu;
 CPU_ZERO(&cur_cpu);
 if (sched_getaffinity(0, sizeof(cur_cpu), &cur_cpu) == -1)
 {
  printf("warning: cound not get cpu affinity, continuing...\n");
  return -1;
 }
 int num = sysconf(_SC_NPROCESSORS_CONF);
 for (int i = 0; i < num; i++)
 {
  if (CPU_ISSET(i, &cur_cpu))
  {
   printf("[%s] this process %d is running processor : %d\n", prefix_, getpid(), i);
  }
 }
    #endif
 
 return 0;
}
 
int set_cpu(int cpu_id_)
{
    #ifdef ENABLE_SET_CPU
 cpu_set_t mask;
 CPU_ZERO(&mask);
 CPU_SET(cpu_id_, &mask);
 if (sched_setaffinity(0, sizeof(mask), &mask) == -1)
 {
  printf("warning: could not set CPU affinity, continuing...\n");
  return -1;
    }
    #endif
 
 return 0;
}
 
void* thd_func1(void* arg_)
{
 set_cpu(0);
 which_cpu("thread 1 start");
    timeval begin_tv;
    gettimeofday(&begin_tv, NULL);
 
    for (int i = 0; i < EXEC_COUNT; i++)
    {
        bits.a += 1;
        int a = bits.a;
    }
 
    timeval end_tv;
    gettimeofday(&end_tv, NULL);
    printf("thd1 perf:[%lu]us\n", (end_tv.tv_sec * 1000 * 1000 + end_tv.tv_usec) - (begin_tv.tv_sec * 1000 * 1000 + begin_tv.tv_usec));
 which_cpu("thread 1 end");
 
    return NULL;
}
 
void* thd_func2(void* arg_)
{
 set_cpu(1);
 which_cpu("thread 2 start");
    timeval begin_tv;
    gettimeofday(&begin_tv, NULL);
 
    for (int i = 0; i < EXEC_COUNT; i++)
    {
        bits.b += 2;
        int b = bits.b;
    }
 
    timeval end_tv;
    gettimeofday(&end_tv, NULL);
    printf("thd2 perf:[%lu]us\n", (end_tv.tv_sec * 1000 * 1000 + end_tv.tv_usec) - (begin_tv.tv_sec * 1000 * 1000 + begin_tv.tv_usec));
 which_cpu("thread 2 end");
 
    return NULL;
}
 
int main(int argc_, char* argv_[])
{
 int num = sysconf(_SC_NPROCESSORS_CONF);
 printf("system has %d processor(s).\n", num);
 cpu_set_t cpu_mask;
 cpu_set_t cur_cpu_info;
 
 memset((void*)&bits, 0, sizeof(bits_t));
 set_cpu(0);
 which_cpu("main thread");
 
    pthread_t pid1;
    pthread_create(&pid1, NULL, thd_func1, NULL);
 
    pthread_t pid2;
    pthread_create(&pid2, NULL, thd_func2, NULL);
 
    pthread_join(pid1, NULL);
    pthread_join(pid2, NULL);
 
    return 0;
}
该程序中会创建两个线程, 分别对全局变量bits的a和b成员进行1亿次加法操作.
在这里我分别针对四种情况进行了测试 -
1. 两个线程分别跑在不同的cpu上, bits_t结构体没有placeholder这64个填充字节.
2. 两个线程分别跑在不同的cpu上, bits_t结构体有placeholder这64个填充字节.
3. 两个线程分别跑在相同的cpu上, bits_t结构体没有placeholder这64个填充字节.
4. 两个线程分别跑在相同的cpu上, bits_t结构体有placeholder这64个填充字节.
程序可以通过set_cpu函数来将线程绑定到指定的cpu上去.
为了大家阅读的方便, 我已将测试结果报告整理成以下四个表格.
情况一测试结果:
 线程id  CPU绑定  有无placeholder  平均耗时(微妙)
 1  cpu0  无  2186931
 2  cpu1  无  2033496

   

情况二测试结果:

线程id  CPU绑定  有无placeholder  平均耗时(微妙)
 1  cpu0  有  402144
 2  cpu1  有  392745

   

我 们先来看情况一和情况二的结果对比, 显然, 后者要比前者效率高得多的多, 可以验证在有 placeholder填充字节之后, bit_t的a和b域被划分到了cache的不同两行, 所以当在cpu0执行的线程1修改a后, cpu1在读b时, 不需要去同步cache. 而情况一因为a和b在cache中的同一行, 导致两个cpu要互相进行大量的cache行同步.

情况三测试结果:
线程id  CPU绑定  有无placeholder  平均耗时(微妙)
 1  cpu0  无  716056
 2  cpu0  无  686804

   

情况四测试结果:
线程id  CPU绑定  有无placeholder  平均耗时(微妙)
 1  cpu0  有  761421
 2  cpu0  有  884969


可以看出, 情况三和四, 因为两个线程运行在同一个cpu上, 有和没有placeholder填充字节在性能上几乎没有什么区别, 因为不存在cache之间行同步的问题, 但是由于是一个cpu在调度切换两个线程, 所以要比情况一慢一点.

从上面测试结果看来, 某些特定情况下, 对于cache的优化还是很重要的, 但是也不能一味地为了追求性能都将所有共享数据加入填充字节, 毕竟cache就那么大, 如果不是某些特定的读写非常频繁的场景下, 没有必要这么做.


PS: 由于不同的硬件架构体系之间会有差别, 例如某些硬件架构同一个cpu下的两个物理核之间共享cache, 所以测试时要试具体环境而定.

建议继续学习:

  1. Buffer和cache的区别是什么?    (阅读:6823)
  2. 谈冷热数据    (阅读:5731)
  3. Linux操作系统中内存buffer和cache的区别    (阅读:5305)
  4. 学习:一个并发的Cache    (阅读:4967)
  5. 关于Linux的文件系统cache    (阅读:4773)
  6. Twitter架构图(cache篇)    (阅读:4719)
  7. 7个示例科普CPU Cache    (阅读:4111)
  8. 详解MyISAM Key Cache(前篇)    (阅读:4064)
  9. [squid] 过期时间在 60 秒内 squid 不 Cache 的问题    (阅读:3960)
  10. 多核与异步并行    (阅读:3862)
QQ技术交流群:445447336,欢迎加入!
扫一扫订阅我的微信号:IT技术博客大学习
© 2009 - 2024 by blogread.cn 微博:@IT技术博客大学习

京ICP备15002552号-1