copy-unaligned.S 1.6 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071
  1. /* SPDX-License-Identifier: GPL-2.0 */
  2. /* Copyright (C) 2023 Rivos Inc. */
  3. #include <linux/linkage.h>
  4. #include <asm/asm.h>
  5. .text
  6. /* void __riscv_copy_words_unaligned(void *, const void *, size_t) */
  7. /* Performs a memcpy without aligning buffers, using word loads and stores. */
  8. /* Note: The size is truncated to a multiple of 8 * SZREG */
  9. SYM_FUNC_START(__riscv_copy_words_unaligned)
  10. andi a4, a2, ~((8*SZREG)-1)
  11. beqz a4, 2f
  12. add a3, a1, a4
  13. 1:
  14. REG_L a4, 0(a1)
  15. REG_L a5, SZREG(a1)
  16. REG_L a6, 2*SZREG(a1)
  17. REG_L a7, 3*SZREG(a1)
  18. REG_L t0, 4*SZREG(a1)
  19. REG_L t1, 5*SZREG(a1)
  20. REG_L t2, 6*SZREG(a1)
  21. REG_L t3, 7*SZREG(a1)
  22. REG_S a4, 0(a0)
  23. REG_S a5, SZREG(a0)
  24. REG_S a6, 2*SZREG(a0)
  25. REG_S a7, 3*SZREG(a0)
  26. REG_S t0, 4*SZREG(a0)
  27. REG_S t1, 5*SZREG(a0)
  28. REG_S t2, 6*SZREG(a0)
  29. REG_S t3, 7*SZREG(a0)
  30. addi a0, a0, 8*SZREG
  31. addi a1, a1, 8*SZREG
  32. bltu a1, a3, 1b
  33. 2:
  34. ret
  35. SYM_FUNC_END(__riscv_copy_words_unaligned)
  36. /* void __riscv_copy_bytes_unaligned(void *, const void *, size_t) */
  37. /* Performs a memcpy without aligning buffers, using only byte accesses. */
  38. /* Note: The size is truncated to a multiple of 8 */
  39. SYM_FUNC_START(__riscv_copy_bytes_unaligned)
  40. andi a4, a2, ~(8-1)
  41. beqz a4, 2f
  42. add a3, a1, a4
  43. 1:
  44. lb a4, 0(a1)
  45. lb a5, 1(a1)
  46. lb a6, 2(a1)
  47. lb a7, 3(a1)
  48. lb t0, 4(a1)
  49. lb t1, 5(a1)
  50. lb t2, 6(a1)
  51. lb t3, 7(a1)
  52. sb a4, 0(a0)
  53. sb a5, 1(a0)
  54. sb a6, 2(a0)
  55. sb a7, 3(a0)
  56. sb t0, 4(a0)
  57. sb t1, 5(a0)
  58. sb t2, 6(a0)
  59. sb t3, 7(a0)
  60. addi a0, a0, 8
  61. addi a1, a1, 8
  62. bltu a1, a3, 1b
  63. 2:
  64. ret
  65. SYM_FUNC_END(__riscv_copy_bytes_unaligned)