1 | // Copyright (c) 2017 Google Inc. |
2 | // |
3 | // Licensed under the Apache License, Version 2.0 (the "License"); |
4 | // you may not use this file except in compliance with the License. |
5 | // You may obtain a copy of the License at |
6 | // |
7 | // http://www.apache.org/licenses/LICENSE-2.0 |
8 | // |
9 | // Unless required by applicable law or agreed to in writing, software |
10 | // distributed under the License is distributed on an "AS IS" BASIS, |
11 | // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
12 | // See the License for the specific language governing permissions and |
13 | // limitations under the License. |
14 | |
15 | #include "source/opt/local_redundancy_elimination.h" |
16 | |
17 | #include "source/opt/value_number_table.h" |
18 | |
19 | namespace spvtools { |
20 | namespace opt { |
21 | |
22 | Pass::Status LocalRedundancyEliminationPass::Process() { |
23 | bool modified = false; |
24 | ValueNumberTable vnTable(context()); |
25 | |
26 | for (auto& func : *get_module()) { |
27 | for (auto& bb : func) { |
28 | // Keeps track of all ids that contain a given value number. We keep |
29 | // track of multiple values because they could have the same value, but |
30 | // different decorations. |
31 | std::map<uint32_t, uint32_t> value_to_ids; |
32 | if (EliminateRedundanciesInBB(&bb, vnTable, &value_to_ids)) |
33 | modified = true; |
34 | } |
35 | } |
36 | return (modified ? Status::SuccessWithChange : Status::SuccessWithoutChange); |
37 | } |
38 | |
39 | bool LocalRedundancyEliminationPass::EliminateRedundanciesInBB( |
40 | BasicBlock* block, const ValueNumberTable& vnTable, |
41 | std::map<uint32_t, uint32_t>* value_to_ids) { |
42 | bool modified = false; |
43 | |
44 | auto func = [this, &vnTable, &modified, value_to_ids](Instruction* inst) { |
45 | if (inst->result_id() == 0) { |
46 | return; |
47 | } |
48 | |
49 | uint32_t value = vnTable.GetValueNumber(inst); |
50 | |
51 | if (value == 0) { |
52 | return; |
53 | } |
54 | |
55 | auto candidate = value_to_ids->insert({value, inst->result_id()}); |
56 | if (!candidate.second) { |
57 | context()->KillNamesAndDecorates(inst); |
58 | context()->ReplaceAllUsesWith(inst->result_id(), candidate.first->second); |
59 | context()->KillInst(inst); |
60 | modified = true; |
61 | } |
62 | }; |
63 | block->ForEachInst(func); |
64 | return modified; |
65 | } |
66 | } // namespace opt |
67 | } // namespace spvtools |
68 | |