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

   1 ; RUN: llc -O0 -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: [[v2_int:%[0-9]+]] = OpTypeVector [[int]] 2
  13 ; CHECK-DAG: [[RWBufferTypeInt:%[0-9]+]] = OpTypeImage [[int]] Buffer 2 0 0 2 R32i {{$}}
  14 ; CHECK-DAG: [[IntBufferPtrType:%[0-9]+]] = OpTypePointer UniformConstant [[RWBufferTypeInt]]
  15 ; CHECK-DAG: [[IntBufferVar]] = OpVariable [[IntBufferPtrType]] UniformConstant
  16
  17 ; CHECK: {{%[0-9]+}} = OpFunction {{%[0-9]+}} DontInline {{%[0-9]+}}
  18 ; CHECK-NEXT: OpLabel
  19 define void @RWBufferLoad_Vec4_I32() #0 {
  20 ; CHECK: [[buffer:%[0-9]+]] = OpLoad [[RWBufferTypeInt]] [[IntBufferVar]]
  21   %buffer0 = call target("spirv.Image", i32, 5, 2, 0, 0, 2, 24)
  22       @llvm.spv.handle.fromBinding.tspirv.Image_i32_5_2_0_0_2_24(
  23           i32 16, i32 7, i32 1, i32 0, i1 false)
  24
  25 ; CHECK: OpImageRead [[v4_int]] [[buffer]] [[zero]]
  26   %data0 = call <4 x i32> @llvm.spv.typedBufferLoad(
  27       target("spirv.Image", i32, 5, 2, 0, 0, 2, 24) %buffer0, i32 0)
  28
  29   ret void
  30 }
  31
  32 ; CHECK: {{%[0-9]+}} = OpFunction {{%[0-9]+}} DontInline {{%[0-9]+}}
  33 ; CHECK-NEXT: OpLabel
  34 define void @RWBufferLoad_I32() #0 {
  35 ; CHECK: [[buffer:%[0-9]+]] = OpLoad [[RWBufferTypeInt]] [[IntBufferVar]]
  36   %buffer1 = call target("spirv.Image", i32, 5, 2, 0, 0, 2, 24)
  37       @llvm.spv.handle.fromBinding.tspirv.Image_i32_5_2_0_0_2_24(
  38           i32 16, i32 7, i32 1, i32 0, i1 false)
  39
  40 ; CHECK: [[V:%[0-9]+]] = OpImageRead [[v4_int]] [[buffer]] [[zero]]
  41 ; CHECK: OpCompositeExtract [[int]] [[V]] 0
  42   %data1 = call i32 @llvm.spv.typedBufferLoad(
  43       target("spirv.Image", i32, 5, 2, 0, 0, 2, 24) %buffer1, i32 0)
  44
  45   ret void
  46 }
  47
  48 ; CHECK: {{%[0-9]+}} = OpFunction {{%[0-9]+}} DontInline {{%[0-9]+}}
  49 ; CHECK-NEXT: OpLabel
  50 define void @RWBufferLoad_Vec2_I32() #0 {
  51 ; CHECK: [[buffer:%[0-9]+]] = OpLoad [[RWBufferTypeInt]] [[IntBufferVar]]
  52   %buffer0 = call target("spirv.Image", i32, 5, 2, 0, 0, 2, 24)
  53       @llvm.spv.handle.fromBinding.tspirv.Image_i32_5_2_0_0_2_24(
  54           i32 16, i32 7, i32 1, i32 0, i1 false)
  55
  56 ; CHECK: [[V:%[0-9]+]] = OpImageRead [[v4_int]] [[buffer]] [[zero]]
  57 ; CHECK: [[e0:%[0-9]+]] = OpCompositeExtract [[int]] [[V]] 0
  58 ; CHECK: [[e1:%[0-9]+]] = OpCompositeExtract [[int]] [[V]] 1
  59 ; CHECK: OpCompositeConstruct [[v2_int]] [[e0]] [[e1]]
  60   %data0 = call <2 x i32> @llvm.spv.typedBufferLoad(
  61       target("spirv.Image", i32, 5, 2, 0, 0, 2, 24) %buffer0, i32 0)
  62
  63   ret void
  64 }
  65
  66 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" }