NEON optimized memcpy.
372 MB/s for large transfers, 440 MB/s for smaller ones down to 1KB. 130 MB/s for very small transfers ( < 32 bytes ) Performance is similar with non-congruent buffers.
This commit is contained in:
parent
33acbf0719
commit
ee223d02d9
@ -28,6 +28,136 @@
|
|||||||
|
|
||||||
#include <machine/cpu-features.h>
|
#include <machine/cpu-features.h>
|
||||||
|
|
||||||
|
#if __ARM_ARCH__ == 7 || defined(__ARM_NEON__)
|
||||||
|
|
||||||
|
.text
|
||||||
|
.fpu neon
|
||||||
|
|
||||||
|
.global memcpy
|
||||||
|
.type memcpy, %function
|
||||||
|
.align 4
|
||||||
|
|
||||||
|
/* a prefetch distance of 32*4 works best experimentally */
|
||||||
|
#define PREFETCH_DISTANCE (32*4)
|
||||||
|
|
||||||
|
memcpy:
|
||||||
|
.fnstart
|
||||||
|
.save {r0, lr}
|
||||||
|
stmfd sp!, {r0, lr}
|
||||||
|
|
||||||
|
/* start preloading as early as possible */
|
||||||
|
pld [r1, #0]
|
||||||
|
pld [r1, #32]
|
||||||
|
|
||||||
|
/* do we have at least 16-bytes to copy (needed for alignment below) */
|
||||||
|
cmp r2, #16
|
||||||
|
blo 5f
|
||||||
|
|
||||||
|
/* align destination to half cache-line for the write-buffer */
|
||||||
|
rsb r3, r0, #0
|
||||||
|
ands r3, r3, #0xF
|
||||||
|
beq 0f
|
||||||
|
|
||||||
|
/* copy up to 15-bytes (count in r3) */
|
||||||
|
sub r2, r2, r3
|
||||||
|
movs ip, r3, lsl #31
|
||||||
|
ldrmib lr, [r1], #1
|
||||||
|
strmib lr, [r0], #1
|
||||||
|
ldrcsb ip, [r1], #1
|
||||||
|
ldrcsb lr, [r1], #1
|
||||||
|
strcsb ip, [r0], #1
|
||||||
|
strcsb lr, [r0], #1
|
||||||
|
movs ip, r3, lsl #29
|
||||||
|
bge 1f
|
||||||
|
// copies 4 bytes, destination 32-bits aligned
|
||||||
|
vld4.8 {d0[0], d1[0], d2[0], d3[0]}, [r1]!
|
||||||
|
vst4.8 {d0[0], d1[0], d2[0], d3[0]}, [r0, :32]!
|
||||||
|
1: bcc 2f
|
||||||
|
// copies 8 bytes, destination 64-bits aligned
|
||||||
|
vld1.8 {d0}, [r1]!
|
||||||
|
vst1.8 {d0}, [r0, :64]!
|
||||||
|
2:
|
||||||
|
|
||||||
|
0: /* preload immediately the next cache line, which we may need */
|
||||||
|
pld [r1, #(32*0)]
|
||||||
|
pld [r1, #(32*1)]
|
||||||
|
pld [r1, #(32*2)]
|
||||||
|
pld [r1, #(32*3)]
|
||||||
|
|
||||||
|
/* make sure we have at least 128 bytes to copy */
|
||||||
|
subs r2, r2, #128
|
||||||
|
blo 2f
|
||||||
|
|
||||||
|
/* preload all the cache lines we need.
|
||||||
|
* NOTE: the number of pld below depends on PREFETCH_DISTANCE,
|
||||||
|
* ideally would would increase the distance in the main loop to
|
||||||
|
* avoid the goofy code below. In practice this doesn't seem to make
|
||||||
|
* a big difference.
|
||||||
|
*/
|
||||||
|
pld [r1, #(PREFETCH_DISTANCE + 32*0)]
|
||||||
|
pld [r1, #(PREFETCH_DISTANCE + 32*1)]
|
||||||
|
pld [r1, #(PREFETCH_DISTANCE + 32*2)]
|
||||||
|
pld [r1, #(PREFETCH_DISTANCE + 32*3)]
|
||||||
|
|
||||||
|
1: /* The main loop copies 128 bytes at a time */
|
||||||
|
vld1.8 {d0 - d3}, [r1]!
|
||||||
|
vld1.8 {d4 - d7}, [r1]!
|
||||||
|
vld1.8 {d16 - d19}, [r1]!
|
||||||
|
vld1.8 {d20 - d23}, [r1]!
|
||||||
|
pld [r1, #(PREFETCH_DISTANCE + 32*0)]
|
||||||
|
pld [r1, #(PREFETCH_DISTANCE + 32*1)]
|
||||||
|
pld [r1, #(PREFETCH_DISTANCE + 32*2)]
|
||||||
|
pld [r1, #(PREFETCH_DISTANCE + 32*3)]
|
||||||
|
subs r2, r2, #128
|
||||||
|
vst1.8 {d0 - d3}, [r0, :128]!
|
||||||
|
vst1.8 {d4 - d7}, [r0, :128]!
|
||||||
|
vst1.8 {d16 - d19}, [r0, :128]!
|
||||||
|
vst1.8 {d20 - d23}, [r0, :128]!
|
||||||
|
bhs 1b
|
||||||
|
|
||||||
|
2: /* fix-up the remaining count and make sure we have >= 32 bytes left */
|
||||||
|
add r2, r2, #128
|
||||||
|
subs r2, r2, #32
|
||||||
|
blo 4f
|
||||||
|
|
||||||
|
3: /* 32 bytes at a time. These cache lines were already preloaded */
|
||||||
|
vld1.8 {d0 - d3}, [r1]!
|
||||||
|
subs r2, r2, #32
|
||||||
|
vst1.8 {d0 - d3}, [r0, :128]!
|
||||||
|
bhs 3b
|
||||||
|
|
||||||
|
4: /* less than 32 left */
|
||||||
|
add r2, r2, #32
|
||||||
|
tst r2, #0x10
|
||||||
|
beq 5f
|
||||||
|
// copies 16 bytes, 128-bits aligned
|
||||||
|
vld1.8 {d0, d1}, [r1]!
|
||||||
|
vst1.8 {d0, d1}, [r0, :128]!
|
||||||
|
|
||||||
|
5: /* copy up to 15-bytes (count in r2) */
|
||||||
|
movs ip, r2, lsl #29
|
||||||
|
bcc 1f
|
||||||
|
vld1.8 {d0}, [r1]!
|
||||||
|
vst1.8 {d0}, [r0]!
|
||||||
|
1: bge 2f
|
||||||
|
vld4.8 {d0[0], d1[0], d2[0], d3[0]}, [r1]!
|
||||||
|
vst4.8 {d0[0], d1[0], d2[0], d3[0]}, [r0]!
|
||||||
|
2: movs ip, r2, lsl #31
|
||||||
|
ldrmib r3, [r1], #1
|
||||||
|
ldrcsb ip, [r1], #1
|
||||||
|
ldrcsb lr, [r1], #1
|
||||||
|
strmib r3, [r0], #1
|
||||||
|
strcsb ip, [r0], #1
|
||||||
|
strcsb lr, [r0], #1
|
||||||
|
|
||||||
|
ldmfd sp!, {r0, lr}
|
||||||
|
bx lr
|
||||||
|
.fnend
|
||||||
|
|
||||||
|
|
||||||
|
#else /* __ARM_ARCH__ < 7 */
|
||||||
|
|
||||||
|
|
||||||
.text
|
.text
|
||||||
|
|
||||||
.global memcpy
|
.global memcpy
|
||||||
@ -385,3 +515,5 @@ copy_last_3_and_return:
|
|||||||
bx lr
|
bx lr
|
||||||
.fnend
|
.fnend
|
||||||
|
|
||||||
|
|
||||||
|
#endif /* __ARM_ARCH__ < 7 */
|
||||||
|
Loading…
x
Reference in New Issue
Block a user