llvm/test/CodeGen/AMDGPU/lshl-add-u64.ll

   1 ; RUN: llc -mtriple=amdgcn -mcpu=gfx940 -verify-machineinstrs < %s | FileCheck -check-prefix=GCN %s
   2
   3 define i64 @lshl_add_u64_v1v(i64 %v, i64 %a) {
   4 ; GCN-LABEL: lshl_add_u64_v1v:
   5 ; GCN: v_lshl_add_u64 v[{{[0-9:]+}}], v[{{[0-9:]+}}], 1, v[{{[0-9:]+}}]
   6   %shl = shl i64 %v, 1
   7   %add = add i64 %shl, %a
   8   ret i64 %add
   9 }
  10
  11 define i64 @lshl_add_u64_v4v(i64 %v, i64 %a) {
  12 ; GCN-LABEL: lshl_add_u64_v4v:
  13 ; GCN: v_lshl_add_u64 v[{{[0-9:]+}}], v[{{[0-9:]+}}], 4, v[{{[0-9:]+}}]
  14   %shl = shl i64 %v, 4
  15   %add = add i64 %shl, %a
  16   ret i64 %add
  17 }
  18
  19 define i64 @lshl_add_u64_v5v(i64 %v, i64 %a) {
  20 ; GCN-LABEL: lshl_add_u64_v5v:
  21 ; GCN:      v_lshlrev_b64
  22 ; GCN-NEXT: v_lshl_add_u64 v[{{[0-9:]+}}], v[{{[0-9:]+}}], 0, v[{{[0-9:]+}}]
  23   %shl = shl i64 %v, 5
  24   %add = add i64 %shl, %a
  25   ret i64 %add
  26 }
  27
  28 define i64 @lshl_add_u64_vvv(i64 %v, i64 %s, i64 %a) {
  29 ; GCN-LABEL: lshl_add_u64_vvv:
  30 ; GCN:      v_lshlrev_b64
  31 ; GCN-NEXT: v_lshl_add_u64 v[{{[0-9:]+}}], v[{{[0-9:]+}}], 0, v[{{[0-9:]+}}]
  32   %shl = shl i64 %v, %s
  33   %add = add i64 %shl, %a
  34   ret i64 %add
  35 }
  36
  37 define amdgpu_kernel void @lshl_add_u64_s2v(i64 %v) {
  38 ; GCN-LABEL: lshl_add_u64_s2v:
  39 ; GCN: v_lshl_add_u64 v[{{[0-9:]+}}], s[{{[0-9:]+}}], 2, v[{{[0-9:]+}}]
  40   %a = load i64, ptr undef
  41   %shl = shl i64 %v, 2
  42   %add = add i64 %shl, %a
  43   store i64 %add, ptr undef
  44   ret void
  45 }
  46
  47 define amdgpu_kernel void @lshl_add_u64_v2s(i64 %a) {
  48 ; GCN-LABEL: lshl_add_u64_v2s:
  49 ; GCN: v_lshl_add_u64 v[{{[0-9:]+}}], v[{{[0-9:]+}}], 2, s[{{[0-9:]+}}]
  50   %v = load i64, ptr undef
  51   %shl = shl i64 %v, 2
  52   %add = add i64 %shl, %a
  53   store i64 %add, ptr undef
  54   ret void
  55 }
  56
  57 define amdgpu_kernel void @lshl_add_u64_s2s(i64 %v, i64 %a) {
  58 ; GCN-LABEL: lshl_add_u64_s2s:
  59 ; GCN:    s_lshl_b64
  60 ; GCN:    s_add_u32
  61 ; GCN:    s_addc_u32
  62   %shl = shl i64 %v, 2
  63   %add = add i64 %shl, %a
  64   store i64 %add, ptr undef
  65   ret void
  66 }
  67
  68 define i64 @add_u64_vv(i64 %v, i64 %a) {
  69 ; GCN-LABEL: add_u64_vv:
  70 ; GCN: v_lshl_add_u64 v[0:1], v[0:1], 0, v[2:3]
  71   %add = add i64 %v, %a
  72   ret i64 %add
  73 }
  74
  75 define amdgpu_kernel void @add_u64_sv(i64 %v) {
  76 ; GCN-LABEL: add_u64_sv:
  77 ; GCN: v_lshl_add_u64 v[0:1], s[0:1], 0, v[0:1]
  78   %a = load i64, ptr undef
  79   %add = add i64 %v, %a
  80   store i64 %add, ptr undef
  81   ret void
  82 }
  83
  84 define amdgpu_kernel void @add_u64_vs(i64 %a) {
  85 ; GCN-LABEL: add_u64_vs:
  86 ; GCN: v_lshl_add_u64 v[0:1], v[0:1], 0, s[0:1]
  87   %v = load i64, ptr undef
  88   %add = add i64 %v, %a
  89   store i64 %add, ptr undef
  90   ret void
  91 }
  92
  93 define amdgpu_kernel void @add_u64_ss(i64 %v, i64 %a) {
  94 ; GCN-LABEL: add_u64_ss:
  95 ; GCN: s_add_u32
  96 ; GCN: s_addc_u32 s1, s1, s3
  97   %add = add i64 %v, %a
  98   store i64 %add, ptr undef
  99   ret void
 100 }
 101
 102 define i32 @lshl_add_u64_gep(ptr %p, i64 %a) {
 103 ; GCN-LABEL: lshl_add_u64_gep:
 104 ; GCN: v_lshl_add_u64 v[0:1], v[2:3], 2, v[0:1]
 105   %gep = getelementptr inbounds i32, ptr %p, i64 %a
 106   %v = load i32, ptr %gep
 107   ret i32 %v
 108 }