[llvm-branch-commits] [compiler-rt] 91d3e76 - [AArch64][SME] Rewrite __arm_sc_memset to remove invalid instruction (#101522)

[Tobias Hieta via llvm-branch-commits]

Author: Kerry McLaughlin
Date: 2024-08-10T11:55:42+02:00
New Revision: 91d3e76a2b670aece2a15c54f3f9893c0ed96ba1

URL: https://github.com/llvm/llvm-project/commit/91d3e76a2b670aece2a15c54f3f9893c0ed96ba1
DIFF: https://github.com/llvm/llvm-project/commit/91d3e76a2b670aece2a15c54f3f9893c0ed96ba1.diff

LOG: [AArch64][SME] Rewrite __arm_sc_memset to remove invalid instruction (#101522)

The implementation of __arm_sc_memset in compiler-rt contains
a Neon dup instruction which is not valid in streaming mode. This
patch rewrites the function, using an SVE mov instruction if available.

(cherry picked from commit d6649f2d4871c4535ae0519920e36100748890c4)

Added: 
    

Modified: 
    compiler-rt/lib/builtins/aarch64/sme-libc-mem-routines.S

Removed: 
    


################################################################################
diff  --git a/compiler-rt/lib/builtins/aarch64/sme-libc-mem-routines.S b/compiler-rt/lib/builtins/aarch64/sme-libc-mem-routines.S
index 926ad3b1b63315..0318d9a6f1ebd2 100644
--- a/compiler-rt/lib/builtins/aarch64/sme-libc-mem-routines.S
+++ b/compiler-rt/lib/builtins/aarch64/sme-libc-mem-routines.S
@@ -252,7 +252,15 @@ DEFINE_COMPILERRT_FUNCTION_ALIAS(__arm_sc_memmove, __arm_sc_memcpy)
 #define zva_val  x5
 
 DEFINE_COMPILERRT_OUTLINE_FUNCTION_UNMANGLED(__arm_sc_memset)
-        dup     v0.16B, valw
+#ifdef __ARM_FEATURE_SVE
+        mov     z0.b, valw
+#else
+        bfi valw, valw, #8, #8
+        bfi valw, valw, #16, #16
+        bfi val, val, #32, #32
+        fmov d0, val
+        fmov v0.d[1], val
+#endif
         add     dstend2, dstin, count
 
         cmp     count, 96