SPIRV-Tools/test/opt/dominator_tree/simple.cpp
Victor Lomuller efc5061929 Dominator analysis interface clean.
Remove the CFG requirement when querying a dominator/post-dominator from an IRContext.

Updated all uses of the function and tests.
2018-04-20 15:41:59 -04:00

176 lines
5.5 KiB
C++

// Copyright (c) 2017 Google 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 <vector>
#include <gmock/gmock.h>
#include "../assembly_builder.h"
#include "../function_utils.h"
#include "../pass_fixture.h"
#include "../pass_utils.h"
#include "opt/dominator_analysis.h"
#include "opt/pass.h"
namespace {
using namespace spvtools;
using ::testing::UnorderedElementsAre;
using PassClassTest = PassTest<::testing::Test>;
/*
Generated from the following GLSL
#version 440 core
layout(location = 0) out vec4 c;
layout(location = 1)in vec4 in_val;
void main(){
if ( in_val.x < 10) {
int z = 0;
int i = 0;
for (i = 0; i < in_val.y; ++i) {
z += i;
}
c = vec4(i,i,i,i);
} else {
c = vec4(1,1,1,1);
}
}
*/
TEST_F(PassClassTest, BasicVisitFromEntryPoint) {
const std::string text = R"(
OpCapability Shader
%1 = OpExtInstImport "GLSL.std.450"
OpMemoryModel Logical GLSL450
OpEntryPoint Fragment %4 "main" %9 %43
OpExecutionMode %4 OriginUpperLeft
OpSource GLSL 440
OpName %4 "main"
OpName %9 "in_val"
OpName %22 "z"
OpName %24 "i"
OpName %43 "c"
OpDecorate %9 Location 1
OpDecorate %43 Location 0
%2 = OpTypeVoid
%3 = OpTypeFunction %2
%6 = OpTypeFloat 32
%7 = OpTypeVector %6 4
%8 = OpTypePointer Input %7
%9 = OpVariable %8 Input
%10 = OpTypeInt 32 0
%11 = OpConstant %10 0
%12 = OpTypePointer Input %6
%15 = OpConstant %6 10
%16 = OpTypeBool
%20 = OpTypeInt 32 1
%21 = OpTypePointer Function %20
%23 = OpConstant %20 0
%32 = OpConstant %10 1
%40 = OpConstant %20 1
%42 = OpTypePointer Output %7
%43 = OpVariable %42 Output
%54 = OpConstant %6 1
%55 = OpConstantComposite %7 %54 %54 %54 %54
%4 = OpFunction %2 None %3
%5 = OpLabel
%22 = OpVariable %21 Function
%24 = OpVariable %21 Function
%13 = OpAccessChain %12 %9 %11
%14 = OpLoad %6 %13
%17 = OpFOrdLessThan %16 %14 %15
OpSelectionMerge %19 None
OpBranchConditional %17 %18 %53
%18 = OpLabel
OpStore %22 %23
OpStore %24 %23
OpStore %24 %23
OpBranch %25
%25 = OpLabel
OpLoopMerge %27 %28 None
OpBranch %29
%29 = OpLabel
%30 = OpLoad %20 %24
%31 = OpConvertSToF %6 %30
%33 = OpAccessChain %12 %9 %32
%34 = OpLoad %6 %33
%35 = OpFOrdLessThan %16 %31 %34
OpBranchConditional %35 %26 %27
%26 = OpLabel
%36 = OpLoad %20 %24
%37 = OpLoad %20 %22
%38 = OpIAdd %20 %37 %36
OpStore %22 %38
OpBranch %28
%28 = OpLabel
%39 = OpLoad %20 %24
%41 = OpIAdd %20 %39 %40
OpStore %24 %41
OpBranch %25
%27 = OpLabel
%44 = OpLoad %20 %24
%45 = OpConvertSToF %6 %44
%46 = OpLoad %20 %24
%47 = OpConvertSToF %6 %46
%48 = OpLoad %20 %24
%49 = OpConvertSToF %6 %48
%50 = OpLoad %20 %24
%51 = OpConvertSToF %6 %50
%52 = OpCompositeConstruct %7 %45 %47 %49 %51
OpStore %43 %52
OpBranch %19
%53 = OpLabel
OpStore %43 %55
OpBranch %19
%19 = OpLabel
OpReturn
OpFunctionEnd
)";
// clang-format on
std::unique_ptr<ir::IRContext> context =
BuildModule(SPV_ENV_UNIVERSAL_1_1, nullptr, text,
SPV_TEXT_TO_BINARY_OPTION_PRESERVE_NUMERIC_IDS);
ir::Module* module = context->module();
EXPECT_NE(nullptr, module) << "Assembling failed for shader:\n"
<< text << std::endl;
const ir::Function* f = spvtest::GetFunction(module, 4);
opt::DominatorAnalysis* analysis = context->GetDominatorAnalysis(f);
const ir::CFG& cfg = *context->cfg();
opt::DominatorTree& tree = analysis->GetDomTree();
EXPECT_EQ(tree.GetRoot()->bb_, cfg.pseudo_entry_block());
EXPECT_TRUE(analysis->Dominates(5, 18));
EXPECT_TRUE(analysis->Dominates(5, 53));
EXPECT_TRUE(analysis->Dominates(5, 19));
EXPECT_TRUE(analysis->Dominates(5, 25));
EXPECT_TRUE(analysis->Dominates(5, 29));
EXPECT_TRUE(analysis->Dominates(5, 27));
EXPECT_TRUE(analysis->Dominates(5, 26));
EXPECT_TRUE(analysis->Dominates(5, 28));
EXPECT_TRUE(analysis->StrictlyDominates(5, 18));
EXPECT_TRUE(analysis->StrictlyDominates(5, 53));
EXPECT_TRUE(analysis->StrictlyDominates(5, 19));
EXPECT_TRUE(analysis->StrictlyDominates(5, 25));
EXPECT_TRUE(analysis->StrictlyDominates(5, 29));
EXPECT_TRUE(analysis->StrictlyDominates(5, 27));
EXPECT_TRUE(analysis->StrictlyDominates(5, 26));
EXPECT_TRUE(analysis->StrictlyDominates(5, 28));
}
} // namespace