llvm/test/CodeGen/SPIRV/hlsl-resources/BufferStore.ll

   1 ; RUN: llc -O3 -verify-machineinstrs -mtriple=spirv-vulkan-library %s -o - | FileCheck %s
   2 ; RUN: %if spirv-tools %{ llc -O0 -mtriple=spirv-vulkan-library %s -o - -filetype=obj | spirv-val %}
   3
   4 ; CHECK-NOT: OpCapability StorageImageReadWithoutFormat
   5
   6 ; CHECK-DAG: OpDecorate [[IntBufferVar:%[0-9]+]] DescriptorSet 16
   7 ; CHECK-DAG: OpDecorate [[IntBufferVar]] Binding 7
   8
   9 ; CHECK-DAG: [[int:%[0-9]+]] = OpTypeInt 32 0
  10 ; CHECK-DAG: [[zero:%[0-9]+]] = OpConstant [[int]] 0
  11 ; CHECK-DAG: [[v4_int:%[0-9]+]] = OpTypeVector [[int]] 4
  12 ; CHECK-DAG: [[RWBufferTypeInt:%[0-9]+]] = OpTypeImage [[int]] Buffer 2 0 0 2 R32i {{$}}
  13 ; CHECK-DAG: [[IntBufferPtrType:%[0-9]+]] = OpTypePointer UniformConstant [[RWBufferTypeInt]]
  14 ; CHECK-DAG: [[IntBufferVar]] = OpVariable [[IntBufferPtrType]] UniformConstant
  15
  16
  17 ; CHECK: {{%[0-9]+}} = OpFunction {{%[0-9]+}} DontInline {{%[0-9]+}}
  18 declare <4 x i32> @get_data() #1
  19
  20 ; CHECK: {{%[0-9]+}} = OpFunction {{%[0-9]+}} DontInline {{%[0-9]+}}
  21 ; CHECK-NEXT: OpLabel
  22 define void @RWBufferStore_Vec4_I32() #0 {
  23 ; CHECK: [[buffer:%[0-9]+]] = OpLoad [[RWBufferTypeInt]] [[IntBufferVar]]
  24   %buffer0 = call target("spirv.Image", i32, 5, 2, 0, 0, 2, 24)
  25       @llvm.spv.handle.fromBinding.tspirv.Image_i32_5_2_0_0_2_24(
  26           i32 16, i32 7, i32 1, i32 0, i1 false)
  27
  28 ; CHECK: [[data:%[0-9]+]] = OpFunctionCall
  29   %data = call <4 x i32> @get_data()
  30 ; CHECK: OpImageWrite [[buffer]] [[zero]] [[data]]
  31   call void @llvm.spv.typedBufferStore(target("spirv.Image", i32, 5, 2, 0, 0, 2, 24) %buffer0, i32 0, <4 x i32> %data)
  32
  33   ret void
  34 }
  35
  36 attributes #0 = { convergent noinline norecurse "frame-pointer"="all" "hlsl.numthreads"="1,1,1" "hlsl.shader"="compute" "no-trapping-math"="true" "stack-protector-buffer-size"="8" }
  37 attributes #1 = { convergent noinline norecurse "frame-pointer"="all" "no-trapping-math"="true" "stack-protector-buffer-size"="8" }