shader: Fix branches to visited virtual blocks

This commit is contained in:
ReinUsesLisp 2021-04-02 01:17:47 -03:00 committed by ameerj
parent d0a529683a
commit b4a5e767d0
2 changed files with 12 additions and 0 deletions

View file

@ -56,6 +56,8 @@ ArgType Arg(EmitContext& ctx, const IR::Value& arg) {
return arg.Label(); return arg.Label();
} else if constexpr (std::is_same_v<ArgType, IR::Attribute>) { } else if constexpr (std::is_same_v<ArgType, IR::Attribute>) {
return arg.Attribute(); return arg.Attribute();
} else if constexpr (std::is_same_v<ArgType, IR::Reg>) {
return arg.Reg();
} }
} }

View file

@ -486,6 +486,16 @@ Block* CFG::AddLabel(Block* block, Stack stack, Location pc, FunctionId function
} }
if (const auto it{function.blocks.find(pc, Compare{})}; it != function.blocks.end()) { if (const auto it{function.blocks.find(pc, Compare{})}; it != function.blocks.end()) {
// Block already exists and it has been visited // Block already exists and it has been visited
if (function.blocks.begin() != it) {
// Check if the previous node is the virtual variant of the label
// This won't exist if a virtual node is not needed or it hasn't been visited
// If it hasn't been visited and a virtual node is needed, this will still behave as
// expected because the node impersonated with its virtual node.
const auto prev{std::prev(it)};
if (it->begin.Virtual() == prev->begin) {
return &*prev;
}
}
return &*it; return &*it;
} }
Block* const new_block{block_pool.Create(Block{ Block* const new_block{block_pool.Create(Block{