Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
8 changes: 5 additions & 3 deletions src/ir/names.h
Original file line number Diff line number Diff line change
Expand Up @@ -49,12 +49,14 @@ inline void ensureNames(Function* func) {
// name will begin there. This can be used to avoid trying the same 0,1,2,..
// etc. names each time (which could lead to quadratic behavior in certain
// cases).
inline Name
getValidName(Name root, std::function<bool(Name)> check, Index hint = 0) {
inline Name getValidName(Name root,
std::function<bool(Name)> check,
Index hint = 0,
std::string separator = "_") {
if (check(root)) {
return root;
}
auto prefixed = std::string(root.str) + '_';
auto prefixed = std::string(root.str) + separator;
Index num = hint;
while (1) {
auto name = prefixed + std::to_string(num);
Expand Down
10 changes: 7 additions & 3 deletions src/wasm-ir-builder.h
Original file line number Diff line number Diff line change
Expand Up @@ -461,9 +461,13 @@ class IRBuilder : public UnifiedExpressionVisitor<IRBuilder, Result<>> {
std::unordered_map<Name, std::vector<Index>> labelDepths;

Name makeFresh(Name label) {
return Names::getValidName(label, [&](Name candidate) {
return labelDepths.insert({candidate, {}}).second;
});
return Names::getValidName(
label,
[&](Name candidate) {
return labelDepths.insert({candidate, {}}).second;
},
0,
"");
}

void pushScope(ScopeCtx scope) {
Expand Down
6 changes: 5 additions & 1 deletion src/wasm/wasm-ir-builder.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -1035,7 +1035,11 @@ Result<Name> IRBuilder::getLabelName(Index label) {
auto& scopeLabel = (*scope)->label;
if (!scopeLabel) {
// The scope does not already have a name, so we need to create one.
scopeLabel = makeFresh("label");
if ((*scope)->getBlock()) {
scopeLabel = makeFresh("block");
} else {
scopeLabel = makeFresh("label");
}
}
(*scope)->labelUsed = true;
return scopeLabel;
Expand Down
76 changes: 38 additions & 38 deletions test/lit/wat-kitchen-sink.wast
Original file line number Diff line number Diff line change
Expand Up @@ -2122,8 +2122,8 @@
;; CHECK-NEXT: (block $l
;; CHECK-NEXT: (try $__delegate__l
;; CHECK-NEXT: (do
;; CHECK-NEXT: (block $l_0
;; CHECK-NEXT: (block $l_1
;; CHECK-NEXT: (block $l0
;; CHECK-NEXT: (block $l1
;; CHECK-NEXT: (try
;; CHECK-NEXT: (do
;; CHECK-NEXT: (nop)
Expand All @@ -2149,13 +2149,13 @@
;; CHECK-NEXT: (block $l
;; CHECK-NEXT: (try $__delegate__l
;; CHECK-NEXT: (do
;; CHECK-NEXT: (block $l_0
;; CHECK-NEXT: (block $l0
;; CHECK-NEXT: (try
;; CHECK-NEXT: (do
;; CHECK-NEXT: (nop)
;; CHECK-NEXT: )
;; CHECK-NEXT: (catch $empty
;; CHECK-NEXT: (block $l_1
;; CHECK-NEXT: (block $l1
;; CHECK-NEXT: (try
;; CHECK-NEXT: (do
;; CHECK-NEXT: (nop)
Expand Down Expand Up @@ -2185,13 +2185,13 @@
;; CHECK-NEXT: (block $l
;; CHECK-NEXT: (try $__delegate__l
;; CHECK-NEXT: (do
;; CHECK-NEXT: (block $l_0
;; CHECK-NEXT: (block $l0
;; CHECK-NEXT: (try
;; CHECK-NEXT: (do
;; CHECK-NEXT: (nop)
;; CHECK-NEXT: )
;; CHECK-NEXT: (catch_all
;; CHECK-NEXT: (block $l_1
;; CHECK-NEXT: (block $l1
;; CHECK-NEXT: (try
;; CHECK-NEXT: (do
;; CHECK-NEXT: (nop)
Expand Down Expand Up @@ -2617,14 +2617,14 @@
;; CHECK-NEXT: (block $l
;; CHECK-NEXT: (br $l)
;; CHECK-NEXT: )
;; CHECK-NEXT: (loop $l_0
;; CHECK-NEXT: (br $l_0)
;; CHECK-NEXT: (loop $l0
;; CHECK-NEXT: (br $l0)
;; CHECK-NEXT: )
;; CHECK-NEXT: (block $l_1
;; CHECK-NEXT: (block $l1
;; CHECK-NEXT: (if
;; CHECK-NEXT: (i32.const 0)
;; CHECK-NEXT: (then
;; CHECK-NEXT: (br $l_1)
;; CHECK-NEXT: (br $l1)
;; CHECK-NEXT: )
;; CHECK-NEXT: )
;; CHECK-NEXT: )
Expand All @@ -2645,17 +2645,17 @@
;; CHECK: (func $label-shadowed (type $void)
;; CHECK-NEXT: (block $l
;; CHECK-NEXT: (br $l)
;; CHECK-NEXT: (loop $l_0
;; CHECK-NEXT: (br $l_0)
;; CHECK-NEXT: (block $l_1
;; CHECK-NEXT: (loop $l0
;; CHECK-NEXT: (br $l0)
;; CHECK-NEXT: (block $l1
;; CHECK-NEXT: (if
;; CHECK-NEXT: (i32.const 0)
;; CHECK-NEXT: (then
;; CHECK-NEXT: (br $l_1)
;; CHECK-NEXT: (br $l1)
;; CHECK-NEXT: )
;; CHECK-NEXT: )
;; CHECK-NEXT: )
;; CHECK-NEXT: (br $l_0)
;; CHECK-NEXT: (br $l0)
;; CHECK-NEXT: )
;; CHECK-NEXT: (br $l)
;; CHECK-NEXT: )
Expand All @@ -2676,14 +2676,14 @@
)

;; CHECK: (func $label-index (type $void)
;; CHECK-NEXT: (block $label_1
;; CHECK-NEXT: (block $label
;; CHECK-NEXT: (block $label_0
;; CHECK-NEXT: (block $block1
;; CHECK-NEXT: (block $block
;; CHECK-NEXT: (block $block0
;; CHECK-NEXT: (block $l
;; CHECK-NEXT: (br $label)
;; CHECK-NEXT: (br $label_0)
;; CHECK-NEXT: (br $block)
;; CHECK-NEXT: (br $block0)
;; CHECK-NEXT: (br $l)
;; CHECK-NEXT: (br $label_1)
;; CHECK-NEXT: (br $block1)
;; CHECK-NEXT: )
;; CHECK-NEXT: )
;; CHECK-NEXT: )
Expand Down Expand Up @@ -2810,8 +2810,8 @@
)

;; CHECK: (func $br_if-index (type $void)
;; CHECK-NEXT: (block $label
;; CHECK-NEXT: (br_if $label
;; CHECK-NEXT: (block $block
;; CHECK-NEXT: (br_if $block
;; CHECK-NEXT: (i32.const 0)
;; CHECK-NEXT: )
;; CHECK-NEXT: )
Expand Down Expand Up @@ -2918,11 +2918,11 @@
)

;; CHECK: (func $br-table-index (type $void)
;; CHECK-NEXT: (block $label
;; CHECK-NEXT: (block $block
;; CHECK-NEXT: (block $l
;; CHECK-NEXT: (block $label_1
;; CHECK-NEXT: (block $label_0
;; CHECK-NEXT: (br_table $label $l $label_0 $label_1
;; CHECK-NEXT: (block $block1
;; CHECK-NEXT: (block $block0
;; CHECK-NEXT: (br_table $block $l $block0 $block1
;; CHECK-NEXT: (i32.const 0)
;; CHECK-NEXT: )
;; CHECK-NEXT: )
Expand Down Expand Up @@ -4098,8 +4098,8 @@

;; CHECK: (func $br-on-non-null (type $8) (param $0 anyref)
;; CHECK-NEXT: (drop
;; CHECK-NEXT: (block $label (result (ref any))
;; CHECK-NEXT: (br_on_non_null $label
;; CHECK-NEXT: (block $block (result (ref any))
;; CHECK-NEXT: (br_on_non_null $block
;; CHECK-NEXT: (local.get $0)
;; CHECK-NEXT: )
;; CHECK-NEXT: (unreachable)
Expand All @@ -4117,10 +4117,10 @@

;; CHECK: (func $br-on-cast (type $8) (param $0 anyref)
;; CHECK-NEXT: (drop
;; CHECK-NEXT: (block $label (result i31ref)
;; CHECK-NEXT: (block $block (result i31ref)
;; CHECK-NEXT: (drop
;; CHECK-NEXT: (block (result (ref any))
;; CHECK-NEXT: (br_on_cast $label anyref i31ref
;; CHECK-NEXT: (br_on_cast $block anyref i31ref
;; CHECK-NEXT: (local.get $0)
;; CHECK-NEXT: )
;; CHECK-NEXT: )
Expand All @@ -4142,10 +4142,10 @@

;; CHECK: (func $br-on-cast-fail (type $8) (param $0 anyref)
;; CHECK-NEXT: (drop
;; CHECK-NEXT: (block $label (result (ref any))
;; CHECK-NEXT: (block $block (result (ref any))
;; CHECK-NEXT: (drop
;; CHECK-NEXT: (block (result i31ref)
;; CHECK-NEXT: (br_on_cast_fail $label anyref i31ref
;; CHECK-NEXT: (br_on_cast_fail $block anyref i31ref
;; CHECK-NEXT: (local.get $0)
;; CHECK-NEXT: )
;; CHECK-NEXT: )
Expand Down Expand Up @@ -5121,14 +5121,14 @@

;; CHECK: (func $resume (type $91) (param $ct (ref $simple-cont))
;; CHECK-NEXT: (local $f f32)
;; CHECK-NEXT: (block $label_1
;; CHECK-NEXT: (block $label
;; CHECK-NEXT: (block
;; CHECK-NEXT: (drop
;; CHECK-NEXT: (block $label (result (ref $to-f32-cont))
;; CHECK-NEXT: (block $block (result (ref $to-f32-cont))
;; CHECK-NEXT: (tuple.drop 3
;; CHECK-NEXT: (block $label_0 (type $36) (result i32 i64 (ref null $simple-cont))
;; CHECK-NEXT: (block $block0 (type $36) (result i32 i64 (ref null $simple-cont))
;; CHECK-NEXT: (local.set $f
;; CHECK-NEXT: (resume $simple-cont (tag $empty $label) (tag $tag-pair-to-pair $label_0)
;; CHECK-NEXT: (resume $simple-cont (tag $empty $block) (tag $tag-pair-to-pair $block0)
;; CHECK-NEXT: (i32.const 0)
;; CHECK-NEXT: (i64.const 1)
;; CHECK-NEXT: (local.get $ct)
Expand All @@ -5140,7 +5140,7 @@
;; CHECK-NEXT: (unreachable)
;; CHECK-NEXT: )
;; CHECK-NEXT: )
;; CHECK-NEXT: (br $label_1)
;; CHECK-NEXT: (br $label)
;; CHECK-NEXT: )
;; CHECK-NEXT: )
;; CHECK-NEXT: )
Expand Down