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