This patch adds the fix_truncv4sfv4hi2 (V4SF->V4HI) pattern which is implemented
using fix_truncv4sfv4si2 (V4SF->V4SI) and then truncv4siv4hi2 (V4SI->V4HI).

        PR target/113882

gcc/ChangeLog:

        * config/aarch64/aarch64-simd.md (fix_truncv4sfv4hi2): New pattern.

gcc/testsuite/ChangeLog:

        * gcc.target/aarch64/fix_trunc2.c: New test.

Signed-off-by: Pengxuan Zheng <quic_pzh...@quicinc.com>
---
 gcc/config/aarch64/aarch64-simd.md            | 13 +++++++++++++
 gcc/testsuite/gcc.target/aarch64/fix_trunc2.c | 14 ++++++++++++++
 2 files changed, 27 insertions(+)
 create mode 100644 gcc/testsuite/gcc.target/aarch64/fix_trunc2.c

diff --git a/gcc/config/aarch64/aarch64-simd.md 
b/gcc/config/aarch64/aarch64-simd.md
index 868f4486218..096f7b56a27 100644
--- a/gcc/config/aarch64/aarch64-simd.md
+++ b/gcc/config/aarch64/aarch64-simd.md
@@ -3032,6 +3032,19 @@ (define_expand 
"<fix_trunc_optab><VHSDF:mode><fcvt_target>2"
   "TARGET_SIMD"
   {})
 
+
+(define_expand "fix_truncv4sfv4hi2"
+  [(match_operand:V4HI 0 "register_operand")
+   (match_operand:V4SF 1 "register_operand")]
+  "TARGET_SIMD"
+  {
+    rtx tmp = gen_reg_rtx (V4SImode);
+    emit_insn (gen_fix_truncv4sfv4si2 (tmp, operands[1]));
+    emit_insn (gen_truncv4siv4hi2 (operands[0], tmp));
+    DONE;
+  }
+)
+
 (define_expand "ftrunc<VHSDF:mode>2"
   [(set (match_operand:VHSDF 0 "register_operand")
        (unspec:VHSDF [(match_operand:VHSDF 1 "register_operand")]
diff --git a/gcc/testsuite/gcc.target/aarch64/fix_trunc2.c 
b/gcc/testsuite/gcc.target/aarch64/fix_trunc2.c
new file mode 100644
index 00000000000..57cc00913a3
--- /dev/null
+++ b/gcc/testsuite/gcc.target/aarch64/fix_trunc2.c
@@ -0,0 +1,14 @@
+/* { dg-do compile } */
+/* { dg-options "-O2" } */
+
+void
+f (short *__restrict a, float *__restrict b)
+{
+  a[0] = b[0];
+  a[1] = b[1];
+  a[2] = b[2];
+  a[3] = b[3];
+}
+
+/* { dg-final { scan-assembler-times {fcvtzs\tv[0-9]+.4s, v[0-9]+.4s} 1 } } */
+/* { dg-final { scan-assembler-times {xtn\tv[0-9]+.4h, v[0-9]+.4s} 1 } } */
-- 
2.17.1

Reply via email to