llvm/test/CodeGen/X86/fold-broadcast.ll

   1 ; NOTE: Assertions have been autogenerated by utils/update_llc_test_checks.py UTC_ARGS: --version 4
   2 ; RUN: llc -mtriple=x86_64-unknown-unknown -mattr=+avx512bf16 < %s | FileCheck %s
   3
   4 define <16 x float> @mm512_dpbf16_ps_broadcast_rhs(<16 x float> noundef %acc, <32 x bfloat> noundef %lhs, ptr nocapture noundef readonly %rhs) {
   5 ; CHECK-LABEL: mm512_dpbf16_ps_broadcast_rhs:
   6 ; CHECK:       # %bb.0: # %entry
   7 ; CHECK-NEXT:    vdpbf16ps (%rdi){1to16}, %zmm1, %zmm0
   8 ; CHECK-NEXT:    retq
   9 entry:
  10   %0 = load float, ptr %rhs, align 4
  11   %vecinit.i = insertelement <16 x float> poison, float %0, i64 0
  12   %vecinit15.i = shufflevector <16 x float> %vecinit.i, <16 x float> poison, <16 x i32> zeroinitializer
  13   %1 = bitcast <16 x float> %vecinit15.i to <32 x bfloat>
  14   %2 = tail call <16 x float> @llvm.x86.avx512bf16.dpbf16ps.512(<16 x float> %acc, <32 x bfloat> %lhs, <32 x bfloat> %1)
  15   ret <16 x float> %2
  16 }
  17
  18 declare <16 x float> @llvm.x86.avx512bf16.dpbf16ps.512(<16 x float>, <32 x bfloat>, <32 x bfloat>)