llvm/test/CodeGen/AArch64/bf16.ll

   1 ; RUN: llc < %s -asm-verbose=0 -mtriple=arm64-eabi -mattr=+bf16 | FileCheck %s
   2 ; RUN: llc < %s -asm-verbose=0 -mtriple=aarch64 -mattr=+bf16 | FileCheck %s
   3
   4 ; test argument passing and simple load/store
   5
   6 define bfloat @test_load(ptr %p) nounwind {
   7 ; CHECK-LABEL: test_load:
   8 ; CHECK-NEXT: ldr h0, [x0]
   9 ; CHECK-NEXT: ret
  10   %tmp1 = load bfloat, ptr %p, align 16
  11   ret bfloat %tmp1
  12 }
  13
  14 define bfloat @test_load_offset1(ptr %p) nounwind {
  15 ; CHECK-LABEL: test_load_offset1:
  16 ; CHECK-NEXT: ldur h0, [x0, #1]
  17 ; CHECK-NEXT: ret
  18   %g = getelementptr inbounds i8, ptr %p, i64 1
  19   %tmp1 = load bfloat, ptr %g, align 2
  20   ret bfloat %tmp1
  21 }
  22
  23 define bfloat @test_load_offset2(ptr %p) nounwind {
  24 ; CHECK-LABEL: test_load_offset2:
  25 ; CHECK-NEXT: ldr h0, [x0, #2]
  26 ; CHECK-NEXT: ret
  27   %g = getelementptr inbounds i8, ptr %p, i64 2
  28   %tmp1 = load bfloat, ptr %g, align 2
  29   ret bfloat %tmp1
  30 }
  31
  32 define <4 x bfloat> @test_vec_load(ptr %p) nounwind {
  33 ; CHECK-LABEL: test_vec_load:
  34 ; CHECK-NEXT: ldr d0, [x0]
  35 ; CHECK-NEXT: ret
  36   %tmp1 = load <4 x bfloat>, ptr %p, align 16
  37   ret <4 x bfloat> %tmp1
  38 }
  39
  40 define void @test_store(ptr %a, bfloat %b) nounwind {
  41 ; CHECK-LABEL: test_store:
  42 ; CHECK-NEXT: str h0, [x0]
  43 ; CHECK-NEXT: ret
  44   store bfloat %b, ptr %a, align 16
  45   ret void
  46 }
  47
  48 define void @test_store_negative_offset(ptr %a, bfloat %b) nounwind {
  49 ; CHECK-LABEL: test_store_negative_offset:
  50 ; CHECK-NEXT: stur    h0, [x0, #-4]
  51 ; CHECK-NEXT: ret
  52 entry:
  53   %0 = getelementptr inbounds bfloat, ptr %a, i64 -2
  54   store bfloat %b, ptr %0, align 2
  55   ret void
  56 }
  57
  58 ; Simple store of v4bf16
  59 define void @test_vec_store(ptr %a, <4 x bfloat> %b) nounwind {
  60 ; CHECK-LABEL: test_vec_store:
  61 ; CHECK-NEXT: str d0, [x0]
  62 ; CHECK-NEXT: ret
  63 entry:
  64   store <4 x bfloat> %b, ptr %a, align 16
  65   ret void
  66 }
  67
  68 define <8 x bfloat> @test_build_vector_const() {
  69 ; CHECK-LABEL: test_build_vector_const:
  70 ; CHECK: mov [[TMP:w[0-9]+]], #16256
  71 ; CHECK: dup v0.8h, [[TMP]]
  72   ret  <8 x bfloat> <bfloat 0xR3F80, bfloat 0xR3F80, bfloat 0xR3F80, bfloat 0xR3F80, bfloat 0xR3F80, bfloat 0xR3F80, bfloat 0xR3F80, bfloat 0xR3F80>
  73 }
  74
  75 define { bfloat, ptr } @test_store_post(bfloat %val, ptr %ptr) {
  76 ; CHECK-LABEL: test_store_post:
  77 ; CHECK: str h0, [x0], #2
  78
  79   store bfloat %val, ptr %ptr
  80   %res.tmp = insertvalue { bfloat, ptr } undef, bfloat %val, 0
  81
  82   %next = getelementptr bfloat, ptr %ptr, i32 1
  83   %res = insertvalue { bfloat, ptr } %res.tmp, ptr %next, 1
  84
  85   ret { bfloat, ptr } %res
  86 }
  87
  88 define { <4 x bfloat>, ptr } @test_store_post_v4bf16(<4 x bfloat> %val, ptr %ptr) {
  89 ; CHECK-LABEL: test_store_post_v4bf16:
  90 ; CHECK: str d0, [x0], #8
  91
  92   store <4 x bfloat> %val, ptr %ptr
  93   %res.tmp = insertvalue { <4 x bfloat>, ptr } undef, <4 x bfloat> %val, 0
  94
  95   %next = getelementptr <4 x bfloat>, ptr %ptr, i32 1
  96   %res = insertvalue { <4 x bfloat>, ptr } %res.tmp, ptr %next, 1
  97
  98   ret { <4 x bfloat>, ptr } %res
  99 }
 100
 101 define { <8 x bfloat>, ptr } @test_store_post_v8bf16(<8 x bfloat> %val, ptr %ptr) {
 102 ; CHECK-LABEL: test_store_post_v8bf16:
 103 ; CHECK: str q0, [x0], #16
 104
 105   store <8 x bfloat> %val, ptr %ptr
 106   %res.tmp = insertvalue { <8 x bfloat>, ptr } undef, <8 x bfloat> %val, 0
 107
 108   %next = getelementptr <8 x bfloat>, ptr %ptr, i32 1
 109   %res = insertvalue { <8 x bfloat>, ptr } %res.tmp, ptr %next, 1
 110
 111   ret { <8 x bfloat>, ptr } %res
 112 }
 113
 114 define bfloat @test_bitcast_halftobfloat(half %a) nounwind {
 115 ; CHECK-LABEL: test_bitcast_halftobfloat:
 116 ; CHECK-NEXT: ret
 117   %r = bitcast half %a to bfloat
 118   ret bfloat %r
 119 }
 120
 121 define half @test_bitcast_bfloattohalf(bfloat %a) nounwind {
 122 ; CHECK-LABEL: test_bitcast_bfloattohalf:
 123 ; CHECK-NEXT: ret
 124   %r = bitcast bfloat %a to half
 125   ret half %r
 126 }