llvm/test/CodeGen/AMDGPU/s_addk_i32.ll

   1 ; RUN: llc -mtriple=amdgcn--amdpal -mcpu=tahiti -verify-machineinstrs < %s | FileCheck -check-prefix=SI %s
   2 ; RUN: llc -mtriple=amdgcn--amdpal -mcpu=tonga -mattr=-flat-for-global,-xnack -verify-machineinstrs < %s | FileCheck -check-prefix=SI %s
   3
   4 ; TODO: Some of those tests fail with OS == amdhsa due to unreasonable register
   5 ;       allocation differences.
   6
   7 ; SI-LABEL: {{^}}s_addk_i32_k0:
   8 ; SI: s_load_dword [[VAL:s[0-9]+]]
   9 ; SI: s_addk_i32 [[VAL]], 0x41
  10 ; SI: v_mov_b32_e32 [[VRESULT:v[0-9]+]], [[VAL]]
  11 ; SI: buffer_store_dword [[VRESULT]]
  12 ; SI: s_endpgm
  13 define amdgpu_kernel void @s_addk_i32_k0(ptr addrspace(1) %out, i32 %b) {
  14   %add = add i32 %b, 65
  15   store i32 %add, ptr addrspace(1) %out
  16   ret void
  17 }
  18
  19 ; SI-LABEL: {{^}}s_addk_i32_k0_x2:
  20 ; SI-DAG: s_addk_i32 {{s[0-9]+}}, 0x41
  21 ; SI-DAG: s_addk_i32 {{s[0-9]+}}, 0x41
  22 ; SI: s_endpgm
  23 define amdgpu_kernel void @s_addk_i32_k0_x2(ptr addrspace(1) %out0, ptr addrspace(1) %out1, i32 %a, i32 %b) {
  24   %add0 = add i32 %a, 65
  25   %add1 = add i32 %b, 65
  26   store i32 %add0, ptr addrspace(1) %out0
  27   store i32 %add1, ptr addrspace(1) %out1
  28   ret void
  29 }
  30
  31 ; SI-LABEL: {{^}}s_addk_i32_k1:
  32 ; SI: s_addk_i32 {{s[0-9]+}}, 0x7fff{{$}}
  33 ; SI: s_endpgm
  34 define amdgpu_kernel void @s_addk_i32_k1(ptr addrspace(1) %out, i32 %b) {
  35   %add = add i32 %b, 32767 ; (1 << 15) - 1
  36   store i32 %add, ptr addrspace(1) %out
  37   ret void
  38 }
  39
  40 ; SI-LABEL: {{^}}s_addk_i32_k2:
  41 ; SI: s_sub_i32 s{{[0-9]+}}, s{{[0-9]+}}, 17
  42 ; SI: s_endpgm
  43 define amdgpu_kernel void @s_addk_i32_k2(ptr addrspace(1) %out, i32 %b) {
  44   %add = add i32 %b, -17
  45   store i32 %add, ptr addrspace(1) %out
  46   ret void
  47 }
  48
  49 ; SI-LABEL: {{^}}s_addk_i32_k3:
  50 ; SI: s_addk_i32 {{s[0-9]+}}, 0xffbf{{$}}
  51 ; SI: s_endpgm
  52 define amdgpu_kernel void @s_addk_i32_k3(ptr addrspace(1) %out, i32 %b) {
  53   %add = add i32 %b, -65
  54   store i32 %add, ptr addrspace(1) %out
  55   ret void
  56 }
  57
  58 ; SI-LABEL: {{^}}s_addk_v2i32_k0:
  59 ; SI-DAG: s_addk_i32 {{s[0-9]+}}, 0x41
  60 ; SI-DAG: s_addk_i32 {{s[0-9]+}}, 0x42
  61 ; SI: s_endpgm
  62 ; Note: dummy argument here to prevent combining of descriptor loads for %out and %b
  63 define amdgpu_kernel void @s_addk_v2i32_k0(ptr addrspace(1) %out, i32 %dummy, <2 x i32> %b) {
  64   %add = add <2 x i32> %b, <i32 65, i32 66>
  65   store <2 x i32> %add, ptr addrspace(1) %out
  66   ret void
  67 }
  68
  69 ; SI-LABEL: {{^}}s_addk_v4i32_k0:
  70 ; SI-DAG: s_addk_i32 {{s[0-9]+}}, 0x41
  71 ; SI-DAG: s_addk_i32 {{s[0-9]+}}, 0x42
  72 ; SI-DAG: s_addk_i32 {{s[0-9]+}}, 0x43
  73 ; SI-DAG: s_addk_i32 {{s[0-9]+}}, 0x44
  74 ; SI: s_endpgm
  75 define amdgpu_kernel void @s_addk_v4i32_k0(ptr addrspace(1) %out, <4 x i32> %b) {
  76   %add = add <4 x i32> %b, <i32 65, i32 66, i32 67, i32 68>
  77   store <4 x i32> %add, ptr addrspace(1) %out
  78   ret void
  79 }
  80
  81 ; SI-LABEL: {{^}}s_addk_v8i32_k0:
  82 ; SI-DAG: s_addk_i32 {{s[0-9]+}}, 0x41
  83 ; SI-DAG: s_addk_i32 {{s[0-9]+}}, 0x42
  84 ; SI-DAG: s_addk_i32 {{s[0-9]+}}, 0x43
  85 ; SI-DAG: s_addk_i32 {{s[0-9]+}}, 0x44
  86 ; SI-DAG: s_addk_i32 {{s[0-9]+}}, 0x45
  87 ; SI-DAG: s_addk_i32 {{s[0-9]+}}, 0x46
  88 ; SI-DAG: s_addk_i32 {{s[0-9]+}}, 0x47
  89 ; SI-DAG: s_addk_i32 {{s[0-9]+}}, 0x48
  90 ; SI: s_endpgm
  91 define amdgpu_kernel void @s_addk_v8i32_k0(ptr addrspace(1) %out, <8 x i32> %b) {
  92   %add = add <8 x i32> %b, <i32 65, i32 66, i32 67, i32 68, i32 69, i32 70, i32 71, i32 72>
  93   store <8 x i32> %add, ptr addrspace(1) %out
  94   ret void
  95 }
  96
  97 ; SI-LABEL: {{^}}no_s_addk_i32_k0:
  98 ; SI: s_add_i32 {{s[0-9]+}}, {{s[0-9]+}}, 0x8000{{$}}
  99 ; SI: s_endpgm
 100 define amdgpu_kernel void @no_s_addk_i32_k0(ptr addrspace(1) %out, i32 %b) {
 101   %add = add i32 %b, 32768 ; 1 << 15
 102   store i32 %add, ptr addrspace(1) %out
 103   ret void
 104 }
 105
 106 @lds = addrspace(3) global [512 x i32] undef, align 4
 107
 108 ; SI-LABEL: {{^}}commute_s_addk_i32:
 109 ; SI: s_addk_i32 s{{[0-9]+}}, 0x800{{$}}
 110 define amdgpu_kernel void @commute_s_addk_i32(ptr addrspace(1) %out, i32 %b) #0 {
 111   %size = call i32 @llvm.amdgcn.groupstaticsize()
 112   %add = add i32 %size, %b
 113   call void asm sideeffect "; foo $0, $1", "v,s"(ptr addrspace(3) @lds, i32 %add)
 114   ret void
 115 }
 116
 117 declare i32 @llvm.amdgcn.groupstaticsize() #1
 118
 119 attributes #0 = { nounwind }
 120 attributes #1 = { nounwind readnone }