[AArch64] Unit test for trunc lowering from i64 to i8

[nilanjana_basu]

Summary

This test is added so that it can be updated when 'trunc' lowering is extended to use AArch64 'tbl' instructions.

[fhahn]

Thanks for the patch! Could you also adda additional tests with trunc to odd types, something like <8 x i11> and <9 x i8>, to make sure we don't crash on those?

[nilanjana_basu]

Added two test cases for <11 x i64> to <11 x i8> & <8 x i19> to <8 x i8>.

[fhahn]

LGTM, thanks!

llvm/test/CodeGen/AArch64/trunc-to-tbl.ll

320

nit: remove newline here

384

nit: remove newline here

464

nit: remove newline here

484

nit: remove newline here

548

nit: remove newline here

[nilanjana_basu]

Removed the extra newlines

[nilanjana_basu]

Removed extra newlines

llvm/test/CodeGen/AArch64/trunc-to-tbl.ll

loop:
   store <8 x i8> %trunc, ptr %gep.dst
   %iv.next = add i64 %iv, 1
   %ec = icmp eq i64 %iv.next, 1000
   br i1 %ec, label %loop, label %exit
 
 exit:
   ret void
 }
+
+define void @trunc_v16i64_to_v16i8_in_loop(ptr %A, ptr %dst) {
+; CHECK-LABEL:  trunc_v16i64_to_v16i8_in_loop:
+; CHECK:    ; %bb.0:                                ; %entry
+; CHECK-NEXT:     mov x8, xzr
+; CHECK-NEXT:   LBB3_1:                                 ; %loop
+; CHECK-NEXT:                                           ; =>This Inner Loop Header: Depth=1
+; CHECK-NEXT:     add x9, x0, x8, lsl #7
+; CHECK-NEXT:     ldp q3, q2, [x9, #96]
+; CHECK-NEXT:     ldp q1, q0, [x9, #32]
+; CHECK-NEXT:     uzp1.4s v2, v3, v2
+; CHECK-NEXT:     ldp q5, q4, [x9, #64]
+; CHECK-NEXT:     uzp1.4s v0, v1, v0
+; CHECK-NEXT:     ldp q3, q6, [x9]
+; CHECK-NEXT:     uzp1.4s v4, v5, v4
+; CHECK-NEXT:     uzp1.8h v2, v4, v2
+; CHECK-NEXT:     uzp1.4s v1, v3, v6
+; CHECK-NEXT:     uzp1.8h v0, v1, v0
+; CHECK-NEXT:     uzp1.16b  v0, v0, v2
+; CHECK-NEXT:     str q0, [x1, x8, lsl  #4]
+; CHECK-NEXT:     add x8, x8, #1
+; CHECK-NEXT:     cmp x8, #1000
+; CHECK-NEXT:     b.eq  LBB3_1
+; CHECK-NEXT:   ; %bb.2:                                ; %exit
+; CHECK-NEXT:     ret
+
+; CHECK-BE-LABEL:  trunc_v16i64_to_v16i8_in_loop:
+; CHECK-BE:  // %bb.0:                               // %entry
+; CHECK-BE-NEXT:      mov x8, xzr
+; CHECK-BE-NEXT:   .LBB3_1:                                // %loop
+; CHECK-BE-NEXT:                                            // =>This Inner Loop Header: Depth=1
+; CHECK-BE-NEXT:      add x9, x0, x8, lsl #7
+; CHECK-BE-NEXT:      add x10, x9, #48
+; CHECK-BE-NEXT:      add x11, x9, #32
+; CHECK-BE-NEXT:      ld1 { v5.2d }, [x9]
+; CHECK-BE-NEXT:      ld1 { v0.2d }, [x10]
+; CHECK-BE-NEXT:      add x10, x9, #80
+; CHECK-BE-NEXT:      ld1 { v1.2d }, [x11]
+; CHECK-BE-NEXT:      add x11, x9, #112
+; CHECK-BE-NEXT:      ld1 { v2.2d }, [x10]
+; CHECK-BE-NEXT:      add x10, x9, #96
+; CHECK-BE-NEXT:      ld1 { v3.2d }, [x11]
+; CHECK-BE-NEXT:      uzp1  v0.4s, v1.4s, v0.4s
+; CHECK-BE-NEXT:      ld1 { v4.2d }, [x10]
+; CHECK-BE-NEXT:      add x10, x9, #64
+; CHECK-BE-NEXT:      add x9, x9, #16
+; CHECK-BE-NEXT:      ld1 { v6.2d }, [x10]
+; CHECK-BE-NEXT:      ld1 { v7.2d }, [x9]
+; CHECK-BE-NEXT:      add x9, x1, x8, lsl #4
+; CHECK-BE-NEXT:      uzp1  v3.4s, v4.4s, v3.4s
+; CHECK-BE-NEXT:      add x8, x8, #1
+; CHECK-BE-NEXT:      cmp x8, #1000
+; CHECK-BE-NEXT:      uzp1  v2.4s, v6.4s, v2.4s
+; CHECK-BE-NEXT:      uzp1  v1.4s, v5.4s, v7.4s
+; CHECK-BE-NEXT:      uzp1  v2.8h, v2.8h, v3.8h
+; CHECK-BE-NEXT:      uzp1  v0.8h, v1.8h, v0.8h
+; CHECK-BE-NEXT:      uzp1  v0.16b, v0.16b, v2.16b
+; CHECK-BE-NEXT:      st1 { v0.16b }, [x9]
+; CHECK-BE-NEXT:      b.eq  .LBB3_1
+; CHECK-BE-NEXT:    // %bb.2:                               // %exit
+; CHECK-BE-NEXT:      ret
+
+entry:
+  br label %loop
+
+loop:
+  %iv = phi i64 [ 0, %entry ], [ %iv.next, %loop ]
+  %gep.A = getelementptr inbounds <16 x i64>, ptr %A, i64 %iv
+  %l.A = load <16 x i64>, ptr %gep.A
+  %trunc = trunc <16 x i64> %l.A to <16 x i8>
+  %gep.dst = getelementptr inbounds <16 x i8>, ptr %dst, i64 %iv
+  store <16 x i8> %trunc, ptr %gep.dst
+  %iv.next = add i64 %iv, 1
+  %ec = icmp eq i64 %iv.next, 1000
+  br i1 %ec, label %loop, label %exit
+
+exit:
+  ret void
+}
+
+define void @trunc_v8i64_to_v8i8_in_loop(ptr %A, ptr %dst) {
+; CHECK-LABEL:  trunc_v8i64_to_v8i8_in_loop:
+; CHECK:    ; %bb.0:                                ; %entry
+; CHECK-NEXT:   mov x8, xzr
+; CHECK-NEXT: LBB4_1:                                 ; %loop
+; CHECK-NEXT:                                          ; =>This Inner Loop Header: Depth=1
+; CHECK-NEXT:   add x9, x0, x8, lsl #6
+; CHECK-NEXT:   ldp q1, q0, [x9, #32]
+; CHECK-NEXT:   ldp q3, q2, [x9]
+; CHECK-NEXT:   uzp1.4s v0, v1, v0
+; CHECK-NEXT:   uzp1.4s v1, v3, v2
+; CHECK-NEXT:   uzp1.8h v0, v1, v0
+; CHECK-NEXT:   xtn.8b  v0, v0
+; CHECK-NEXT:   str d0, [x1, x8, lsl  #3]
+; CHECK-NEXT:   add x8, x8, #1
+; CHECK-NEXT:   cmp x8, #1000
+; CHECK-NEXT:   b.eq  LBB4_1
+; CHECK-NEXT:  ; %bb.2:                                ; %exit
+; CHECK-NEXT:   ret
+
+; CHECK-BE-LABEL:  trunc_v8i64_to_v8i8_in_loop:
+; CHECK-BE:  // %bb.0:                               // %entry
+; CHECK-BE-NEXT:    mov x8, xzr
+; CHECK-BE-NEXT: .LBB4_1:                                // %loop
+; CHECK-BE-NEXT:                                         // =>This Inner Loop Header: Depth=1
+; CHECK-BE-NEXT:   add  x9, x0, x8, lsl #6
+; CHECK-BE-NEXT:   add  x10, x9, #48
+; CHECK-BE-NEXT:   ld1  { v1.2d }, [x9]
+; CHECK-BE-NEXT:   ld1  { v0.2d }, [x10]
+; CHECK-BE-NEXT:   add  x10, x9, #32
+; CHECK-BE-NEXT:   add  x9, x9, #16
+; CHECK-BE-NEXT:   ld1  { v2.2d }, [x10]
+; CHECK-BE-NEXT:   ld1  { v3.2d }, [x9]
+; CHECK-BE-NEXT:   add  x9, x1, x8, lsl #3
+; CHECK-BE-NEXT:   add  x8, x8, #1
+; CHECK-BE-NEXT:   cmp  x8, #1000
+; CHECK-BE-NEXT:   uzp1 v0.4s, v2.4s, v0.4s
+; CHECK-BE-NEXT:   uzp1 v1.4s, v1.4s, v3.4s
+; CHECK-BE-NEXT:   uzp1 v0.8h, v1.8h, v0.8h
+; CHECK-BE-NEXT:   xtn  v0.8b, v0.8h
+; CHECK-BE-NEXT:   st1  { v0.8b }, [x9]
+; CHECK-BE-NEXT:   b.eq .LBB4_1
+; CHECK-BE-NEXT:   // %bb.2:                               // %exit
+; CHECK-BE-NEXT:    ret
+
+entry:
+  br label %loop
+
+loop:
+  %iv = phi i64 [ 0, %entry ], [ %iv.next, %loop ]
+  %gep.A = getelementptr inbounds <8 x i64>, ptr %A, i64 %iv
+  %l.A = load <8 x i64>, ptr %gep.A
+  %trunc = trunc <8 x i64> %l.A to <8 x i8>
+  %gep.dst = getelementptr inbounds <8 x i8>, ptr %dst, i64 %iv
+  store <8 x i8> %trunc, ptr %gep.dst
+  %iv.next = add i64 %iv, 1
+  %ec = icmp eq i64 %iv.next, 1000
+  br i1 %ec, label %loop, label %exit
+
+exit:
+  ret void
+}
+
+define void @trunc_v8i19_to_v8i8_in_loop(ptr %A, ptr %dst) {
+; CHECK-LABEL:  trunc_v8i19_to_v8i8_in_loop:
+; CHECK:    ; %bb.0:                                ; %entry
+; CHECK-NEXT:   mov x8, xzr
+; CHECK-NEXT:  LBB5_1:                                 ; %loop
+; CHECK-NEXT:                                          ; =>This Inner Loop Header: Depth=1
+; CHECK-NEXT:   ldp x10, x9, [x0]
+; CHECK-NEXT:   ldrb  w11, [x0, #18]
+; CHECK-NEXT:   ldrh  w13, [x0, #16]
+; CHECK-NEXT:   add x0, x0, #32
+; CHECK-NEXT:   lsr x14, x10, #19
+; CHECK-NEXT:   fmov  s0, w10
+; CHECK-NEXT:   ubfx  x12, x9, #12, #20
+; CHECK-NEXT:   lsr x15, x9, #31
+; CHECK-NEXT:   bfi w13, w11, #16, #8
+; CHECK-NEXT:   lsr x11, x9, #50
+; CHECK-NEXT:   mov.s v0[1], w14
+; CHECK-NEXT:   fmov  s1, w12
+; CHECK-NEXT:   lsr x12, x10, #38
+; CHECK-NEXT:   bfi w11, w13, #14, #18
+; CHECK-NEXT:   lsr x10, x10, #57
+; CHECK-NEXT:   bfi w10, w9, #7, #25
+; CHECK-NEXT:   lsr w9, w13, #5
+; CHECK-NEXT:   mov.s v1[1], w15
+; CHECK-NEXT:   mov.s v0[2], w12
+; CHECK-NEXT:   mov.s v1[2], w11
+; CHECK-NEXT:   mov.s v0[3], w10
+; CHECK-NEXT:   mov.s v1[3], w9
+; CHECK-NEXT:   uzp1.8h v0, v0, v1
+; CHECK-NEXT:   xtn.8b  v0, v0
+; CHECK-NEXT:   str d0, [x1, x8, lsl  #3]
+; CHECK-NEXT:   add x8, x8, #1
+; CHECK-NEXT:   cmp x8, #1000
+; CHECK-NEXT:   b.eq  LBB5_1
+; CHECK-NEXT:  ; %bb.2:                                ; %exit
+; CHECK-NEXT:   ret
+
+; CHECK-BE-LABEL:  trunc_v8i19_to_v8i8_in_loop:
+; CHECK-BE:  // %bb.0:                               // %entry
+; CHECK-BE-NEXT:    mov x8, xzr
+; CHECK-BE-NEXT:   .LBB5_1:                                // %loop
+; CHECK-BE-NEXT:                                           // =>This Inner Loop Header: Depth=1
+; CHECK-BE-NEXT:    ldp x10, x9, [x0]
+; CHECK-BE-NEXT:    ldrh  w15, [x0, #16]
+; CHECK-BE-NEXT:    lsr x12, x10, #40
+; CHECK-BE-NEXT:    lsr x13, x10, #45
+; CHECK-BE-NEXT:    lsr x11, x9, #40
+; CHECK-BE-NEXT:    ubfx  x14, x9, #33, #7
+; CHECK-BE-NEXT:    ubfx  x16, x10, #26, #14
+; CHECK-BE-NEXT:    bfi w16, w12, #14, #18
+; CHECK-BE-NEXT:    ubfx  x12, x9, #14, #18
+; CHECK-BE-NEXT:    bfi w14, w11, #7, #24
+; CHECK-BE-NEXT:    ldrb  w11, [x0, #18]
+; CHECK-BE-NEXT:    fmov  s0, w13
+; CHECK-BE-NEXT:    add x0, x0, #32
+; CHECK-BE-NEXT:    fmov  s1, w14
+; CHECK-BE-NEXT:    bfi w11, w15, #8, #16
+; CHECK-BE-NEXT:    mov v0.s[1], w16
+; CHECK-BE-NEXT:    mov v1.s[1], w12
+; CHECK-BE-NEXT:    extr  x12, x10, x9, #40
+; CHECK-BE-NEXT:    lsl x9, x9, #24
+; CHECK-BE-NEXT:    ubfx  x10, x10, #7, #25
+; CHECK-BE-NEXT:    orr w9, w11, w9
+; CHECK-BE-NEXT:    lsr w9, w9, #19
+; CHECK-BE-NEXT:    mov v0.s[2], w10
+; CHECK-BE-NEXT:    ubfx  x10, x12, #12, #20
+; CHECK-BE-NEXT:    mov v1.s[2], w9
+; CHECK-BE-NEXT:    add x9, x1, x8, lsl #3
+; CHECK-BE-NEXT:    add x8, x8, #1
+; CHECK-BE-NEXT:    mov v0.s[3], w10
+; CHECK-BE-NEXT:    cmp x8, #1000
+; CHECK-BE-NEXT:    mov v1.s[3], w11
+; CHECK-BE-NEXT:    uzp1  v0.8h, v0.8h, v1.8h
+; CHECK-BE-NEXT:    xtn v0.8b, v0.8h
+; CHECK-BE-NEXT:    st1 { v0.8b }, [x9]
+; CHECK-BE-NEXT:    b.eq  .LBB5_1
+; CHECK-BE-NEXT:   // %bb.2:                               // %exit
+; CHECK-BE-NEXT:    ret
+
+entry:
+  br label %loop
+
+loop:
+  %iv = phi i64 [ 0, %entry ], [ %iv.next, %loop ]
+  %gep.A = getelementptr inbounds <8 x i19>, ptr %A, i64 %iv
+  %l.A = load <8 x i19>, ptr %gep.A
+  %trunc = trunc <8 x i19> %l.A to <8 x i8>
+  %gep.dst = getelementptr inbounds <8 x i8>, ptr %dst, i64 %iv
+  store <8 x i8> %trunc, ptr %gep.dst
+  %iv.next = add i64 %iv, 1
+  %ec = icmp eq i64 %iv.next, 1000
+  br i1 %ec, label %loop, label %exit
+
+exit:
+  ret void
+}
+
+define void @trunc_v11i64_to_v11i8_in_loop(ptr %A, ptr %dst) {
+; CHECK-LABEL:  trunc_v11i64_to_v11i8_in_loop:
+; CHECK:    ; %bb.0:                                ; %entry
+; CHECK-NEXT:   mov w8, #1000
+; CHECK-NEXT:  LBB6_1:                                 ; %loop
+; CHECK-NEXT:                                          ; =>This Inner Loop Header: Depth=1
+; CHECK-NEXT:   ldp q1, q0, [x0, #32]
+; CHECK-NEXT:   add x9, x1, #8
+; CHECK-NEXT:   add x10, x1, #10
+; CHECK-NEXT:   subs  x8, x8, #1
+; CHECK-NEXT:   ldp q3, q2, [x0]
+; CHECK-NEXT:   uzp1.4s v0, v1, v0
+; CHECK-NEXT:   ldr d4, [x0, #80]
+; CHECK-NEXT:   ldr q1, [x0, #64]
+; CHECK-NEXT:   add x0, x0, #128
+; CHECK-NEXT:   uzp1.4s v2, v3, v2
+; CHECK-NEXT:   uzp1.4s v1, v1, v4
+; CHECK-NEXT:   uzp1.8h v0, v2, v0
+; CHECK-NEXT:   xtn.4h  v1, v1
+; CHECK-NEXT:   uzp1.16b  v0, v0, v1
+; CHECK-NEXT:   xtn.8b  v1, v1
+; CHECK-NEXT:   st1.b { v1 }[2], [x10]
+; CHECK-NEXT:   str d0, [x1], #16
+; CHECK-NEXT:   st1.h { v0 }[4], [x9]
+; CHECK-NEXT:   b.eq  LBB6_1
+; CHECK-NEXT:  ; %bb.2:                                ; %exit
+; CHECK-NEXT:   ret
+
+; CHECK-BE-LABEL: trunc_v11i64_to_v11i8_in_loop: 
+; CHECK-BE:  // %bb.0:                               // %entry
+; CHECK-BE-NEXT:  mov w8, #1000
+; CHECK-BE-NEXT:.LBB6_1:                                // %loop
+; CHECK-BE-NEXT:                                        // =>This Inner Loop Header: Depth=1
+; CHECK-BE-NEXT:  add x9, x0, #48
+; CHECK-BE-NEXT:  add x10, x0, #32
+; CHECK-BE-NEXT:  ld1 { v0.2d }, [x0]
+; CHECK-BE-NEXT:  subs  x8, x8, #1
+; CHECK-BE-NEXT:  ld1 { v1.2d }, [x9]
+; CHECK-BE-NEXT:  add x9, x0, #16
+; CHECK-BE-NEXT:  ld1 { v2.2d }, [x10]
+; CHECK-BE-NEXT:  add x10, x0, #64
+; CHECK-BE-NEXT:  ld1 { v3.2d }, [x9]
+; CHECK-BE-NEXT:  add x9, x1, #10
+; CHECK-BE-NEXT:  ld1 { v4.2d }, [x10]
+; CHECK-BE-NEXT:  add x10, x1, #8
+; CHECK-BE-NEXT:  uzp1  v1.4s, v2.4s, v1.4s
+; CHECK-BE-NEXT:  ldr d2, [x0, #80]
+; CHECK-BE-NEXT:  add x0, x0, #128
+; CHECK-BE-NEXT:  uzp1  v0.4s, v0.4s, v3.4s
+; CHECK-BE-NEXT:  uzp1  v2.4s, v4.4s, v2.4s
+; CHECK-BE-NEXT:  uzp1  v0.8h, v0.8h, v1.8h
+; CHECK-BE-NEXT:  xtn v1.4h, v2.4s
+; CHECK-BE-NEXT:  uzp1  v0.16b, v0.16b, v1.16b
+; CHECK-BE-NEXT:  xtn v1.8b, v1.8h
+; CHECK-BE-NEXT:  st1 { v1.b }[2], [x9]
+; CHECK-BE-NEXT:  rev64 v2.16b, v0.16b
+; CHECK-BE-NEXT:  rev16 v0.16b, v0.16b
+; CHECK-BE-NEXT:  str d2, [x1], #16
+; CHECK-BE-NEXT:  st1 { v0.h }[4], [x10]
+; CHECK-BE-NEXT:  b.eq  .LBB6_1
+; CHECK-BE-NEXT:// %bb.2:                               // %exit
+; CHECK-BE-NEXT:  ret
+
+entry:
+  br label %loop
+
+loop:
+  %iv = phi i64 [ 0, %entry ], [ %iv.next, %loop ]
+  %gep.A = getelementptr inbounds <11 x i64>, ptr %A, i64 %iv
+  %l.A = load <11 x i64>, ptr %gep.A
+  %trunc = trunc <11 x i64> %l.A to <11 x i8>
+  %gep.dst = getelementptr inbounds <11 x i8>, ptr %dst, i64 %iv
+  store <11 x i8> %trunc, ptr %gep.dst
+  %iv.next = add i64 %iv, 1
+  %ec = icmp eq i64 %iv.next, 1000
+  br i1 %ec, label %loop, label %exit
+
+exit:
+  ret void
+}