i386: Fixed vec_init_dup_v16bf [PR106887]

gcc/ChangeLog:

	PR target/106887
	* config/i386/i386-expand.cc (ix86_expand_vector_init_duplicate):
	Fixed V16BF mode case.

gcc/testsuite/ChangeLog:

	PR target/106887
	* gcc.target/i386/vect-bfloat16-2c.c: New test.

diff --git a/gcc/config/i386/i386-expand.cc b/gcc/config/i386/i386-expand.cc
index d7b49c99dc8399a8f689b1c9e17c5a7cd0114731..5334363e235bd37bd352edf11cbcdbd5363e8863 100644 (file)
--- a/gcc/config/i386/i386-expand.cc
+++ b/gcc/config/i386/i386-expand.cc
@@ -15109,9 +15109,24 @@ ix86_expand_vector_init_duplicate (bool mmx_ok, machine_mode mode,
        return ix86_vector_duplicate_value (mode, target, val);
       else
        {
-         machine_mode hvmode = (mode == V16HImode ? V8HImode
-                                : mode == V16HFmode ? V8HFmode
-                                : V16QImode);
+         machine_mode hvmode;
+         switch (mode)
+           {
+           case V16HImode:
+             hvmode = V8HImode;
+             break;
+           case V16HFmode:
+             hvmode = V8HFmode;
+             break;
+           case V16BFmode:
+             hvmode = V8BFmode;
+             break;
+           case V32QImode:
+             hvmode = V16QImode;
+             break;
+           default:
+             gcc_unreachable ();
+           }
          rtx x = gen_reg_rtx (hvmode);
 
          ok = ix86_expand_vector_init_duplicate (false, hvmode, x, val);
@@ -15130,10 +15145,24 @@ ix86_expand_vector_init_duplicate (bool mmx_ok, machine_mode mode,
        return ix86_vector_duplicate_value (mode, target, val);
       else
        {
-         machine_mode hvmode = (mode == V32HImode ? V16HImode
-                                : mode == V32HFmode ? V16HFmode
-                                : mode == V32BFmode ? V16BFmode
-                                : V32QImode);
+         machine_mode hvmode;
+         switch (mode)
+           {
+           case V32HImode:
+             hvmode = V16HImode;
+             break;
+           case V32HFmode:
+             hvmode = V16HFmode;
+             break;
+           case V32BFmode:
+             hvmode = V16BFmode;
+             break;
+           case V64QImode:
+             hvmode = V32QImode;
+             break;
+           default:
+             gcc_unreachable ();
+           }
          rtx x = gen_reg_rtx (hvmode);
 
          ok = ix86_expand_vector_init_duplicate (false, hvmode, x, val);

diff --git a/gcc/testsuite/gcc.target/i386/vect-bfloat16-2c.c b/gcc/testsuite/gcc.target/i386/vect-bfloat16-2c.c
new file mode 100644 (file)
index 0000000..bead94e
--- /dev/null
+++ b/gcc/testsuite/gcc.target/i386/vect-bfloat16-2c.c
@@ -0,0 +1,76 @@
+/* { dg-do compile } */
+/* { dg-options "-mf16c -msse2 -mno-avx2 -O2" } */
+
+typedef __bf16 v8bf __attribute__ ((__vector_size__ (16)));
+typedef __bf16 v16bf __attribute__ ((__vector_size__ (32)));
+
+#define VEC_EXTRACT(V,S,IDX)                   \
+  S                                            \
+  __attribute__((noipa))                       \
+  vec_extract_##V##_##IDX (V v)                        \
+  {                                            \
+    return v[IDX];                             \
+  }
+
+#define VEC_SET(V,S,IDX)                       \
+  V                                            \
+  __attribute__((noipa))                       \
+  vec_set_##V##_##IDX (V v, S s)               \
+  {                                            \
+    v[IDX] = s;                                \
+    return v;                                  \
+  }
+
+v8bf
+vec_init_v8bf (__bf16 a1, __bf16 a2, __bf16 a3, __bf16 a4,
+              __bf16 a5,  __bf16 a6, __bf16 a7, __bf16 a8)
+{
+    return __extension__ (v8bf) {a1, a2, a3, a4, a5, a6, a7, a8};
+}
+
+v16bf
+vec_init_v16bf (__bf16 a1, __bf16 a2, __bf16 a3, __bf16 a4,
+              __bf16 a5,  __bf16 a6, __bf16 a7, __bf16 a8,
+              __bf16 a9,  __bf16 a10, __bf16 a11, __bf16 a12,
+              __bf16 a13,  __bf16 a14, __bf16 a15, __bf16 a16)
+{
+    return __extension__ (v16bf) {a1, a2, a3, a4, a5, a6, a7, a8,
+                                 a9, a10, a11, a12, a13, a14, a15, a16};
+}
+
+v8bf
+vec_init_dup_v8bf (__bf16 a1)
+{
+    return __extension__ (v8bf) {a1, a1, a1, a1, a1, a1, a1, a1};
+}
+
+v16bf
+vec_init_dup_v16bf (__bf16 a1)
+{
+    return __extension__ (v16bf) {a1, a1, a1, a1, a1, a1, a1, a1,
+                                 a1, a1, a1, a1, a1, a1, a1, a1};
+}
+
+/* { dg-final { scan-assembler-times "vpunpcklwd" 12 } } */
+/* { dg-final { scan-assembler-times "vpunpckldq" 6 } } */
+/* { dg-final { scan-assembler-times "vpunpcklqdq" 3 } } */
+
+VEC_EXTRACT (v8bf, __bf16, 0);
+VEC_EXTRACT (v8bf, __bf16, 4);
+VEC_EXTRACT (v16bf, __bf16, 0);
+VEC_EXTRACT (v16bf, __bf16, 3);
+VEC_EXTRACT (v16bf, __bf16, 8);
+VEC_EXTRACT (v16bf, __bf16, 15);
+/* { dg-final { scan-assembler-times "vpsrldq\[\t ]*\\\$8" 1 } } */
+/* { dg-final { scan-assembler-times "vpsrldq\[\t ]*\\\$6" 1 } } */
+/* { dg-final { scan-assembler-times "vpsrldq\[\t ]*\\\$14" 1 } } */
+/* { dg-final { scan-assembler-times "vextract" 4 } } */
+
+VEC_SET (v8bf, __bf16, 4);
+VEC_SET (v16bf, __bf16, 3);
+VEC_SET (v16bf, __bf16, 8);
+VEC_SET (v16bf, __bf16, 15);
+/* { dg-final { scan-assembler-times "vpblendw" 3 { target { ! ia32 } } } } */
+
+/* { dg-final { scan-assembler-times "vpinsrw" 30 { target ia32 } } } */
+