158 lines
4.5 KiB
C++
158 lines
4.5 KiB
C++
// Copyright (c) 2023 The Khronos Group 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.
|
|
|
|
#include <string>
|
|
|
|
#include "gmock/gmock.h"
|
|
#include "source/opt/licm_pass.h"
|
|
#include "test/opt/pass_fixture.h"
|
|
|
|
namespace spvtools {
|
|
namespace opt {
|
|
namespace {
|
|
|
|
using PassClassTest = PassTest<::testing::Test>;
|
|
|
|
/*
|
|
Tests for the LICM pass to check it handles access chains correctly
|
|
|
|
Generated from the following GLSL fragment shader
|
|
--eliminate-local-multi-store has also been run on the spv binary
|
|
#version 460
|
|
void main() {
|
|
for (uint i = 0; i < 123u; ++i) {
|
|
vec2 do_not_hoist_store = vec2(0.0f);
|
|
float do_not_hoist_access_chain_load = do_not_hoist_store.x;
|
|
}
|
|
}
|
|
*/
|
|
|
|
TEST_F(PassClassTest, HoistAccessChains) {
|
|
const std::string before_hoist = R"(OpCapability Shader
|
|
%1 = OpExtInstImport "GLSL.std.450"
|
|
OpMemoryModel Logical GLSL450
|
|
OpEntryPoint Fragment %main "main"
|
|
OpExecutionMode %main OriginUpperLeft
|
|
OpSource GLSL 460
|
|
OpName %main "main"
|
|
OpName %i "i"
|
|
OpName %do_not_hoist_store "do_not_hoist_store"
|
|
OpName %do_not_hoist_access_chain_load "do_not_hoist_access_chain_load"
|
|
%void = OpTypeVoid
|
|
%7 = OpTypeFunction %void
|
|
%uint = OpTypeInt 32 0
|
|
%_ptr_Function_uint = OpTypePointer Function %uint
|
|
%uint_0 = OpConstant %uint 0
|
|
%uint_123 = OpConstant %uint 123
|
|
%bool = OpTypeBool
|
|
%float = OpTypeFloat 32
|
|
%v2float = OpTypeVector %float 2
|
|
%_ptr_Function_v2float = OpTypePointer Function %v2float
|
|
%float_0 = OpConstant %float 0
|
|
%17 = OpConstantComposite %v2float %float_0 %float_0
|
|
%_ptr_Function_float = OpTypePointer Function %float
|
|
%int = OpTypeInt 32 1
|
|
%int_1 = OpConstant %int 1
|
|
%main = OpFunction %void None %7
|
|
%21 = OpLabel
|
|
%i = OpVariable %_ptr_Function_uint Function
|
|
%do_not_hoist_store = OpVariable %_ptr_Function_v2float Function
|
|
%do_not_hoist_access_chain_load = OpVariable %_ptr_Function_float Function
|
|
OpStore %i %uint_0
|
|
OpBranch %22
|
|
%22 = OpLabel
|
|
OpLoopMerge %23 %24 None
|
|
OpBranch %25
|
|
%25 = OpLabel
|
|
%26 = OpLoad %uint %i
|
|
%27 = OpULessThan %bool %26 %uint_123
|
|
OpBranchConditional %27 %28 %23
|
|
%28 = OpLabel
|
|
OpStore %do_not_hoist_store %17
|
|
%29 = OpAccessChain %_ptr_Function_float %do_not_hoist_store %uint_0
|
|
%30 = OpLoad %float %29
|
|
OpStore %do_not_hoist_access_chain_load %30
|
|
OpBranch %24
|
|
%24 = OpLabel
|
|
%31 = OpLoad %uint %i
|
|
%32 = OpIAdd %uint %31 %int_1
|
|
OpStore %i %32
|
|
OpBranch %22
|
|
%23 = OpLabel
|
|
OpReturn
|
|
OpFunctionEnd
|
|
)";
|
|
|
|
const std::string after_hoist = R"(OpCapability Shader
|
|
%1 = OpExtInstImport "GLSL.std.450"
|
|
OpMemoryModel Logical GLSL450
|
|
OpEntryPoint Fragment %main "main"
|
|
OpExecutionMode %main OriginUpperLeft
|
|
OpSource GLSL 460
|
|
OpName %main "main"
|
|
OpName %i "i"
|
|
OpName %do_not_hoist_store "do_not_hoist_store"
|
|
OpName %do_not_hoist_access_chain_load "do_not_hoist_access_chain_load"
|
|
%void = OpTypeVoid
|
|
%7 = OpTypeFunction %void
|
|
%uint = OpTypeInt 32 0
|
|
%_ptr_Function_uint = OpTypePointer Function %uint
|
|
%uint_0 = OpConstant %uint 0
|
|
%uint_123 = OpConstant %uint 123
|
|
%bool = OpTypeBool
|
|
%float = OpTypeFloat 32
|
|
%v2float = OpTypeVector %float 2
|
|
%_ptr_Function_v2float = OpTypePointer Function %v2float
|
|
%float_0 = OpConstant %float 0
|
|
%17 = OpConstantComposite %v2float %float_0 %float_0
|
|
%_ptr_Function_float = OpTypePointer Function %float
|
|
%int = OpTypeInt 32 1
|
|
%int_1 = OpConstant %int 1
|
|
%main = OpFunction %void None %7
|
|
%21 = OpLabel
|
|
%i = OpVariable %_ptr_Function_uint Function
|
|
%do_not_hoist_store = OpVariable %_ptr_Function_v2float Function
|
|
%do_not_hoist_access_chain_load = OpVariable %_ptr_Function_float Function
|
|
OpStore %i %uint_0
|
|
%29 = OpAccessChain %_ptr_Function_float %do_not_hoist_store %uint_0
|
|
OpBranch %22
|
|
%22 = OpLabel
|
|
OpLoopMerge %23 %24 None
|
|
OpBranch %25
|
|
%25 = OpLabel
|
|
%26 = OpLoad %uint %i
|
|
%27 = OpULessThan %bool %26 %uint_123
|
|
OpBranchConditional %27 %28 %23
|
|
%28 = OpLabel
|
|
OpStore %do_not_hoist_store %17
|
|
%30 = OpLoad %float %29
|
|
OpStore %do_not_hoist_access_chain_load %30
|
|
OpBranch %24
|
|
%24 = OpLabel
|
|
%31 = OpLoad %uint %i
|
|
%32 = OpIAdd %uint %31 %int_1
|
|
OpStore %i %32
|
|
OpBranch %22
|
|
%23 = OpLabel
|
|
OpReturn
|
|
OpFunctionEnd
|
|
)";
|
|
|
|
SinglePassRunAndCheck<LICMPass>(before_hoist, after_hoist, true);
|
|
}
|
|
|
|
} // namespace
|
|
} // namespace opt
|
|
} // namespace spvtools
|