在 c++ 中使用 x86 DIV 的这个 asm 块有啥用?
Posted
技术标签:
【中文标题】在 c++ 中使用 x86 DIV 的这个 asm 块有啥用?【英文标题】:What is the use of this asm block using x86 DIV in c++?在 c++ 中使用 x86 DIV 的这个 asm 块有什么用? 【发布时间】:2018-10-03 20:33:10 【问题描述】:有人可以帮助我了解使用 asm 块进行 unsigned long long
乘法在性能方面的好处。它与竞争性编程优化有关。我猜它使乘法更快,但我实际上无法理解代码。
const int md = 998244353;
inline int mul(int a, int b)
#if !defined(_WIN32) || defined(_WIN64)
return (int) ((long long) a * b % md);
#endif
unsigned long long x = (long long) a * b;
unsigned xh = (unsigned) (x >> 32), xl = (unsigned) x, d, m;
asm(
"divl %4; \n\t"
: "=a" (d), "=d" (m)
: "d" (xh), "a" (xl), "r" (md)
);
return m;
【问题讨论】:
不管它是如何工作的,这段代码在使用现代编译器时很可能已经过时了,因为它们应该从原始版本生成更好或可比较的输出。 专业提示:编译器是由很多真正聪明的人编写的。如果你给他们明显的代码,他们会很好地为你提供最好的机器代码。这意味着可理解的代码通常会变成高性能代码。 FWIW,GCC 8.2 is able to boil it down to justimul
calls, while Clang still uses idiv
.
@BartekBanachewicz:那是因为您禁用了优化!有关来自 clang 和 gcc -O3
的更好代码,请参阅 godbolt.org/z/qtQkxc。所有现代编译器都知道如何使用乘法逆进行除以编译时常数,但 gcc 是唯一一个即使在 -O0
, IIRC 也这样做的编译器。 Why does GCC use multiplication by a strange number in implementing integer division?
@NathanOliver:如果那是真的!编译器通常会生成好的代码,但几乎不可能是最好的。
【参考方案1】:
此代码实际上是 32 位的加速(其中 64x64 => 128 乘法不可用,因此编译器使用实际除法,但在 64 位上损失严重,其中编译器使用乘法逆来完全避免缓慢的硬件除法.Why does GCC use multiplication by a strange number in implementing integer division?
此外,它应该真正使用__builtin_constant_p
仅在内联和常量传播后任一输入不是编译时常量时才使用内联汇编。
但无论如何,x86's div
instruction 确实是EDX:EAX / (src)
=> 商(EAX)和除数(EDX)。见When and why do we sign extend and use cdq with mul/div?
"a"
和 "d"
约束分别要求 EAX 和 EDX 中 64 位乘积的低半部分和高半部分作为输入。
来自Godbolt compiler explorer:
const int md = 998244353;
int mul(int a, int b)
#ifdef __x86_64__ // FIXME: just use the asm if defined(i386) to exclude all others
return (int) ((long long) a * b % md);
#else
if(__builtin_constant_p(a) && __builtin_constant_p(b))
return (int) ((long long) a * b % md);
// clang's builtin_constant_p is evaled before inlining, derp
unsigned long long x = (long long) a * b;
unsigned xh = (unsigned) (x >> 32), xl = (unsigned) x, d, m;
asm(
"divl %4; \n\t"
: "=a" (d), "=d" (m)
: "d" (xh), "a" (xl), "r" (md)
);
return m;
#endif
int main()
return mul(1,2);
使用gcc8.2 -O3 -m32
编译如下:
mul(int, int):
mov eax, DWORD PTR [esp+8]
mov ecx, 998244353
imul DWORD PTR [esp+4] # one-operand imul does EDX:EAX = EAX * src
divl ecx; # EDX:EAX / ecx => EAX and EDX
mov eax, edx # return the remainder
ret
main:
mov eax, 2 # builtin_constant_p used the pure C, allowing constant-propagation
ret
注意 div
是 unsigned 除法,所以这与 C 不匹配。C 正在执行有符号乘法和有符号除法。 这可能应该使用idiv
,或将输入转换为无符号。或者,也许出于某种奇怪的原因,他们确实想要带有负输入的奇怪结果。
那么为什么编译器不能在没有内联汇编的情况下自己发出这个?因为如果商溢出目标寄存器 (al/ax/eax/rax),它会出现 #DE(除法异常)错误,而不是像所有其他整数指令一样静默截断。
64 位 / 32 位 => 32 位除法只有在您知道除数足够大以用于可能的除数时才是安全的。 (但即使是这样,gcc 仍然不知道寻找这种优化。例如,如果 a
是 32,如果使用单个 mul
和 div
完成,a * 7ULL / 9
不可能导致 #DE -bit 类型。但 gcc 仍会发出对 libgcc 辅助函数的调用。)
【讨论】:
很好的答案,如果可以的话,我会给它两票。 感谢@PaulSanders 的出色解释。当我开始学习 asm 时,你能推荐我一些好的资源吗? 回答问题的是彼得!你只需要谷歌一下,抱歉,我没有具体的建议。 @JohnMcLane: ***.com/tags/x86/info 有指南和手册的链接。以上是关于在 c++ 中使用 x86 DIV 的这个 asm 块有啥用?的主要内容,如果未能解决你的问题,请参考以下文章