blob: 7912a7d927ae6666d595daa5d99759b2dd81ad41 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
|
/*
* Copyright 2018 WebAssembly Community Group participants
*
* 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.
*/
#ifndef wasm_passes_opt_utils_h
#define wasm_passes_opt_utils_h
#include <functional>
#include <unordered_set>
#include <pass.h>
#include <wasm.h>
namespace wasm {
namespace OptUtils {
// Run useful optimizations after inlining new code into a set
// of functions.
inline void optimizeAfterInlining(std::unordered_set<Function*>& funcs,
Module* module,
PassRunner* parentRunner) {
// save the full list of functions on the side
std::vector<std::unique_ptr<Function>> all;
all.swap(module->functions);
module->updateMaps();
for (auto& func : funcs) {
module->addFunction(func);
}
PassRunner runner(module, parentRunner->options);
runner.setIsNested(true);
runner.setValidateGlobally(false); // not a full valid module
// this is especially useful after inlining
runner.add("precompute-propagate");
runner.addDefaultFunctionOptimizationPasses(); // do all the usual stuff
runner.run();
// restore all the funcs
for (auto& func : module->functions) {
func.release();
}
all.swap(module->functions);
module->updateMaps();
}
struct FunctionRefReplacer
: public WalkerPass<PostWalker<FunctionRefReplacer>> {
bool isFunctionParallel() override { return true; }
using MaybeReplace = std::function<void(Name&)>;
FunctionRefReplacer(MaybeReplace maybeReplace) : maybeReplace(maybeReplace) {}
FunctionRefReplacer* create() override {
return new FunctionRefReplacer(maybeReplace);
}
void visitCall(Call* curr) { maybeReplace(curr->target); }
void visitRefFunc(RefFunc* curr) { maybeReplace(curr->func); }
private:
MaybeReplace maybeReplace;
};
inline void replaceFunctions(PassRunner* runner,
Module& module,
const std::map<Name, Name>& replacements) {
auto maybeReplace = [&](Name& name) {
auto iter = replacements.find(name);
if (iter != replacements.end()) {
name = iter->second;
}
};
// replace direct calls
FunctionRefReplacer(maybeReplace).run(runner, &module);
// replace in table
for (auto& segment : module.table.segments) {
for (auto& name : segment.data) {
maybeReplace(name);
}
}
// replace in start
if (module.start.is()) {
maybeReplace(module.start);
}
// replace in exports
for (auto& exp : module.exports) {
maybeReplace(exp->value);
}
}
} // namespace OptUtils
} // namespace wasm
#endif // wasm_passes_opt_utils_h
|
