external/vulkancts/modules/vulkan/spirv_assembly/vktSpvAsmUboMatrixPaddingTests.cpp

   1 /*-------------------------------------------------------------------------
   2  * Vulkan Conformance Tests
   3  * ------------------------
   4  *
   5  * Copyright (c) 2017 Google Inc.
   6  *
   7  * Licensed under the Apache License, Version 2.0 (the "License");
   8  * you may not use this file except in compliance with the License.
   9  * You may obtain a copy of the License at
  10  *
  11  *      http://www.apache.org/licenses/LICENSE-2.0
  12  *
  13  * Unless required by applicable law or agreed to in writing, software
  14  * distributed under the License is distributed on an "AS IS" BASIS,
  15  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  16  * See the License for the specific language governing permissions and
  17  * limitations under the License.
  18  *
  19  *//*!
  20  * \file
  21  * \brief SPIR-V Assembly Tests for UBO matrix padding.
  22  *//*--------------------------------------------------------------------*/
  23
  24 #include "vktSpvAsmUboMatrixPaddingTests.hpp"
  25 #include "vktSpvAsmComputeShaderCase.hpp"
  26 #include "vktSpvAsmComputeShaderTestUtil.hpp"
  27 #include "vktSpvAsmGraphicsShaderTestUtil.hpp"
  28
  29 namespace vkt
  30 {
  31 namespace SpirVAssembly
  32 {
  33
  34 using namespace vk;
  35 using std::map;
  36 using std::string;
  37 using std::vector;
  38 using tcu::IVec3;
  39 using tcu::RGBA;
  40 using tcu::Vec4;
  41
  42 namespace
  43 {
  44
  45 void addComputeUboMatrixPaddingTest (tcu::TestCaseGroup* group)
  46 {
  47         tcu::TestContext&       testCtx                 = group->getTestContext();
  48         de::Random                      rnd                             (deStringHash(group->getName()));
  49         const int                       numElements             = 128;
  50
  51         // Read input UBO containing and array of mat2x2 using no padding inside matrix. Output
  52         // into output buffer containing floats. The input and output buffer data should match.
  53         const string            shaderSource    =
  54                 "                       OpCapability Shader\n"
  55                 "                  %1 = OpExtInstImport \"GLSL.std.450\"\n"
  56                 "                       OpMemoryModel Logical GLSL450\n"
  57                 "                       OpEntryPoint GLCompute %main \"main\" %id\n"
  58                 "                       OpExecutionMode %main LocalSize 1 1 1\n"
  59                 "                       OpSource GLSL 430\n"
  60                 "                       OpDecorate %id BuiltIn GlobalInvocationId\n"
  61                 "                       OpDecorate %_arr_v4 ArrayStride 16\n"
  62                 "                       OpMemberDecorate %Output 0 Offset 0\n"
  63                 "                       OpDecorate %Output BufferBlock\n"
  64                 "                       OpDecorate %dataOutput DescriptorSet 0\n"
  65                 "                       OpDecorate %dataOutput Binding 1\n"
  66                 "                       OpDecorate %_arr_mat2v2 ArrayStride 16\n"
  67                 "                       OpMemberDecorate %Input 0 ColMajor\n"
  68                 "                       OpMemberDecorate %Input 0 Offset 0\n"
  69                 "                       OpMemberDecorate %Input 0 MatrixStride 8\n"
  70                 "                       OpDecorate %Input Block\n"
  71                 "                       OpDecorate %dataInput DescriptorSet 0\n"
  72                 "                       OpDecorate %dataInput Binding 0\n"
  73                 "               %void = OpTypeVoid\n"
  74                 "                  %3 = OpTypeFunction %void\n"
  75                 "                %u32 = OpTypeInt 32 0\n"
  76                 " %_ptr_Function_uint = OpTypePointer Function %u32\n"
  77                 "             %v3uint = OpTypeVector %u32 3\n"
  78                 "  %_ptr_Input_v3uint = OpTypePointer Input %v3uint\n"
  79                 "                 %id = OpVariable %_ptr_Input_v3uint Input\n"
  80                 "                %i32 = OpTypeInt 32 1\n"
  81                 "              %int_0 = OpConstant %i32 0\n"
  82                 "              %int_1 = OpConstant %i32 1\n"
  83                 "             %uint_0 = OpConstant %u32 0\n"
  84                 "             %uint_1 = OpConstant %u32 1\n"
  85                 "             %uint_2 = OpConstant %u32 2\n"
  86                 "             %uint_3 = OpConstant %u32 3\n"
  87                 "    %_ptr_Input_uint = OpTypePointer Input %u32\n"
  88                 "                %f32 = OpTypeFloat 32\n"
  89                 "            %v4float = OpTypeVector %f32 4\n"
  90                 "           %uint_128 = OpConstant %u32 128\n"
  91                 "            %_arr_v4 = OpTypeArray %v4float %uint_128\n"
  92                 "             %Output = OpTypeStruct %_arr_v4\n"
  93                 "%_ptr_Uniform_Output = OpTypePointer Uniform %Output\n"
  94                 "         %dataOutput = OpVariable %_ptr_Uniform_Output Uniform\n"
  95                 "            %v2float = OpTypeVector %f32 2\n"
  96                 "        %mat2v2float = OpTypeMatrix %v2float 2\n"
  97                 "        %_arr_mat2v2 = OpTypeArray %mat2v2float %uint_128\n"
  98                 "              %Input = OpTypeStruct %_arr_mat2v2\n"
  99                 " %_ptr_Uniform_Input = OpTypePointer Uniform %Input\n"
 100                 "          %dataInput = OpVariable %_ptr_Uniform_Input Uniform\n"
 101                 " %_ptr_Uniform_float = OpTypePointer Uniform %f32\n"
 102                 "               %main = OpFunction %void None %3\n"
 103                 "                  %5 = OpLabel\n"
 104                 "                  %i = OpVariable %_ptr_Function_uint Function\n"
 105                 "                 %14 = OpAccessChain %_ptr_Input_uint %id %uint_0\n"
 106                 "                 %15 = OpLoad %u32 %14\n"
 107                 "                       OpStore %i %15\n"
 108                 "                %idx = OpLoad %u32 %i\n"
 109                 "                 %34 = OpAccessChain %_ptr_Uniform_float %dataInput %int_0 %idx %int_0 %uint_0\n"
 110                 "                 %35 = OpLoad %f32 %34\n"
 111                 "                 %36 = OpAccessChain %_ptr_Uniform_float %dataOutput %int_0 %idx %uint_0\n"
 112                 "                       OpStore %36 %35\n"
 113                 "                 %40 = OpAccessChain %_ptr_Uniform_float %dataInput %int_0 %idx %int_0 %uint_1\n"
 114                 "                 %41 = OpLoad %f32 %40\n"
 115                 "                 %42 = OpAccessChain %_ptr_Uniform_float %dataOutput %int_0 %idx %uint_1\n"
 116                 "                       OpStore %42 %41\n"
 117                 "                 %46 = OpAccessChain %_ptr_Uniform_float %dataInput %int_0 %idx %int_1 %uint_0\n"
 118                 "                 %47 = OpLoad %f32 %46\n"
 119                 "                 %49 = OpAccessChain %_ptr_Uniform_float %dataOutput %int_0 %idx %uint_2\n"
 120                 "                       OpStore %49 %47\n"
 121                 "                 %52 = OpAccessChain %_ptr_Uniform_float %dataInput %int_0 %idx %int_1 %uint_1\n"
 122                 "                 %53 = OpLoad %f32 %52\n"
 123                 "                 %55 = OpAccessChain %_ptr_Uniform_float %dataOutput %int_0 %idx %uint_3\n"
 124                 "                       OpStore %55 %53\n"
 125                 "                       OpReturn\n"
 126                 "                       OpFunctionEnd\n";
 127
 128                 vector<tcu::Vec4>               inputData;
 129                 ComputeShaderSpec               spec;
 130
 131                 inputData.reserve(numElements);
 132                 for (deUint32 numIdx = 0; numIdx < numElements; ++numIdx)
 133                         inputData.push_back(tcu::Vec4(rnd.getFloat(), rnd.getFloat(), rnd.getFloat(), rnd.getFloat()));
 134
 135                 spec.assembly                   = shaderSource;
 136                 spec.numWorkGroups              = IVec3(numElements, 1, 1);
 137                 spec.inputTypes[0]              = VK_DESCRIPTOR_TYPE_UNIFORM_BUFFER;
 138
 139                 spec.inputs.push_back(BufferSp(new Vec4Buffer(inputData)));
 140                 // Shader is expected to pass the input data by treating the input vec4 as mat2x2
 141                 spec.outputs.push_back(BufferSp(new Vec4Buffer(inputData)));
 142
 143                 group->addChild(new SpvAsmComputeShaderCase(testCtx, "mat2x2", "Tests mat2x2 member in UBO struct without padding (treated as vec4).", spec));
 144         }
 145 }
 146
 147 void addGraphicsUboMatrixPaddingTest (tcu::TestCaseGroup* group)
 148 {
 149         de::Random                              rnd                                     (deStringHash(group->getName()));
 150         map<string, string>             fragments;
 151         const deUint32                  numDataPoints           = 128;
 152         RGBA                                    defaultColors[4];
 153         GraphicsResources               resources;
 154         vector<tcu::Vec4>               inputData;
 155
 156         inputData.reserve(numDataPoints);
 157         for (deUint32 numIdx = 0; numIdx < numDataPoints; ++numIdx)
 158                 inputData.push_back(tcu::Vec4(rnd.getFloat(), rnd.getFloat(), rnd.getFloat(), rnd.getFloat()));
 159
 160         resources.inputs.push_back(std::make_pair(VK_DESCRIPTOR_TYPE_UNIFORM_BUFFER, BufferSp(new Vec4Buffer(inputData))));
 161         // Shader is expected to pass the input data by treating the input vec4 as mat2x2
 162         resources.outputs.push_back(std::make_pair(VK_DESCRIPTOR_TYPE_STORAGE_BUFFER, BufferSp(new Vec4Buffer(inputData))));
 163
 164         getDefaultColors(defaultColors);
 165
 166         fragments["pre_main"]   =
 167                 "             %uint_128 = OpConstant %u32 128\n"
 168                 "    %_arr_v4f_uint_128 = OpTypeArray %v4f32 %uint_128\n"
 169                 "               %Output = OpTypeStruct %_arr_v4f_uint_128\n"
 170                 "  %_ptr_Uniform_Output = OpTypePointer Uniform %Output\n"
 171                 "           %dataOutput = OpVariable %_ptr_Uniform_Output Uniform\n"
 172                 "              %mat2v2f = OpTypeMatrix %v2f32 2\n"
 173                 "%_arr_mat2v2f_uint_128 = OpTypeArray %mat2v2f %uint_128\n"
 174                 "                %Input = OpTypeStruct %_arr_mat2v2f_uint_128\n"
 175                 "   %_ptr_Uniform_Input = OpTypePointer Uniform %Input\n"
 176                 "            %dataInput = OpVariable %_ptr_Uniform_Input Uniform\n"
 177                 "       %_ptr_Uniform_f = OpTypePointer Uniform %f32\n"
 178                 "            %c_i32_128 = OpConstant %i32 128\n";
 179
 180         fragments["decoration"] =
 181                 "                         OpDecorate %_arr_v4f_uint_128 ArrayStride 16\n"
 182                 "                         OpMemberDecorate %Output 0 Offset 0\n"
 183                 "                         OpDecorate %Output BufferBlock\n"
 184                 "                         OpDecorate %dataOutput DescriptorSet 0\n"
 185                 "                         OpDecorate %dataOutput Binding 1\n"
 186                 "                         OpDecorate %_arr_mat2v2f_uint_128 ArrayStride 16\n"
 187                 "                         OpMemberDecorate %Input 0 ColMajor\n"
 188                 "                         OpMemberDecorate %Input 0 Offset 0\n"
 189                 "                         OpMemberDecorate %Input 0 MatrixStride 8\n"
 190                 "                         OpDecorate %Input Block\n"
 191                 "                         OpDecorate %dataInput DescriptorSet 0\n"
 192                 "                         OpDecorate %dataInput Binding 0\n";
 193
 194         // Read input UBO containing and array of mat2x2 using no padding inside matrix. Output
 195         // into output buffer containing floats. The input and output buffer data should match.
 196         // The whole array is handled inside a for loop.
 197         fragments["testfun"]    =
 198                 "            %test_code = OpFunction %v4f32 None %v4f32_function\n"
 199                 "                %param = OpFunctionParameter %v4f32\n"
 200
 201                 "                %entry = OpLabel\n"
 202                 "                    %i = OpVariable %fp_i32 Function\n"
 203                 "                         OpStore %i %c_i32_0\n"
 204                 "                         OpBranch %loop\n"
 205
 206                 "                 %loop = OpLabel\n"
 207                 "                   %15 = OpLoad %i32 %i\n"
 208                 "                   %lt = OpSLessThan %bool %15 %c_i32_128\n"
 209                 "                         OpLoopMerge %merge %inc None\n"
 210                 "                         OpBranchConditional %lt %write %merge\n"
 211
 212                 "                %write = OpLabel\n"
 213                 "                   %30 = OpLoad %i32 %i\n"
 214                 "                   %34 = OpAccessChain %_ptr_Uniform_f %dataInput %c_i32_0 %30 %c_i32_0 %c_u32_0\n"
 215                 "                   %35 = OpLoad %f32 %34\n"
 216                 "                   %36 = OpAccessChain %_ptr_Uniform_f %dataOutput %c_i32_0 %30 %c_u32_0\n"
 217                 "                         OpStore %36 %35\n"
 218                 "                   %40 = OpAccessChain %_ptr_Uniform_f %dataInput %c_i32_0 %30 %c_i32_0 %c_u32_1\n"
 219                 "                   %41 = OpLoad %f32 %40\n"
 220                 "                   %42 = OpAccessChain %_ptr_Uniform_f %dataOutput %c_i32_0 %30 %c_u32_1\n"
 221                 "                         OpStore %42 %41\n"
 222                 "                   %46 = OpAccessChain %_ptr_Uniform_f %dataInput %c_i32_0 %30 %c_i32_1 %c_u32_0\n"
 223                 "                   %47 = OpLoad %f32 %46\n"
 224                 "                   %49 = OpAccessChain %_ptr_Uniform_f %dataOutput %c_i32_0 %30 %c_u32_2\n"
 225                 "                         OpStore %49 %47\n"
 226                 "                   %52 = OpAccessChain %_ptr_Uniform_f %dataInput %c_i32_0 %30 %c_i32_1 %c_u32_1\n"
 227                 "                   %53 = OpLoad %f32 %52\n"
 228                 "                   %55 = OpAccessChain %_ptr_Uniform_f %dataOutput %c_i32_0 %30 %c_u32_3\n"
 229                 "                         OpStore %55 %53\n"
 230                 "                         OpBranch %inc\n"
 231
 232                 "                  %inc = OpLabel\n"
 233                 "                   %37 = OpLoad %i32 %i\n"
 234                 "                   %39 = OpIAdd %i32 %37 %c_i32_1\n"
 235                 "                         OpStore %i %39\n"
 236                 "                         OpBranch %loop\n"
 237
 238                 "                %merge = OpLabel\n"
 239                 "                         OpReturnValue %param\n"
 240
 241                 "                         OpFunctionEnd\n";
 242
 243         resources.inputs.back().first   = VK_DESCRIPTOR_TYPE_UNIFORM_BUFFER;
 244
 245         createTestsForAllStages("mat2x2", defaultColors, defaultColors, fragments, resources, vector<string>(), group);
 246 }
 247
 248 tcu::TestCaseGroup* createUboMatrixPaddingComputeGroup (tcu::TestContext& testCtx)
 249 {
 250         de::MovePtr<tcu::TestCaseGroup> group           (new tcu::TestCaseGroup(testCtx, "ubo_padding", "Compute tests for UBO struct member packing."));
 251         addComputeUboMatrixPaddingTest(group.get());
 252
 253         return group.release();
 254 }
 255
 256 tcu::TestCaseGroup* createUboMatrixPaddingGraphicsGroup (tcu::TestContext& testCtx)
 257 {
 258         de::MovePtr<tcu::TestCaseGroup> group           (new tcu::TestCaseGroup(testCtx, "ubo_padding", "Graphics tests for UBO struct member packing."));
 259         addGraphicsUboMatrixPaddingTest(group.get());
 260
 261         return group.release();
 262 }
 263
 264 } // SpirVAssembly
 265 } // vkt