Add OpSwitch

This commit is contained in:
ReinUsesLisp 2019-01-05 23:58:43 -03:00
parent e7971b4451
commit 2b0a59d890
2 changed files with 21 additions and 0 deletions

View file

@ -192,6 +192,11 @@ class Module {
std::uint32_t true_weight = 0, std::uint32_t true_weight = 0,
std::uint32_t false_weight = 0); std::uint32_t false_weight = 0);
/// Multi-way branch to one of the operand label.
Id OpSwitch(Id selector, Id default_label,
const std::vector<Literal>& literals,
const std::vector<Id>& labels);
/// Returns with no value from a function with void return type. /// Returns with no value from a function with void return type.
Id OpReturn(); Id OpReturn();

View file

@ -7,6 +7,7 @@
#include "common_types.h" #include "common_types.h"
#include "op.h" #include "op.h"
#include "sirit/sirit.h" #include "sirit/sirit.h"
#include <cassert>
#include <vector> #include <vector>
namespace Sirit { namespace Sirit {
@ -51,6 +52,21 @@ Id Module::OpBranchConditional(Id condition, Id true_label, Id false_label,
return AddCode(std::move(op)); return AddCode(std::move(op));
} }
Id Module::OpSwitch(Id selector, Id default_label,
const std::vector<Literal>& literals,
const std::vector<Id>& labels) {
const std::size_t size = literals.size();
assert(literals.size() == labels.size());
auto op{std::make_unique<Op>(spv::Op::OpSwitch)};
op->Add(selector);
op->Add(default_label);
for (std::size_t i = 0; i < size; ++i) {
op->Add(literals[i]);
op->Add(labels[i]);
}
return AddCode(std::move(op));
}
Id Module::OpReturn() { return AddCode(spv::Op::OpReturn); } Id Module::OpReturn() { return AddCode(spv::Op::OpReturn); }
Id Module::OpReturnValue(Id value) { Id Module::OpReturnValue(Id value) {