blob: ef94384330dfad6f43ff9cf1ffc27fee70ed1211 [file] [log] [blame] [edit]
// Copyright 2023 The Tint Authors.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// 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 "src/tint/lang/spirv/writer/common/helper_test.h"
using namespace tint::core::number_suffixes; // NOLINT
namespace tint::spirv::writer {
namespace {
TEST_F(SpirvWriterTest, Switch_Basic) {
auto* func = b.Function("foo", ty.void_());
b.Append(func->Block(), [&] {
auto* swtch = b.Switch(42_i);
auto* def_case = b.Case(swtch, Vector{core::ir::Switch::CaseSelector()});
b.Append(def_case, [&] { //
b.ExitSwitch(swtch);
});
b.Return(func);
});
ASSERT_TRUE(Generate()) << Error() << output_;
EXPECT_INST(R"(
%4 = OpLabel
OpSelectionMerge %8 None
OpSwitch %int_42 %5
%5 = OpLabel
OpBranch %8
%8 = OpLabel
OpReturn
OpFunctionEnd
)");
}
TEST_F(SpirvWriterTest, Switch_MultipleCases) {
auto* func = b.Function("foo", ty.void_());
b.Append(func->Block(), [&] {
auto* swtch = b.Switch(42_i);
auto* case_a = b.Case(swtch, Vector{core::ir::Switch::CaseSelector{b.Constant(1_i)}});
b.Append(case_a, [&] { //
b.ExitSwitch(swtch);
});
auto* case_b = b.Case(swtch, Vector{core::ir::Switch::CaseSelector{b.Constant(2_i)}});
b.Append(case_b, [&] { //
b.ExitSwitch(swtch);
});
auto* def_case = b.Case(swtch, Vector{core::ir::Switch::CaseSelector()});
b.Append(def_case, [&] { //
b.ExitSwitch(swtch);
});
b.Return(func);
});
ASSERT_TRUE(Generate()) << Error() << output_;
EXPECT_INST(R"(
%4 = OpLabel
OpSelectionMerge %10 None
OpSwitch %int_42 %5 1 %8 2 %9
%8 = OpLabel
OpBranch %10
%9 = OpLabel
OpBranch %10
%5 = OpLabel
OpBranch %10
%10 = OpLabel
OpReturn
OpFunctionEnd
)");
}
TEST_F(SpirvWriterTest, Switch_MultipleSelectorsPerCase) {
auto* func = b.Function("foo", ty.void_());
b.Append(func->Block(), [&] {
auto* swtch = b.Switch(42_i);
auto* case_a = b.Case(swtch, Vector{core::ir::Switch::CaseSelector{b.Constant(1_i)},
core::ir::Switch::CaseSelector{b.Constant(3_i)}});
b.Append(case_a, [&] { //
b.ExitSwitch(swtch);
});
auto* case_b = b.Case(swtch, Vector{core::ir::Switch::CaseSelector{b.Constant(2_i)},
core::ir::Switch::CaseSelector{b.Constant(4_i)}});
b.Append(case_b, [&] { //
b.ExitSwitch(swtch);
});
auto* def_case = b.Case(swtch, Vector{core::ir::Switch::CaseSelector{b.Constant(5_i)},
core::ir::Switch::CaseSelector()});
b.Append(def_case, [&] { //
b.ExitSwitch(swtch);
});
b.Return(func);
});
ASSERT_TRUE(Generate()) << Error() << output_;
EXPECT_INST(R"(
%4 = OpLabel
OpSelectionMerge %10 None
OpSwitch %int_42 %5 1 %8 3 %8 2 %9 4 %9 5 %5
%8 = OpLabel
OpBranch %10
%9 = OpLabel
OpBranch %10
%5 = OpLabel
OpBranch %10
%10 = OpLabel
OpReturn
OpFunctionEnd
)");
}
TEST_F(SpirvWriterTest, Switch_AllCasesReturn) {
auto* func = b.Function("foo", ty.void_());
b.Append(func->Block(), [&] {
auto* swtch = b.Switch(42_i);
auto* case_a = b.Case(swtch, Vector{core::ir::Switch::CaseSelector{b.Constant(1_i)}});
b.Append(case_a, [&] { //
b.Return(func);
});
auto* case_b = b.Case(swtch, Vector{core::ir::Switch::CaseSelector{b.Constant(2_i)}});
b.Append(case_b, [&] { //
b.Return(func);
});
auto* def_case = b.Case(swtch, Vector{core::ir::Switch::CaseSelector()});
b.Append(def_case, [&] { //
b.Return(func);
});
b.Unreachable();
});
ASSERT_TRUE(Generate()) << Error() << output_;
EXPECT_INST(R"(
%4 = OpLabel
OpSelectionMerge %10 None
OpSwitch %int_42 %5 1 %8 2 %9
%8 = OpLabel
OpBranch %10
%9 = OpLabel
OpBranch %10
%5 = OpLabel
OpBranch %10
%10 = OpLabel
OpReturn
OpFunctionEnd
)");
}
TEST_F(SpirvWriterTest, Switch_ConditionalBreak) {
auto* func = b.Function("foo", ty.void_());
b.Append(func->Block(), [&] {
auto* swtch = b.Switch(42_i);
auto* case_a = b.Case(swtch, Vector{core::ir::Switch::CaseSelector{b.Constant(1_i)}});
b.Append(case_a, [&] {
auto* cond_break = b.If(true);
b.Append(cond_break->True(), [&] { //
b.ExitSwitch(swtch);
});
b.Append(cond_break->False(), [&] { //
b.ExitIf(cond_break);
});
b.Return(func);
});
auto* def_case = b.Case(swtch, Vector{core::ir::Switch::CaseSelector()});
b.Append(def_case, [&] { //
b.ExitSwitch(swtch);
});
b.Return(func);
});
ASSERT_TRUE(Generate()) << Error() << output_;
EXPECT_INST(R"(
%4 = OpLabel
OpSelectionMerge %9 None
OpSwitch %int_42 %5 1 %8
%8 = OpLabel
OpSelectionMerge %10 None
OpBranchConditional %true %11 %10
%11 = OpLabel
OpBranch %9
%10 = OpLabel
OpBranch %9
%5 = OpLabel
OpBranch %9
%9 = OpLabel
OpReturn
OpFunctionEnd
)");
}
TEST_F(SpirvWriterTest, Switch_Phi_SingleValue) {
auto* func = b.Function("foo", ty.i32());
b.Append(func->Block(), [&] {
auto* s = b.Switch(42_i);
s->SetResults(b.InstructionResult(ty.i32()));
auto* case_a = b.Case(s, Vector{core::ir::Switch::CaseSelector{b.Constant(1_i)},
core::ir::Switch::CaseSelector{nullptr}});
b.Append(case_a, [&] { //
b.ExitSwitch(s, 10_i);
});
auto* case_b = b.Case(s, Vector{core::ir::Switch::CaseSelector{b.Constant(2_i)}});
b.Append(case_b, [&] { //
b.ExitSwitch(s, 20_i);
});
b.Return(func, s);
});
ASSERT_TRUE(Generate()) << Error() << output_;
EXPECT_INST(R"(
%4 = OpLabel
OpSelectionMerge %8 None
OpSwitch %int_42 %5 1 %5 2 %7
%5 = OpLabel
OpBranch %8
%7 = OpLabel
OpBranch %8
%8 = OpLabel
%9 = OpPhi %int %int_10 %5 %int_20 %7
OpReturnValue %9
OpFunctionEnd
)");
}
TEST_F(SpirvWriterTest, Switch_Phi_SingleValue_CaseReturn) {
auto* func = b.Function("foo", ty.i32());
b.Append(func->Block(), [&] {
auto* s = b.Switch(42_i);
s->SetResults(b.InstructionResult(ty.i32()));
auto* case_a = b.Case(s, Vector{core::ir::Switch::CaseSelector{b.Constant(1_i)},
core::ir::Switch::CaseSelector{nullptr}});
b.Append(case_a, [&] { //
b.Return(func, 10_i);
});
auto* case_b = b.Case(s, Vector{core::ir::Switch::CaseSelector{b.Constant(2_i)}});
b.Append(case_b, [&] { //
b.ExitSwitch(s, 20_i);
});
b.Return(func, s);
});
ASSERT_TRUE(Generate()) << Error() << output_;
EXPECT_INST(R"(
%4 = OpLabel
%return_value = OpVariable %_ptr_Function_int Function %7
%continue_execution = OpVariable %_ptr_Function_bool Function
OpStore %continue_execution %true
OpSelectionMerge %15 None
OpSwitch %int_42 %12 1 %12 2 %14
%12 = OpLabel
OpStore %continue_execution %false
OpStore %return_value %int_10
OpBranch %15
%14 = OpLabel
OpBranch %15
%15 = OpLabel
%18 = OpPhi %int %19 %12 %int_20 %14
%21 = OpLoad %bool %continue_execution
OpSelectionMerge %22 None
OpBranchConditional %21 %23 %22
%23 = OpLabel
OpStore %return_value %18
OpBranch %22
%22 = OpLabel
%24 = OpLoad %int %return_value
OpReturnValue %24
OpFunctionEnd
)");
}
TEST_F(SpirvWriterTest, Switch_Phi_MultipleValue_0) {
auto* func = b.Function("foo", ty.i32());
b.Append(func->Block(), [&] {
auto* s = b.Switch(42_i);
s->SetResults(b.InstructionResult(ty.i32()), b.InstructionResult(ty.bool_()));
auto* case_a = b.Case(s, Vector{core::ir::Switch::CaseSelector{b.Constant(1_i)},
core::ir::Switch::CaseSelector{nullptr}});
b.Append(case_a, [&] { //
b.ExitSwitch(s, 10_i, true);
});
auto* case_b = b.Case(s, Vector{core::ir::Switch::CaseSelector{b.Constant(2_i)}});
b.Append(case_b, [&] { //
b.ExitSwitch(s, 20_i, false);
});
b.Return(func, s->Result(0));
});
ASSERT_TRUE(Generate()) << Error() << output_;
EXPECT_INST(R"(
%4 = OpLabel
OpSelectionMerge %8 None
OpSwitch %int_42 %5 1 %5 2 %7
%5 = OpLabel
OpBranch %8
%7 = OpLabel
OpBranch %8
%8 = OpLabel
%9 = OpPhi %int %int_10 %5 %int_20 %7
%13 = OpPhi %bool %true %5 %false %7
OpReturnValue %9
OpFunctionEnd
)");
}
TEST_F(SpirvWriterTest, Switch_Phi_MultipleValue_1) {
auto* func = b.Function("foo", ty.bool_());
b.Append(func->Block(), [&] {
auto* s = b.Switch(b.Constant(42_i));
s->SetResults(b.InstructionResult(ty.i32()), b.InstructionResult(ty.bool_()));
auto* case_a = b.Case(s, Vector{core::ir::Switch::CaseSelector{b.Constant(1_i)},
core::ir::Switch::CaseSelector{nullptr}});
b.Append(case_a, [&] { //
b.ExitSwitch(s, 10_i, true);
});
auto* case_b = b.Case(s, Vector{core::ir::Switch::CaseSelector{b.Constant(2_i)}});
b.Append(case_b, [&] { //
b.ExitSwitch(s, 20_i, false);
});
b.Return(func, s->Result(1));
});
ASSERT_TRUE(Generate()) << Error() << output_;
EXPECT_INST(R"(
%4 = OpLabel
OpSelectionMerge %9 None
OpSwitch %int_42 %5 1 %5 2 %8
%5 = OpLabel
OpBranch %9
%8 = OpLabel
OpBranch %9
%9 = OpLabel
%10 = OpPhi %int %int_10 %5 %int_20 %8
%13 = OpPhi %bool %true %5 %false %8
OpReturnValue %13
OpFunctionEnd
)");
}
TEST_F(SpirvWriterTest, Switch_Phi_NestedIf) {
auto* func = b.Function("foo", ty.i32());
b.Append(func->Block(), [&] {
auto* s = b.Switch(42_i);
s->SetResults(b.InstructionResult(ty.i32()));
auto* case_a = b.Case(s, Vector{core::ir::Switch::CaseSelector{b.Constant(1_i)},
core::ir::Switch::CaseSelector{nullptr}});
b.Append(case_a, [&] { //
auto* inner = b.If(true);
inner->SetResults(b.InstructionResult(ty.i32()));
b.Append(inner->True(), [&] { //
b.ExitIf(inner, 10_i);
});
b.Append(inner->False(), [&] { //
b.ExitIf(inner, 20_i);
});
b.ExitSwitch(s, inner->Result());
});
auto* case_b = b.Case(s, Vector{core::ir::Switch::CaseSelector{b.Constant(2_i)}});
b.Append(case_b, [&] { //
b.ExitSwitch(s, 20_i);
});
b.Return(func, s);
});
ASSERT_TRUE(Generate()) << Error() << output_;
EXPECT_INST(R"(
%4 = OpLabel
OpSelectionMerge %8 None
OpSwitch %int_42 %5 1 %5 2 %7
%5 = OpLabel
OpSelectionMerge %9 None
OpBranchConditional %true %10 %11
%10 = OpLabel
OpBranch %9
%11 = OpLabel
OpBranch %9
%9 = OpLabel
%14 = OpPhi %int %int_10 %10 %int_20 %11
OpBranch %8
%7 = OpLabel
OpBranch %8
%8 = OpLabel
%17 = OpPhi %int %int_20 %7 %14 %9
OpReturnValue %17
OpFunctionEnd
)");
}
TEST_F(SpirvWriterTest, Switch_Phi_NestedSwitch) {
auto* func = b.Function("foo", ty.i32());
b.Append(func->Block(), [&] {
auto* outer = b.Switch(42_i);
outer->SetResults(b.InstructionResult(ty.i32()));
auto* case_a = b.Case(outer, Vector{core::ir::Switch::CaseSelector{b.Constant(1_i)},
core::ir::Switch::CaseSelector{nullptr}});
b.Append(case_a, [&] { //
auto* inner = b.Switch(42_i);
auto* case_inner = b.Case(inner, Vector{core::ir::Switch::CaseSelector{b.Constant(2_i)},
core::ir::Switch::CaseSelector{nullptr}});
b.Append(case_inner, [&] { //
b.ExitSwitch(inner);
});
b.ExitSwitch(outer, 10_i);
});
auto* case_b = b.Case(outer, Vector{core::ir::Switch::CaseSelector{b.Constant(2_i)}});
b.Append(case_b, [&] { //
b.ExitSwitch(outer, 20_i);
});
b.Return(func, outer);
});
ASSERT_TRUE(Generate()) << Error() << output_;
EXPECT_INST(R"(
%4 = OpLabel
OpSelectionMerge %8 None
OpSwitch %int_42 %5 1 %5 2 %7
%5 = OpLabel
OpSelectionMerge %10 None
OpSwitch %int_42 %9 2 %9
%9 = OpLabel
OpBranch %10
%10 = OpLabel
OpBranch %8
%7 = OpLabel
OpBranch %8
%8 = OpLabel
%11 = OpPhi %int %int_20 %7 %int_10 %10
OpReturnValue %11
OpFunctionEnd
)");
}
} // namespace
} // namespace tint::spirv::writer