nb_kernel102_ia32_sse.intel_syntax.s
来自「最著名最快的分子模拟软件」· S 代码 · 共 2,198 行 · 第 1/5 页
S
2,198 行
mov ecx, [edx + 8] mov edx, [edx + 12] ;# eax-edx=jnr1-4 add dword ptr [esp + nb102nf_innerjjnr], 16 ;# advance pointer (unrolled 4) mov esi, [ebp + nb102nf_pos] ;# base of pos[] lea eax, [eax + eax*2] ;# replace jnr with j3 lea ebx, [ebx + ebx*2] lea ecx, [ecx + ecx*2] ;# replace jnr with j3 lea edx, [edx + edx*2] ;# move j coordinates to local temp variables movlps xmm2, [esi + eax*4] movlps xmm3, [esi + eax*4 + 12] movlps xmm4, [esi + eax*4 + 24] movlps xmm5, [esi + ebx*4] movlps xmm6, [esi + ebx*4 + 12] movlps xmm7, [esi + ebx*4 + 24] movhps xmm2, [esi + ecx*4] movhps xmm3, [esi + ecx*4 + 12] movhps xmm4, [esi + ecx*4 + 24] movhps xmm5, [esi + edx*4] movhps xmm6, [esi + edx*4 + 12] movhps xmm7, [esi + edx*4 + 24] ;# current state: ;# xmm2= jxOa jyOa jxOc jyOc ;# xmm3= jxH1a jyH1a jxH1c jyH1c ;# xmm4= jxH2a jyH2a jxH2c jyH2c ;# xmm5= jxOb jyOb jxOd jyOd ;# xmm6= jxH1b jyH1b jxH1d jyH1d ;# xmm7= jxH2b jyH2b jxH2d jyH2d movaps xmm0, xmm2 movaps xmm1, xmm3 unpcklps xmm0, xmm5 ;# xmm0= jxOa jxOb jyOa jyOb unpcklps xmm1, xmm6 ;# xmm1= jxH1a jxH1b jyH1a jyH1b unpckhps xmm2, xmm5 ;# xmm2= jxOc jxOd jyOc jyOd unpckhps xmm3, xmm6 ;# xmm3= jxH1c jxH1d jyH1c jyH1d movaps xmm5, xmm4 movaps xmm6, xmm0 unpcklps xmm4, xmm7 ;# xmm4= jxH2a jxH2b jyH2a jyH2b unpckhps xmm5, xmm7 ;# xmm5= jxH2c jxH2d jyH2c jyH2d movaps xmm7, xmm1 movlhps xmm0, xmm2 ;# xmm0= jxOa jxOb jxOc jxOd movaps [esp + nb102nf_jxO], xmm0 movhlps xmm2, xmm6 ;# xmm2= jyOa jyOb jyOc jyOd movaps [esp + nb102nf_jyO], xmm2 movlhps xmm1, xmm3 movaps [esp + nb102nf_jxH1], xmm1 movhlps xmm3, xmm7 movaps xmm6, xmm4 movaps [esp + nb102nf_jyH1], xmm3 movlhps xmm4, xmm5 movaps [esp + nb102nf_jxH2], xmm4 movhlps xmm5, xmm6 movaps [esp + nb102nf_jyH2], xmm5 movss xmm0, [esi + eax*4 + 8] movss xmm1, [esi + eax*4 + 20] movss xmm2, [esi + eax*4 + 32] movss xmm3, [esi + ecx*4 + 8] movss xmm4, [esi + ecx*4 + 20] movss xmm5, [esi + ecx*4 + 32] movhps xmm0, [esi + ebx*4 + 4] movhps xmm1, [esi + ebx*4 + 16] movhps xmm2, [esi + ebx*4 + 28] movhps xmm3, [esi + edx*4 + 4] movhps xmm4, [esi + edx*4 + 16] movhps xmm5, [esi + edx*4 + 28] shufps xmm0, xmm3, 204 ;# constant 11001100 shufps xmm1, xmm4, 204 ;# constant 11001100 shufps xmm2, xmm5, 204 ;# constant 11001100 movaps [esp + nb102nf_jzO], xmm0 movaps [esp + nb102nf_jzH1], xmm1 movaps [esp + nb102nf_jzH2], xmm2 movaps xmm0, [esp + nb102nf_ixO] movaps xmm1, [esp + nb102nf_iyO] movaps xmm2, [esp + nb102nf_izO] movaps xmm3, [esp + nb102nf_ixO] movaps xmm4, [esp + nb102nf_iyO] movaps xmm5, [esp + nb102nf_izO] subps xmm0, [esp + nb102nf_jxO] subps xmm1, [esp + nb102nf_jyO] subps xmm2, [esp + nb102nf_jzO] subps xmm3, [esp + nb102nf_jxH1] subps xmm4, [esp + nb102nf_jyH1] subps xmm5, [esp + nb102nf_jzH1] mulps xmm0, xmm0 mulps xmm1, xmm1 mulps xmm2, xmm2 mulps xmm3, xmm3 mulps xmm4, xmm4 mulps xmm5, xmm5 addps xmm0, xmm1 addps xmm0, xmm2 addps xmm3, xmm4 addps xmm3, xmm5 movaps [esp + nb102nf_rsqOO], xmm0 movaps [esp + nb102nf_rsqOH1], xmm3 movaps xmm0, [esp + nb102nf_ixO] movaps xmm1, [esp + nb102nf_iyO] movaps xmm2, [esp + nb102nf_izO] movaps xmm3, [esp + nb102nf_ixH1] movaps xmm4, [esp + nb102nf_iyH1] movaps xmm5, [esp + nb102nf_izH1] subps xmm0, [esp + nb102nf_jxH2] subps xmm1, [esp + nb102nf_jyH2] subps xmm2, [esp + nb102nf_jzH2] subps xmm3, [esp + nb102nf_jxO] subps xmm4, [esp + nb102nf_jyO] subps xmm5, [esp + nb102nf_jzO] mulps xmm0, xmm0 mulps xmm1, xmm1 mulps xmm2, xmm2 mulps xmm3, xmm3 mulps xmm4, xmm4 mulps xmm5, xmm5 addps xmm0, xmm1 addps xmm0, xmm2 addps xmm3, xmm4 addps xmm3, xmm5 movaps [esp + nb102nf_rsqOH2], xmm0 movaps [esp + nb102nf_rsqH1O], xmm3 movaps xmm0, [esp + nb102nf_ixH1] movaps xmm1, [esp + nb102nf_iyH1] movaps xmm2, [esp + nb102nf_izH1] movaps xmm3, [esp + nb102nf_ixH1] movaps xmm4, [esp + nb102nf_iyH1] movaps xmm5, [esp + nb102nf_izH1] subps xmm0, [esp + nb102nf_jxH1] subps xmm1, [esp + nb102nf_jyH1] subps xmm2, [esp + nb102nf_jzH1] subps xmm3, [esp + nb102nf_jxH2] subps xmm4, [esp + nb102nf_jyH2] subps xmm5, [esp + nb102nf_jzH2] mulps xmm0, xmm0 mulps xmm1, xmm1 mulps xmm2, xmm2 mulps xmm3, xmm3 mulps xmm4, xmm4 mulps xmm5, xmm5 addps xmm0, xmm1 addps xmm0, xmm2 addps xmm3, xmm4 addps xmm3, xmm5 movaps [esp + nb102nf_rsqH1H1], xmm0 movaps [esp + nb102nf_rsqH1H2], xmm3 movaps xmm0, [esp + nb102nf_ixH2] movaps xmm1, [esp + nb102nf_iyH2] movaps xmm2, [esp + nb102nf_izH2] movaps xmm3, [esp + nb102nf_ixH2] movaps xmm4, [esp + nb102nf_iyH2] movaps xmm5, [esp + nb102nf_izH2] subps xmm0, [esp + nb102nf_jxO] subps xmm1, [esp + nb102nf_jyO] subps xmm2, [esp + nb102nf_jzO] subps xmm3, [esp + nb102nf_jxH1] subps xmm4, [esp + nb102nf_jyH1] subps xmm5, [esp + nb102nf_jzH1] mulps xmm0, xmm0 mulps xmm1, xmm1 mulps xmm2, xmm2 mulps xmm3, xmm3 mulps xmm4, xmm4 mulps xmm5, xmm5 addps xmm0, xmm1 addps xmm0, xmm2 addps xmm4, xmm3 addps xmm4, xmm5 movaps [esp + nb102nf_rsqH2O], xmm0 movaps [esp + nb102nf_rsqH2H1], xmm4 movaps xmm0, [esp + nb102nf_ixH2] movaps xmm1, [esp + nb102nf_iyH2] movaps xmm2, [esp + nb102nf_izH2] subps xmm0, [esp + nb102nf_jxH2] subps xmm1, [esp + nb102nf_jyH2] subps xmm2, [esp + nb102nf_jzH2] mulps xmm0, xmm0 mulps xmm1, xmm1 mulps xmm2, xmm2 addps xmm0, xmm1 addps xmm0, xmm2 movaps [esp + nb102nf_rsqH2H2], xmm0 ;# start doing invsqrt use rsq values in xmm0, xmm4 rsqrtps xmm1, xmm0 rsqrtps xmm5, xmm4 movaps xmm2, xmm1 movaps xmm6, xmm5 mulps xmm1, xmm1 mulps xmm5, xmm5 movaps xmm3, [esp + nb102nf_three] movaps xmm7, xmm3 mulps xmm1, xmm0 mulps xmm5, xmm4 subps xmm3, xmm1 subps xmm7, xmm5 mulps xmm3, xmm2 mulps xmm7, xmm6 mulps xmm3, [esp + nb102nf_half] ;# rinvH2H2 mulps xmm7, [esp + nb102nf_half] ;# rinvH2H1 movaps [esp + nb102nf_rinvH2H2], xmm3 movaps [esp + nb102nf_rinvH2H1], xmm7 rsqrtps xmm1, [esp + nb102nf_rsqOO] rsqrtps xmm5, [esp + nb102nf_rsqOH1] movaps xmm2, xmm1 movaps xmm6, xmm5 mulps xmm1, xmm1 mulps xmm5, xmm5 movaps xmm3, [esp + nb102nf_three] movaps xmm7, xmm3 mulps xmm1, [esp + nb102nf_rsqOO] mulps xmm5, [esp + nb102nf_rsqOH1] subps xmm3, xmm1 subps xmm7, xmm5 mulps xmm3, xmm2 mulps xmm7, xmm6 mulps xmm3, [esp + nb102nf_half] mulps xmm7, [esp + nb102nf_half] movaps [esp + nb102nf_rinvOO], xmm3 movaps [esp + nb102nf_rinvOH1], xmm7 rsqrtps xmm1, [esp + nb102nf_rsqOH2] rsqrtps xmm5, [esp + nb102nf_rsqH1O] movaps xmm2, xmm1 movaps xmm6, xmm5 mulps xmm1, xmm1 mulps xmm5, xmm5 movaps xmm3, [esp + nb102nf_three] movaps xmm7, xmm3 mulps xmm1, [esp + nb102nf_rsqOH2] mulps xmm5, [esp + nb102nf_rsqH1O] subps xmm3, xmm1 subps xmm7, xmm5 mulps xmm3, xmm2 mulps xmm7, xmm6 mulps xmm3, [esp + nb102nf_half] mulps xmm7, [esp + nb102nf_half] movaps [esp + nb102nf_rinvOH2], xmm3 movaps [esp + nb102nf_rinvH1O], xmm7 rsqrtps xmm1, [esp + nb102nf_rsqH1H1] rsqrtps xmm5, [esp + nb102nf_rsqH1H2] movaps xmm2, xmm1 movaps xmm6, xmm5 mulps xmm1, xmm1 mulps xmm5, xmm5 movaps xmm3, [esp + nb102nf_three] movaps xmm7, xmm3 mulps xmm1, [esp + nb102nf_rsqH1H1] mulps xmm5, [esp + nb102nf_rsqH1H2] subps xmm3, xmm1 subps xmm7, xmm5 mulps xmm3, xmm2 mulps xmm7, xmm6 mulps xmm3, [esp + nb102nf_half] mulps xmm7, [esp + nb102nf_half] movaps [esp + nb102nf_rinvH1H1], xmm3 movaps [esp + nb102nf_rinvH1H2], xmm7 rsqrtps xmm1, [esp + nb102nf_rsqH2O] movaps xmm2, xmm1 mulps xmm1, xmm1 movaps xmm3, [esp + nb102nf_three] mulps xmm1, [esp + nb102nf_rsqH2O] subps xmm3, xmm1 mulps xmm3, xmm2 mulps xmm3, [esp + nb102nf_half] movaps [esp + nb102nf_rinvH2O], xmm3 ;# sum OO pot in xmm0, OH in xmm1 HH in xmm2 movaps xmm0, [esp + nb102nf_rinvOO] movaps xmm1, [esp + nb102nf_rinvOH1] movaps xmm2, [esp + nb102nf_rinvH1H1] addps xmm1, [esp + nb102nf_rinvOH2] addps xmm2, [esp + nb102nf_rinvH1H2] addps xmm1, [esp + nb102nf_rinvH1O] addps xmm2, [esp + nb102nf_rinvH2H1] addps xmm1, [esp + nb102nf_rinvH2O] addps xmm2, [esp + nb102nf_rinvH2H2] mulps xmm0, [esp + nb102nf_qqOO] mulps xmm1, [esp + nb102nf_qqOH] mulps xmm2, [esp + nb102nf_qqHH] addps xmm0, [esp + nb102nf_vctot] addps xmm1, xmm2 addps xmm0, xmm1 movaps [esp + nb102nf_vctot], xmm0 ;# should we do one more iteration? sub dword ptr [esp + nb102nf_innerk], 4 jl .nb102nf_single_check jmp .nb102nf_unroll_loop.nb102nf_single_check: add dword ptr [esp + nb102nf_innerk], 4 jnz .nb102nf_single_loop jmp .nb102nf_updateouterdata.nb102nf_single_loop: mov edx, [esp + nb102nf_innerjjnr] ;# pointer to jjnr[k] mov eax, [edx] add dword ptr [esp + nb102nf_innerjjnr], 4 mov esi, [ebp + nb102nf_pos] lea eax, [eax + eax*2] ;# fetch j coordinates xorps xmm3, xmm3 xorps xmm4, xmm4 xorps xmm5, xmm5 movss xmm3, [esi + eax*4] ;# jxO - - - movss xmm4, [esi + eax*4 + 4] ;# jyO - - - movss xmm5, [esi + eax*4 + 8] ;# jzO - - - movlps xmm6, [esi + eax*4 + 12] ;# xmm6 = jxH1 jyH1 - - movss xmm7, [esi + eax*4 + 20] ;# xmm7 = jzH1 - - - movhps xmm6, [esi + eax*4 + 24] ;# xmm6 = jxH1 jyH1 jxH2 jyH2 movss xmm2, [esi + eax*4 + 32] ;# xmm2 = jzH2 - - - ;# have all coords, time for some shuffling. shufps xmm6, xmm6, 216 ;# constant 11011000 ;# xmm6 = jxH1 jxH2 jyH1 jyH2 unpcklps xmm7, xmm2 ;# xmm7 = jzH1 jzH2 - - movaps xmm0, [esp + nb102nf_ixO] movaps xmm1, [esp + nb102nf_iyO] movaps xmm2, [esp + nb102nf_izO] movlhps xmm3, xmm6 ;# xmm3 = jxO 0 jxH1 jxH2 shufps xmm4, xmm6, 228 ;# constant 11100100 ;# xmm4 = jyO 0 jyH1 jyH2 shufps xmm5, xmm7, 68 ;# constant 01000100 ;# xmm5 = jzO 0 jzH1 jzH2 ;# store all j coordinates in jO movaps [esp + nb102nf_jxO], xmm3 movaps [esp + nb102nf_jyO], xmm4 movaps [esp + nb102nf_jzO], xmm5 subps xmm0, xmm3 subps xmm1, xmm4 subps xmm2, xmm5 mulps xmm0, xmm0 mulps xmm1, xmm1 mulps xmm2, xmm2 addps xmm0, xmm1 addps xmm0, xmm2 ;# have rsq in xmm0 ;# do invsqrt rsqrtps xmm1, xmm0 movaps xmm2, xmm1 mulps xmm1, xmm1 movaps xmm3, [esp + nb102nf_three] mulps xmm1, xmm0 subps xmm3, xmm1 mulps xmm3, xmm2 mulps xmm3, [esp + nb102nf_half] ;# rinv iO - j water xorps xmm1, xmm1 xorps xmm4, xmm4 ;# fetch charges to xmm4 (temporary) movss xmm4, [esp + nb102nf_qqOO] movhps xmm4, [esp + nb102nf_qqOH] mulps xmm3, xmm4 ;# xmm3=vcoul addps xmm3, [esp + nb102nf_vctot] movaps [esp + nb102nf_vctot], xmm3 ;# done with i O Now do i H1 & H2 simultaneously: movaps xmm0, [esp + nb102nf_ixH1] movaps xmm1, [esp + nb102nf_iyH1] movaps xmm2, [esp + nb102nf_izH1] movaps xmm3, [esp + nb102nf_ixH2] movaps xmm4, [esp + nb102nf_iyH2] movaps xmm5, [esp + nb102nf_izH2] subps xmm0, [esp + nb102nf_jxO] subps xmm1, [esp + nb102nf_jyO] subps xmm2, [esp + nb102nf_jzO] subps xmm3, [esp + nb102nf_jxO] subps xmm4, [esp + nb102nf_jyO] subps xmm5, [esp + nb102nf_jzO] mulps xmm0, xmm0 mulps xmm1, xmm1 mulps xmm2, xmm2 mulps xmm3, xmm3 mulps xmm4, xmm4 mulps xmm5, xmm5 addps xmm0, xmm1 addps xmm4, xmm3 addps xmm0, xmm2 ;# have rsqH1 in xmm0 addps xmm4, xmm5 ;# have rsqH2 in xmm4 ;# do invsqrt rsqrtps xmm1, xmm0 rsqrtps xmm5, xmm4 movaps xmm2, xmm1 movaps xmm6, xmm5 mulps xmm1, xmm1 mulps xmm5, xmm5 movaps xmm3, [esp + nb102nf_three] movaps xmm7, xmm3 mulps xmm1, xmm0 mulps xmm5, xmm4 subps xmm3, xmm1 subps xmm7, xmm5 mulps xmm3, xmm2 mulps xmm7, xmm6 mulps xmm3, [esp + nb102nf_half] ;# rinv H1 - j water mulps xmm7, [esp + nb102nf_half] ;# rinv H2 - j water ;# assemble charges in xmm6 xorps xmm6, xmm6 ;# do coulomb interaction movaps x⌨️ 快捷键说明
复制代码Ctrl + C
搜索代码Ctrl + F
全屏模式F11
增大字号Ctrl + =
减小字号Ctrl + -
显示快捷键?