Vulkan: support entrypoint command (#188)

Fixes #30

1 files changed, 173 insertions, 0 deletions

diff --git a/tests/cases/compute_ssbo_with_entrypoint_command.amber b/tests/cases/compute_ssbo_with_entrypoint_command.amber
new file mode 100644
index 0000000..eae3aeb
--- /dev/null
+++ b/tests/cases/compute_ssbo_with_entrypoint_command.amber
@@ -0,0 +1,173 @@
+# Copyright 2018 The Amber Authors.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     https://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+[comment]
+Source code in GLSL:
+
+#version 430
+
+layout(set = 0, binding = 0) buffer block0 {
+  float data_set0_binding0[3];
+};
+
+layout(set = 1, binding = 2) buffer block1 {
+  float data_set1_binding2[3];
+};
+
+layout(set = 2, binding = 1) buffer block2 {
+  float data_set2_binding1[3];
+};
+
+layout(set = 2, binding = 3) buffer block3 {
+  float data_set2_binding3[3];
+};
+
+[compute shader spirv]
+; SPIR-V
+; Version: 1.0
+; Generator: Khronos Glslang Reference Front End; 7
+; Bound: 71
+; Schema: 0
+               OpCapability Shader
+          %1 = OpExtInstImport "GLSL.std.450"
+               OpMemoryModel Logical GLSL450
+               OpEntryPoint GLCompute %main "foo" %gl_WorkGroupID
+               OpExecutionMode %main LocalSize 1 1 1
+               OpSource GLSL 430
+               OpName %main "foo"
+               OpName %index "index"
+               OpName %gl_WorkGroupID "gl_WorkGroupID"
+               OpName %block0 "block0"
+               OpMemberName %block0 0 "data_set0_binding0"
+               OpName %_ ""
+               OpName %block1 "block1"
+               OpMemberName %block1 0 "data_set1_binding2"
+               OpName %__0 ""
+               OpName %block2 "block2"
+               OpMemberName %block2 0 "data_set2_binding1"
+               OpName %__1 ""
+               OpName %block3 "block3"
+               OpMemberName %block3 0 "data_set2_binding3"
+               OpName %__2 ""
+               OpDecorate %gl_WorkGroupID BuiltIn WorkgroupId
+               OpDecorate %_arr_float_uint_3 ArrayStride 4
+               OpMemberDecorate %block0 0 Offset 0
+               OpDecorate %block0 BufferBlock
+               OpDecorate %_ DescriptorSet 0
+               OpDecorate %_ Binding 0
+               OpDecorate %_arr_float_uint_3_0 ArrayStride 4
+               OpMemberDecorate %block1 0 Offset 0
+               OpDecorate %block1 BufferBlock
+               OpDecorate %__0 DescriptorSet 1
+               OpDecorate %__0 Binding 2
+               OpDecorate %_arr_float_uint_3_1 ArrayStride 4
+               OpMemberDecorate %block2 0 Offset 0
+               OpDecorate %block2 BufferBlock
+               OpDecorate %__1 DescriptorSet 2
+               OpDecorate %__1 Binding 1
+               OpDecorate %_arr_float_uint_3_2 ArrayStride 4
+               OpMemberDecorate %block3 0 Offset 0
+               OpDecorate %block3 BufferBlock
+               OpDecorate %__2 DescriptorSet 2
+               OpDecorate %__2 Binding 3
+       %void = OpTypeVoid
+          %3 = OpTypeFunction %void
+       %uint = OpTypeInt 32 0
+%_ptr_Function_uint = OpTypePointer Function %uint
+     %v3uint = OpTypeVector %uint 3
+%_ptr_Input_v3uint = OpTypePointer Input %v3uint
+%gl_WorkGroupID = OpVariable %_ptr_Input_v3uint Input
+     %uint_0 = OpConstant %uint 0
+%_ptr_Input_uint = OpTypePointer Input %uint
+      %float = OpTypeFloat 32
+     %uint_3 = OpConstant %uint 3
+%_arr_float_uint_3 = OpTypeArray %float %uint_3
+     %block0 = OpTypeStruct %_arr_float_uint_3
+%_ptr_Uniform_block0 = OpTypePointer Uniform %block0
+          %_ = OpVariable %_ptr_Uniform_block0 Uniform
+        %int = OpTypeInt 32 1
+      %int_0 = OpConstant %int 0
+%_ptr_Uniform_float = OpTypePointer Uniform %float
+    %float_1 = OpConstant %float 1
+%_arr_float_uint_3_0 = OpTypeArray %float %uint_3
+     %block1 = OpTypeStruct %_arr_float_uint_3_0
+%_ptr_Uniform_block1 = OpTypePointer Uniform %block1
+        %__0 = OpVariable %_ptr_Uniform_block1 Uniform
+%_arr_float_uint_3_1 = OpTypeArray %float %uint_3
+     %block2 = OpTypeStruct %_arr_float_uint_3_1
+%_ptr_Uniform_block2 = OpTypePointer Uniform %block2
+        %__1 = OpVariable %_ptr_Uniform_block2 Uniform
+   %float_10 = OpConstant %float 10
+%_arr_float_uint_3_2 = OpTypeArray %float %uint_3
+     %block3 = OpTypeStruct %_arr_float_uint_3_2
+%_ptr_Uniform_block3 = OpTypePointer Uniform %block3
+        %__2 = OpVariable %_ptr_Uniform_block3 Uniform
+   %float_30 = OpConstant %float 30
+       %main = OpFunction %void None %3
+          %5 = OpLabel
+      %index = OpVariable %_ptr_Function_uint Function
+         %14 = OpAccessChain %_ptr_Input_uint %gl_WorkGroupID %uint_0
+         %15 = OpLoad %uint %14
+               OpStore %index %15
+         %24 = OpLoad %uint %index
+         %25 = OpLoad %uint %index
+         %27 = OpAccessChain %_ptr_Uniform_float %_ %int_0 %25
+         %28 = OpLoad %float %27
+         %30 = OpFAdd %float %28 %float_1
+         %31 = OpAccessChain %_ptr_Uniform_float %_ %int_0 %24
+               OpStore %31 %30
+         %36 = OpLoad %uint %index
+         %41 = OpLoad %uint %index
+         %42 = OpAccessChain %_ptr_Uniform_float %__1 %int_0 %41
+         %43 = OpLoad %float %42
+         %44 = OpLoad %uint %index
+         %45 = OpAccessChain %_ptr_Uniform_float %__0 %int_0 %44
+         %46 = OpLoad %float %45
+         %47 = OpFSub %float %43 %46
+         %48 = OpAccessChain %_ptr_Uniform_float %__0 %int_0 %36
+               OpStore %48 %47
+         %49 = OpLoad %uint %index
+         %55 = OpLoad %uint %index
+         %56 = OpAccessChain %_ptr_Uniform_float %__2 %int_0 %55
+         %57 = OpLoad %float %56
+         %58 = OpFMul %float %float_10 %57
+         %59 = OpLoad %uint %index
+         %60 = OpAccessChain %_ptr_Uniform_float %__1 %int_0 %59
+         %61 = OpLoad %float %60
+         %62 = OpFAdd %float %58 %61
+         %63 = OpAccessChain %_ptr_Uniform_float %__1 %int_0 %49
+               OpStore %63 %62
+         %64 = OpLoad %uint %index
+         %66 = OpLoad %uint %index
+         %67 = OpAccessChain %_ptr_Uniform_float %__2 %int_0 %66
+         %68 = OpLoad %float %67
+         %69 = OpFMul %float %float_30 %68
+         %70 = OpAccessChain %_ptr_Uniform_float %__2 %int_0 %64
+               OpStore %70 %69
+               OpReturn
+               OpFunctionEnd
+[test]
+ssbo 0:0 subdata vec3  0  1.0  2.0  3.0
+ssbo 1:2 subdata vec3  0  4.0  5.0  6.0
+ssbo 2:1 subdata vec3  0 21.0 22.0 23.0
+ssbo 2:3 subdata vec3  0  0.7  0.8  0.9
+
+compute entrypoint foo
+
+compute 3 1 1
+
+probe ssbo vec3 0:0 0 ~=  2.0  3.0  4.0
+probe ssbo vec3 1:2 0 ~= 17.0 17.0 17.0
+probe ssbo vec3 2:1 0 ~= 28.0 30.0 32.0
+probe ssbo vec3 2:3 0 ~= 21.0 24.0 27.0