Updated SM3性能优化 (markdown)

Sun Yimin 2021-03-10 17:17:12 +08:00
parent 80dd222342
commit 22d8920487

@ -1,20 +1,23 @@
最近对SM3的性能进行了优化主要包括 最近对SM3的性能进行了优化主要包括
* 纯golang版本减少WW'预计算;避免使用条件判断。 * 纯golang版本减少WW'预计算;避免使用条件判断。
* 三月初实现了针对AMD64架构的ASM版本。 * 三月初实现了针对AMD64架构的ASM版本。
下一步看看能否实现使用AVX2指令集进行优化。 下一步看看能否实现使用AVX2指令集进行优化。
# SHA256 SIMD Execution
下面是SHA256 message scheduler SIMD实现四轮算出下一个4个DWORDs. 下面是SHA256 message scheduler SIMD实现四轮算出下一个4个DWORDs.
// Wt = SIGMA1(Wt-2) + Wt-7 + SIGMA0(Wt-15) + Wt-16; for 16 <= t <= 63 // Wt = SIGMA1(Wt-2) + Wt-7 + SIGMA0(Wt-15) + Wt-16; for 16 <= t <= 63
// SIGMA0(x) = ROTR(7,x) XOR ROTR(18,x) XOR SHR(3,x) // SIGMA0(x) = ROTR(7,x) XOR ROTR(18,x) XOR SHR(3,x)
// SIGMA1(x) = ROTR(17,x) XOR ROTR(19,x) XOR SHR(10,x) // SIGMA1(x) = ROTR(17,x) XOR ROTR(19,x) XOR SHR(10,x)
// Transpose data into high/low parts // Transpose data into high/low parts
VPERM2I128 $0x20, XTMP2, XTMP0, XDWORD0 // w3, w2, w1, w0 VPERM2I128 $0x20, XTMP2, XTMP0, XDWORD0 // w3, w2, w1, w0
VPERM2I128 $0x31, XTMP2, XTMP0, XDWORD1 // w7, w6, w5, w4 VPERM2I128 $0x31, XTMP2, XTMP0, XDWORD1 // w7, w6, w5, w4
VPERM2I128 $0x20, XTMP3, XTMP1, XDWORD2 // w11, w10, w9, w8 VPERM2I128 $0x20, XTMP3, XTMP1, XDWORD2 // w11, w10, w9, w8
VPERM2I128 $0x31, XTMP3, XTMP1, XDWORD3 // w15, w14, w13, w12 VPERM2I128 $0x31, XTMP3, XTMP1, XDWORD3 // w15, w14, w13, w12
VPALIGNR $4, XDWORD2, XDWORD3, XTMP0; \ // XTMP0 = W[-7] = {w12,w11,w10,w9} VPALIGNR $4, XDWORD2, XDWORD3, XTMP0; \ // XTMP0 = W[-7] = {w12,w11,w10,w9}
@ -51,20 +54,23 @@
VPADDD XTMP0, XTMP5, XDWORD0; \ // XDWORD0 = {W[3], W[2], W[1], W[0]} VPADDD XTMP0, XTMP5, XDWORD0; \ // XDWORD0 = {W[3], W[2], W[1], W[0]}
SM3的message scheduler有两个显著差别 SM3的message scheduler有两个显著差别
1. Wi+3依赖Wi所以也和SHA256类似不能一次生成四个DWORDs但和SHA256不完全一样。 1. Wi+3依赖Wi所以也和SHA256类似不能一次生成四个DWORDs但和SHA256不完全一样。
1. 比SHA256需要多算4个DWORDs。 1. 比SHA256需要多算4个DWORDs。
Intel 指令参考: Intel 指令参考:
https://software.intel.com/sites/landingpage/IntrinsicsGuide/ https://software.intel.com/sites/landingpage/IntrinsicsGuide/
SM3的第一版比SHA256复杂 # SM3 SIMD Execution
SM3的第一版比SHA256复杂不知道有没有继续优化的空间。
// Wj ← P1(Wj16 ⊕ Wj9 ⊕ (Wj3 ≪ 15)) ⊕ (Wj13 ≪ 7) ⊕ Wj6 // Wj ← P1(Wj16 ⊕ Wj9 ⊕ (Wj3 ≪ 15)) ⊕ (Wj13 ≪ 7) ⊕ Wj6
// Transpose data into high/low parts // Transpose data into high/low parts
VPERM2I128 $0x20, XTMP2, XTMP0, XDWORD0 // w3, w2, w1, w0 VPERM2I128 $0x20, XTMP2, XTMP0, XDWORD0 // w3, w2, w1, w0
VPERM2I128 $0x31, XTMP2, XTMP0, XDWORD1 // w7, w6, w5, w4 VPERM2I128 $0x31, XTMP2, XTMP0, XDWORD1 // w7, w6, w5, w4
VPERM2I128 $0x20, XTMP3, XTMP1, XDWORD2 // w11, w10, w9, w8 VPERM2I128 $0x20, XTMP3, XTMP1, XDWORD2 // w11, w10, w9, w8
VPERM2I128 $0x31, XTMP3, XTMP1, XDWORD3 // w15, w14, w13, w12 VPERM2I128 $0x31, XTMP3, XTMP1, XDWORD3 // w15, w14, w13, w12
VPALIGNR $12, XDWORD0, XDWORD1, XTMP0; \ // XTMP0 = W[-13] = {w6,w5,w4,w3} VPALIGNR $12, XDWORD0, XDWORD1, XTMP0; \ // XTMP0 = W[-13] = {w6,w5,w4,w3}
VPSLLD $7, XTMP0, XTMP1; \ VPSLLD $7, XTMP0, XTMP1; \
@ -107,3 +113,47 @@ SM3的第一版比SHA256复杂
VPALIGNR $8, XTMP1, XTMP2, XTMP3; \ // XTMP3 = {W[1], W[0], W[3], W[2]} VPALIGNR $8, XTMP1, XTMP2, XTMP3; \ // XTMP3 = {W[1], W[0], W[3], W[2]}
VPSHUFD $0x4E, XTMP3, XDWORD0; \ // XDWORD0 = {W[3], W[2], W[1], W[0]} VPSHUFD $0x4E, XTMP3, XDWORD0; \ // XDWORD0 = {W[3], W[2], W[1], W[0]}
由于要算**52**个DWORDs所以
| Ingteger Execution | SIMD Execution |
| ---------------------- | ---------------------------- |
| Execute Rounds 0...15 | Compute Message DWORDs 16…31 |
| Execute Rounds 16...31 | Compute Message DWORDs 32…47 |
| Execute Rounds 32...47 | Compute Message DWORDs 48…63 |
| Execute Rounds 48...63 | Compute Message DWORDs 64…67 |
# 其它SIMD Execution
SHA 256的64轮压缩计算中对Kt + Wt 进行了并行计算
VPADDD 0*32(TBL)(SRND*1), XDWORD0, XFER
VMOVDQU XFER, (_XFER + 0*32)(SP)(SRND*1)
SM3可以对Wt XOR Wt+4进行并行计算
VPXOR XDWORD0, XDWORD1, XFER
VMOVDQU XFER, (_XFER + 0*32)(SP)(SRND*1)
但其依然需要Wt, 如何传入和使用Wt呢难道把Wt和Wt'一起放到栈譬如W0,W1,W2,W3,W0',W1',W2',W3'
VMOVDQU XDWORD0, (_XFER + 0*32)(SP)(SRND*1)
VPXOR XDWORD0, XDWORD1, XFER
VMOVDQU XFER, (_XFER + 1*32)(SP)(SRND*1)
没有大改变SHA 256 AVX2的主流程但是栈的处理需要改变。
# T常量处理
这个比较麻烦要是作为参数传入那就没法和SHA 256一样循环处理必须把所有压缩轮都列出来。
# RORXL
这是循环右移SM3使用循环左移没看到有ROLXL这样的指令应该能传入负数来实现
实在不行只能用MOVL + ROLL来处理但性能就没啥提高了。
# 主流程
如果只有(只剩)一个block (64字节)则只处理这个block否则2个block(128字节)处理两个block并行处理message scheduler。