xref: /llvm-project/llvm/test/CodeGen/SPIRV/hlsl-resources/BufferLoad.ll (revision aa07f922103ebe8e78c8da4c754b43af3c129f3e)
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
19define 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.resource.handlefrombinding.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.resource.load.typedbuffer(
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
34define 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.resource.handlefrombinding.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.resource.load.typedbuffer(
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
50define 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.resource.handlefrombinding.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.resource.load.typedbuffer(
61      target("spirv.Image", i32, 5, 2, 0, 0, 2, 24) %buffer0, i32 0)
62
63  ret void
64}
65
66attributes #0 = { convergent noinline norecurse "frame-pointer"="all" "hlsl.numthreads"="1,1,1" "hlsl.shader"="compute" "no-trapping-math"="true" "stack-protector-buffer-size"="8" }
67