summaryrefslogtreecommitdiff
path: root/src/passes/opt-utils.h
diff options
context:
space:
mode:
Diffstat (limited to 'src/passes/opt-utils.h')
-rw-r--r--src/passes/opt-utils.h7
1 files changed, 2 insertions, 5 deletions
diff --git a/src/passes/opt-utils.h b/src/passes/opt-utils.h
index 27a3c5c25..b333779f7 100644
--- a/src/passes/opt-utils.h
+++ b/src/passes/opt-utils.h
@@ -20,6 +20,7 @@
#include <functional>
#include <unordered_set>
+#include <ir/element-utils.h>
#include <pass.h>
#include <wasm.h>
@@ -86,11 +87,7 @@ inline void replaceFunctions(PassRunner* runner,
// replace direct calls
FunctionRefReplacer(maybeReplace).run(runner, &module);
// replace in table
- for (auto& segment : module.elementSegments) {
- for (auto& name : segment->data) {
- maybeReplace(name);
- }
- }
+ ElementUtils::iterAllElementFunctionNames(&module, maybeReplace);
// replace in start
if (module.start.is()) {