| // Copyright (c) 2017-2022 Valve Corporation |
| // Copyright (c) 2017-2022 LunarG Inc. |
| // |
| // 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 |
| // |
| // http://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. |
| |
| // Bindless Check Instrumentation Tests. |
| |
| #include <string> |
| #include <vector> |
| |
| #include "test/opt/assembly_builder.h" |
| #include "test/opt/pass_fixture.h" |
| #include "test/opt/pass_utils.h" |
| |
| namespace spvtools { |
| namespace opt { |
| namespace { |
| |
| using InstBindlessTest = PassTest<::testing::Test>; |
| |
| static const std::string kOutputDecorations = R"( |
| ; CHECK: OpDecorate [[output_buffer_type:%inst_bindless_OutputBuffer]] Block |
| ; CHECK: OpMemberDecorate [[output_buffer_type]] 0 Offset 0 |
| ; CHECK: OpMemberDecorate [[output_buffer_type]] 1 Offset 4 |
| ; CHECK: OpDecorate [[output_buffer_var:%\w+]] DescriptorSet 7 |
| ; CHECK: OpDecorate [[output_buffer_var]] Binding 0 |
| )"; |
| |
| static const std::string kOutputGlobals = R"( |
| ; CHECK: [[output_buffer_type]] = OpTypeStruct %uint %uint %_runtimearr_uint |
| ; CHECK: [[output_ptr_type:%\w+]] = OpTypePointer StorageBuffer [[output_buffer_type]] |
| ; CHECK: [[output_buffer_var]] = OpVariable [[output_ptr_type]] StorageBuffer |
| )"; |
| |
| static const std::string kStreamWrite4Begin = R"( |
| ; CHECK: %inst_bindless_stream_write_4 = OpFunction %void None {{%\w+}} |
| ; CHECK: [[param_1:%\w+]] = OpFunctionParameter %uint |
| ; CHECK: [[param_2:%\w+]] = OpFunctionParameter %uint |
| ; CHECK: [[param_3:%\w+]] = OpFunctionParameter %uint |
| ; CHECK: [[param_4:%\w+]] = OpFunctionParameter %uint |
| ; CHECK: {{%\w+}} = OpLabel |
| ; CHECK: {{%\w+}} = OpAccessChain %_ptr_StorageBuffer_uint [[output_buffer_var]] %uint_1 |
| ; CHECK: {{%\w+}} = OpAtomicIAdd %uint {{%\w+}} %uint_4 %uint_0 %uint_10 |
| ; CHECK: {{%\w+}} = OpIAdd %uint {{%\w+}} %uint_10 |
| ; CHECK: {{%\w+}} = OpArrayLength %uint [[output_buffer_var]] 2 |
| ; CHECK: {{%\w+}} = OpULessThanEqual %bool {{%\w+}} {{%\w+}} |
| ; CHECK: OpSelectionMerge {{%\w+}} None |
| ; CHECK: OpBranchConditional {{%\w+}} {{%\w+}} {{%\w+}} |
| ; CHECK: {{%\w+}} = OpLabel |
| ; CHECK: {{%\w+}} = OpIAdd %uint {{%\w+}} %uint_0 |
| ; CHECK: {{%\w+}} = OpAccessChain %_ptr_StorageBuffer_uint [[output_buffer_var]] %uint_2 {{%\w+}} |
| ; CHECK: OpStore {{%\w+}} %uint_10 |
| ; CHECK: {{%\w+}} = OpIAdd %uint {{%\w+}} %uint_1 |
| ; CHECK: {{%\w+}} = OpAccessChain %_ptr_StorageBuffer_uint [[output_buffer_var]] %uint_2 {{%\w+}} |
| ; CHECK: OpStore {{%\w+}} %uint_23 |
| ; CHECK: {{%\w+}} = OpIAdd %uint {{%\w+}} %uint_2 |
| ; CHECK: {{%\w+}} = OpAccessChain %_ptr_StorageBuffer_uint [[output_buffer_var]] %uint_2 {{%\w+}} |
| ; CHECK: OpStore {{%\w+}} [[param_1]] |
| )"; |
| |
| static const std::string kStreamWrite4End = R"( |
| ; CHECK: {{%\w+}} = OpIAdd %uint {{%\w+}} %uint_7 |
| ; CHECK: {{%\w+}} = OpAccessChain %_ptr_StorageBuffer_uint [[output_buffer_var]] %uint_2 {{%\w+}} |
| ; CHECK: OpStore {{%\w+}} [[param_2]] |
| ; CHECK: {{%\w+}} = OpIAdd %uint {{%\w+}} %uint_8 |
| ; CHECK: {{%\w+}} = OpAccessChain %_ptr_StorageBuffer_uint [[output_buffer_var]] %uint_2 {{%\w+}} |
| ; CHECK: OpStore {{%\w+}} [[param_3]] |
| ; CHECK: {{%\w+}} = OpIAdd %uint {{%\w+}} %uint_9 |
| ; CHECK: {{%\w+}} = OpAccessChain %_ptr_StorageBuffer_uint [[output_buffer_var]] %uint_2 {{%\w+}} |
| ; CHECK: OpStore {{%\w+}} [[param_4]] |
| ; CHECK: OpBranch {{%\w+}} |
| ; CHECK: {{%\w+}} = OpLabel |
| ; CHECK: OpReturn |
| ; CHECK: OpFunctionEnd |
| )"; |
| |
| // clang-format off |
| static const std::string kStreamWrite4Frag = kStreamWrite4Begin + R"( |
| ; CHECK: {{%\w+}} = OpIAdd %uint {{%\w+}} %uint_3 |
| ; CHECK: {{%\w+}} = OpAccessChain %_ptr_StorageBuffer_uint [[output_buffer_var]] %uint_2 {{%\w+}} |
| ; CHECK: OpStore {{%\w+}} %uint_4 |
| ; CHECK: {{%\w+}} = OpLoad %v4float %gl_FragCoord |
| ; CHECK: {{%\w+}} = OpBitcast %v4uint {{%\w+}} |
| ; CHECK: {{%\w+}} = OpCompositeExtract %uint {{%\w+}} 0 |
| ; CHECK: {{%\w+}} = OpIAdd %uint {{%\w+}} %uint_4 |
| ; CHECK: {{%\w+}} = OpAccessChain %_ptr_StorageBuffer_uint [[output_buffer_var]] %uint_2 {{%\w+}} |
| ; CHECK: OpStore {{%\w+}} {{%\w+}} |
| ; CHECK: {{%\w+}} = OpCompositeExtract %uint {{%\w+}} 1 |
| ; CHECK: {{%\w+}} = OpIAdd %uint {{%\w+}} %uint_5 |
| ; CHECK: {{%\w+}} = OpAccessChain %_ptr_StorageBuffer_uint [[output_buffer_var]] %uint_2 {{%\w+}} |
| ; CHECK: OpStore {{%\w+}} {{%\w+}} |
| )" + kStreamWrite4End; |
| |
| static const std::string kStreamWrite4Tese = kStreamWrite4Begin + R"( |
| ; CHECK: {{%\w+}} = OpIAdd %uint {{%\w+}} %uint_3 |
| ; CHECK: {{%\w+}} = OpAccessChain %_ptr_StorageBuffer_uint [[output_buffer_var]] %uint_2 {{%\w+}} |
| ; CHECK: OpStore {{%\w+}} %uint_2 |
| ; CHECK: {{%\w+}} = OpLoad %uint %gl_PrimitiveID |
| ; CHECK: {{%\w+}} = OpIAdd %uint {{%\w+}} %uint_4 |
| ; CHECK: {{%\w+}} = OpAccessChain %_ptr_StorageBuffer_uint [[output_buffer_var]] %uint_2 {{%\w+}} |
| ; CHECK: OpStore {{%\w+}} {{%\w+}} |
| ; CHECK: {{%\w+}} = OpLoad %v3float %gl_TessCoord |
| ; CHECK: {{%\w+}} = OpBitcast %v3uint {{%\w+}} |
| ; CHECK: {{%\w+}} = OpCompositeExtract %uint {{%\w+}} 0 |
| ; CHECK: {{%\w+}} = OpCompositeExtract %uint {{%\w+}} 1 |
| ; CHECK: {{%\w+}} = OpIAdd %uint {{%\w+}} %uint_5 |
| ; CHECK: {{%\w+}} = OpAccessChain %_ptr_StorageBuffer_uint [[output_buffer_var]] %uint_2 {{%\w+}} |
| ; CHECK: OpStore {{%\w+}} {{%\w+}} |
| ; CHECK: {{%\w+}} = OpIAdd %uint {{%\w+}} %uint_6 |
| ; CHECK: {{%\w+}} = OpAccessChain %_ptr_StorageBuffer_uint [[output_buffer_var]] %uint_2 {{%\w+}} |
| ; CHECK: OpStore {{%\w+}} {{%\w+}} |
| )" + kStreamWrite4End; |
| |
| static const std::string kStreamWrite4Vert = kStreamWrite4Begin + R"( |
| ; CHECK: {{%\w+}} = OpIAdd %uint {{%\w+}} %uint_3 |
| ; CHECK: {{%\w+}} = OpAccessChain %_ptr_StorageBuffer_uint [[output_buffer_var]] %uint_2 {{%\w+}} |
| ; CHECK: OpStore {{%\w+}} %uint_0 |
| ; CHECK: {{%\w+}} = OpLoad %uint %gl_VertexIndex |
| ; CHECK: {{%\w+}} = OpIAdd %uint {{%\w+}} %uint_4 |
| ; CHECK: {{%\w+}} = OpAccessChain %_ptr_StorageBuffer_uint [[output_buffer_var]] %uint_2 {{%\w+}} |
| ; CHECK: OpStore {{%\w+}} {{%\w+}} |
| ; CHECK: {{%\w+}} = OpLoad %uint %gl_InstanceIndex |
| ; CHECK: {{%\w+}} = OpIAdd %uint {{%\w+}} %uint_5 |
| ; CHECK: {{%\w+}} = OpAccessChain %_ptr_StorageBuffer_uint [[output_buffer_var]] %uint_2 {{%\w+}} |
| ; CHECK: OpStore {{%\w+}} {{%\w+}} |
| )" + kStreamWrite4End; |
| |
| static const std::string kStreamWrite4Compute = kStreamWrite4Begin + R"( |
| ; CHECK: {{%\w+}} = OpIAdd %uint {{%\w+}} %uint_3 |
| ; CHECK: {{%\w+}} = OpAccessChain %_ptr_StorageBuffer_uint [[output_buffer_var]] %uint_2 {{%\w+}} |
| ; CHECK: OpStore {{%\w+}} %uint_5 |
| ; CHECK: {{%\w+}} = OpLoad %v3uint %gl_GlobalInvocationID |
| ; CHECK: {{%\w+}} = OpCompositeExtract %uint {{%\w+}} 0 |
| ; CHECK: {{%\w+}} = OpCompositeExtract %uint {{%\w+}} 1 |
| ; CHECK: {{%\w+}} = OpCompositeExtract %uint {{%\w+}} 2 |
| ; CHECK: {{%\w+}} = OpIAdd %uint {{%\w+}} %uint_4 |
| ; CHECK: {{%\w+}} = OpAccessChain %_ptr_StorageBuffer_uint [[output_buffer_var]] %uint_2 {{%\w+}} |
| ; CHECK: OpStore {{%\w+}} {{%\w+}} |
| ; CHECK: {{%\w+}} = OpIAdd %uint {{%\w+}} %uint_5 |
| ; CHECK: {{%\w+}} = OpAccessChain %_ptr_StorageBuffer_uint [[output_buffer_var]] %uint_2 {{%\w+}} |
| ; CHECK: OpStore {{%\w+}} {{%\w+}} |
| ; CHECK: {{%\w+}} = OpIAdd %uint {{%\w+}} %uint_6 |
| ; CHECK: {{%\w+}} = OpAccessChain %_ptr_StorageBuffer_uint [[output_buffer_var]] %uint_2 {{%\w+}} |
| ; CHECK: OpStore {{%\w+}} {{%\w+}} |
| )" + kStreamWrite4End; |
| |
| static const std::string kStreamWrite4Ray = kStreamWrite4Begin + R"( |
| ; CHECK: {{%\w+}} = OpIAdd %uint {{%\w+}} %uint_3 |
| ; CHECK: {{%\w+}} = OpAccessChain %_ptr_StorageBuffer_uint [[output_buffer_var]] %uint_2 {{%\w+}} |
| ; CHECK: OpStore {{%\w+}} {{%\w+}} |
| ; CHECK: {{%\w+}} = OpLoad %v3uint {{%\w+}} |
| ; CHECK: {{%\w+}} = OpCompositeExtract %uint %90 0 |
| ; CHECK: {{%\w+}} = OpCompositeExtract %uint %90 1 |
| ; CHECK: {{%\w+}} = OpCompositeExtract %uint %90 2 |
| ; CHECK: {{%\w+}} = OpIAdd %uint {{%\w+}} %uint_4 |
| ; CHECK: {{%\w+}} = OpAccessChain %_ptr_StorageBuffer_uint [[output_buffer_var]] %uint_2 {{%\w+}} |
| ; CHECK: OpStore {{%\w+}} {{%\w+}} |
| ; CHECK: {{%\w+}} = OpIAdd %uint {{%\w+}} %uint_5 |
| ; CHECK: {{%\w+}} = OpAccessChain %_ptr_StorageBuffer_uint [[output_buffer_var]] %uint_2 {{%\w+}} |
| ; CHECK: OpStore {{%\w+}} {{%\w+}} |
| ; CHECK: {{%\w+}} = OpIAdd %uint {{%\w+}} %uint_6 |
| ; CHECK: {{%\w+}} = OpAccessChain %_ptr_StorageBuffer_uint [[output_buffer_var]] %uint_2 {{%\w+}} |
| ; CHECK: OpStore {{%\w+}} {{%\w+}} |
| )" + kStreamWrite4End; |
| // clang-format on |
| |
| static const std::string kStreamWrite5Begin = R"( |
| ; CHECK: %inst_bindless_stream_write_5 = OpFunction %void None {{%\w+}} |
| ; CHECK: [[param_1:%\w+]] = OpFunctionParameter %uint |
| ; CHECK: [[param_2:%\w+]] = OpFunctionParameter %uint |
| ; CHECK: [[param_3:%\w+]] = OpFunctionParameter %uint |
| ; CHECK: [[param_4:%\w+]] = OpFunctionParameter %uint |
| ; CHECK: [[param_5:%\w+]] = OpFunctionParameter %uint |
| ; CHECK: {{%\w+}} = OpLabel |
| ; CHECK: {{%\w+}} = OpAccessChain %_ptr_StorageBuffer_uint [[output_buffer_var]] %uint_1 |
| ; CHECK: {{%\w+}} = OpAtomicIAdd %uint {{%\w+}} %uint_4 %uint_0 %uint_11 |
| ; CHECK: {{%\w+}} = OpIAdd %uint {{%\w+}} %uint_11 |
| ; CHECK: {{%\w+}} = OpArrayLength %uint [[output_buffer_var]] 2 |
| ; CHECK: {{%\w+}} = OpULessThanEqual %bool {{%\w+}} {{%\w+}} |
| ; CHECK: OpSelectionMerge {{%\w+}} None |
| ; CHECK: OpBranchConditional {{%\w+}} {{%\w+}} {{%\w+}} |
| ; CHECK: {{%\w+}} = OpLabel |
| ; CHECK: {{%\w+}} = OpIAdd %uint {{%\w+}} %uint_0 |
| ; CHECK: {{%\w+}} = OpAccessChain %_ptr_StorageBuffer_uint [[output_buffer_var]] %uint_2 {{%\w+}} |
| ; CHECK: OpStore {{%\w+}} %uint_11 |
| ; CHECK: {{%\w+}} = OpIAdd %uint {{%\w+}} %uint_1 |
| ; CHECK: {{%\w+}} = OpAccessChain %_ptr_StorageBuffer_uint [[output_buffer_var]] %uint_2 {{%\w+}} |
| ; CHECK: OpStore {{%\w+}} %uint_23 |
| ; CHECK: {{%\w+}} = OpIAdd %uint {{%\w+}} %uint_2 |
| ; CHECK: {{%\w+}} = OpAccessChain %_ptr_StorageBuffer_uint [[output_buffer_var]] %uint_2 {{%\w+}} |
| ; CHECK: OpStore {{%\w+}} [[param_1]] |
| )"; |
| |
| static const std::string kStreamWrite5End = R"( |
| ; CHECK: {{%\w+}} = OpIAdd %uint {{%\w+}} %uint_7 |
| ; CHECK: {{%\w+}} = OpAccessChain %_ptr_StorageBuffer_uint [[output_buffer_var]] %uint_2 {{%\w+}} |
| ; CHECK: OpStore {{%\w+}} [[param_2]] |
| ; CHECK: {{%\w+}} = OpIAdd %uint {{%\w+}} %uint_8 |
| ; CHECK: {{%\w+}} = OpAccessChain %_ptr_StorageBuffer_uint [[output_buffer_var]] %uint_2 {{%\w+}} |
| ; CHECK: OpStore {{%\w+}} [[param_3]] |
| ; CHECK: {{%\w+}} = OpIAdd %uint {{%\w+}} %uint_9 |
| ; CHECK: {{%\w+}} = OpAccessChain %_ptr_StorageBuffer_uint [[output_buffer_var]] %uint_2 {{%\w+}} |
| ; CHECK: OpStore {{%\w+}} [[param_4]] |
| ; CHECK: {{%\w+}} = OpIAdd %uint {{%\w+}} %uint_10 |
| ; CHECK: {{%\w+}} = OpAccessChain %_ptr_StorageBuffer_uint [[output_buffer_var]] %uint_2 {{%\w+}} |
| ; CHECK: OpStore {{%\w+}} [[param_5]] |
| ; CHECK: OpBranch {{%\w+}} |
| ; CHECK: {{%\w+}} = OpLabel |
| ; CHECK: OpReturn |
| ; CHECK: OpFunctionEnd |
| )"; |
| |
| // clang-format off |
| static const std::string kStreamWrite5Frag = kStreamWrite5Begin + R"( |
| ; CHECK: {{%\w+}} = OpIAdd %uint {{%\w+}} %uint_3 |
| ; CHECK: {{%\w+}} = OpAccessChain %_ptr_StorageBuffer_uint [[output_buffer_var]] %uint_2 {{%\w+}} |
| ; CHECK: OpStore {{%\w+}} %uint_4 |
| ; CHECK: {{%\w+}} = OpLoad %v4float %gl_FragCoord |
| ; CHECK: {{%\w+}} = OpBitcast %v4uint {{%\w+}} |
| ; CHECK: {{%\w+}} = OpCompositeExtract %uint {{%\w+}} 0 |
| ; CHECK: {{%\w+}} = OpIAdd %uint {{%\w+}} %uint_4 |
| ; CHECK: {{%\w+}} = OpAccessChain %_ptr_StorageBuffer_uint [[output_buffer_var]] %uint_2 {{%\w+}} |
| ; CHECK: OpStore {{%\w+}} {{%\w+}} |
| ; CHECK: {{%\w+}} = OpCompositeExtract %uint {{%\w+}} 1 |
| ; CHECK: {{%\w+}} = OpIAdd %uint {{%\w+}} %uint_5 |
| ; CHECK: {{%\w+}} = OpAccessChain %_ptr_StorageBuffer_uint [[output_buffer_var]] %uint_2 {{%\w+}} |
| ; CHECK: OpStore {{%\w+}} {{%\w+}} |
| )" + kStreamWrite4End; |
| |
| static const std::string kStreamWrite5Vert = kStreamWrite5Begin + R"( |
| ; CHECK: {{%\w+}} = OpIAdd %uint {{%\w+}} %uint_3 |
| ; CHECK: {{%\w+}} = OpAccessChain %_ptr_StorageBuffer_uint [[output_buffer_var]] %uint_2 {{%\w+}} |
| ; CHECK: OpStore {{%\w+}} %uint_0 |
| ; CHECK: {{%\w+}} = OpLoad %uint %gl_VertexIndex |
| ; CHECK: {{%\w+}} = OpIAdd %uint {{%\w+}} %uint_4 |
| ; CHECK: {{%\w+}} = OpAccessChain %_ptr_StorageBuffer_uint [[output_buffer_var]] %uint_2 {{%\w+}} |
| ; CHECK: OpStore {{%\w+}} {{%\w+}} |
| ; CHECK: {{%\w+}} = OpLoad %uint %gl_InstanceIndex |
| ; CHECK: {{%\w+}} = OpIAdd %uint {{%\w+}} %uint_5 |
| ; CHECK: {{%\w+}} = OpAccessChain %_ptr_StorageBuffer_uint [[output_buffer_var]] %uint_2 {{%\w+}} |
| ; CHECK: OpStore {{%\w+}} {{%\w+}} |
| )" + kStreamWrite5End; |
| // clang-format on |
| |
| static const std::string kInputDecorations = R"( |
| ; CHECK: OpDecorate [[input_buffer_type:%inst_bindless_InputBuffer]] Block |
| ; CHECK: OpMemberDecorate [[input_buffer_type]] 0 Offset 0 |
| ; CHECK: OpDecorate [[input_buffer_var:%\w+]] DescriptorSet 7 |
| ; CHECK: OpDecorate [[input_buffer_var]] Binding 1 |
| )"; |
| |
| static const std::string kInputGlobals = R"( |
| ; CHECK: [[input_buffer_type]] = OpTypeStruct %_runtimearr_uint |
| ; CHECK: [[input_ptr_type:%\w+]] = OpTypePointer StorageBuffer [[input_buffer_type]] |
| ; CHECK: [[input_buffer_var]] = OpVariable [[input_ptr_type]] StorageBuffer |
| )"; |
| |
| static const std::string kDirectRead2 = R"( |
| ; CHECK: %inst_bindless_direct_read_2 = OpFunction %uint None {{%\w+}} |
| ; CHECK: [[param_1:%\w+]] = OpFunctionParameter %uint |
| ; CHECK: [[param_2:%\w+]] = OpFunctionParameter %uint |
| ; CHECK: {{%\w+}} = OpLabel |
| ; CHECK: {{%\w+}} = OpAccessChain %_ptr_StorageBuffer_uint [[input_buffer_var]] %uint_0 [[param_1]] |
| ; CHECK: {{%\w+}} = OpLoad %uint {{%\w+}} |
| ; CHECK: {{%\w+}} = OpIAdd %uint {{%\w+}} [[param_2]] |
| ; CHECK: {{%\w+}} = OpAccessChain %_ptr_StorageBuffer_uint [[input_buffer_var]] %uint_0 {{%\w+}} |
| ; CHECK: {{%\w+}} = OpLoad %uint {{%\w+}} |
| ; CHECK: OpReturnValue {{%\w+}} |
| ; CHECK: OpFunctionEnd |
| )"; |
| |
| static const std::string kDirectRead3 = R"( |
| ;CHECK: %inst_bindless_direct_read_3 = OpFunction %uint None {{%\w+}} |
| ; CHECK: [[param_1:%\w+]] = OpFunctionParameter %uint |
| ; CHECK: [[param_2:%\w+]] = OpFunctionParameter %uint |
| ; CHECK: [[param_3:%\w+]] = OpFunctionParameter %uint |
| ;CHECK: {{%\w+}} = OpLabel |
| ;CHECK: {{%\w+}} = OpAccessChain %_ptr_StorageBuffer_uint [[input_buffer_var]] %uint_0 [[param_1]] |
| ;CHECK: {{%\w+}} = OpLoad %uint {{%\w+}} |
| ;CHECK: {{%\w+}} = OpIAdd %uint {{%\w+}} [[param_2]] |
| ;CHECK: {{%\w+}} = OpAccessChain %_ptr_StorageBuffer_uint [[input_buffer_var]] %uint_0 {{%\w+}} |
| ;CHECK: {{%\w+}} = OpLoad %uint {{%\w+}} |
| ;CHECK: {{%\w+}} = OpIAdd %uint {{%\w+}} [[param_3]] |
| ;CHECK: {{%\w+}} = OpAccessChain %_ptr_StorageBuffer_uint [[input_buffer_var]] %uint_0 {{%\w+}} |
| ;CHECK: {{%\w+}} = OpLoad %uint {{%\w+}} |
| ;CHECK: OpReturnValue {{%\w+}} |
| ;CHECK: OpFunctionEnd |
| )"; |
| |
| static const std::string kDirectRead4 = R"( |
| ; CHECK: %inst_bindless_direct_read_4 = OpFunction %uint None {{%\w+}} |
| ; CHECK: [[param_1:%\w+]] = OpFunctionParameter %uint |
| ; CHECK: [[param_2:%\w+]] = OpFunctionParameter %uint |
| ; CHECK: [[param_3:%\w+]] = OpFunctionParameter %uint |
| ; CHECK: [[param_4:%\w+]] = OpFunctionParameter %uint |
| ; CHECK: {{%\w+}} = OpLabel |
| ; CHECK: {{%\w+}} = OpAccessChain %_ptr_StorageBuffer_uint [[input_buffer_var]] %uint_0 [[param_1]] |
| ; CHECK: {{%\w+}} = OpLoad %uint {{%\w+}} |
| ; CHECK: {{%\w+}} = OpIAdd %uint {{%\w+}} [[param_2]] |
| ; CHECK: {{%\w+}} = OpAccessChain %_ptr_StorageBuffer_uint [[input_buffer_var]] %uint_0 {{%\w+}} |
| ; CHECK: {{%\w+}} = OpLoad %uint {{%\w+}} |
| ; CHECK: {{%\w+}} = OpIAdd %uint {{%\w+}} [[param_3]] |
| ; CHECK: {{%\w+}} = OpAccessChain %_ptr_StorageBuffer_uint [[input_buffer_var]] %uint_0 {{%\w+}} |
| ; CHECK: {{%\w+}} = OpLoad %uint {{%\w+}} |
| ; CHECK: {{%\w+}} = OpIAdd %uint {{%\w+}} [[param_4]] |
| ; CHECK: {{%\w+}} = OpAccessChain %_ptr_StorageBuffer_uint [[input_buffer_var]] %uint_0 {{%\w+}} |
| ; CHECK: {{%\w+}} = OpLoad %uint {{%\w+}} |
| ; CHECK: OpReturnValue {{%\w+}} |
| ; CHECK: OpFunctionEnd |
| )"; |
| |
| TEST_F(InstBindlessTest, NoInstrumentConstIndexInbounds) { |
| // Texture2D g_tColor[128]; |
| // |
| // SamplerState g_sAniso; |
| // |
| // struct PS_INPUT |
| // { |
| // float2 vTextureCoords : TEXCOORD2; |
| // }; |
| // |
| // struct PS_OUTPUT |
| // { |
| // float4 vColor : SV_Target0; |
| // }; |
| // |
| // PS_OUTPUT MainPs(PS_INPUT i) |
| // { |
| // PS_OUTPUT ps_output; |
| // |
| // ps_output.vColor = g_tColor[ 37 ].Sample(g_sAniso, i.vTextureCoords.xy); |
| // return ps_output; |
| // } |
| |
| const std::string before = |
| R"(OpCapability Shader |
| %1 = OpExtInstImport "GLSL.std.450" |
| OpMemoryModel Logical GLSL450 |
| OpEntryPoint Fragment %MainPs "MainPs" %i_vTextureCoords %_entryPointOutput_vColor |
| OpExecutionMode %MainPs OriginUpperLeft |
| OpSource HLSL 500 |
| OpName %MainPs "MainPs" |
| OpName %g_tColor "g_tColor" |
| OpName %g_sAniso "g_sAniso" |
| OpName %i_vTextureCoords "i.vTextureCoords" |
| OpName %_entryPointOutput_vColor "@entryPointOutput.vColor" |
| OpDecorate %g_tColor DescriptorSet 3 |
| OpDecorate %g_tColor Binding 0 |
| OpDecorate %g_sAniso DescriptorSet 0 |
| OpDecorate %i_vTextureCoords Location 0 |
| OpDecorate %_entryPointOutput_vColor Location 0 |
| %void = OpTypeVoid |
| %8 = OpTypeFunction %void |
| %float = OpTypeFloat 32 |
| %v2float = OpTypeVector %float 2 |
| %v4float = OpTypeVector %float 4 |
| %int = OpTypeInt 32 1 |
| %int_0 = OpConstant %int 0 |
| %int_37 = OpConstant %int 37 |
| %15 = OpTypeImage %float 2D 0 0 0 1 Unknown |
| %uint = OpTypeInt 32 0 |
| %uint_128 = OpConstant %uint 128 |
| %_arr_15_uint_128 = OpTypeArray %15 %uint_128 |
| %_ptr_UniformConstant__arr_15_uint_128 = OpTypePointer UniformConstant %_arr_15_uint_128 |
| %g_tColor = OpVariable %_ptr_UniformConstant__arr_15_uint_128 UniformConstant |
| %_ptr_UniformConstant_15 = OpTypePointer UniformConstant %15 |
| %21 = OpTypeSampler |
| %_ptr_UniformConstant_21 = OpTypePointer UniformConstant %21 |
| %g_sAniso = OpVariable %_ptr_UniformConstant_21 UniformConstant |
| %23 = OpTypeSampledImage %15 |
| %_ptr_Input_v2float = OpTypePointer Input %v2float |
| %i_vTextureCoords = OpVariable %_ptr_Input_v2float Input |
| %_ptr_Output_v4float = OpTypePointer Output %v4float |
| %_entryPointOutput_vColor = OpVariable %_ptr_Output_v4float Output |
| %MainPs = OpFunction %void None %8 |
| %26 = OpLabel |
| %27 = OpLoad %v2float %i_vTextureCoords |
| %28 = OpAccessChain %_ptr_UniformConstant_15 %g_tColor %int_37 |
| %29 = OpLoad %15 %28 |
| %30 = OpLoad %21 %g_sAniso |
| %31 = OpSampledImage %23 %29 %30 |
| %32 = OpImageSampleImplicitLod %v4float %31 %27 |
| OpStore %_entryPointOutput_vColor %32 |
| OpReturn |
| OpFunctionEnd |
| )"; |
| |
| SetAssembleOptions(SPV_TEXT_TO_BINARY_OPTION_PRESERVE_NUMERIC_IDS); |
| SinglePassRunAndCheck<InstBindlessCheckPass>( |
| before, before, true, true, 7u, 23u, false, false, false, false, false); |
| } |
| |
| TEST_F(InstBindlessTest, NoInstrumentNonBindless) { |
| // This test verifies that the pass will correctly not instrument vanilla |
| // texture sample. |
| // |
| // Texture2D g_tColor; |
| // |
| // SamplerState g_sAniso; |
| // |
| // struct PS_INPUT |
| // { |
| // float2 vTextureCoords : TEXCOORD2; |
| // }; |
| // |
| // struct PS_OUTPUT |
| // { |
| // float4 vColor : SV_Target0; |
| // }; |
| // |
| // PS_OUTPUT MainPs(PS_INPUT i) |
| // { |
| // PS_OUTPUT ps_output; |
| // ps_output.vColor = |
| // g_tColor.Sample(g_sAniso, i.vTextureCoords.xy); |
| // return ps_output; |
| // } |
| |
| const std::string whole_file = |
| R"(OpCapability Shader |
| %1 = OpExtInstImport "GLSL.std.450" |
| OpMemoryModel Logical GLSL450 |
| OpEntryPoint Fragment %MainPs "MainPs" %i_vTextureCoords %_entryPointOutput_vColor |
| OpExecutionMode %MainPs OriginUpperLeft |
| OpSource HLSL 500 |
| OpName %MainPs "MainPs" |
| OpName %g_tColor "g_tColor" |
| OpName %g_sAniso "g_sAniso" |
| OpName %i_vTextureCoords "i.vTextureCoords" |
| OpName %_entryPointOutput_vColor "@entryPointOutput.vColor" |
| OpDecorate %g_tColor DescriptorSet 0 |
| OpDecorate %g_tColor Binding 0 |
| OpDecorate %g_sAniso DescriptorSet 0 |
| OpDecorate %g_sAniso Binding 0 |
| OpDecorate %i_vTextureCoords Location 0 |
| OpDecorate %_entryPointOutput_vColor Location 0 |
| %void = OpTypeVoid |
| %8 = OpTypeFunction %void |
| %float = OpTypeFloat 32 |
| %v2float = OpTypeVector %float 2 |
| %v4float = OpTypeVector %float 4 |
| %12 = OpTypeImage %float 2D 0 0 0 1 Unknown |
| %_ptr_UniformConstant_12 = OpTypePointer UniformConstant %12 |
| %g_tColor = OpVariable %_ptr_UniformConstant_12 UniformConstant |
| %14 = OpTypeSampler |
| %_ptr_UniformConstant_14 = OpTypePointer UniformConstant %14 |
| %g_sAniso = OpVariable %_ptr_UniformConstant_14 UniformConstant |
| %16 = OpTypeSampledImage %12 |
| %_ptr_Input_v2float = OpTypePointer Input %v2float |
| %i_vTextureCoords = OpVariable %_ptr_Input_v2float Input |
| %_ptr_Output_v4float = OpTypePointer Output %v4float |
| %_entryPointOutput_vColor = OpVariable %_ptr_Output_v4float Output |
| %MainPs = OpFunction %void None %8 |
| %19 = OpLabel |
| %20 = OpLoad %v2float %i_vTextureCoords |
| %21 = OpLoad %12 %g_tColor |
| %22 = OpLoad %14 %g_sAniso |
| %23 = OpSampledImage %16 %21 %22 |
| %24 = OpImageSampleImplicitLod %v4float %23 %20 |
| OpStore %_entryPointOutput_vColor %24 |
| OpReturn |
| OpFunctionEnd |
| )"; |
| |
| // SetAssembleOptions(SPV_TEXT_TO_BINARY_OPTION_PRESERVE_NUMERIC_IDS); |
| SinglePassRunAndCheck<InstBindlessCheckPass>(whole_file, whole_file, true, |
| true, 7u, 23u, false, false, |
| false, false, false); |
| } |
| |
| TEST_F(InstBindlessTest, Simple) { |
| // Texture2D g_tColor[128]; |
| // |
| // layout(push_constant) cbuffer PerViewConstantBuffer_t |
| // { |
| // uint g_nDataIdx; |
| // }; |
| // |
| // SamplerState g_sAniso; |
| // |
| // struct PS_INPUT |
| // { |
| // float2 vTextureCoords : TEXCOORD2; |
| // }; |
| // |
| // struct PS_OUTPUT |
| // { |
| // float4 vColor : SV_Target0; |
| // }; |
| // |
| // PS_OUTPUT MainPs(PS_INPUT i) |
| // { |
| // PS_OUTPUT ps_output; |
| // ps_output.vColor = |
| // g_tColor[ g_nDataIdx ].Sample(g_sAniso, i.vTextureCoords.xy); |
| // return ps_output; |
| // } |
| |
| const std::string entry = R"( |
| OpCapability Shader |
| ; CHECK: OpExtension "SPV_KHR_storage_buffer_storage_class" |
| %1 = OpExtInstImport "GLSL.std.450" |
| OpMemoryModel Logical GLSL450 |
| OpEntryPoint Fragment %MainPs "MainPs" %i_vTextureCoords %_entryPointOutput_vColor |
| ; CHECK: OpEntryPoint Fragment %MainPs "MainPs" %i_vTextureCoords %_entryPointOutput_vColor %gl_FragCoord |
| OpExecutionMode %MainPs OriginUpperLeft |
| OpSource HLSL 500 |
| )"; |
| |
| // clang-format off |
| const std::string names_annots = R"( |
| OpName %MainPs "MainPs" |
| OpName %g_tColor "g_tColor" |
| OpName %PerViewConstantBuffer_t "PerViewConstantBuffer_t" |
| OpMemberName %PerViewConstantBuffer_t 0 "g_nDataIdx" |
| OpName %_ "" |
| OpName %g_sAniso "g_sAniso" |
| OpName %i_vTextureCoords "i.vTextureCoords" |
| OpName %_entryPointOutput_vColor "@entryPointOutput.vColor" |
| OpDecorate %g_tColor DescriptorSet 3 |
| OpDecorate %g_tColor Binding 0 |
| OpMemberDecorate %PerViewConstantBuffer_t 0 Offset 0 |
| OpDecorate %PerViewConstantBuffer_t Block |
| OpDecorate %g_sAniso DescriptorSet 0 |
| OpDecorate %i_vTextureCoords Location 0 |
| OpDecorate %_entryPointOutput_vColor Location 0 |
| ; CHECK: OpDecorate %_runtimearr_uint ArrayStride 4 |
| )" + kOutputDecorations + R"( |
| ; CHECK: OpDecorate %gl_FragCoord BuiltIn FragCoord |
| )"; |
| |
| const std::string consts_types_vars = R"( |
| %void = OpTypeVoid |
| %10 = OpTypeFunction %void |
| %float = OpTypeFloat 32 |
| %v2float = OpTypeVector %float 2 |
| %v4float = OpTypeVector %float 4 |
| %int = OpTypeInt 32 1 |
| %int_0 = OpConstant %int 0 |
| %16 = OpTypeImage %float 2D 0 0 0 1 Unknown |
| %uint = OpTypeInt 32 0 |
| %uint_128 = OpConstant %uint 128 |
| %_arr_16_uint_128 = OpTypeArray %16 %uint_128 |
| %_ptr_UniformConstant__arr_16_uint_128 = OpTypePointer UniformConstant %_arr_16_uint_128 |
| %g_tColor = OpVariable %_ptr_UniformConstant__arr_16_uint_128 UniformConstant |
| %PerViewConstantBuffer_t = OpTypeStruct %uint |
| %_ptr_PushConstant_PerViewConstantBuffer_t = OpTypePointer PushConstant %PerViewConstantBuffer_t |
| %_ = OpVariable %_ptr_PushConstant_PerViewConstantBuffer_t PushConstant |
| %_ptr_PushConstant_uint = OpTypePointer PushConstant %uint |
| %_ptr_UniformConstant_16 = OpTypePointer UniformConstant %16 |
| %24 = OpTypeSampler |
| %_ptr_UniformConstant_24 = OpTypePointer UniformConstant %24 |
| %g_sAniso = OpVariable %_ptr_UniformConstant_24 UniformConstant |
| %26 = OpTypeSampledImage %16 |
| %_ptr_Input_v2float = OpTypePointer Input %v2float |
| %i_vTextureCoords = OpVariable %_ptr_Input_v2float Input |
| %_ptr_Output_v4float = OpTypePointer Output %v4float |
| %_entryPointOutput_vColor = OpVariable %_ptr_Output_v4float Output |
| ; CHECK: %bool = OpTypeBool |
| ; CHECK: %48 = OpTypeFunction %void %uint %uint %uint %uint |
| ; CHECK: %_runtimearr_uint = OpTypeRuntimeArray %uint |
| )" + kOutputGlobals + R"( |
| ; CHECK: %_ptr_StorageBuffer_uint = OpTypePointer StorageBuffer %uint |
| ; CHECK: %_ptr_Input_v4float = OpTypePointer Input %v4float |
| ; CHECK: %gl_FragCoord = OpVariable %_ptr_Input_v4float Input |
| ; CHECK: %v4uint = OpTypeVector %uint 4 |
| ; CHECK: %103 = OpConstantNull %v4float |
| )"; |
| // clang-format on |
| |
| const std::string main_func = R"( |
| %MainPs = OpFunction %void None %10 |
| %29 = OpLabel |
| %30 = OpLoad %v2float %i_vTextureCoords |
| %31 = OpAccessChain %_ptr_PushConstant_uint %_ %int_0 |
| %32 = OpLoad %uint %31 |
| %33 = OpAccessChain %_ptr_UniformConstant_16 %g_tColor %32 |
| %34 = OpLoad %16 %33 |
| %35 = OpLoad %24 %g_sAniso |
| %36 = OpSampledImage %26 %34 %35 |
| %37 = OpImageSampleImplicitLod %v4float %36 %30 |
| OpStore %_entryPointOutput_vColor %37 |
| ; CHECK-NOT: %37 = OpImageSampleImplicitLod %v4float %36 %30 |
| ; CHECK-NOT: OpStore %_entryPointOutput_vColor %37 |
| ; CHECK: %40 = OpULessThan %bool %32 %uint_128 |
| ; CHECK: OpSelectionMerge %41 None |
| ; CHECK: OpBranchConditional %40 %42 %43 |
| ; CHECK: %42 = OpLabel |
| ; CHECK: %44 = OpLoad %16 %33 |
| ; CHECK: %45 = OpSampledImage %26 %44 %35 |
| ; CHECK: %46 = OpImageSampleImplicitLod %v4float %45 %30 |
| ; CHECK: OpBranch %41 |
| ; CHECK: %43 = OpLabel |
| ; CHECK: %102 = OpFunctionCall %void %inst_bindless_stream_write_4 %uint_56 %uint_0 %32 %uint_128 |
| ; CHECK: OpBranch %41 |
| ; CHECK: %41 = OpLabel |
| ; CHECK: %104 = OpPhi %v4float %46 %42 %103 %43 |
| ; CHECK: OpStore %_entryPointOutput_vColor %104 |
| OpReturn |
| OpFunctionEnd |
| )"; |
| |
| const std::string output_func = kStreamWrite4Frag; |
| |
| SetAssembleOptions(SPV_TEXT_TO_BINARY_OPTION_PRESERVE_NUMERIC_IDS); |
| SinglePassRunAndMatch<InstBindlessCheckPass, uint32_t, uint32_t, bool, bool>( |
| entry + names_annots + consts_types_vars + main_func + output_func, true, |
| 7u, 23u, false, false, false, false, false); |
| } |
| |
| TEST_F(InstBindlessTest, InstrumentMultipleInstructions) { |
| // Texture2D g_tColor[128]; |
| // |
| // layout(push_constant) cbuffer PerViewConstantBuffer_t |
| // { |
| // uint g_nDataIdx; |
| // uint g_nDataIdx2; |
| // }; |
| // |
| // SamplerState g_sAniso; |
| // |
| // struct PS_INPUT |
| // { |
| // float2 vTextureCoords : TEXCOORD2; |
| // }; |
| // |
| // struct PS_OUTPUT |
| // { |
| // float4 vColor : SV_Target0; |
| // }; |
| // |
| // PS_OUTPUT MainPs(PS_INPUT i) |
| // { |
| // PS_OUTPUT ps_output; |
| // |
| // float t = g_tColor[g_nDataIdx ].Sample(g_sAniso, i.vTextureCoords.xy); |
| // float t2 = g_tColor[g_nDataIdx2].Sample(g_sAniso, i.vTextureCoords.xy); |
| // ps_output.vColor = t + t2; |
| // return ps_output; |
| // } |
| |
| // clang-format off |
| const std::string defs = R"( |
| OpCapability Shader |
| ; CHECK: OpExtension "SPV_KHR_storage_buffer_storage_class" |
| %1 = OpExtInstImport "GLSL.std.450" |
| OpMemoryModel Logical GLSL450 |
| OpEntryPoint Fragment %MainPs "MainPs" %i_vTextureCoords %_entryPointOutput_vColor |
| ; CHECK: OpEntryPoint Fragment %MainPs "MainPs" %i_vTextureCoords %_entryPointOutput_vColor %gl_FragCoord |
| OpExecutionMode %MainPs OriginUpperLeft |
| OpSource HLSL 500 |
| OpName %MainPs "MainPs" |
| OpName %g_tColor "g_tColor" |
| OpName %PerViewConstantBuffer_t "PerViewConstantBuffer_t" |
| OpMemberName %PerViewConstantBuffer_t 0 "g_nDataIdx" |
| OpName %_ "" |
| OpName %g_sAniso "g_sAniso" |
| OpName %i_vTextureCoords "i.vTextureCoords" |
| OpName %_entryPointOutput_vColor "@entryPointOutput.vColor" |
| OpDecorate %g_tColor DescriptorSet 3 |
| OpDecorate %g_tColor Binding 0 |
| OpMemberDecorate %PerViewConstantBuffer_t 0 Offset 0 |
| OpMemberDecorate %PerViewConstantBuffer_t 1 Offset 4 |
| OpDecorate %PerViewConstantBuffer_t Block |
| OpDecorate %g_sAniso DescriptorSet 0 |
| OpDecorate %i_vTextureCoords Location 0 |
| OpDecorate %_entryPointOutput_vColor Location 0 |
| ; CHECK: OpDecorate %_runtimearr_uint ArrayStride 4 |
| )" + kOutputDecorations + R"( |
| ; CHECK: OpDecorate %gl_FragCoord BuiltIn FragCoord |
| %void = OpTypeVoid |
| %10 = OpTypeFunction %void |
| %float = OpTypeFloat 32 |
| %v2float = OpTypeVector %float 2 |
| %v4float = OpTypeVector %float 4 |
| %int = OpTypeInt 32 1 |
| %int_0 = OpConstant %int 0 |
| %int_1 = OpConstant %int 1 |
| %17 = OpTypeImage %float 2D 0 0 0 1 Unknown |
| %uint = OpTypeInt 32 0 |
| %uint_128 = OpConstant %uint 128 |
| %_arr_17_uint_128 = OpTypeArray %17 %uint_128 |
| %_ptr_UniformConstant__arr_17_uint_128 = OpTypePointer UniformConstant %_arr_17_uint_128 |
| %g_tColor = OpVariable %_ptr_UniformConstant__arr_17_uint_128 UniformConstant |
| %PerViewConstantBuffer_t = OpTypeStruct %uint %uint |
| %_ptr_PushConstant_PerViewConstantBuffer_t = OpTypePointer PushConstant %PerViewConstantBuffer_t |
| %_ = OpVariable %_ptr_PushConstant_PerViewConstantBuffer_t PushConstant |
| %_ptr_PushConstant_uint = OpTypePointer PushConstant %uint |
| %_ptr_UniformConstant_17 = OpTypePointer UniformConstant %17 |
| %25 = OpTypeSampler |
| %_ptr_UniformConstant_25 = OpTypePointer UniformConstant %25 |
| %g_sAniso = OpVariable %_ptr_UniformConstant_25 UniformConstant |
| %27 = OpTypeSampledImage %17 |
| %_ptr_Input_v2float = OpTypePointer Input %v2float |
| %i_vTextureCoords = OpVariable %_ptr_Input_v2float Input |
| %_ptr_Output_v4float = OpTypePointer Output %v4float |
| %_entryPointOutput_vColor = OpVariable %_ptr_Output_v4float Output |
| ; CHECK: %bool = OpTypeBool |
| ; CHECK: %56 = OpTypeFunction %void %uint %uint %uint %uint |
| ; CHECK: %_runtimearr_uint = OpTypeRuntimeArray %uint |
| )" + kOutputGlobals + R"( |
| ; CHECK: %_ptr_StorageBuffer_uint = OpTypePointer StorageBuffer %uint |
| ; CHECK: %_ptr_Input_v4float = OpTypePointer Input %v4float |
| ; CHECK: %gl_FragCoord = OpVariable %_ptr_Input_v4float Input |
| ; CHECK: %v4uint = OpTypeVector %uint 4 |
| ; CHECK: %111 = OpConstantNull %v4float |
| )"; |
| // clang-format on |
| |
| const std::string main_func = |
| R"(%MainPs = OpFunction %void None %10 |
| %30 = OpLabel |
| %31 = OpLoad %v2float %i_vTextureCoords |
| %32 = OpAccessChain %_ptr_PushConstant_uint %_ %int_0 |
| %33 = OpLoad %uint %32 |
| %34 = OpAccessChain %_ptr_UniformConstant_17 %g_tColor %33 |
| %35 = OpLoad %17 %34 |
| %36 = OpLoad %25 %g_sAniso |
| %37 = OpSampledImage %27 %35 %36 |
| %38 = OpImageSampleImplicitLod %v4float %37 %31 |
| ; CHECK-NOT: %38 = OpImageSampleImplicitLod %v4float %37 %31 |
| ; CHECK: %48 = OpULessThan %bool %33 %uint_128 |
| ; CHECK: OpSelectionMerge %49 None |
| ; CHECK: OpBranchConditional %48 %50 %51 |
| ; CHECK: %50 = OpLabel |
| ; CHECK: %52 = OpLoad %17 %34 |
| ; CHECK: %53 = OpSampledImage %27 %52 %36 |
| ; CHECK: %54 = OpImageSampleImplicitLod %v4float %53 %31 |
| ; CHECK: OpBranch %49 |
| ; CHECK: %51 = OpLabel |
| ; CHECK: %110 = OpFunctionCall %void %inst_bindless_stream_write_4 %uint_58 %uint_0 %33 %uint_128 |
| ; CHECK: OpBranch %49 |
| ; CHECK: %49 = OpLabel |
| ; CHECK: %112 = OpPhi %v4float %54 %50 %111 %51 |
| %39 = OpAccessChain %_ptr_PushConstant_uint %_ %int_1 |
| %40 = OpLoad %uint %39 |
| %41 = OpAccessChain %_ptr_UniformConstant_17 %g_tColor %40 |
| %42 = OpLoad %17 %41 |
| %43 = OpSampledImage %27 %42 %36 |
| %44 = OpImageSampleImplicitLod %v4float %43 %31 |
| %45 = OpFAdd %v4float %38 %44 |
| ; CHECK-NOT: %44 = OpImageSampleImplicitLod %v4float %43 %31 |
| ; CHECK-NOT: %45 = OpFAdd %v4float %38 %44 |
| ; CHECK: %113 = OpULessThan %bool %40 %uint_128 |
| ; CHECK: OpSelectionMerge %114 None |
| ; CHECK: OpBranchConditional %113 %115 %116 |
| ; CHECK: %115 = OpLabel |
| ; CHECK: %117 = OpLoad %17 %41 |
| ; CHECK: %118 = OpSampledImage %27 %117 %36 |
| ; CHECK: %119 = OpImageSampleImplicitLod %v4float %118 %31 |
| ; CHECK: OpBranch %114 |
| ; CHECK: %116 = OpLabel |
| ; CHECK: %121 = OpFunctionCall %void %inst_bindless_stream_write_4 %uint_64 %uint_0 %40 %uint_128 |
| ; CHECK: OpBranch %114 |
| ; CHECK: %114 = OpLabel |
| ; CHECK: %122 = OpPhi %v4float %119 %115 %111 %116 |
| ; CHECK: %45 = OpFAdd %v4float %112 %122 |
| OpStore %_entryPointOutput_vColor %45 |
| OpReturn |
| OpFunctionEnd |
| )"; |
| |
| const std::string output_func = kStreamWrite4Frag; |
| |
| SetAssembleOptions(SPV_TEXT_TO_BINARY_OPTION_PRESERVE_NUMERIC_IDS); |
| SinglePassRunAndMatch<InstBindlessCheckPass>(defs + main_func + output_func, |
| true, 7u, 23u, false, false, |
| false, false, false); |
| } |
| |
| TEST_F(InstBindlessTest, InstrumentOpImage) { |
| // This test verifies that the pass will correctly instrument shader |
| // using OpImage. This test was created by editing the SPIR-V |
| // from the Simple test. |
| |
| // clang-format off |
| const std::string defs = R"( |
| OpCapability Shader |
| OpCapability StorageImageReadWithoutFormat |
| ; CHECK: OpExtension "SPV_KHR_storage_buffer_storage_class" |
| %1 = OpExtInstImport "GLSL.std.450" |
| OpMemoryModel Logical GLSL450 |
| OpEntryPoint Fragment %MainPs "MainPs" %i_vTextureCoords %_entryPointOutput_vColor |
| ; CHECK: OpEntryPoint Fragment %MainPs "MainPs" %i_vTextureCoords %_entryPointOutput_vColor %gl_FragCoord |
| OpExecutionMode %MainPs OriginUpperLeft |
| OpSource HLSL 500 |
| OpName %MainPs "MainPs" |
| OpName %g_tColor "g_tColor" |
| OpName %PerViewConstantBuffer_t "PerViewConstantBuffer_t" |
| OpMemberName %PerViewConstantBuffer_t 0 "g_nDataIdx" |
| OpName %_ "" |
| OpName %i_vTextureCoords "i.vTextureCoords" |
| OpName %_entryPointOutput_vColor "@entryPointOutput.vColor" |
| OpDecorate %g_tColor DescriptorSet 3 |
| OpDecorate %g_tColor Binding 0 |
| OpMemberDecorate %PerViewConstantBuffer_t 0 Offset 0 |
| OpDecorate %PerViewConstantBuffer_t Block |
| OpDecorate %i_vTextureCoords Location 0 |
| OpDecorate %_entryPointOutput_vColor Location 0 |
| ; CHECK: OpDecorate %_runtimearr_uint ArrayStride 4 |
| )" + kOutputDecorations + R"( |
| ; CHECK: OpDecorate %gl_FragCoord BuiltIn FragCoord |
| %void = OpTypeVoid |
| %3 = OpTypeFunction %void |
| %float = OpTypeFloat 32 |
| %v4float = OpTypeVector %float 4 |
| %int = OpTypeInt 32 1 |
| %v2int = OpTypeVector %int 2 |
| %int_0 = OpConstant %int 0 |
| %20 = OpTypeImage %float 2D 0 0 0 0 Unknown |
| %uint = OpTypeInt 32 0 |
| %uint_128 = OpConstant %uint 128 |
| %39 = OpTypeSampledImage %20 |
| %_arr_39_uint_128 = OpTypeArray %39 %uint_128 |
| %_ptr_UniformConstant__arr_39_uint_128 = OpTypePointer UniformConstant %_arr_39_uint_128 |
| %g_tColor = OpVariable %_ptr_UniformConstant__arr_39_uint_128 UniformConstant |
| %PerViewConstantBuffer_t = OpTypeStruct %uint |
| %_ptr_PushConstant_PerViewConstantBuffer_t = OpTypePointer PushConstant %PerViewConstantBuffer_t |
| %_ = OpVariable %_ptr_PushConstant_PerViewConstantBuffer_t PushConstant |
| %_ptr_PushConstant_uint = OpTypePointer PushConstant %uint |
| %_ptr_UniformConstant_39 = OpTypePointer UniformConstant %39 |
| %_ptr_Input_v2int = OpTypePointer Input %v2int |
| %i_vTextureCoords = OpVariable %_ptr_Input_v2int Input |
| %_ptr_Output_v4float = OpTypePointer Output %v4float |
| %_entryPointOutput_vColor = OpVariable %_ptr_Output_v4float Output |
| ; CHECK: uint_0 = OpConstant %uint 0 |
| ; CHECK: bool = OpTypeBool |
| ; CHECK: %86 = OpTypeFunction %void %uint %uint %uint %uint |
| ; CHECK: %_runtimearr_uint = OpTypeRuntimeArray %uint |
| )" + kOutputGlobals + R"( |
| ; CHECK: %_ptr_StorageBuffer_uint = OpTypePointer StorageBuffer %uint |
| ; CHECK: %_ptr_Input_v4float = OpTypePointer Input %v4float |
| ; CHECK: %gl_FragCoord = OpVariable %_ptr_Input_v4float Input |
| ; CHECK: %v4uint = OpTypeVector %uint 4 |
| ; CHECK: %141 = OpConstantNull %v4float |
| )"; |
| // clang-format on |
| |
| const std::string main_func = R"( |
| %MainPs = OpFunction %void None %3 |
| %5 = OpLabel |
| %53 = OpLoad %v2int %i_vTextureCoords |
| %63 = OpAccessChain %_ptr_PushConstant_uint %_ %int_0 |
| %64 = OpLoad %uint %63 |
| %65 = OpAccessChain %_ptr_UniformConstant_39 %g_tColor %64 |
| %66 = OpLoad %39 %65 |
| %75 = OpImage %20 %66 |
| %71 = OpImageRead %v4float %75 %53 |
| OpStore %_entryPointOutput_vColor %71 |
| ; CHECK-NOT: %71 = OpImageRead %v4float %75 %53 |
| ; CHECK-NOT: OpStore %_entryPointOutput_vColor %71 |
| ; CHECK: %78 = OpULessThan %bool %64 %uint_128 |
| ; CHECK: OpSelectionMerge %79 None |
| ; CHECK: OpBranchConditional %78 %80 %81 |
| ; CHECK: %80 = OpLabel |
| ; CHECK: %82 = OpLoad %39 %65 |
| ; CHECK: %83 = OpImage %20 %82 |
| ; CHECK: %84 = OpImageRead %v4float %83 %53 |
| ; CHECK: OpBranch %79 |
| ; CHECK: %81 = OpLabel |
| ; CHECK: %140 = OpFunctionCall %void %inst_bindless_stream_write_4 %uint_51 %uint_0 %64 %uint_128 |
| ; CHECK: OpBranch %79 |
| ; CHECK: %79 = OpLabel |
| ; CHECK: %142 = OpPhi %v4float %84 %80 %141 %81 |
| ; CHECK: OpStore %_entryPointOutput_vColor %142 |
| OpReturn |
| OpFunctionEnd |
| )"; |
| |
| const std::string output_func = kStreamWrite4Frag; |
| |
| SetAssembleOptions(SPV_TEXT_TO_BINARY_OPTION_PRESERVE_NUMERIC_IDS); |
| SinglePassRunAndMatch<InstBindlessCheckPass>(defs + main_func + output_func, |
| true, 7u, 23u, false, false, |
| false, false, false); |
| } |
| |
| TEST_F(InstBindlessTest, InstrumentSampledImage) { |
| // This test verifies that the pass will correctly instrument shader |
| // using sampled image. This test was created by editing the SPIR-V |
| // from the Simple test. |
| |
| // clang-format off |
| const std::string defs = R"( |
| OpCapability Shader |
| ; CHECK: OpExtension "SPV_KHR_storage_buffer_storage_class" |
| %1 = OpExtInstImport "GLSL.std.450" |
| OpMemoryModel Logical GLSL450 |
| OpEntryPoint Fragment %MainPs "MainPs" %i_vTextureCoords %_entryPointOutput_vColor |
| ; CHECK: OpEntryPoint Fragment %MainPs "MainPs" %i_vTextureCoords %_entryPointOutput_vColor %gl_FragCoord |
| OpExecutionMode %MainPs OriginUpperLeft |
| OpSource HLSL 500 |
| OpName %MainPs "MainPs" |
| OpName %g_tColor "g_tColor" |
| OpName %PerViewConstantBuffer_t "PerViewConstantBuffer_t" |
| OpMemberName %PerViewConstantBuffer_t 0 "g_nDataIdx" |
| OpName %_ "" |
| OpName %i_vTextureCoords "i.vTextureCoords" |
| OpName %_entryPointOutput_vColor "@entryPointOutput.vColor" |
| OpDecorate %g_tColor DescriptorSet 3 |
| OpDecorate %g_tColor Binding 0 |
| OpMemberDecorate %PerViewConstantBuffer_t 0 Offset 0 |
| OpDecorate %PerViewConstantBuffer_t Block |
| OpDecorate %i_vTextureCoords Location 0 |
| OpDecorate %_entryPointOutput_vColor Location 0 |
| )" + kOutputDecorations + R"( |
| ; CHECK: OpDecorate %gl_FragCoord BuiltIn FragCoord |
| %void = OpTypeVoid |
| %3 = OpTypeFunction %void |
| %float = OpTypeFloat 32 |
| %v2float = OpTypeVector %float 2 |
| %v4float = OpTypeVector %float 4 |
| %int = OpTypeInt 32 1 |
| %int_0 = OpConstant %int 0 |
| %20 = OpTypeImage %float 2D 0 0 0 1 Unknown |
| %uint = OpTypeInt 32 0 |
| %uint_128 = OpConstant %uint 128 |
| %39 = OpTypeSampledImage %20 |
| %_arr_39_uint_128 = OpTypeArray %39 %uint_128 |
| %_ptr_UniformConstant__arr_39_uint_128 = OpTypePointer UniformConstant %_arr_39_uint_128 |
| %g_tColor = OpVariable %_ptr_UniformConstant__arr_39_uint_128 UniformConstant |
| %PerViewConstantBuffer_t = OpTypeStruct %uint |
| %_ptr_PushConstant_PerViewConstantBuffer_t = OpTypePointer PushConstant %PerViewConstantBuffer_t |
| %_ = OpVariable %_ptr_PushConstant_PerViewConstantBuffer_t PushConstant |
| %_ptr_PushConstant_uint = OpTypePointer PushConstant %uint |
| %_ptr_UniformConstant_39 = OpTypePointer UniformConstant %39 |
| %_ptr_Input_v2float = OpTypePointer Input %v2float |
| %i_vTextureCoords = OpVariable %_ptr_Input_v2float Input |
| %_ptr_Output_v4float = OpTypePointer Output %v4float |
| %_entryPointOutput_vColor = OpVariable %_ptr_Output_v4float Output |
| ; CHECK: uint_0 = OpConstant %uint 0 |
| ; CHECK: bool = OpTypeBool |
| ; CHECK: %81 = OpTypeFunction %void %uint %uint %uint %uint |
| ; CHECK: %_runtimearr_uint = OpTypeRuntimeArray %uint |
| )" + kOutputGlobals + R"( |
| ; CHECK: %_ptr_StorageBuffer_uint = OpTypePointer StorageBuffer %uint |
| ; CHECK: %_ptr_Input_v4float = OpTypePointer Input %v4float |
| ; CHECK: %gl_FragCoord = OpVariable %_ptr_Input_v4float Input |
| ; CHECK: %v4uint = OpTypeVector %uint 4 |
| ; CHECK: %136 = OpConstantNull %v4float |
| )"; |
| // clang-format on |
| |
| const std::string main_func = R"( |
| %MainPs = OpFunction %void None %3 |
| %5 = OpLabel |
| %53 = OpLoad %v2float %i_vTextureCoords |
| %63 = OpAccessChain %_ptr_PushConstant_uint %_ %int_0 |
| %64 = OpLoad %uint %63 |
| %65 = OpAccessChain %_ptr_UniformConstant_39 %g_tColor %64 |
| %66 = OpLoad %39 %65 |
| %71 = OpImageSampleImplicitLod %v4float %66 %53 |
| OpStore %_entryPointOutput_vColor %71 |
| ; CHECK-NOT: %71 = OpImageSampleImplicitLod %v4float %66 %53 |
| ; CHECK-NOT: OpStore %_entryPointOutput_vColor %71 |
| ; CHECK: %74 = OpULessThan %bool %64 %uint_128 |
| ; CHECK: OpSelectionMerge %75 None |
| ; CHECK: OpBranchConditional %74 %76 %77 |
| ; CHECK: %76 = OpLabel |
| ; CHECK: %78 = OpLoad %39 %65 |
| ; CHECK: %79 = OpImageSampleImplicitLod %v4float %78 %53 |
| ; CHECK: OpBranch %75 |
| ; CHECK: %77 = OpLabel |
| ; CHECK: %135 = OpFunctionCall %void %inst_bindless_stream_write_4 %uint_49 %uint_0 %64 %uint_128 |
| ; CHECK: OpBranch %75 |
| ; CHECK: %75 = OpLabel |
| ; CHECK: %137 = OpPhi %v4float %79 %76 %136 %77 |
| ; CHECK: OpStore %_entryPointOutput_vColor %137 |
| OpReturn |
| OpFunctionEnd |
| )"; |
| |
| const std::string output_func = kStreamWrite4Frag; |
| |
| SetAssembleOptions(SPV_TEXT_TO_BINARY_OPTION_PRESERVE_NUMERIC_IDS); |
| SinglePassRunAndMatch<InstBindlessCheckPass>(defs + main_func + output_func, |
| true, 7u, 23u, false, false, |
| false, false, false); |
| } |
| |
| TEST_F(InstBindlessTest, InstrumentImageWrite) { |
| // This test verifies that the pass will correctly instrument shader |
| // doing bindless image write. This test was created by editing the SPIR-V |
| // from the Simple test. |
| |
| // clang-format off |
| const std::string defs = R"( |
| OpCapability Shader |
| OpCapability StorageImageWriteWithoutFormat |
| ; CHECK: OpExtension "SPV_KHR_storage_buffer_storage_class" |
| %1 = OpExtInstImport "GLSL.std.450" |
| OpMemoryModel Logical GLSL450 |
| OpEntryPoint Fragment %MainPs "MainPs" %i_vTextureCoords %_entryPointOutput_vColor |
| ; CHECK: OpEntryPoint Fragment %MainPs "MainPs" %i_vTextureCoords %_entryPointOutput_vColor %gl_FragCoord |
| OpExecutionMode %MainPs OriginUpperLeft |
| OpSource HLSL 500 |
| OpName %MainPs "MainPs" |
| OpName %g_tColor "g_tColor" |
| OpName %PerViewConstantBuffer_t "PerViewConstantBuffer_t" |
| OpMemberName %PerViewConstantBuffer_t 0 "g_nDataIdx" |
| OpName %_ "" |
| OpName %i_vTextureCoords "i.vTextureCoords" |
| OpName %_entryPointOutput_vColor "@entryPointOutput.vColor" |
| OpDecorate %g_tColor DescriptorSet 3 |
| OpDecorate %g_tColor Binding 0 |
| OpMemberDecorate %PerViewConstantBuffer_t 0 Offset 0 |
| OpDecorate %PerViewConstantBuffer_t Block |
| OpDecorate %i_vTextureCoords Location 0 |
| OpDecorate %_entryPointOutput_vColor Location 0 |
| ; CHECK: OpDecorate %_runtimearr_uint ArrayStride 4 |
| )" + kOutputDecorations + R"( |
| ; CHECK: OpDecorate %gl_FragCoord BuiltIn FragCoord |
| %void = OpTypeVoid |
| %3 = OpTypeFunction %void |
| %float = OpTypeFloat 32 |
| %v2float = OpTypeVector %float 2 |
| %v4float = OpTypeVector %float 4 |
| %int = OpTypeInt 32 1 |
| %v2int = OpTypeVector %int 2 |
| %int_0 = OpConstant %int 0 |
| %20 = OpTypeImage %float 2D 0 0 0 0 Unknown |
| %uint = OpTypeInt 32 0 |
| %uint_128 = OpConstant %uint 128 |
| %80 = OpConstantNull %v4float |
| %_arr_20_uint_128 = OpTypeArray %20 %uint_128 |
| %_ptr_UniformConstant__arr_20_uint_128 = OpTypePointer UniformConstant %_arr_20_uint_128 |
| %g_tColor = OpVariable %_ptr_UniformConstant__arr_20_uint_128 UniformConstant |
| %PerViewConstantBuffer_t = OpTypeStruct %uint |
| %_ptr_PushConstant_PerViewConstantBuffer_t = OpTypePointer PushConstant %PerViewConstantBuffer_t |
| %_ = OpVariable %_ptr_PushConstant_PerViewConstantBuffer_t PushConstant |
| %_ptr_PushConstant_uint = OpTypePointer PushConstant %uint |
| %_ptr_UniformConstant_20 = OpTypePointer UniformConstant %20 |
| %_ptr_Input_v2int = OpTypePointer Input %v2int |
| %i_vTextureCoords = OpVariable %_ptr_Input_v2int Input |
| %_ptr_Output_v4float = OpTypePointer Output %v4float |
| %_entryPointOutput_vColor = OpVariable %_ptr_Output_v4float Output |
| ; CHECK: uint_0 = OpConstant %uint 0 |
| ; CHECK: bool = OpTypeBool |
| ; CHECK: %41 = OpTypeFunction %void %uint %uint %uint %uint |
| ; CHECK: _runtimearr_uint = OpTypeRuntimeArray %uint |
| )" + kOutputGlobals + R"( |
| ; CHECK: %_ptr_StorageBuffer_uint = OpTypePointer StorageBuffer %uint |
| ; CHECK: %_ptr_Input_v4float = OpTypePointer Input %v4float |
| ; CHECK: %gl_FragCoord = OpVariable %_ptr_Input_v4float Input |
| ; CHECK: %v4uint = OpTypeVector %uint 4 |
| )"; |
| // clang-format on |
| |
| const std::string main_func = R"( |
| %MainPs = OpFunction %void None %3 |
| %5 = OpLabel |
| %53 = OpLoad %v2int %i_vTextureCoords |
| %63 = OpAccessChain %_ptr_PushConstant_uint %_ %int_0 |
| %64 = OpLoad %uint %63 |
| %65 = OpAccessChain %_ptr_UniformConstant_20 %g_tColor %64 |
| %66 = OpLoad %20 %65 |
| OpImageWrite %66 %53 %80 |
| OpStore %_entryPointOutput_vColor %80 |
| ; CHECK-NOT: OpImageWrite %66 %53 %80 |
| ; CHECK-NOT: OpStore %_entryPointOutput_vColor %80 |
| ; CHECK: %35 = OpULessThan %bool %30 %uint_128 |
| ; CHECK: OpSelectionMerge %36 None |
| ; CHECK: OpBranchConditional %35 %37 %38 |
| ; CHECK: %37 = OpLabel |
| ; CHECK: %39 = OpLoad %16 %31 |
| ; CHECK: OpImageWrite %39 %28 %19 |
| ; CHECK: OpBranch %36 |
| ; CHECK: %38 = OpLabel |
| ; CHECK: %95 = OpFunctionCall %void %inst_bindless_stream_write_4 %uint_51 %uint_0 %30 %uint_128 |
| ; CHECK: OpBranch %36 |
| ; CHECK: %36 = OpLabel |
| ; CHECK: OpStore %_entryPointOutput_vColor %19 |
| OpReturn |
| OpFunctionEnd |
| )"; |
| |
| const std::string output_func = kStreamWrite4Frag; |
| |
| // SetAssembleOptions(SPV_TEXT_TO_BINARY_OPTION_PRESERVE_NUMERIC_IDS); |
| SinglePassRunAndMatch<InstBindlessCheckPass>(defs + main_func + output_func, |
| true, 7u, 23u, false, false, |
| false, false, false); |
| } |
| |
| TEST_F(InstBindlessTest, InstrumentVertexSimple) { |
| // This test verifies that the pass will correctly instrument shader |
| // doing bindless image write. This test was created by editing the SPIR-V |
| // from the Simple test. |
| |
| // clang-format off |
| const std::string defs = R"( |
| OpCapability Shader |
| OpCapability Sampled1D |
| ; CHECK: OpExtension "SPV_KHR_storage_buffer_storage_class" |
| %1 = OpExtInstImport "GLSL.std.450" |
| OpMemoryModel Logical GLSL450 |
| OpEntryPoint Vertex %main "main" %_ %coords2D |
| OpSource GLSL 450 |
| OpName %main "main" |
| OpName %lod "lod" |
| OpName %coords1D "coords1D" |
| OpName %gl_PerVertex "gl_PerVertex" |
| OpMemberName %gl_PerVertex 0 "gl_Position" |
| OpMemberName %gl_PerVertex 1 "gl_PointSize" |
| OpMemberName %gl_PerVertex 2 "gl_ClipDistance" |
| OpMemberName %gl_PerVertex 3 "gl_CullDistance" |
| OpName %_ "" |
| OpName %texSampler1D "texSampler1D" |
| OpName %foo "foo" |
| OpMemberName %foo 0 "g_idx" |
| OpName %__0 "" |
| OpName %coords2D "coords2D" |
| ; CHECK: OpDecorate %_runtimearr_uint ArrayStride 4 |
| )" + kOutputDecorations + R"( |
| ; CHECK: OpDecorate %gl_VertexIndex BuiltIn VertexIndex |
| ; CHECK: OpDecorate %gl_InstanceIndex BuiltIn InstanceIndex |
| OpMemberDecorate %gl_PerVertex 0 BuiltIn Position |
| OpMemberDecorate %gl_PerVertex 1 BuiltIn PointSize |
| OpMemberDecorate %gl_PerVertex 2 BuiltIn ClipDistance |
| OpMemberDecorate %gl_PerVertex 3 BuiltIn CullDistance |
| OpDecorate %gl_PerVertex Block |
| OpDecorate %texSampler1D DescriptorSet 0 |
| OpDecorate %texSampler1D Binding 3 |
| OpMemberDecorate %foo 0 Offset 0 |
| OpDecorate %foo Block |
| OpDecorate %__0 DescriptorSet 0 |
| OpDecorate %__0 Binding 5 |
| OpDecorate %coords2D Location 0 |
| %void = OpTypeVoid |
| %3 = OpTypeFunction %void |
| %float = OpTypeFloat 32 |
| %_ptr_Function_float = OpTypePointer Function %float |
| %float_3 = OpConstant %float 3 |
| %float_1_78900003 = OpConstant %float 1.78900003 |
| %v4float = OpTypeVector %float 4 |
| %uint = OpTypeInt 32 0 |
| %uint_1 = OpConstant %uint 1 |
| %_arr_float_uint_1 = OpTypeArray %float %uint_1 |
| %gl_PerVertex = OpTypeStruct %v4float %float %_arr_float_uint_1 %_arr_float_uint_1 |
| %_ptr_Output_gl_PerVertex = OpTypePointer Output %gl_PerVertex |
| %_ = OpVariable %_ptr_Output_gl_PerVertex Output |
| %int = OpTypeInt 32 1 |
| %int_0 = OpConstant %int 0 |
| %21 = OpTypeImage %float 1D 0 0 0 1 Unknown |
| %22 = OpTypeSampledImage %21 |
| %uint_128 = OpConstant %uint 128 |
| %_arr_22_uint_128 = OpTypeArray %22 %uint_128 |
| %_ptr_UniformConstant__arr_22_uint_128 = OpTypePointer UniformConstant %_arr_22_uint_128 |
| %texSampler1D = OpVariable %_ptr_UniformConstant__arr_22_uint_128 UniformConstant |
| %foo = OpTypeStruct %int |
| %_ptr_Uniform_foo = OpTypePointer Uniform %foo |
| %__0 = OpVariable %_ptr_Uniform_foo Uniform |
| %_ptr_Uniform_int = OpTypePointer Uniform %int |
| %_ptr_UniformConstant_22 = OpTypePointer UniformConstant %22 |
| %_ptr_Output_v4float = OpTypePointer Output %v4float |
| %v2float = OpTypeVector %float 2 |
| %_ptr_Input_v2float = OpTypePointer Input %v2float |
| %coords2D = OpVariable %_ptr_Input_v2float Input |
| ; CHECK: %bool = OpTypeBool |
| ; CHECK: %54 = OpTypeFunction %void %uint %uint %uint %uint |
| ; CHECK: %_runtimearr_uint = OpTypeRuntimeArray %uint |
| )" + kOutputGlobals + R"( |
| ; CHECK: %_ptr_StorageBuffer_uint = OpTypePointer StorageBuffer %uint |
| ; CHECK: %_ptr_Input_uint = OpTypePointer Input %uint |
| ; CHECK: %gl_VertexIndex = OpVariable %_ptr_Input_uint Input |
| ; CHECK: %gl_InstanceIndex = OpVariable %_ptr_Input_uint Input |
| ; CHECK: %106 = OpConstantNull %v4float |
| )"; |
| // clang-format on |
| |
| const std::string main_func = R"( |
| %main = OpFunction %void None %3 |
| %5 = OpLabel |
| %lod = OpVariable %_ptr_Function_float Function |
| %coords1D = OpVariable %_ptr_Function_float Function |
| OpStore %lod %float_3 |
| OpStore %coords1D %float_1_78900003 |
| %31 = OpAccessChain %_ptr_Uniform_int %__0 %int_0 |
| %32 = OpLoad %int %31 |
| %34 = OpAccessChain %_ptr_UniformConstant_22 %texSampler1D %32 |
| %35 = OpLoad %22 %34 |
| %36 = OpLoad %float %coords1D |
| %37 = OpLoad %float %lod |
| %38 = OpImageSampleExplicitLod %v4float %35 %36 Lod %37 |
| %40 = OpAccessChain %_ptr_Output_v4float %_ %int_0 |
| OpStore %40 %38 |
| ; CHECK-NOT: %38 = OpImageSampleExplicitLod %v4float %35 %36 Lod %37 |
| ; CHECK-NOT: %40 = OpAccessChain %_ptr_Output_v4float %_ %int_0 |
| ; CHECK-NOT: OpStore %40 %38 |
| ; CHECK: %46 = OpULessThan %bool %37 %uint_128 |
| ; CHECK: OpSelectionMerge %47 None |
| ; CHECK: OpBranchConditional %46 %48 %49 |
| ; CHECK: %48 = OpLabel |
| ; CHECK: %50 = OpLoad %25 %38 |
| ; CHECK: %51 = OpImageSampleExplicitLod %v4float %50 %40 Lod %41 |
| ; CHECK: OpBranch %47 |
| ; CHECK: %49 = OpLabel |
| ; CHECK: %52 = OpBitcast %uint %37 |
| ; CHECK: %105 = OpFunctionCall %void %inst_bindless_stream_write_4 %uint_74 %uint_0 %52 %uint_128 |
| ; CHECK: OpBranch %47 |
| ; CHECK: %47 = OpLabel |
| ; CHECK: %107 = OpPhi %v4float %51 %48 %106 %49 |
| ; CHECK: %43 = OpAccessChain %_ptr_Output_v4float %_ %int_0 |
| ; CHECK: OpStore %43 %107 |
| OpReturn |
| OpFunctionEnd |
| )"; |
| |
| const std::string output_func = kStreamWrite4Vert; |
| |
| // SetAssembleOptions(SPV_TEXT_TO_BINARY_OPTION_PRESERVE_NUMERIC_IDS); |
| SinglePassRunAndMatch<InstBindlessCheckPass>(defs + main_func + output_func, |
| true, 7u, 23u, false, false, |
| false, false, false); |
| } |
| |
| TEST_F(InstBindlessTest, InstrumentTeseSimple) { |
| // This test verifies that the pass will correctly instrument tessellation |
| // evaluation shader doing bindless buffer load. |
| // |
| // clang-format off |
| // |
| // #version 450 |
| // #extension GL_EXT_nonuniform_qualifier : enable |
| // |
| // layout(std140, set = 0, binding = 0) uniform ufoo { uint index; } uniform_index_buffer; |
| // |
| // layout(set = 0, binding = 1) buffer bfoo { vec4 val; } adds[11]; |
| // |
| // layout(triangles, equal_spacing, cw) in; |
| // |
| // void main() { |
| // gl_Position = adds[uniform_index_buffer.index].val; |
| // } |
| // |
| |
| const std::string defs = R"( |
| OpCapability Tessellation |
| %1 = OpExtInstImport "GLSL.std.450" |
| OpMemoryModel Logical GLSL450 |
| OpEntryPoint TessellationEvaluation %main "main" %_ |
| ; CHECK: OpEntryPoint TessellationEvaluation %main "main" %_ %gl_PrimitiveID %gl_TessCoord |
| OpExecutionMode %main Triangles |
| OpExecutionMode %main SpacingEqual |
| OpExecutionMode %main VertexOrderCw |
| OpSource GLSL 450 |
| OpSourceExtension "GL_EXT_nonuniform_qualifier" |
| OpName %main "main" |
| OpName %gl_PerVertex "gl_PerVertex" |
| OpMemberName %gl_PerVertex 0 "gl_Position" |
| OpMemberName %gl_PerVertex 1 "gl_PointSize" |
| OpMemberName %gl_PerVertex 2 "gl_ClipDistance" |
| OpMemberName %gl_PerVertex 3 "gl_CullDistance" |
| OpName %_ "" |
| OpName %bfoo "bfoo" |
| OpMemberName %bfoo 0 "val" |
| OpName %adds "adds" |
| OpName %ufoo "ufoo" |
| OpMemberName %ufoo 0 "index" |
| OpName %uniform_index_buffer "uniform_index_buffer" |
| OpMemberDecorate %gl_PerVertex 0 BuiltIn Position |
| OpMemberDecorate %gl_PerVertex 1 BuiltIn PointSize |
| OpMemberDecorate %gl_PerVertex 2 BuiltIn ClipDistance |
| OpMemberDecorate %gl_PerVertex 3 BuiltIn CullDistance |
| OpDecorate %gl_PerVertex Block |
| OpMemberDecorate %bfoo 0 Offset 0 |
| OpDecorate %bfoo Block |
| OpDecorate %adds DescriptorSet 0 |
| OpDecorate %adds Binding 1 |
| OpMemberDecorate %ufoo 0 Offset 0 |
| OpDecorate %ufoo Block |
| OpDecorate %uniform_index_buffer DescriptorSet 0 |
| OpDecorate %uniform_index_buffer Binding 0 |
| ; CHECK: OpDecorate %_runtimearr_uint ArrayStride 4 |
| )" + kOutputDecorations + R"( |
| ; CHECK: OpDecorate %gl_PrimitiveID BuiltIn PrimitiveId |
| ; CHECK: OpDecorate %gl_TessCoord BuiltIn TessCoord |
| %void = OpTypeVoid |
| %3 = OpTypeFunction %void |
| %float = OpTypeFloat 32 |
| %v4float = OpTypeVector %float 4 |
| %uint = OpTypeInt 32 0 |
| %uint_1 = OpConstant %uint 1 |
| %_arr_float_uint_1 = OpTypeArray %float %uint_1 |
| %gl_PerVertex = OpTypeStruct %v4float %float %_arr_float_uint_1 %_arr_float_uint_1 |
| %_ptr_Output_gl_PerVertex = OpTypePointer Output %gl_PerVertex |
| %_ = OpVariable %_ptr_Output_gl_PerVertex Output |
| %int = OpTypeInt 32 1 |
| %int_0 = OpConstant %int 0 |
| %bfoo = OpTypeStruct %v4float |
| %uint_11 = OpConstant %uint 11 |
| %_arr_bfoo_uint_11 = OpTypeArray %bfoo %uint_11 |
| %_ptr_StorageBuffer__arr_bfoo_uint_11 = OpTypePointer StorageBuffer %_arr_bfoo_uint_11 |
| %adds = OpVariable %_ptr_StorageBuffer__arr_bfoo_uint_11 StorageBuffer |
| %ufoo = OpTypeStruct %uint |
| %_ptr_Uniform_ufoo = OpTypePointer Uniform %ufoo |
| %uniform_index_buffer = OpVariable %_ptr_Uniform_ufoo Uniform |
| %_ptr_Uniform_uint = OpTypePointer Uniform %uint |
| %_ptr_StorageBuffer_v4float = OpTypePointer StorageBuffer %v4float |
| %_ptr_Output_v4float = OpTypePointer Output %v4float |
| ; CHECK: %bool = OpTypeBool |
| ; CHECK: %40 = OpTypeFunction %void %uint %uint %uint %uint |
| ; CHECK: %_runtimearr_uint = OpTypeRuntimeArray %uint |
| )" + kOutputGlobals + R"( |
| ; CHECK: %_ptr_StorageBuffer_uint = OpTypePointer StorageBuffer %uint |
| ; CHECK: %_ptr_Input_uint = OpTypePointer Input %uint |
| ; CHECK: %gl_PrimitiveID = OpVariable %_ptr_Input_uint Input |
| ; CHECK: %v3float = OpTypeVector %float 3 |
| ; CHECK: %_ptr_Input_v3float = OpTypePointer Input %v3float |
| ; CHECK: %gl_TessCoord = OpVariable %_ptr_Input_v3float Input |
| ; CHECK: %v3uint = OpTypeVector %uint 3 |
| ; CHECK: %101 = OpConstantNull %v4float |
| )"; |
| // clang-format on |
| |
| const std::string main_func = R"( |
| %main = OpFunction %void None %3 |
| %5 = OpLabel |
| %25 = OpAccessChain %_ptr_Uniform_uint %uniform_index_buffer %int_0 |
| %26 = OpLoad %uint %25 |
| %28 = OpAccessChain %_ptr_StorageBuffer_v4float %adds %26 %int_0 |
| %29 = OpLoad %v4float %28 |
| ; CHECK-NOT: %29 = OpLoad %v4float %28 |
| ; CHECK: %34 = OpULessThan %bool %28 %uint_11 |
| ; CHECK: OpSelectionMerge %35 None |
| ; CHECK: OpBranchConditional %34 %36 %37 |
| ; CHECK: %36 = OpLabel |
| ; CHECK: %38 = OpLoad %v4float %29 |
| ; CHECK: OpBranch %35 |
| ; CHECK: %37 = OpLabel |
| ; CHECK: %100 = OpFunctionCall %void %inst_bindless_stream_write_4 %uint_63 %uint_0 %28 %uint_11 |
| ; CHECK: OpBranch %35 |
| ; CHECK: %35 = OpLabel |
| ; CHECK: %102 = OpPhi %v4float %38 %36 %101 %37 |
| %31 = OpAccessChain %_ptr_Output_v4float %_ %int_0 |
| OpStore %31 %29 |
| ; CHECK-NOT: OpStore %31 %29 |
| ; CHECK: OpStore %31 %102 |
| OpReturn |
| OpFunctionEnd |
| )"; |
| |
| const std::string output_func = kStreamWrite4Tese; |
| |
| // SetAssembleOptions(SPV_TEXT_TO_BINARY_OPTION_PRESERVE_NUMERIC_IDS); |
| SinglePassRunAndMatch<InstBindlessCheckPass>(defs + main_func + output_func, |
| true, 7u, 23u, false, false, |
| false, false, false); |
| } |
| |
| TEST_F(InstBindlessTest, MultipleDebugFunctions) { |
| // Same source as Simple, but compiled -g and not optimized, especially not |
| // inlined. The OpSource has had the source extracted for the sake of brevity. |
| |
| // clang-format off |
| const std::string defs = R"( |
| OpCapability Shader |
| ; CHECK: OpExtension "SPV_KHR_storage_buffer_storage_class" |
| %2 = OpExtInstImport "GLSL.std.450" |
| OpMemoryModel Logical GLSL450 |
| OpEntryPoint Fragment %MainPs "MainPs" %i_vTextureCoords %_entryPointOutput_vColor |
| ; CHECK: OpEntryPoint Fragment %MainPs "MainPs" %i_vTextureCoords %_entryPointOutput_vColor %gl_FragCoord |
| OpExecutionMode %MainPs OriginUpperLeft |
| %1 = OpString "foo5.frag" |
| OpSource HLSL 500 %1 |
| OpName %MainPs "MainPs" |
| OpName %PS_INPUT "PS_INPUT" |
| OpMemberName %PS_INPUT 0 "vTextureCoords" |
| OpName %PS_OUTPUT "PS_OUTPUT" |
| OpMemberName %PS_OUTPUT 0 "vColor" |
| OpName %_MainPs_struct_PS_INPUT_vf21_ "@MainPs(struct-PS_INPUT-vf21;" |
| OpName %i "i" |
| OpName %ps_output "ps_output" |
| OpName %g_tColor "g_tColor" |
| OpName %PerViewConstantBuffer_t "PerViewConstantBuffer_t" |
| OpMemberName %PerViewConstantBuffer_t 0 "g_nDataIdx" |
| OpName %_ "" |
| OpName %g_sAniso "g_sAniso" |
| OpName %i_0 "i" |
| OpName %i_vTextureCoords "i.vTextureCoords" |
| OpName %_entryPointOutput_vColor "@entryPointOutput.vColor" |
| OpName %param "param" |
| OpDecorate %g_tColor DescriptorSet 0 |
| OpDecorate %g_tColor Binding 0 |
| OpMemberDecorate %PerViewConstantBuffer_t 0 Offset 0 |
| OpDecorate %PerViewConstantBuffer_t Block |
| OpDecorate %g_sAniso DescriptorSet 0 |
| OpDecorate %g_sAniso Binding 1 |
| OpDecorate %i_vTextureCoords Location 0 |
| OpDecorate %_entryPointOutput_vColor Location 0 |
| ; CHECK: OpDecorate %_runtimearr_uint ArrayStride 4 |
| )" + kOutputDecorations + R"( |
| ; CHECK: OpDecorate %gl_FragCoord BuiltIn FragCoord |
| %void = OpTypeVoid |
| %4 = OpTypeFunction %void |
| %float = OpTypeFloat 32 |
| %v2float = OpTypeVector %float 2 |
| %PS_INPUT = OpTypeStruct %v2float |
| %_ptr_Function_PS_INPUT = OpTypePointer Function %PS_INPUT |
| %v4float = OpTypeVector %float 4 |
| %PS_OUTPUT = OpTypeStruct %v4float |
| %13 = OpTypeFunction %PS_OUTPUT %_ptr_Function_PS_INPUT |
| %_ptr_Function_PS_OUTPUT = OpTypePointer Function %PS_OUTPUT |
| %int = OpTypeInt 32 1 |
| %int_0 = OpConstant %int 0 |
| %21 = OpTypeImage %float 2D 0 0 0 1 Unknown |
| %uint = OpTypeInt 32 0 |
| %uint_128 = OpConstant %uint 128 |
| %_arr_21_uint_128 = OpTypeArray %21 %uint_128 |
| %_ptr_UniformConstant__arr_21_uint_128 = OpTypePointer UniformConstant %_arr_21_uint_128 |
| %g_tColor = OpVariable %_ptr_UniformConstant__arr_21_uint_128 UniformConstant |
| %PerViewConstantBuffer_t = OpTypeStruct %uint |
| %_ptr_PushConstant_PerViewConstantBuffer_t = OpTypePointer PushConstant %PerViewConstantBuffer_t |
| %_ = OpVariable %_ptr_PushConstant_PerViewConstantBuffer_t PushConstant |
| %_ptr_PushConstant_uint = OpTypePointer PushConstant %uint |
| %_ptr_UniformConstant_21 = OpTypePointer UniformConstant %21 |
| %36 = OpTypeSampler |
| %_ptr_UniformConstant_36 = OpTypePointer UniformConstant %36 |
| %g_sAniso = OpVariable %_ptr_UniformConstant_36 UniformConstant |
| %40 = OpTypeSampledImage %21 |
| %_ptr_Function_v2float = OpTypePointer Function %v2float |
| %_ptr_Function_v4float = OpTypePointer Function %v4float |
| %_ptr_Input_v2float = OpTypePointer Input %v2float |
| %i_vTextureCoords = OpVariable %_ptr_Input_v2float Input |
| %_ptr_Output_v4float = OpTypePointer Output %v4float |
| %_entryPointOutput_vColor = OpVariable %_ptr_Output_v4float Output |
| ; CHECK: %bool = OpTypeBool |
| ; CHECK: %70 = OpTypeFunction %void %uint %uint %uint %uint |
| ; CHECK: %_runtimearr_uint = OpTypeRuntimeArray %uint |
| )" + kOutputGlobals + R"( |
| ; CHECK: %_ptr_StorageBuffer_uint = OpTypePointer StorageBuffer %uint |
| ; CHECK: %_ptr_Input_v4float = OpTypePointer Input %v4float |
| ; CHECK: %gl_FragCoord = OpVariable %_ptr_Input_v4float Input |
| ; CHECK: %v4uint = OpTypeVector %uint 4 |
| ; CHECK: %125 = OpConstantNull %v4float |
| )"; |
| // clang-format on |
| |
| const std::string func1 = R"( |
| %MainPs = OpFunction %void None %4 |
| %6 = OpLabel |
| %i_0 = OpVariable %_ptr_Function_PS_INPUT Function |
| %param = OpVariable %_ptr_Function_PS_INPUT Function |
| OpLine %1 21 0 |
| %54 = OpLoad %v2float %i_vTextureCoords |
| %55 = OpAccessChain %_ptr_Function_v2float %i_0 %int_0 |
| OpStore %55 %54 |
| %59 = OpLoad %PS_INPUT %i_0 |
| OpStore %param %59 |
| %60 = OpFunctionCall %PS_OUTPUT %_MainPs_struct_PS_INPUT_vf21_ %param |
| %61 = OpCompositeExtract %v4float %60 0 |
| OpStore %_entryPointOutput_vColor %61 |
| OpReturn |
| OpFunctionEnd |
| )"; |
| |
| const std::string func2 = R"( |
| %_MainPs_struct_PS_INPUT_vf21_ = OpFunction %PS_OUTPUT None %13 |
| %i = OpFunctionParameter %_ptr_Function_PS_INPUT |
| %16 = OpLabel |
| %ps_output = OpVariable %_ptr_Function_PS_OUTPUT Function |
| OpLine %1 24 0 |
| %31 = OpAccessChain %_ptr_PushConstant_uint %_ %int_0 |
| %32 = OpLoad %uint %31 |
| %34 = OpAccessChain %_ptr_UniformConstant_21 %g_tColor %32 |
| %35 = OpLoad %21 %34 |
| %39 = OpLoad %36 %g_sAniso |
| %41 = OpSampledImage %40 %35 %39 |
| %43 = OpAccessChain %_ptr_Function_v2float %i %int_0 |
| %44 = OpLoad %v2float %43 |
| %45 = OpImageSampleImplicitLod %v4float %41 %44 |
| ; CHECK-NOT: %45 = OpImageSampleImplicitLod %v4float %41 %44 |
| ; CHECK: OpNoLine |
| ; CHECK: %62 = OpULessThan %bool %50 %uint_128 |
| ; CHECK: OpSelectionMerge %63 None |
| ; CHECK: OpBranchConditional %62 %64 %65 |
| ; CHECK: %64 = OpLabel |
| ; CHECK: %66 = OpLoad %27 %51 |
| ; CHECK: %67 = OpSampledImage %37 %66 %53 |
| ; CHECK: OpLine %5 24 0 |
| ; CHECK: %68 = OpImageSampleImplicitLod %v4float %67 %56 |
| ; CHECK: OpNoLine |
| ; CHECK: OpBranch %63 |
| ; CHECK: %65 = OpLabel |
| ; CHECK: %124 = OpFunctionCall %void %inst_bindless_stream_write_4 %uint_109 %uint_0 %50 %uint_128 |
| ; CHECK: OpBranch %63 |
| ; CHECK: %63 = OpLabel |
| ; CHECK: %126 = OpPhi %v4float %68 %64 %125 %65 |
| ; CHECK: OpLine %5 24 0 |
| %47 = OpAccessChain %_ptr_Function_v4float %ps_output %int_0 |
| OpStore %47 %45 |
| OpLine %1 25 0 |
| %48 = OpLoad %PS_OUTPUT %ps_output |
| OpReturnValue %48 |
| OpFunctionEnd |
| )"; |
| |
| const std::string output_func = kStreamWrite4Frag; |
| |
| // SetAssembleOptions(SPV_TEXT_TO_BINARY_OPTION_PRESERVE_NUMERIC_IDS); |
| SinglePassRunAndMatch<InstBindlessCheckPass>( |
| defs + func1 + func2 + output_func, true, 7u, 23u, false, false, false, |
| false, false); |
| } |
| |
| TEST_F(InstBindlessTest, RuntimeArray) { |
| // This test verifies that the pass will correctly instrument shader |
| // with runtime descriptor array. This test was created by editing the |
| // SPIR-V from the Simple test. |
| |
| // clang-format off |
| const std::string defs = R"( |
| OpCapability Shader |
| OpCapability RuntimeDescriptorArray |
| OpExtension "SPV_EXT_descriptor_indexing" |
| ; CHECK: OpExtension "SPV_KHR_storage_buffer_storage_class" |
| %1 = OpExtInstImport "GLSL.std.450" |
| OpMemoryModel Logical GLSL450 |
| OpEntryPoint Fragment %MainPs "MainPs" %i_vTextureCoords %_entryPointOutput_vColor |
| ; CHECK: OpEntryPoint Fragment %MainPs "MainPs" %i_vTextureCoords %_entryPointOutput_vColor %gl_FragCoord |
| OpExecutionMode %MainPs OriginUpperLeft |
| OpSource HLSL 500 |
| OpName %MainPs "MainPs" |
| OpName %g_tColor "g_tColor" |
| OpName %PerViewConstantBuffer_t "PerViewConstantBuffer_t" |
| OpMemberName %PerViewConstantBuffer_t 0 "g_nDataIdx" |
| OpName %_ "" |
| OpName %g_sAniso "g_sAniso" |
| OpName %i_vTextureCoords "i.vTextureCoords" |
| OpName %_entryPointOutput_vColor "@entryPointOutput.vColor" |
| OpDecorate %g_tColor DescriptorSet 1 |
| OpDecorate %g_tColor Binding 2 |
| OpMemberDecorate %PerViewConstantBuffer_t 0 Offset 0 |
| OpDecorate %PerViewConstantBuffer_t Block |
| OpDecorate %g_sAniso DescriptorSet 1 |
| OpDecorate %g_sAniso Binding 0 |
| OpDecorate %i_vTextureCoords Location 0 |
| OpDecorate %_entryPointOutput_vColor Location 0 |
| ; CHECK: OpDecorate %_runtimearr_uint ArrayStride 4 |
| )" + kInputDecorations + kOutputDecorations + R"( |
| ; CHECK: OpDecorate %gl_FragCoord BuiltIn FragCoord |
| %void = OpTypeVoid |
| %3 = OpTypeFunction %void |
| %float = OpTypeFloat 32 |
| %v2float = OpTypeVector %float 2 |
| %v4float = OpTypeVector %float 4 |
| %int = OpTypeInt 32 1 |
| %int_0 = OpConstant %int 0 |
| %20 = OpTypeImage %float 2D 0 0 0 1 Unknown |
| %uint = OpTypeInt 32 0 |
| %uint_1 = OpConstant %uint 1 |
| %_rarr_20 = OpTypeRuntimeArray %20 |
| %_ptr_UniformConstant__arr_20 = OpTypePointer UniformConstant %_rarr_20 |
| %g_tColor = OpVariable %_ptr_UniformConstant__arr_20 UniformConstant |
| %PerViewConstantBuffer_t = OpTypeStruct %uint |
| %_ptr_PushConstant_PerViewConstantBuffer_t = OpTypePointer PushConstant %PerViewConstantBuffer_t |
| %_ = OpVariable %_ptr_PushConstant_PerViewConstantBuffer_t PushConstant |
| %_ptr_PushConstant_uint = OpTypePointer PushConstant %uint |
| %_ptr_UniformConstant_20 = OpTypePointer UniformConstant %20 |
| %35 = OpTypeSampler |
| %_ptr_UniformConstant_35 = OpTypePointer UniformConstant %35 |
| %g_sAniso = OpVariable %_ptr_UniformConstant_35 UniformConstant |
| %39 = OpTypeSampledImage %20 |
| %_ptr_Input_v2float = OpTypePointer Input %v2float |
| %i_vTextureCoords = OpVariable %_ptr_Input_v2float Input |
| %_ptr_Output_v4float = OpTypePointer Output %v4float |
| %_entryPointOutput_vColor = OpVariable %_ptr_Output_v4float Output |
| ; CHECK: %41 = OpTypeFunction %uint %uint %uint |
| ; CHECK: %_runtimearr_uint = OpTypeRuntimeArray %uint |
| )" + kInputGlobals + R"( |
| ; CHECK: %_ptr_StorageBuffer_uint = OpTypePointer StorageBuffer %uint |
| ; CHECK: %bool = OpTypeBool |
| ; CHECK: %65 = OpTypeFunction %void %uint %uint %uint %uint |
| )" + kOutputGlobals + R"( |
| ; CHECK: %_ptr_Input_v4float = OpTypePointer Input %v4float |
| ; CHECK: %gl_FragCoord = OpVariable %_ptr_Input_v4float Input |
| ; CHECK: %v4uint = OpTypeVector %uint 4 |
| ; CHECK: %116 = OpConstantNull %v4float |
| ; CHECK: %119 = OpTypeFunction %uint %uint %uint %uint %uint |
| )"; |
| // clang-format on |
| |
| const std::string main_func = R"( |
| %MainPs = OpFunction %void None %3 |
| %5 = OpLabel |
| %53 = OpLoad %v2float %i_vTextureCoords |
| %63 = OpAccessChain %_ptr_PushConstant_uint %_ %int_0 |
| %64 = OpLoad %uint %63 |
| %65 = OpAccessChain %_ptr_UniformConstant_20 %g_tColor %64 |
| %66 = OpLoad %20 %65 |
| %67 = OpLoad %35 %g_sAniso |
| %68 = OpSampledImage %39 %66 %67 |
| %71 = OpImageSampleImplicitLod %v4float %68 %53 |
| OpStore %_entryPointOutput_vColor %71 |
| ; CHECK-NOT: %71 = OpImageSampleImplicitLod %v4float %68 %53 |
| ; CHECK-NOT: OpStore %_entryPointOutput_vColor %71 |
| ; CHECK: %55 = OpFunctionCall %uint %inst_bindless_direct_read_2 %uint_2 %uint_2 |
| ; CHECK: %57 = OpULessThan %bool %32 %55 |
| ; CHECK: OpSelectionMerge %58 None |
| ; CHECK: OpBranchConditional %57 %59 %60 |
| ; CHECK: %59 = OpLabel |
| ; CHECK: %61 = OpLoad %16 %33 |
| ; CHECK: %62 = OpSampledImage %26 %61 %35 |
| ; CHECK: %136 = OpFunctionCall %uint %inst_bindless_direct_read_4 %uint_0 %uint_1 %uint_2 %32 |
| ; CHECK: %137 = OpULessThan %bool %uint_0 %136 |
| ; CHECK: OpSelectionMerge %138 None |
| ; CHECK: OpBranchConditional %137 %139 %140 |
| ; CHECK: %139 = OpLabel |
| ; CHECK: %141 = OpLoad %16 %33 |
| ; CHECK: %142 = OpSampledImage %26 %141 %35 |
| ; CHECK: %143 = OpImageSampleImplicitLod %v4float %142 %30 |
| ; CHECK: OpBranch %138 |
| ; CHECK: %140 = OpLabel |
| ; CHECK: %144 = OpFunctionCall %void %inst_bindless_stream_write_4 %uint_59 %uint_1 %32 %uint_0 |
| ; CHECK: OpBranch %138 |
| ; CHECK: %138 = OpLabel |
| ; CHECK: %145 = OpPhi %v4float %143 %139 %116 %140 |
| ; CHECK: OpBranch %58 |
| ; CHECK: %60 = OpLabel |
| ; CHECK: %115 = OpFunctionCall %void %inst_bindless_stream_write_4 %uint_59 %uint_0 %32 %55 |
| ; CHECK: OpBranch %58 |
| ; CHECK: %58 = OpLabel |
| ; CHECK: %117 = OpPhi %v4float %145 %138 %116 %60 |
| ; CHECK: OpStore %_entryPointOutput_vColor %117 |
| OpReturn |
| OpFunctionEnd |
| )"; |
| |
| const std::string new_funcs = kDirectRead2 + kStreamWrite4Frag + kDirectRead4; |
| |
| // SetAssembleOptions(SPV_TEXT_TO_BINARY_OPTION_PRESERVE_NUMERIC_IDS); |
| SinglePassRunAndMatch<InstBindlessCheckPass>(defs + main_func + new_funcs, |
| true, 7u, 23u, true, true, false, |
| false, false); |
| } |
| |
| TEST_F(InstBindlessTest, InstrumentInitCheckOnScalarDescriptor) { |
| // This test verifies that the pass will correctly instrument vanilla |
| // texture sample on a scalar descriptor with an initialization check if the |
| // input_init_enable argument is set to true. This can happen when the |
| // descriptor indexing extension is enabled in the API but the SPIR-V |
| // does not have the extension enabled because it does not contain a |
| // runtime array. This is the same shader as NoInstrumentNonBindless. |
| |
| // clang-format off |
| const std::string defs = R"( |
| OpCapability Shader |
| ; CHECK: OpExtension "SPV_KHR_storage_buffer_storage_class" |
| %1 = OpExtInstImport "GLSL.std.450" |
| OpMemoryModel Logical GLSL450 |
| OpEntryPoint Fragment %MainPs "MainPs" %i_vTextureCoords %_entryPointOutput_vColor |
| ; CHECK: OpEntryPoint Fragment %MainPs "MainPs" %i_vTextureCoords %_entryPointOutput_vColor %gl_FragCoord |
| OpExecutionMode %MainPs OriginUpperLeft |
| OpSource HLSL 500 |
| OpName %MainPs "MainPs" |
| OpName %g_tColor "g_tColor" |
| OpName %g_sAniso "g_sAniso" |
| OpName %i_vTextureCoords "i.vTextureCoords" |
| OpName %_entryPointOutput_vColor "@entryPointOutput.vColor" |
| OpDecorate %g_tColor DescriptorSet 0 |
| OpDecorate %g_tColor Binding 0 |
| OpDecorate %g_sAniso DescriptorSet 0 |
| OpDecorate %g_sAniso Binding 0 |
| OpDecorate %i_vTextureCoords Location 0 |
| OpDecorate %_entryPointOutput_vColor Location 0 |
| ; check: OpDecorate %_runtimearr_uint ArrayStride 4 |
| )" + kInputDecorations + kOutputDecorations + R"( |
| ; check: OpDecorate %gl_FragCoord BuiltIn FragCoord |
| %void = OpTypeVoid |
| %8 = OpTypeFunction %void |
| %float = OpTypeFloat 32 |
| %v2float = OpTypeVector %float 2 |
| %v4float = OpTypeVector %float 4 |
| %12 = OpTypeImage %float 2D 0 0 0 1 Unknown |
| %_ptr_UniformConstant_12 = OpTypePointer UniformConstant %12 |
| %g_tColor = OpVariable %_ptr_UniformConstant_12 UniformConstant |
| %14 = OpTypeSampler |
| %_ptr_UniformConstant_14 = OpTypePointer UniformConstant %14 |
| %g_sAniso = OpVariable %_ptr_UniformConstant_14 UniformConstant |
| %16 = OpTypeSampledImage %12 |
| %_ptr_Input_v2float = OpTypePointer Input %v2float |
| %i_vTextureCoords = OpVariable %_ptr_Input_v2float Input |
| %_ptr_Output_v4float = OpTypePointer Output %v4float |
| %_entryPointOutput_vColor = OpVariable %_ptr_Output_v4float Output |
| ; CHECK: %uint = OpTypeInt 32 0 |
| ; CHECK: %28 = OpTypeFunction %uint %uint %uint %uint %uint |
| ; CHECK: %_runtimearr_uint = OpTypeRuntimeArray %uint |
| )" + kInputGlobals + R"( |
| ; CHECK: %_ptr_StorageBuffer_uint = OpTypePointer StorageBuffer %uint |
| ; CHECK: %bool = OpTypeBool |
| ; CHECK: %61 = OpTypeFunction %void %uint %uint %uint %uint |
| )" + kOutputGlobals + R"( |
| ; CHECK: %_ptr_Input_v4float = OpTypePointer Input %v4float |
| ; CHECK: %gl_FragCoord = OpVariable %_ptr_Input_v4float Input |
| ; CHECK: %v4uint = OpTypeVector %uint 4 |
| ; CHECK: %113 = OpConstantNull %v4float |
| )"; |
| // clang-format on |
| |
| const std::string main_func = R"( |
| %MainPs = OpFunction %void None %8 |
| %19 = OpLabel |
| %20 = OpLoad %v2float %i_vTextureCoords |
| %21 = OpLoad %12 %g_tColor |
| %22 = OpLoad %14 %g_sAniso |
| %23 = OpSampledImage %16 %21 %22 |
| %24 = OpImageSampleImplicitLod %v4float %23 %20 |
| OpStore %_entryPointOutput_vColor %24 |
| ; CHECK-NOT: %24 = OpImageSampleImplicitLod %v4float %23 %20 |
| ; CHECK-NOT: OpStore %_entryPointOutput_vColor %24 |
| ; CHECK: %50 = OpFunctionCall %uint %inst_bindless_direct_read_4 %uint_0 %uint_0 %uint_0 %uint_0 |
| ; CHECK: %52 = OpULessThan %bool %uint_0 %50 |
| ; CHECK: OpSelectionMerge %54 None |
| ; CHECK: OpBranchConditional %52 %55 %56 |
| ; CHECK: %55 = OpLabel |
| ; CHECK: %57 = OpLoad %12 %g_tColor |
| ; CHECK: %58 = OpSampledImage %16 %57 %22 |
| ; CHECK: %59 = OpImageSampleImplicitLod %v4float %58 %20 |
| ; CHECK: OpBranch %54 |
| ; CHECK: %56 = OpLabel |
| ; CHECK: %112 = OpFunctionCall %void %inst_bindless_stream_write_4 %uint_39 %uint_1 %uint_0 %uint_0 |
| ; CHECK: OpBranch %54 |
| ; CHECK: %54 = OpLabel |
| ; CHECK: %114 = OpPhi %v4float %59 %55 %113 %56 |
| ; CHECK: OpStore %_entryPointOutput_vColor %114 |
| OpReturn |
| OpFunctionEnd |
| )"; |
| |
| const std::string new_funcs = kDirectRead4 + kStreamWrite4Frag; |
| |
| // SetAssembleOptions(SPV_TEXT_TO_BINARY_OPTION_PRESERVE_NUMERIC_IDS); |
| SinglePassRunAndMatch<InstBindlessCheckPass>(defs + main_func + new_funcs, |
| true, 7u, 23u, true, true, false, |
| false, false); |
| } |
| |
| TEST_F(InstBindlessTest, SPV14AddToEntryPoint) { |
| const std::string text = R"( |
| ; CHECK: OpEntryPoint Fragment {{%\w+}} "foo" {{%\w+}} {{%\w+}} {{%\w+}} [[v1:%\w+]] [[v2:%\w+]] |
| ; CHECK: OpDecorate [[v1]] DescriptorSet 7 |
| ; CHECK: OpDecorate [[v2]] DescriptorSet 7 |
| ; CHECK: [[v1]] = OpVariable {{%\w+}} StorageBuffer |
| ; CHECK: [[v2]] = OpVariable {{%\w+}} StorageBuffer |
| OpCapability Shader |
| OpExtension "SPV_EXT_descriptor_indexing" |
| OpMemoryModel Logical GLSL450 |
| OpEntryPoint Fragment %foo "foo" %gid %image_var %sampler_var |
| OpExecutionMode %foo OriginUpperLeft |
| OpDecorate %image_var DescriptorSet 0 |
| OpDecorate %image_var Binding 0 |
| OpDecorate %sampler_var DescriptorSet 0 |
| OpDecorate %sampler_var Binding 1 |
| OpDecorate %gid DescriptorSet 0 |
| OpDecorate %gid Binding 2 |
| OpDecorate %struct Block |
| OpMemberDecorate %struct 0 Offset 0 |
| %void = OpTypeVoid |
| %int = OpTypeInt 32 0 |
| %int_0 = OpConstant %int 0 |
| %v3int = OpTypeVector %int 3 |
| %float = OpTypeFloat 32 |
| %v3float = OpTypeVector %float 3 |
| %v4float = OpTypeVector %float 4 |
| %struct = OpTypeStruct %v3int |
| %ptr_ssbo_struct = OpTypePointer StorageBuffer %struct |
| %ptr_ssbo_v3int = OpTypePointer StorageBuffer %v3int |
| %gid = OpVariable %ptr_ssbo_struct StorageBuffer |
| %image = OpTypeImage %float 3D 0 0 0 1 Unknown |
| %ptr_uc_image = OpTypePointer UniformConstant %image |
| %sampler = OpTypeSampler |
| %ptr_uc_sampler = OpTypePointer UniformConstant %sampler |
| %image_var = OpVariable %ptr_uc_image UniformConstant |
| %sampler_var = OpVariable %ptr_uc_sampler UniformConstant |
| %sampled = OpTypeSampledImage %image |
| %void_fn = OpTypeFunction %void |
| %foo = OpFunction %void None %void_fn |
| %entry = OpLabel |
| %ld_image = OpLoad %image %image_var |
| %ld_sampler = OpLoad %sampler %sampler_var |
| %gep = OpAccessChain %ptr_ssbo_v3int %gid %int_0 |
| %ld_gid = OpLoad %v3int %gep |
| %convert = OpConvertUToF %v3float %ld_gid |
| %sampled_image = OpSampledImage %sampled %ld_image %ld_sampler |
| %sample = OpImageSampleImplicitLod %v4float %sampled_image %convert |
| OpReturn |
| OpFunctionEnd |
| )"; |
| |
| SetTargetEnv(SPV_ENV_VULKAN_1_1_SPIRV_1_4); |
| SinglePassRunAndMatch<InstBindlessCheckPass>(text, true, 7u, 23u, true, true, |
| false, false, false); |
| } |
| |
| TEST_F(InstBindlessTest, SPV14AddToEntryPoints) { |
| const std::string text = R"( |
| ; CHECK: OpEntryPoint Fragment {{%\w+}} "foo" {{%\w+}} {{%\w+}} {{%\w+}} [[v1:%\w+]] [[v2:%\w+]] |
| ; CHECK: OpEntryPoint Fragment {{%\w+}} "bar" {{%\w+}} {{%\w+}} {{%\w+}} [[v1:%\w+]] [[v2:%\w+]] |
| ; CHECK: OpDecorate [[v1]] DescriptorSet 7 |
| ; CHECK: OpDecorate [[v2]] DescriptorSet 7 |
| ; CHECK: [[v1]] = OpVariable {{%\w+}} StorageBuffer |
| ; CHECK: [[v2]] = OpVariable {{%\w+}} StorageBuffer |
| OpCapability Shader |
| OpExtension "SPV_EXT_descriptor_indexing" |
| OpMemoryModel Logical GLSL450 |
| OpEntryPoint Fragment %foo "foo" %gid %image_var %sampler_var |
| OpEntryPoint Fragment %foo "bar" %gid %image_var %sampler_var |
| OpExecutionMode %foo OriginUpperLeft |
| OpDecorate %image_var DescriptorSet 0 |
| OpDecorate %image_var Binding 0 |
| OpDecorate %sampler_var DescriptorSet 0 |
| OpDecorate %sampler_var Binding 1 |
| OpDecorate %gid DescriptorSet 0 |
| OpDecorate %gid Binding 2 |
| OpDecorate %struct Block |
| OpMemberDecorate %struct 0 Offset 0 |
| %void = OpTypeVoid |
| %int = OpTypeInt 32 0 |
| %int_0 = OpConstant %int 0 |
| %v3int = OpTypeVector %int 3 |
| %float = OpTypeFloat 32 |
| %v3float = OpTypeVector %float 3 |
| %v4float = OpTypeVector %float 4 |
| %struct = OpTypeStruct %v3int |
| %ptr_ssbo_struct = OpTypePointer StorageBuffer %struct |
| %ptr_ssbo_v3int = OpTypePointer StorageBuffer %v3int |
| %gid = OpVariable %ptr_ssbo_struct StorageBuffer |
| %image = OpTypeImage %float 3D 0 0 0 1 Unknown |
| %ptr_uc_image = OpTypePointer UniformConstant %image |
| %sampler = OpTypeSampler |
| %ptr_uc_sampler = OpTypePointer UniformConstant %sampler |
| %image_var = OpVariable %ptr_uc_image UniformConstant |
| %sampler_var = OpVariable %ptr_uc_sampler UniformConstant |
| %sampled = OpTypeSampledImage %image |
| %void_fn = OpTypeFunction %void |
| %foo = OpFunction %void None %void_fn |
| %entry = OpLabel |
| %ld_image = OpLoad %image %image_var |
| %ld_sampler = OpLoad %sampler %sampler_var |
| %gep = OpAccessChain %ptr_ssbo_v3int %gid %int_0 |
| %ld_gid = OpLoad %v3int %gep |
| %convert = OpConvertUToF %v3float %ld_gid |
| %sampled_image = OpSampledImage %sampled %ld_image %ld_sampler |
| %sample = OpImageSampleImplicitLod %v4float %sampled_image %convert |
| OpReturn |
| OpFunctionEnd |
| )"; |
| |
| SetTargetEnv(SPV_ENV_VULKAN_1_1_SPIRV_1_4); |
| SinglePassRunAndMatch<InstBindlessCheckPass>(text, true, 7u, 23u, true, true, |
| false, false, false); |
| } |
| |
| TEST_F(InstBindlessTest, InstBoundsAndInitLoadUnsizedUBOArray) { |
| // #version 450 |
| // #extension GL_EXT_nonuniform_qualifier : enable |
| // |
| // layout(location=0) in nonuniformEXT flat int nu_ii; |
| // layout(location=0) out float b; |
| // |
| // layout(binding=3) uniform uname { float a; } uniformBuffer[]; |
| // |
| // void main() |
| // { |
| // b = uniformBuffer[nu_ii].a; |
| // } |
| |
| // clang-format off |
| const std::string defs = R"( |
| OpCapability Shader |
| OpCapability ShaderNonUniform |
| OpCapability RuntimeDescriptorArray |
| OpCapability UniformBufferArrayNonUniformIndexing |
| OpExtension "SPV_EXT_descriptor_indexing" |
| ; CHECK: OpExtension "SPV_KHR_storage_buffer_storage_class" |
| %1 = OpExtInstImport "GLSL.std.450" |
| OpMemoryModel Logical GLSL450 |
| OpEntryPoint Fragment %main "main" %b %nu_ii |
| ; CHECK: OpEntryPoint Fragment %main "main" %b %nu_ii %gl_FragCoord |
| OpExecutionMode %main OriginUpperLeft |
| OpSource GLSL 450 |
| OpSourceExtension "GL_EXT_nonuniform_qualifier" |
| OpName %main "main" |
| OpName %b "b" |
| OpName %uname "uname" |
| OpMemberName %uname 0 "a" |
| OpName %uniformBuffer "uniformBuffer" |
| OpName %nu_ii "nu_ii" |
| OpDecorate %b Location 0 |
| OpMemberDecorate %uname 0 Offset 0 |
| OpDecorate %uname Block |
| OpDecorate %uniformBuffer DescriptorSet 0 |
| OpDecorate %uniformBuffer Binding 3 |
| OpDecorate %nu_ii Flat |
| OpDecorate %nu_ii Location 0 |
| OpDecorate %nu_ii NonUniform |
| OpDecorate %16 NonUniform |
| OpDecorate %20 NonUniform |
| ; CHECK: OpDecorate %_runtimearr_uint ArrayStride 4 |
| )" + kInputDecorations + R"( |
| ; CHECK: OpDecorate %130 NonUniform |
| )" + kOutputDecorations + R"( |
| ; CHECK: OpDecorate %gl_FragCoord BuiltIn FragCoord |
| ; CHECK: OpDecorate %127 NonUniform |
| %void = OpTypeVoid |
| %3 = OpTypeFunction %void |
| %float = OpTypeFloat 32 |
| %_ptr_Output_float = OpTypePointer Output %float |
| %b = OpVariable %_ptr_Output_float Output |
| %uname = OpTypeStruct %float |
| %_runtimearr_uname = OpTypeRuntimeArray %uname |
| %_ptr_Uniform__runtimearr_uname = OpTypePointer Uniform %_runtimearr_uname |
| %uniformBuffer = OpVariable %_ptr_Uniform__runtimearr_uname Uniform |
| %int = OpTypeInt 32 1 |
| %_ptr_Input_int = OpTypePointer Input %int |
| %nu_ii = OpVariable %_ptr_Input_int Input |
| %int_0 = OpConstant %int 0 |
| %_ptr_Uniform_float = OpTypePointer Uniform %float |
| ; CHECK: %uint = OpTypeInt 32 0 |
| ; CHECK: %26 = OpTypeFunction %uint %uint %uint |
| ; CHECK: %_runtimearr_uint = OpTypeRuntimeArray %uint |
| )" + kInputGlobals + R"( |
| ; CHECK: %_ptr_StorageBuffer_uint = OpTypePointer StorageBuffer %uint |
| ; CHECK: %bool = OpTypeBool |
| ; CHECK: %49 = OpTypeFunction %void %uint %uint %uint %uint |
| )" + kOutputGlobals + R"( |
| ; CHECK: %v4float = OpTypeVector %float 4 |
| ; CHECK: %_ptr_Input_v4float = OpTypePointer Input %v4float |
| ; CHECK: %gl_FragCoord = OpVariable %_ptr_Input_v4float Input |
| ; CHECK: %v4uint = OpTypeVector %uint 4 |
| ; CHECK: %101 = OpConstantNull %float |
| ; CHECK: %105 = OpTypeFunction %uint %uint %uint %uint %uint |
| )"; |
| // clang-format on |
| |
| const std::string main_func = R"( |
| %main = OpFunction %void None %3 |
| %5 = OpLabel |
| %16 = OpLoad %int %nu_ii |
| %19 = OpAccessChain %_ptr_Uniform_float %uniformBuffer %16 %int_0 |
| %20 = OpLoad %float %19 |
| OpStore %b %20 |
| ; CHECK-NOT: %20 = OpLoad %float %19 |
| ; CHECK-NOT: OpStore %b %20 |
| ; CHECK: %40 = OpFunctionCall %uint %inst_bindless_direct_read_2 %uint_1 %uint_3 |
| ; CHECK: %42 = OpULessThan %bool %7 %40 |
| ; CHECK: OpSelectionMerge %43 None |
| ; CHECK: OpBranchConditional %42 %44 %45 |
| ; CHECK: %44 = OpLabel |
| ; CHECK: %103 = OpBitcast %uint %7 |
| ; CHECK: %122 = OpFunctionCall %uint %inst_bindless_direct_read_4 %uint_0 %uint_0 %uint_3 %103 |
| ; CHECK: %123 = OpULessThan %bool %uint_0 %122 |
| ; CHECK: OpSelectionMerge %124 None |
| ; CHECK: OpBranchConditional %123 %125 %126 |
| ; CHECK: %125 = OpLabel |
| ; CHECK: %127 = OpLoad %float %20 |
| ; CHECK: OpBranch %124 |
| ; CHECK: %126 = OpLabel |
| ; CHECK: %128 = OpBitcast %uint %7 |
| ; CHECK: %129 = OpFunctionCall %void %inst_bindless_stream_write_4 %uint_45 %uint_1 %128 %uint_0 |
| ; CHECK: OpBranch %124 |
| ; CHECK: %124 = OpLabel |
| ; CHECK: %130 = OpPhi %float %127 %125 %101 %126 |
| ; CHECK: OpBranch %43 |
| ; CHECK: %45 = OpLabel |
| ; CHECK: %47 = OpBitcast %uint %7 |
| ; CHECK: %100 = OpFunctionCall %void %inst_bindless_stream_write_4 %uint_45 %uint_0 %47 %40 |
| ; CHECK: OpBranch %43 |
| ; CHECK: %43 = OpLabel |
| ; CHECK: %102 = OpPhi %float %130 %124 %101 %45 |
| ; CHECK: OpStore %b %102 |
| OpReturn |
| OpFunctionEnd |
| )"; |
| |
| const std::string new_funcs = kDirectRead2 + kStreamWrite4Frag + kDirectRead4; |
| |
| // SetAssembleOptions(SPV_TEXT_TO_BINARY_OPTION_PRESERVE_NUMERIC_IDS); |
| SinglePassRunAndMatch<InstBindlessCheckPass>(defs + main_func + new_funcs, |
| true, 7u, 23u, true, true, false, |
| false, false); |
| } |
| |
| TEST_F(InstBindlessTest, InstBoundsAndInitLoadUnsizedSSBOArrayDeprecated) { |
| // #version 450 |
| // #extension GL_EXT_nonuniform_qualifier : enable |
| // |
| // layout(location=0) in nonuniformEXT flat int nu_ii; |
| // layout(location=0) out float b; |
| // |
| // layout(binding=3) buffer bname { float b; } storageBuffer[]; |
| // |
| // void main() |
| // { |
| // b = storageBuffer[nu_ii].b; |
| // } |
| |
| // clang-format off |
| const std::string defs = R"( |
| OpCapability Shader |
| OpCapability ShaderNonUniform |
| OpCapability RuntimeDescriptorArray |
| OpCapability StorageBufferArrayNonUniformIndexing |
| OpExtension "SPV_EXT_descriptor_indexing" |
| ; CHECK: OpExtension "SPV_KHR_storage_buffer_storage_class" |
| %1 = OpExtInstImport "GLSL.std.450" |
| OpMemoryModel Logical GLSL450 |
| OpEntryPoint Fragment %main "main" %b %nu_ii |
| ; CHECK: OpEntryPoint Fragment %main "main" %b %nu_ii %gl_FragCoord |
| OpExecutionMode %main OriginUpperLeft |
| OpSource GLSL 450 |
| OpSourceExtension "GL_EXT_nonuniform_qualifier" |
| OpName %main "main" |
| OpName %b "b" |
| OpName %bname "bname" |
| OpMemberName %bname 0 "a" |
| OpName %storageBuffer "storageBuffer" |
| OpName %nu_ii "nu_ii" |
| OpDecorate %b Location 0 |
| OpMemberDecorate %bname 0 Offset 0 |
| OpDecorate %bname Block |
| OpDecorate %storageBuffer DescriptorSet 0 |
| OpDecorate %storageBuffer Binding 3 |
| OpDecorate %nu_ii Flat |
| OpDecorate %nu_ii Location 0 |
| OpDecorate %nu_ii NonUniform |
| OpDecorate %16 NonUniform |
| OpDecorate %20 NonUniform |
| ; CHECK: OpDecorate %_runtimearr_uint ArrayStride 4 |
| )" + kInputDecorations + R"( |
| ; CHECK: OpDecorate %130 NonUniform |
| )" + kOutputDecorations + R"( |
| ; CHECK: OpDecorate %gl_FragCoord BuiltIn FragCoord |
| ; CHECK: OpDecorate %127 NonUniform |
| %void = OpTypeVoid |
| %3 = OpTypeFunction %void |
| %float = OpTypeFloat 32 |
| %_ptr_Output_float = OpTypePointer Output %float |
| %b = OpVariable %_ptr_Output_float Output |
| %bname = OpTypeStruct %float |
| %_runtimearr_bname = OpTypeRuntimeArray %bname |
| %_ptr_StorageBuffer__runtimearr_bname = OpTypePointer StorageBuffer %_runtimearr_bname |
| %storageBuffer = OpVariable %_ptr_StorageBuffer__runtimearr_bname StorageBuffer |
| %int = OpTypeInt 32 1 |
| %_ptr_Input_int = OpTypePointer Input %int |
| %nu_ii = OpVariable %_ptr_Input_int Input |
| %int_0 = OpConstant %int 0 |
| %_ptr_StorageBuffer_float = OpTypePointer StorageBuffer %float |
| ; CHECK: %uint = OpTypeInt 32 0 |
| ; CHECK: %26 = OpTypeFunction %uint %uint %uint |
| ; CHECK: %_runtimearr_uint = OpTypeRuntimeArray %uint |
| )" + kInputGlobals + R"( |
| ; CHECK: %_ptr_StorageBuffer_uint = OpTypePointer StorageBuffer %uint |
| ; CHECK: %bool = OpTypeBool |
| ; CHECK: %49 = OpTypeFunction %void %uint %uint %uint %uint |
| )" + kOutputGlobals + R"( |
| ; CHECK: %v4float = OpTypeVector %float 4 |
| ; CHECK: %_ptr_Input_v4float = OpTypePointer Input %v4float |
| ; CHECK: %gl_FragCoord = OpVariable %_ptr_Input_v4float Input |
| ; CHECK: %v4uint = OpTypeVector %uint 4 |
| ; CHECK: %101 = OpConstantNull %float |
| ; CHECK: %105 = OpTypeFunction %uint %uint %uint %uint %uint |
| )"; |
| // clang-format on |
| |
| const std::string main_func = R"( |
| %main = OpFunction %void None %3 |
| %5 = OpLabel |
| %16 = OpLoad %int %nu_ii |
| %19 = OpAccessChain %_ptr_StorageBuffer_float %storageBuffer %16 %int_0 |
| %20 = OpLoad %float %19 |
| OpStore %b %20 |
| ; CHECK-NOT: %20 = OpLoad %float %19 |
| ; CHECK-NOT: OpStore %b %20 |
| ; CHECK: %40 = OpFunctionCall %uint %inst_bindless_direct_read_2 %uint_1 %uint_3 |
| ; CHECK: %42 = OpULessThan %bool %7 %40 |
| ; CHECK: OpSelectionMerge %43 None |
| ; CHECK: OpBranchConditional %42 %44 %45 |
| ; CHECK: %44 = OpLabel |
| ; CHECK: %103 = OpBitcast %uint %7 |
| ; CHECK: %122 = OpFunctionCall %uint %inst_bindless_direct_read_4 %uint_0 %uint_0 %uint_3 %103 |
| ; CHECK: %123 = OpULessThan %bool %uint_0 %122 |
| ; CHECK: OpSelectionMerge %124 None |
| ; CHECK: OpBranchConditional %123 %125 %126 |
| ; CHECK: %125 = OpLabel |
| ; CHECK: %127 = OpLoad %float %20 |
| ; CHECK: OpBranch %124 |
| ; CHECK: %126 = OpLabel |
| ; CHECK: %128 = OpBitcast %uint %7 |
| ; CHECK: %129 = OpFunctionCall %void %inst_bindless_stream_write_4 %uint_45 %uint_1 %128 %uint_0 |
| ; CHECK: OpBranch %124 |
| ; CHECK: %124 = OpLabel |
| ; CHECK: %130 = OpPhi %float %127 %125 %101 %126 |
| ; CHECK: OpBranch %43 |
| ; CHECK: %45 = OpLabel |
| ; CHECK: %47 = OpBitcast %uint %7 |
| ; CHECK: %100 = OpFunctionCall %void %inst_bindless_stream_write_4 %uint_45 %uint_0 %47 %40 |
| ; CHECK: OpBranch %43 |
| ; CHECK: %43 = OpLabel |
| ; CHECK: %102 = OpPhi %float %130 %124 %101 %45 |
| ; CHECK: OpStore %b %102 |
| OpReturn |
| OpFunctionEnd |
| )"; |
| |
| const std::string new_funcs = kDirectRead2 + kStreamWrite4Frag + kDirectRead4; |
| |
| // SetAssembleOptions(SPV_TEXT_TO_BINARY_OPTION_PRESERVE_NUMERIC_IDS); |
| SinglePassRunAndMatch<InstBindlessCheckPass>(defs + main_func + new_funcs, |
| true, 7u, 23u, true, true, false, |
| false, false); |
| } |
| |
| TEST_F(InstBindlessTest, InstBoundsAndInitLoadUnsizedSSBOArray) { |
| // Same as Deprecated but declaring as StorageBuffer Block |
| |
| // clang-format off |
| const std::string defs = R"( |
| OpCapability Shader |
| OpCapability ShaderNonUniform |
| OpCapability RuntimeDescriptorArray |
| OpCapability StorageBufferArrayNonUniformIndexing |
| OpExtension "SPV_EXT_descriptor_indexing" |
| ; CHECK: OpExtension "SPV_KHR_storage_buffer_storage_class" |
| %1 = OpExtInstImport "GLSL.std.450" |
| OpMemoryModel Logical GLSL450 |
| OpEntryPoint Fragment %main "main" %b %nu_ii |
| ; CHECK: OpEntryPoint Fragment %main "main" %b %nu_ii %gl_FragCoord |
| OpExecutionMode %main OriginUpperLeft |
| OpSource GLSL 450 |
| OpSourceExtension "GL_EXT_nonuniform_qualifier" |
| OpName %main "main" |
| OpName %b "b" |
| OpName %bname "bname" |
| OpMemberName %bname 0 "a" |
| OpName %storageBuffer "storageBuffer" |
| OpName %nu_ii "nu_ii" |
| OpDecorate %b Location 0 |
| OpMemberDecorate %bname 0 Offset 0 |
| OpDecorate %bname Block |
| OpDecorate %storageBuffer DescriptorSet 0 |
| OpDecorate %storageBuffer Binding 3 |
| OpDecorate %nu_ii Flat |
| OpDecorate %nu_ii Location 0 |
| OpDecorate %nu_ii NonUniform |
| OpDecorate %16 NonUniform |
| OpDecorate %20 NonUniform |
| ; CHECK: OpDecorate %_runtimearr_uint ArrayStride 4 |
| )" + kInputDecorations + R"( |
| ; CHECK: OpDecorate %130 NonUniform |
| )" + kOutputDecorations + R"( |
| ; CHECK: OpDecorate %gl_FragCoord BuiltIn FragCoord |
| ; CHECK: OpDecorate %127 NonUniform |
| %void = OpTypeVoid |
| %3 = OpTypeFunction %void |
| %float = OpTypeFloat 32 |
| %_ptr_Output_float = OpTypePointer Output %float |
| %b = OpVariable %_ptr_Output_float Output |
| %bname = OpTypeStruct %float |
| %_runtimearr_bname = OpTypeRuntimeArray %bname |
| %_ptr_StorageBuffer__runtimearr_bname = OpTypePointer StorageBuffer %_runtimearr_bname |
| %storageBuffer = OpVariable %_ptr_StorageBuffer__runtimearr_bname StorageBuffer |
| %int = OpTypeInt 32 1 |
| %_ptr_Input_int = OpTypePointer Input %int |
| %nu_ii = OpVariable %_ptr_Input_int Input |
| %int_0 = OpConstant %int 0 |
| %_ptr_StorageBuffer_float = OpTypePointer StorageBuffer %float |
| ; CHECK: %uint = OpTypeInt 32 0 |
| ; CHECK: %26 = OpTypeFunction %uint %uint %uint |
| ; CHECK: %_runtimearr_uint = OpTypeRuntimeArray %uint |
| )" + kInputGlobals + R"( |
| ; CHECK: %_ptr_StorageBuffer_uint = OpTypePointer StorageBuffer %uint |
| ; CHECK: %bool = OpTypeBool |
| ; CHECK: %49 = OpTypeFunction %void %uint %uint %uint %uint |
| )" + kOutputGlobals + R"( |
| ; CHECK: %v4float = OpTypeVector %float 4 |
| ; CHECK: %_ptr_Input_v4float = OpTypePointer Input %v4float |
| ; CHECK: %gl_FragCoord = OpVariable %_ptr_Input_v4float Input |
| ; CHECK: %v4uint = OpTypeVector %uint 4 |
| ; CHECK: %101 = OpConstantNull %float |
| ; CHECK: %105 = OpTypeFunction %uint %uint %uint %uint %uint |
| )"; |
| // clang-format on |
| |
| const std::string main_func = R"( |
| %main = OpFunction %void None %3 |
| %5 = OpLabel |
| %16 = OpLoad %int %nu_ii |
| %19 = OpAccessChain %_ptr_StorageBuffer_float %storageBuffer %16 %int_0 |
| %20 = OpLoad %float %19 |
| OpStore %b %20 |
| ; CHECK-NOT: %20 = OpLoad %float %19 |
| ; CHECK-NOT: OpStore %b %20 |
| ; CHECK: %40 = OpFunctionCall %uint %inst_bindless_direct_read_2 %uint_1 %uint_3 |
| ; CHECK: %42 = OpULessThan %bool %7 %40 |
| ; CHECK: OpSelectionMerge %43 None |
| ; CHECK: OpBranchConditional %42 %44 %45 |
| ; CHECK: %44 = OpLabel |
| ; CHECK: %103 = OpBitcast %uint %7 |
| ; CHECK: %122 = OpFunctionCall %uint %inst_bindless_direct_read_4 %uint_0 %uint_0 %uint_3 %103 |
| ; CHECK: %123 = OpULessThan %bool %uint_0 %122 |
| ; CHECK: OpSelectionMerge %124 None |
| ; CHECK: OpBranchConditional %123 %125 %126 |
| ; CHECK: %125 = OpLabel |
| ; CHECK: %127 = OpLoad %float %20 |
| ; CHECK: OpBranch %124 |
| ; CHECK: %126 = OpLabel |
| ; CHECK: %128 = OpBitcast %uint %7 |
| ; CHECK: %129 = OpFunctionCall %void %inst_bindless_stream_write_4 %uint_45 %uint_1 %128 %uint_0 |
| ; CHECK: OpBranch %124 |
| ; CHECK: %124 = OpLabel |
| ; CHECK: %130 = OpPhi %float %127 %125 %101 %126 |
| ; CHECK: OpBranch %43 |
| ; CHECK: %45 = OpLabel |
| ; CHECK: %47 = OpBitcast %uint %7 |
| ; CHECK: %100 = OpFunctionCall %void %inst_bindless_stream_write_4 %uint_45 %uint_0 %47 %40 |
| ; CHECK: OpBranch %43 |
| ; CHECK: %43 = OpLabel |
| ; CHECK: %102 = OpPhi %float %130 %124 %101 %45 |
| ; CHECK: OpStore %b %102 |
| OpReturn |
| OpFunctionEnd |
| )"; |
| |
| const std::string new_funcs = kDirectRead2 + kStreamWrite4Frag + kDirectRead4; |
| |
| // SetAssembleOptions(SPV_TEXT_TO_BINARY_OPTION_PRESERVE_NUMERIC_IDS); |
| SinglePassRunAndMatch<InstBindlessCheckPass>(defs + main_func + new_funcs, |
| true, 7u, 23u, true, true, false, |
| false, false); |
| } |
| |
| TEST_F(InstBindlessTest, InstInitLoadUBOScalar) { |
| // #version 450 |
| // #extension GL_EXT_nonuniform_qualifier : enable |
| // |
| // layout(location=0) out float b; |
| // layout(binding=3) uniform uname { float a; } uniformBuffer; |
| // |
| // void main() |
| // { |
| // b = uniformBuffer.a; |
| // } |
| |
| // clang-format off |
| const std::string defs = R"( |
| OpCapability Shader |
| OpExtension "SPV_EXT_descriptor_indexing" |
| ; CHECK: OpExtension "SPV_KHR_storage_buffer_storage_class" |
| %1 = OpExtInstImport "GLSL.std.450" |
| OpMemoryModel Logical GLSL450 |
| OpEntryPoint Fragment %main "main" %b |
| ; CHECK: OpEntryPoint Fragment %main "main" %b %gl_FragCoord |
| OpExecutionMode %main OriginUpperLeft |
| OpSource GLSL 450 |
| OpSourceExtension "GL_EXT_nonuniform_qualifier" |
| OpName %main "main" |
| OpName %b "b" |
| OpName %uname "uname" |
| OpMemberName %uname 0 "a" |
| OpName %uniformBuffer "uniformBuffer" |
| OpDecorate %b Location 0 |
| OpMemberDecorate %uname 0 Offset 0 |
| OpDecorate %uname Block |
|
