c++ - 如何在 C++ 中将指针指定为“thread_local”存储?
问题描述
我目前正在优化一段数学运算代码,它在指针存储上循环并将结果保存在原地。我注意到,在每次分配时,编译器都会发出一条内存存储指令,如下所示(vmovaps
需要注意):
114 [1] top_data_c[pc] += w1 * bottom_data_hwc[o1 + pc];
0x55555558bf70 <+ 2624> c4 c1 78 10 0c 02 vmovups (%r10,%rax,1),%xmm1
0x55555558bf76 <+ 2630> 48 83 c1 01 add $0x1,%rcx
0x55555558bf7a <+ 2634> c4 c3 75 18 4c 02 10 01 vinsertf128 $0x1,0x10(%r10,%rax,1),%ymm1,%ymm1
0x55555558bf82 <+ 2642> c4 c2 25 a8 0c 04 vfmadd213ps (%r12,%rax,1),%ymm11,%ymm1
0x55555558bf88 <+ 2648> c4 c1 7c 29 0c 04 vmovaps %ymm1,(%r12,%rax,1)
115 [1] top_data_c[pc] += w2 * bottom_data_hwc[o2 + pc];
0x55555558bf8e <+ 2654> c4 c1 78 10 04 01 vmovups (%r9,%rax,1),%xmm0
0x55555558bf94 <+ 2660> c4 c3 7d 18 44 01 10 01 vinsertf128 $0x1,0x10(%r9,%rax,1),%ymm0,%ymm0
0x55555558bf9c <+ 2668> c4 c2 7d b8 ca vfmadd231ps %ymm10,%ymm0,%ymm1
0x55555558bfa1 <+ 2673> c4 c1 7c 29 0c 04 vmovaps %ymm1,(%r12,%rax,1)
116 [1] top_data_c[pc] += w3 * bottom_data_hwc[o3 + pc];
0x55555558bfa7 <+ 2679> c4 c1 78 10 04 00 vmovups (%r8,%rax,1),%xmm0
0x55555558bfad <+ 2685> c4 c3 7d 18 44 00 10 01 vinsertf128 $0x1,0x10(%r8,%rax,1),%ymm0,%ymm0
0x55555558bfb5 <+ 2693> c4 c2 75 98 c1 vfmadd132ps %ymm9,%ymm1,%ymm0
0x55555558bfba <+ 2698> c4 c1 7c 29 04 04 vmovaps %ymm0,(%r12,%rax,1)
117 [1] top_data_c[pc] += w4 * bottom_data_hwc[o4 + pc];
0x55555558bfc0 <+ 2704> c5 f8 10 0c 07 vmovups (%rdi,%rax,1),%xmm1
0x55555558bfc5 <+ 2709> c4 e3 75 18 4c 07 10 01 vinsertf128 $0x1,0x10(%rdi,%rax,1),%ymm1,%ymm1
0x55555558bfcd <+ 2717> c4 c2 75 b8 c0 vfmadd231ps %ymm8,%ymm1,%ymm0
0x55555558bfd2 <+ 2722> c4 c1 7c 29 04 04 vmovaps %ymm0,(%r12,%rax,1)
0x55555558bfd8 <+ 2728> 48 83 c0 20 add $0x20,%rax
0x55555558bfdc <+ 2732> 48 39 4d c0 cmp %rcx,-0x40(%rbp)
0x55555558bfe0 <+ 2736> 77 8e ja 0x55555558bf70
但是,当我将指针更改为本地“堆栈数组”变量时,即T top_data_c[1024]
存储指令仅出现在循环结束时:
114 [1] top_data_c[pc] += w1 * bottom_data_hwc[o1 + pc];
0x55555558bbe0 <+ 1712> c5 f8 10 0c 03 vmovups (%rbx,%rax,1),%xmm1
0x55555558bbe5 <+ 1717> 48 83 c1 01 add $0x1,%rcx
0x55555558bbe9 <+ 1721> c4 e3 75 18 4c 03 10 01 vinsertf128 $0x1,0x10(%rbx,%rax,1),%ymm1,%ymm1
0x55555558bbf1 <+ 1729> c4 c2 25 a8 0c 04 vfmadd213ps (%r12,%rax,1),%ymm11,%ymm1
0x55555558bbf7 <+ 1735> c5 fc 28 c1 vmovaps %ymm1,%ymm0
115 [1] top_data_c[pc] += w2 * bottom_data_hwc[o2 + pc];
0x55555558bbfb <+ 1739> c4 c1 78 10 0c 03 vmovups (%r11,%rax,1),%xmm1
0x55555558bc01 <+ 1745> c4 c3 75 18 4c 03 10 01 vinsertf128 $0x1,0x10(%r11,%rax,1),%ymm1,%ymm1
0x55555558bc09 <+ 1753> c4 c2 7d 98 ca vfmadd132ps %ymm10,%ymm0,%ymm1
116 [1] top_data_c[pc] += w3 * bottom_data_hwc[o3 + pc];
0x55555558bc0e <+ 1758> c4 c1 78 10 04 02 vmovups (%r10,%rax,1),%xmm0
0x55555558bc14 <+ 1764> c4 c3 7d 18 44 02 10 01 vinsertf128 $0x1,0x10(%r10,%rax,1),%ymm0,%ymm0
0x55555558bc1c <+ 1772> c4 e2 35 b8 c8 vfmadd231ps %ymm0,%ymm9,%ymm1
117 [1] top_data_c[pc] += w4 * bottom_data_hwc[o4 + pc];
0x55555558bc21 <+ 1777> c4 c1 78 10 04 01 vmovups (%r9,%rax,1),%xmm0
0x55555558bc27 <+ 1783> c4 c3 7d 18 44 01 10 01 vinsertf128 $0x1,0x10(%r9,%rax,1),%ymm0,%ymm0
0x55555558bc2f <+ 1791> c4 c2 75 98 c0 vfmadd132ps %ymm8,%ymm1,%ymm0
0x55555558bc34 <+ 1796> c4 c1 7c 29 04 04 vmovaps %ymm0,(%r12,%rax,1)
0x55555558bc3a <+ 1802> 48 83 c0 20 add $0x20,%rax
0x55555558bc3e <+ 1806> 48 3b 8d c8 fb ff ff cmp -0x438(%rbp),%rcx
0x55555558bc45 <+ 1813> 72 99 jb 0x55555558bbe0
由于其线程不安全,编译器使指针存储操作远离优化。
在这样的实现中声明堆栈数组或临时变量复制看起来很脏,有没有办法让这种指针存储对编译器来说是线程安全的?当然,这样的计算是完全线程安全的(它的工作原理与 GPU 非常相似)。
解决方案
除非您出于某种原因将top_data_c
指针声明为volatile
,否则编译器可以自由假设它写入的数据没有发生外部(包括并发)修改,因此缺乏优化不应归咎于线程(也不能通过线程局部语义来解决) )。
这里真正的问题是别名 - 编译器不能假设它top_data_c[pc]
不在内部的某个地方bottom_data_hwc
(它可能与 相同bottom_data_hwc[o4 + pc]
,谁知道?),所以它必须进行存储。在“堆栈数组”情况下缓解这个问题的是(如果我没记错的话)top_data_c
是数组而不是指针的事实(不是它在堆栈上或它是线程本地的)。
对严格别名规则的简短介绍:编译器可以假定指向不兼容类型(例如int
and double
)的指针不能指向相同的位置。如果你在里面void foo(int* x, double* y)
,那么在严格的别名规则下,写入x
不能改变你从中读取的内容y
(反之亦然),所以编译器可以自由地重新排序或删除存储,并根据需要在这个函数中读取x
和读取。y
但在里面void foo(double* x, double* y)
,这个保证就消失了。如果x == y
(或x == y + n
)然后写入和读取x
并y
可能相互交互,编译器别无选择,只能执行每个加载/存储。
您应该查看restrict
-related 关键字,该关键字向编译器发出信号“x
仅通过此x
函数中的(及其副本)进行访问”。一些材料:
https://cellperformance.beyond3d.com/articles/2006/05/demystifying-the-restrict-keyword.html
http://assemblyrequired.crashworks.org/load-hit-stores-and-the-__restrict-keyword/
推荐阅读
- c# - c# - 添加附件到 Discord Webhook 消息
- sql - 在 biq 查询中将字符串转换为表列
- ios - 我可以将输出的一个 GLSL 着色器帧用作下一帧的输入吗?
- amazon-dynamodb - Dynamo 中如何处理提示切换
- python - 我想使用 python KIVY 在 Button 上添加文本
- javascript - canload guard Angular2+ 中的订阅
- emacs - 如何调用 elisp 命令从自定义命令中编辑当前缓冲区
- c# - 如何在 c# 中为以下代码编写 nunit 测试用例?如何为没有参数的方法编写 nunit 测试用例?
- css - 使用部分透明的 rgba() 值时,如何消除框阴影和 div 背景之间的间隙
- c# - 从 16 位获取颜色