Browse Source

Import updated asm for cpu mining based on ufasoft assembly.

nfactor-troky
Con Kolivas 14 years ago
parent
commit
4800832b1d
  1. 235
      x86_64/sha256_xmm_amd64.asm

235
x86_64/sha256_xmm_amd64.asm

@ -4,6 +4,10 @@
; Version 2011 ; Version 2011
; This software is Public Domain ; This software is Public Domain
; Significant re-write/optimisation and reordering by,
; Neil Kettle <mu-b@digit-labs.org>
; ~18% performance improvement
; SHA-256 CPU SSE cruncher for Bitcoin Miner ; SHA-256 CPU SSE cruncher for Bitcoin Miner
ALIGN 32 ALIGN 32
@ -13,6 +17,12 @@ BITS 64
%define data rsi %define data rsi
%define init rdx %define init rdx
; 0 = (1024 - 256) (mod (LAB_CALC_UNROLL*LAB_CALC_PARA*16))
%define LAB_CALC_PARA 2
%define LAB_CALC_UNROLL 8
%define LAB_LOOP_UNROLL 8
extern g_4sha256_k extern g_4sha256_k
global CalcSha256_x64 global CalcSha256_x64
@ -22,87 +32,118 @@ CalcSha256_x64:
push rbx push rbx
LAB_NEXT_NONCE: LAB_NEXT_NONCE:
mov r11, data
; mov rax, pnonce mov rcx, 64*4 ; 256 - rcx is # of SHA-2 rounds
; mov eax, [rax] mov rax, 16*4 ; 64 - rax is where we expand to
; mov [rbx+3*16], eax
; inc eax
; mov [rbx+3*16+4], eax
; inc eax
; mov [rbx+3*16+8], eax
; inc eax
; mov [rbx+3*16+12], eax
mov rcx, 64*4 ;rcx is # of SHA-2 rounds
mov rax, 16*4 ;rax is where we expand to
LAB_SHA: LAB_SHA:
push rcx push rcx
lea rcx, qword [r11+rcx*4] lea rcx, qword [data+rcx*4] ; + 1024
lea r11, qword [r11+rax*4] lea r11, qword [data+rax*4] ; + 256
LAB_CALC:
movdqa xmm0, [r11-15*16]
movdqa xmm2, xmm0 ; (Rotr32(w_15, 7) ^ Rotr32(w_15, 18) ^ (w_15 >> 3))
psrld xmm0, 3
movdqa xmm1, xmm0
pslld xmm2, 14
psrld xmm1, 4
pxor xmm0, xmm1
pxor xmm0, xmm2
pslld xmm2, 11
psrld xmm1, 11
pxor xmm0, xmm1
pxor xmm0, xmm2
paddd xmm0, [r11-16*16] LAB_CALC:
%macro lab_calc_blk 1
movdqa xmm3, [r11-2*16] movdqa xmm0, [r11-(15-%1)*16] ; xmm0 = W[I-15]
movdqa xmm2, xmm3 ; (Rotr32(w_2, 17) ^ Rotr32(w_2, 19) ^ (w_2 >> 10)) movdqa xmm4, [r11-(15-(%1+1))*16] ; xmm4 = W[I-15+1]
psrld xmm3, 10 movdqa xmm2, xmm0 ; xmm2 = W[I-15]
movdqa xmm1, xmm3 movdqa xmm6, xmm4 ; xmm6 = W[I-15+1]
pslld xmm2, 13 psrld xmm0, 3 ; xmm0 = W[I-15] >> 3
psrld xmm1, 7 psrld xmm4, 3 ; xmm4 = W[I-15+1] >> 3
pxor xmm3, xmm1 movdqa xmm1, xmm0 ; xmm1 = W[I-15] >> 3
pxor xmm3, xmm2 movdqa xmm5, xmm4 ; xmm5 = W[I-15+1] >> 3
pslld xmm2, 2 pslld xmm2, 14 ; xmm2 = W[I-15] << 14
psrld xmm1, 2 pslld xmm6, 14 ; xmm6 = W[I-15+1] << 14
pxor xmm3, xmm1 psrld xmm1, 4 ; xmm1 = W[I-15] >> 7
pxor xmm3, xmm2 psrld xmm5, 4 ; xmm5 = W[I-15+1] >> 7
paddd xmm0, xmm3 pxor xmm0, xmm1 ; xmm0 = (W[I-15] >> 3) ^ (W[I-15] >> 7)
pxor xmm4, xmm5 ; xmm4 = (W[I-15+1] >> 3) ^ (W[I-15+1] >> 7)
paddd xmm0, [r11-7*16] psrld xmm1, 11 ; xmm1 = W[I-15] >> 18
movdqa [r11], xmm0 psrld xmm5, 11 ; xmm5 = W[I-15+1] >> 18
add r11, 16 pxor xmm0, xmm2 ; xmm0 = (W[I-15] >> 3) ^ (W[I-15] >> 7) ^ (W[I-15] << 14)
pxor xmm4, xmm6 ; xmm4 = (W[I-15+1] >> 3) ^ (W[I-15+1] >> 7) ^ (W[I-15+1] << 14)
pslld xmm2, 11 ; xmm2 = W[I-15] << 25
pslld xmm6, 11 ; xmm6 = W[I-15+1] << 25
pxor xmm0, xmm1 ; xmm0 = (W[I-15] >> 3) ^ (W[I-15] >> 7) ^ (W[I-15] << 14) ^ (W[I-15] >> 18)
pxor xmm4, xmm5 ; xmm4 = (W[I-15+1] >> 3) ^ (W[I-15+1] >> 7) ^ (W[I-15+1] << 14) ^ (W[I-15+1] >> 18)
pxor xmm0, xmm2 ; xmm0 = (W[I-15] >> 3) ^ (W[I-15] >> 7) ^ (W[I-15] << 14) ^ (W[I-15] >> 18) ^ (W[I-15] << 25)
pxor xmm4, xmm6 ; xmm4 = (W[I-15+1] >> 3) ^ (W[I-15+1] >> 7) ^ (W[I-15+1] << 14) ^ (W[I-15+1] >> 18) ^ (W[I-15+1] << 25)
movdqa xmm3, [r11-(2-%1)*16] ; xmm3 = W[I-2]
movdqa xmm7, [r11-(2-(%1+1))*16] ; xmm7 = W[I-2+1]
paddd xmm0, [r11-(16-%1)*16] ; xmm0 = s0(W[I-15]) + W[I-16]
paddd xmm4, [r11-(16-(%1+1))*16] ; xmm4 = s0(W[I-15+1]) + W[I-16+1]
;;;;;;;;;;;;;;;;;;
movdqa xmm2, xmm3 ; xmm2 = W[I-2]
movdqa xmm6, xmm7 ; xmm6 = W[I-2+1]
psrld xmm3, 10 ; xmm3 = W[I-2] >> 10
psrld xmm7, 10 ; xmm7 = W[I-2+1] >> 10
movdqa xmm1, xmm3 ; xmm1 = W[I-2] >> 10
movdqa xmm5, xmm7 ; xmm5 = W[I-2+1] >> 10
paddd xmm0, [r11-(7-%1)*16] ; xmm0 = s0(W[I-15]) + W[I-16] + W[I-7]
pslld xmm2, 13 ; xmm2 = W[I-2] << 13
pslld xmm6, 13 ; xmm6 = W[I-2+1] << 13
psrld xmm1, 7 ; xmm1 = W[I-2] >> 17
psrld xmm5, 7 ; xmm5 = W[I-2+1] >> 17
paddd xmm4, [r11-(7-(%1+1))*16] ; xmm4 = s0(W[I-15+1]) + W[I-16+1] + W[I-7+1]
pxor xmm3, xmm1 ; xmm3 = (W[I-2] >> 10) ^ (W[I-2] >> 17)
pxor xmm7, xmm5 ; xmm7 = (W[I-2+1] >> 10) ^ (W[I-2+1] >> 17)
psrld xmm1, 2 ; xmm1 = W[I-2] >> 19
psrld xmm5, 2 ; xmm5 = W[I-2+1] >> 19
pxor xmm3, xmm2 ; xmm3 = (W[I-2] >> 10) ^ (W[I-2] >> 17) ^ (W[I-2] << 13)
pxor xmm7, xmm6 ; xmm7 = (W[I-2+1] >> 10) ^ (W[I-2+1] >> 17) ^ (W[I-2+1] << 13)
pslld xmm2, 2 ; xmm2 = W[I-2] << 15
pslld xmm6, 2 ; xmm6 = W[I-2+1] << 15
pxor xmm3, xmm1 ; xmm3 = (W[I-2] >> 10) ^ (W[I-2] >> 17) ^ (W[I-2] << 13) ^ (W[I-2] >> 19)
pxor xmm7, xmm5 ; xmm7 = (W[I-2+1] >> 10) ^ (W[I-2+1] >> 17) ^ (W[I-2+1] << 13) ^ (W[I-2+1] >> 19)
pxor xmm3, xmm2 ; xmm3 = (W[I-2] >> 10) ^ (W[I-2] >> 17) ^ (W[I-2] << 13) ^ (W[I-2] >> 19) ^ (W[I-2] << 15)
pxor xmm7, xmm6 ; xmm7 = (W[I-2+1] >> 10) ^ (W[I-2+1] >> 17) ^ (W[I-2+1] << 13) ^ (W[I-2+1] >> 19) ^ (W[I-2+1] << 15)
paddd xmm0, xmm3 ; xmm0 = s0(W[I-15]) + W[I-16] + s1(W[I-2]) + W[I-7]
paddd xmm4, xmm7 ; xmm4 = s0(W[I-15+1]) + W[I-16+1] + s1(W[I-2+1]) + W[I-7+1]
movdqa [r11+(%1*16)], xmm0
movdqa [r11+((%1+1)*16)], xmm4
%endmacro
%assign i 0
%rep LAB_CALC_UNROLL
lab_calc_blk i
%assign i i+LAB_CALC_PARA
%endrep
add r11, LAB_CALC_UNROLL*LAB_CALC_PARA*16
cmp r11, rcx cmp r11, rcx
jb LAB_CALC jb LAB_CALC
pop rcx
pop rcx
mov rax, 0 mov rax, 0
; Load the init values of the message into the hash. ; Load the init values of the message into the hash.
movd xmm0, dword [rdx+4*4] ; xmm0 == e movdqa xmm7, [init]
pshufd xmm0, xmm0, 0 pshufd xmm5, xmm7, 0x55 ; xmm5 == b
movd xmm3, dword [rdx+3*4] ; xmm3 == d pshufd xmm4, xmm7, 0xAA ; xmm4 == c
pshufd xmm3, xmm3, 0 pshufd xmm3, xmm7, 0xFF ; xmm3 == d
movd xmm4, dword [rdx+2*4] ; xmm4 == c pshufd xmm7, xmm7, 0 ; xmm7 == a
pshufd xmm4, xmm4, 0
movd xmm5, dword [rdx+1*4] ; xmm5 == b movdqa xmm0, [init+4*4]
pshufd xmm5, xmm5, 0 pshufd xmm8, xmm0, 0x55 ; xmm8 == f
movd xmm7, dword [rdx+0*4] ; xmm7 == a pshufd xmm9, xmm0, 0xAA ; xmm9 == g
pshufd xmm7, xmm7, 0 pshufd xmm10, xmm0, 0xFF ; xmm10 == h
movd xmm8, dword [rdx+5*4] ; xmm8 == f pshufd xmm0, xmm0, 0 ; xmm0 == e
pshufd xmm8, xmm8, 0
movd xmm9, dword [rdx+6*4] ; xmm9 == g
pshufd xmm9, xmm9, 0
movd xmm10, dword [rdx+7*4] ; xmm10 == h
pshufd xmm10, xmm10, 0
LAB_LOOP: LAB_LOOP:
;; T t1 = h + (Rotr32(e, 6) ^ Rotr32(e, 11) ^ Rotr32(e, 25)) + ((e & f) ^ AndNot(e, g)) + Expand32<T>(g_sha256_k[j]) + w[j] ;; T t1 = h + (Rotr32(e, 6) ^ Rotr32(e, 11) ^ Rotr32(e, 25)) + ((e & f) ^ AndNot(e, g)) + Expand32<T>(g_sha256_k[j]) + w[j]
movdqa xmm6, [rsi+rax*4] %macro lab_loop_blk 0
movdqa xmm6, [data+rax*4]
paddd xmm6, g_4sha256_k[rax*4] paddd xmm6, g_4sha256_k[rax*4]
add rax, 4 add rax, 4
@ -166,53 +207,49 @@ LAB_LOOP:
pslld xmm2, 11 pslld xmm2, 11
pxor xmm7, xmm2 pxor xmm7, xmm2
paddd xmm7, xmm6 ; a = t1 + (Rotr32(a, 2) ^ Rotr32(a, 13) ^ Rotr32(a, 22)) + ((a & c) ^ (a & d) ^ (c & d)); paddd xmm7, xmm6 ; a = t1 + (Rotr32(a, 2) ^ Rotr32(a, 13) ^ Rotr32(a, 22)) + ((a & c) ^ (a & d) ^ (c & d));
%endmacro
%assign i 0
%rep LAB_LOOP_UNROLL
lab_loop_blk
%assign i i+1
%endrep
cmp rax, rcx cmp rax, rcx
jb LAB_LOOP jb LAB_LOOP
; Finished the 64 rounds, calculate hash and save ; Finished the 64 rounds, calculate hash and save
movd xmm1, dword [rdx+0*4] movdqa xmm1, [rdx]
pshufd xmm2, xmm1, 0x55
pshufd xmm6, xmm1, 0xAA
pshufd xmm11, xmm1, 0xFF
pshufd xmm1, xmm1, 0 pshufd xmm1, xmm1, 0
paddd xmm7, xmm1
movd xmm1, dword [rdx+1*4] paddd xmm5, xmm2
pshufd xmm1, xmm1, 0 paddd xmm4, xmm6
paddd xmm5, xmm1 paddd xmm3, xmm11
paddd xmm7, xmm1
movd xmm1, dword [rdx+2*4]
pshufd xmm1, xmm1, 0
paddd xmm4, xmm1
movd xmm1, dword [rdx+3*4] movdqa xmm1, [rdx+4*4]
pshufd xmm2, xmm1, 0x55
pshufd xmm6, xmm1, 0xAA
pshufd xmm11, xmm1, 0xFF
pshufd xmm1, xmm1, 0 pshufd xmm1, xmm1, 0
paddd xmm3, xmm1
movd xmm1, dword [rdx+4*4] paddd xmm8, xmm2
pshufd xmm1, xmm1, 0 paddd xmm9, xmm6
paddd xmm10, xmm11
paddd xmm0, xmm1 paddd xmm0, xmm1
movd xmm1, dword [rdx+5*4] movdqa [hash+0*16], xmm7
pshufd xmm1, xmm1, 0 movdqa [hash+1*16], xmm5
paddd xmm8, xmm1 movdqa [hash+2*16], xmm4
movdqa [hash+3*16], xmm3
movd xmm1, dword [rdx+6*4] movdqa [hash+4*16], xmm0
pshufd xmm1, xmm1, 0 movdqa [hash+5*16], xmm8
paddd xmm9, xmm1 movdqa [hash+6*16], xmm9
movdqa [hash+7*16], xmm10
movd xmm1, dword [rdx+7*4]
pshufd xmm1, xmm1, 0
paddd xmm10, xmm1
debug_me:
movdqa [rdi+0*16], xmm7
movdqa [rdi+1*16], xmm5
movdqa [rdi+2*16], xmm4
movdqa [rdi+3*16], xmm3
movdqa [rdi+4*16], xmm0
movdqa [rdi+5*16], xmm8
movdqa [rdi+6*16], xmm9
movdqa [rdi+7*16], xmm10
LAB_RET: LAB_RET:
pop rbx pop rbx

Loading…
Cancel
Save