llvm/test/CodeGen/AMDGPU/llvm.amdgcn.image.store.a16.ll

   1 ; NOTE: Assertions have been autogenerated by utils/update_llc_test_checks.py
   2 ; RUN: llc -march=amdgcn -mcpu=gfx900 -verify-machineinstrs < %s | FileCheck -check-prefixes=GFX9 %s
   3 ; RUN: llc -march=amdgcn -mcpu=gfx1010 -verify-machineinstrs < %s | FileCheck -check-prefixes=GFX10 %s
   4 ; RUN: llc -march=amdgcn -mcpu=gfx1100 -verify-machineinstrs < %s | FileCheck -check-prefixes=GFX11 %s
   5
   6 define amdgpu_ps void @store_f32_1d(<8 x i32> inreg %rsrc, <2 x i16> %coords, <4 x float> %val) {
   7 ; GFX9-LABEL: store_f32_1d:
   8 ; GFX9:       ; %bb.0: ; %main_body
   9 ; GFX9-NEXT:    image_store v[1:4], v0, s[0:7] dmask:0x1 unorm a16
  10 ; GFX9-NEXT:    s_endpgm
  11 ;
  12 ; GFX10-LABEL: store_f32_1d:
  13 ; GFX10:       ; %bb.0: ; %main_body
  14 ; GFX10-NEXT:    image_store v[1:4], v0, s[0:7] dmask:0x1 dim:SQ_RSRC_IMG_1D unorm a16
  15 ; GFX10-NEXT:    s_endpgm
  16 ;
  17 ; GFX11-LABEL: store_f32_1d:
  18 ; GFX11:       ; %bb.0: ; %main_body
  19 ; GFX11-NEXT:    image_store v[1:4], v0, s[0:7] dmask:0x1 dim:SQ_RSRC_IMG_1D unorm a16
  20 ; GFX11-NEXT:    s_nop 0
  21 ; GFX11-NEXT:    s_sendmsg sendmsg(MSG_DEALLOC_VGPRS)
  22 ; GFX11-NEXT:    s_endpgm
  23 main_body:
  24   %x = extractelement <2 x i16> %coords, i32 0
  25   call void @llvm.amdgcn.image.store.1d.v4f32.i16(<4 x float> %val, i32 1, i16 %x, <8 x i32> %rsrc, i32 0, i32 0)
  26   ret void
  27 }
  28
  29 define amdgpu_ps void @store_v2f32_1d(<8 x i32> inreg %rsrc, <2 x i16> %coords, <4 x float> %val) {
  30 ; GFX9-LABEL: store_v2f32_1d:
  31 ; GFX9:       ; %bb.0: ; %main_body
  32 ; GFX9-NEXT:    image_store v[1:4], v0, s[0:7] dmask:0x3 unorm a16
  33 ; GFX9-NEXT:    s_endpgm
  34 ;
  35 ; GFX10-LABEL: store_v2f32_1d:
  36 ; GFX10:       ; %bb.0: ; %main_body
  37 ; GFX10-NEXT:    image_store v[1:4], v0, s[0:7] dmask:0x3 dim:SQ_RSRC_IMG_1D unorm a16
  38 ; GFX10-NEXT:    s_endpgm
  39 ;
  40 ; GFX11-LABEL: store_v2f32_1d:
  41 ; GFX11:       ; %bb.0: ; %main_body
  42 ; GFX11-NEXT:    image_store v[1:4], v0, s[0:7] dmask:0x3 dim:SQ_RSRC_IMG_1D unorm a16
  43 ; GFX11-NEXT:    s_nop 0
  44 ; GFX11-NEXT:    s_sendmsg sendmsg(MSG_DEALLOC_VGPRS)
  45 ; GFX11-NEXT:    s_endpgm
  46 main_body:
  47   %x = extractelement <2 x i16> %coords, i32 0
  48   call void @llvm.amdgcn.image.store.1d.v4f32.i16(<4 x float> %val, i32 3, i16 %x, <8 x i32> %rsrc, i32 0, i32 0)
  49   ret void
  50 }
  51
  52 define amdgpu_ps void @store_v3f32_1d(<8 x i32> inreg %rsrc, <2 x i16> %coords, <4 x float> %val) {
  53 ; GFX9-LABEL: store_v3f32_1d:
  54 ; GFX9:       ; %bb.0: ; %main_body
  55 ; GFX9-NEXT:    image_store v[1:4], v0, s[0:7] dmask:0x7 unorm a16
  56 ; GFX9-NEXT:    s_endpgm
  57 ;
  58 ; GFX10-LABEL: store_v3f32_1d:
  59 ; GFX10:       ; %bb.0: ; %main_body
  60 ; GFX10-NEXT:    image_store v[1:4], v0, s[0:7] dmask:0x7 dim:SQ_RSRC_IMG_1D unorm a16
  61 ; GFX10-NEXT:    s_endpgm
  62 ;
  63 ; GFX11-LABEL: store_v3f32_1d:
  64 ; GFX11:       ; %bb.0: ; %main_body
  65 ; GFX11-NEXT:    image_store v[1:4], v0, s[0:7] dmask:0x7 dim:SQ_RSRC_IMG_1D unorm a16
  66 ; GFX11-NEXT:    s_nop 0
  67 ; GFX11-NEXT:    s_sendmsg sendmsg(MSG_DEALLOC_VGPRS)
  68 ; GFX11-NEXT:    s_endpgm
  69 main_body:
  70   %x = extractelement <2 x i16> %coords, i32 0
  71   call void @llvm.amdgcn.image.store.1d.v4f32.i16(<4 x float> %val, i32 7, i16 %x, <8 x i32> %rsrc, i32 0, i32 0)
  72   ret void
  73 }
  74
  75 define amdgpu_ps void @store_v4f32_1d(<8 x i32> inreg %rsrc, <2 x i16> %coords, <4 x float> %val) {
  76 ; GFX9-LABEL: store_v4f32_1d:
  77 ; GFX9:       ; %bb.0: ; %main_body
  78 ; GFX9-NEXT:    image_store v[1:4], v0, s[0:7] dmask:0xf unorm a16
  79 ; GFX9-NEXT:    s_endpgm
  80 ;
  81 ; GFX10-LABEL: store_v4f32_1d:
  82 ; GFX10:       ; %bb.0: ; %main_body
  83 ; GFX10-NEXT:    image_store v[1:4], v0, s[0:7] dmask:0xf dim:SQ_RSRC_IMG_1D unorm a16
  84 ; GFX10-NEXT:    s_endpgm
  85 ;
  86 ; GFX11-LABEL: store_v4f32_1d:
  87 ; GFX11:       ; %bb.0: ; %main_body
  88 ; GFX11-NEXT:    image_store v[1:4], v0, s[0:7] dmask:0xf dim:SQ_RSRC_IMG_1D unorm a16
  89 ; GFX11-NEXT:    s_nop 0
  90 ; GFX11-NEXT:    s_sendmsg sendmsg(MSG_DEALLOC_VGPRS)
  91 ; GFX11-NEXT:    s_endpgm
  92 main_body:
  93   %x = extractelement <2 x i16> %coords, i32 0
  94   call void @llvm.amdgcn.image.store.1d.v4f32.i16(<4 x float> %val, i32 15, i16 %x, <8 x i32> %rsrc, i32 0, i32 0)
  95   ret void
  96 }
  97
  98 define amdgpu_ps void @store_f32_2d(<8 x i32> inreg %rsrc, <2 x i16> %coords, <4 x float> %val) {
  99 ; GFX9-LABEL: store_f32_2d:
 100 ; GFX9:       ; %bb.0: ; %main_body
 101 ; GFX9-NEXT:    image_store v[1:4], v0, s[0:7] dmask:0x1 unorm a16
 102 ; GFX9-NEXT:    s_endpgm
 103 ;
 104 ; GFX10-LABEL: store_f32_2d:
 105 ; GFX10:       ; %bb.0: ; %main_body
 106 ; GFX10-NEXT:    image_store v[1:4], v0, s[0:7] dmask:0x1 dim:SQ_RSRC_IMG_2D unorm a16
 107 ; GFX10-NEXT:    s_endpgm
 108 ;
 109 ; GFX11-LABEL: store_f32_2d:
 110 ; GFX11:       ; %bb.0: ; %main_body
 111 ; GFX11-NEXT:    image_store v[1:4], v0, s[0:7] dmask:0x1 dim:SQ_RSRC_IMG_2D unorm a16
 112 ; GFX11-NEXT:    s_nop 0
 113 ; GFX11-NEXT:    s_sendmsg sendmsg(MSG_DEALLOC_VGPRS)
 114 ; GFX11-NEXT:    s_endpgm
 115 main_body:
 116   %x = extractelement <2 x i16> %coords, i32 0
 117   %y = extractelement <2 x i16> %coords, i32 1
 118   call void @llvm.amdgcn.image.store.2d.v4f32.i16(<4 x float> %val, i32 1, i16 %x, i16 %y, <8 x i32> %rsrc, i32 0, i32 0)
 119   ret void
 120 }
 121
 122 define amdgpu_ps void @store_v2f32_2d(<8 x i32> inreg %rsrc, <2 x i16> %coords, <4 x float> %val) {
 123 ; GFX9-LABEL: store_v2f32_2d:
 124 ; GFX9:       ; %bb.0: ; %main_body
 125 ; GFX9-NEXT:    image_store v[1:4], v0, s[0:7] dmask:0x3 unorm a16
 126 ; GFX9-NEXT:    s_endpgm
 127 ;
 128 ; GFX10-LABEL: store_v2f32_2d:
 129 ; GFX10:       ; %bb.0: ; %main_body
 130 ; GFX10-NEXT:    image_store v[1:4], v0, s[0:7] dmask:0x3 dim:SQ_RSRC_IMG_2D unorm a16
 131 ; GFX10-NEXT:    s_endpgm
 132 ;
 133 ; GFX11-LABEL: store_v2f32_2d:
 134 ; GFX11:       ; %bb.0: ; %main_body
 135 ; GFX11-NEXT:    image_store v[1:4], v0, s[0:7] dmask:0x3 dim:SQ_RSRC_IMG_2D unorm a16
 136 ; GFX11-NEXT:    s_nop 0
 137 ; GFX11-NEXT:    s_sendmsg sendmsg(MSG_DEALLOC_VGPRS)
 138 ; GFX11-NEXT:    s_endpgm
 139 main_body:
 140   %x = extractelement <2 x i16> %coords, i32 0
 141   %y = extractelement <2 x i16> %coords, i32 1
 142   call void @llvm.amdgcn.image.store.2d.v4f32.i16(<4 x float> %val, i32 3, i16 %x, i16 %y, <8 x i32> %rsrc, i32 0, i32 0)
 143   ret void
 144 }
 145
 146 define amdgpu_ps void @store_v3f32_2d(<8 x i32> inreg %rsrc, <2 x i16> %coords, <4 x float> %val) {
 147 ; GFX9-LABEL: store_v3f32_2d:
 148 ; GFX9:       ; %bb.0: ; %main_body
 149 ; GFX9-NEXT:    image_store v[1:4], v0, s[0:7] dmask:0x7 unorm a16
 150 ; GFX9-NEXT:    s_endpgm
 151 ;
 152 ; GFX10-LABEL: store_v3f32_2d:
 153 ; GFX10:       ; %bb.0: ; %main_body
 154 ; GFX10-NEXT:    image_store v[1:4], v0, s[0:7] dmask:0x7 dim:SQ_RSRC_IMG_2D unorm a16
 155 ; GFX10-NEXT:    s_endpgm
 156 ;
 157 ; GFX11-LABEL: store_v3f32_2d:
 158 ; GFX11:       ; %bb.0: ; %main_body
 159 ; GFX11-NEXT:    image_store v[1:4], v0, s[0:7] dmask:0x7 dim:SQ_RSRC_IMG_2D unorm a16
 160 ; GFX11-NEXT:    s_nop 0
 161 ; GFX11-NEXT:    s_sendmsg sendmsg(MSG_DEALLOC_VGPRS)
 162 ; GFX11-NEXT:    s_endpgm
 163 main_body:
 164   %x = extractelement <2 x i16> %coords, i32 0
 165   %y = extractelement <2 x i16> %coords, i32 1
 166   call void @llvm.amdgcn.image.store.2d.v4f32.i16(<4 x float> %val, i32 7, i16 %x, i16 %y, <8 x i32> %rsrc, i32 0, i32 0)
 167   ret void
 168 }
 169
 170 define amdgpu_ps void @store_v4f32_2d(<8 x i32> inreg %rsrc, <2 x i16> %coords, <4 x float> %val) {
 171 ; GFX9-LABEL: store_v4f32_2d:
 172 ; GFX9:       ; %bb.0: ; %main_body
 173 ; GFX9-NEXT:    image_store v[1:4], v0, s[0:7] dmask:0xf unorm a16
 174 ; GFX9-NEXT:    s_endpgm
 175 ;
 176 ; GFX10-LABEL: store_v4f32_2d:
 177 ; GFX10:       ; %bb.0: ; %main_body
 178 ; GFX10-NEXT:    image_store v[1:4], v0, s[0:7] dmask:0xf dim:SQ_RSRC_IMG_2D unorm a16
 179 ; GFX10-NEXT:    s_endpgm
 180 ;
 181 ; GFX11-LABEL: store_v4f32_2d:
 182 ; GFX11:       ; %bb.0: ; %main_body
 183 ; GFX11-NEXT:    image_store v[1:4], v0, s[0:7] dmask:0xf dim:SQ_RSRC_IMG_2D unorm a16
 184 ; GFX11-NEXT:    s_nop 0
 185 ; GFX11-NEXT:    s_sendmsg sendmsg(MSG_DEALLOC_VGPRS)
 186 ; GFX11-NEXT:    s_endpgm
 187 main_body:
 188   %x = extractelement <2 x i16> %coords, i32 0
 189   %y = extractelement <2 x i16> %coords, i32 1
 190   call void @llvm.amdgcn.image.store.2d.v4f32.i16(<4 x float> %val, i32 15, i16 %x, i16 %y, <8 x i32> %rsrc, i32 0, i32 0)
 191   ret void
 192 }
 193
 194 define amdgpu_ps void @store_f32_3d(<8 x i32> inreg %rsrc, <2 x i16> %coords_lo, <2 x i16> %coords_hi, <4 x float> %val) {
 195 ; GFX9-LABEL: store_f32_3d:
 196 ; GFX9:       ; %bb.0: ; %main_body
 197 ; GFX9-NEXT:    image_store v[2:5], v[0:1], s[0:7] dmask:0x1 unorm a16
 198 ; GFX9-NEXT:    s_endpgm
 199 ;
 200 ; GFX10-LABEL: store_f32_3d:
 201 ; GFX10:       ; %bb.0: ; %main_body
 202 ; GFX10-NEXT:    image_store v[2:5], v[0:1], s[0:7] dmask:0x1 dim:SQ_RSRC_IMG_3D unorm a16
 203 ; GFX10-NEXT:    s_endpgm
 204 ;
 205 ; GFX11-LABEL: store_f32_3d:
 206 ; GFX11:       ; %bb.0: ; %main_body
 207 ; GFX11-NEXT:    image_store v[2:5], v[0:1], s[0:7] dmask:0x1 dim:SQ_RSRC_IMG_3D unorm a16
 208 ; GFX11-NEXT:    s_nop 0
 209 ; GFX11-NEXT:    s_sendmsg sendmsg(MSG_DEALLOC_VGPRS)
 210 ; GFX11-NEXT:    s_endpgm
 211 main_body:
 212   %x = extractelement <2 x i16> %coords_lo, i32 0
 213   %y = extractelement <2 x i16> %coords_lo, i32 1
 214   %z = extractelement <2 x i16> %coords_hi, i32 0
 215   call void @llvm.amdgcn.image.store.3d.v4f32.i16(<4 x float> %val, i32 1, i16 %x, i16 %y, i16 %z, <8 x i32> %rsrc, i32 0, i32 0)
 216   ret void
 217 }
 218
 219 define amdgpu_ps void @store_v2f32_3d(<8 x i32> inreg %rsrc, <2 x i16> %coords_lo, <2 x i16> %coords_hi, <4 x float> %val) {
 220 ; GFX9-LABEL: store_v2f32_3d:
 221 ; GFX9:       ; %bb.0: ; %main_body
 222 ; GFX9-NEXT:    image_store v[2:5], v[0:1], s[0:7] dmask:0x3 unorm a16
 223 ; GFX9-NEXT:    s_endpgm
 224 ;
 225 ; GFX10-LABEL: store_v2f32_3d:
 226 ; GFX10:       ; %bb.0: ; %main_body
 227 ; GFX10-NEXT:    image_store v[2:5], v[0:1], s[0:7] dmask:0x3 dim:SQ_RSRC_IMG_3D unorm a16
 228 ; GFX10-NEXT:    s_endpgm
 229 ;
 230 ; GFX11-LABEL: store_v2f32_3d:
 231 ; GFX11:       ; %bb.0: ; %main_body
 232 ; GFX11-NEXT:    image_store v[2:5], v[0:1], s[0:7] dmask:0x3 dim:SQ_RSRC_IMG_3D unorm a16
 233 ; GFX11-NEXT:    s_nop 0
 234 ; GFX11-NEXT:    s_sendmsg sendmsg(MSG_DEALLOC_VGPRS)
 235 ; GFX11-NEXT:    s_endpgm
 236 main_body:
 237   %x = extractelement <2 x i16> %coords_lo, i32 0
 238   %y = extractelement <2 x i16> %coords_lo, i32 1
 239   %z = extractelement <2 x i16> %coords_hi, i32 0
 240   call void @llvm.amdgcn.image.store.3d.v4f32.i16(<4 x float> %val, i32 3, i16 %x, i16 %y, i16 %z, <8 x i32> %rsrc, i32 0, i32 0)
 241   ret void
 242 }
 243
 244 define amdgpu_ps void @store_v3f32_3d(<8 x i32> inreg %rsrc, <2 x i16> %coords_lo, <2 x i16> %coords_hi, <4 x float> %val) {
 245 ; GFX9-LABEL: store_v3f32_3d:
 246 ; GFX9:       ; %bb.0: ; %main_body
 247 ; GFX9-NEXT:    image_store v[2:5], v[0:1], s[0:7] dmask:0x7 unorm a16
 248 ; GFX9-NEXT:    s_endpgm
 249 ;
 250 ; GFX10-LABEL: store_v3f32_3d:
 251 ; GFX10:       ; %bb.0: ; %main_body
 252 ; GFX10-NEXT:    image_store v[2:5], v[0:1], s[0:7] dmask:0x7 dim:SQ_RSRC_IMG_3D unorm a16
 253 ; GFX10-NEXT:    s_endpgm
 254 ;
 255 ; GFX11-LABEL: store_v3f32_3d:
 256 ; GFX11:       ; %bb.0: ; %main_body
 257 ; GFX11-NEXT:    image_store v[2:5], v[0:1], s[0:7] dmask:0x7 dim:SQ_RSRC_IMG_3D unorm a16
 258 ; GFX11-NEXT:    s_nop 0
 259 ; GFX11-NEXT:    s_sendmsg sendmsg(MSG_DEALLOC_VGPRS)
 260 ; GFX11-NEXT:    s_endpgm
 261 main_body:
 262   %x = extractelement <2 x i16> %coords_lo, i32 0
 263   %y = extractelement <2 x i16> %coords_lo, i32 1
 264   %z = extractelement <2 x i16> %coords_hi, i32 0
 265   call void @llvm.amdgcn.image.store.3d.v4f32.i16(<4 x float> %val, i32 7, i16 %x, i16 %y, i16 %z, <8 x i32> %rsrc, i32 0, i32 0)
 266   ret void
 267 }
 268
 269 define amdgpu_ps void @store_v4f32_3d(<8 x i32> inreg %rsrc, <2 x i16> %coords_lo, <2 x i16> %coords_hi, <4 x float> %val) {
 270 ; GFX9-LABEL: store_v4f32_3d:
 271 ; GFX9:       ; %bb.0: ; %main_body
 272 ; GFX9-NEXT:    image_store v[2:5], v[0:1], s[0:7] dmask:0xf unorm a16
 273 ; GFX9-NEXT:    s_endpgm
 274 ;
 275 ; GFX10-LABEL: store_v4f32_3d:
 276 ; GFX10:       ; %bb.0: ; %main_body
 277 ; GFX10-NEXT:    image_store v[2:5], v[0:1], s[0:7] dmask:0xf dim:SQ_RSRC_IMG_3D unorm a16
 278 ; GFX10-NEXT:    s_endpgm
 279 ;
 280 ; GFX11-LABEL: store_v4f32_3d:
 281 ; GFX11:       ; %bb.0: ; %main_body
 282 ; GFX11-NEXT:    image_store v[2:5], v[0:1], s[0:7] dmask:0xf dim:SQ_RSRC_IMG_3D unorm a16
 283 ; GFX11-NEXT:    s_nop 0
 284 ; GFX11-NEXT:    s_sendmsg sendmsg(MSG_DEALLOC_VGPRS)
 285 ; GFX11-NEXT:    s_endpgm
 286 main_body:
 287   %x = extractelement <2 x i16> %coords_lo, i32 0
 288   %y = extractelement <2 x i16> %coords_lo, i32 1
 289   %z = extractelement <2 x i16> %coords_hi, i32 0
 290   call void @llvm.amdgcn.image.store.3d.v4f32.i16(<4 x float> %val, i32 15, i16 %x, i16 %y, i16 %z, <8 x i32> %rsrc, i32 0, i32 0)
 291   ret void
 292 }
 293
 294 declare void @llvm.amdgcn.image.store.1d.v4f32.i16(<4 x float>, i32, i16, <8 x i32>, i32, i32) #2
 295 declare void @llvm.amdgcn.image.store.2d.v4f32.i16(<4 x float>, i32, i16, i16, <8 x i32>, i32, i32) #2
 296 declare void @llvm.amdgcn.image.store.3d.v4f32.i16(<4 x float>, i32, i16, i16, i16, <8 x i32>, i32, i32) #2
 297
 298 attributes #0 = { nounwind }
 299 attributes #1 = { nounwind readonly }