Support atomic struct accessors (#7155)

Implement support for both sequentially consistent and acquire-release
variants of `struct.atomic.get` and `struct.atomic.set`, as proposed by
shared-everything-threads. Introduce a new `MemoryOrdering` enum for
describing different levels of atomicity (or the lack thereof). This new
enum should eventually be adopted by linear memory atomic accessors as
well to support acquire-release semantics, but for now just use it in
`StructGet` and `StructSet`.

In addition to implementing parsing and emitting for the instructions,
validate that shared-everything is enabled to use them, mark them as
having synchronization side effects, and lightly optimize them by
relaxing acquire-release accesses to non-shared structs to normal,
unordered accesses. This is valid because such accesses cannot possibly
synchronize with other threads. Also update Precompute to avoid
optimizing out synchronization points.

There are probably other passes that need to be updated to avoid
incorrectly optimizing synchronizing accesses, but identifying and
fixing them is left as future work.

1 files changed, 149 insertions, 0 deletions

diff --git a/test/lit/basic/gc-atomics.wast b/test/lit/basic/gc-atomics.wast
new file mode 100644
index 000000000..c454b4c99
--- /dev/null
+++ b/test/lit/basic/gc-atomics.wast
@@ -0,0 +1,149 @@
+;; NOTE: Assertions have been generated by update_lit_checks.py and should not be edited.
+
+;; RUN: wasm-opt -all %s -S -o - | filecheck %s
+;; RUN: wasm-opt -all %s --roundtrip -S -o - | filecheck %s
+
+(module
+  ;; CHECK:      (type $struct (struct (field (mut i32))))
+  (type $struct (struct (field (mut i32))))
+  ;; CHECK:      (type $packed (struct (field (mut i8))))
+  (type $packed (struct (field (mut i8))))
+
+  ;; CHECK:      (func $get (type $3) (param $0 (ref null $struct)) (result i32)
+  ;; CHECK-NEXT:  (struct.atomic.get $struct 0
+  ;; CHECK-NEXT:   (local.get $0)
+  ;; CHECK-NEXT:  )
+  ;; CHECK-NEXT: )
+  (func $get (param (ref null $struct)) (result i32)
+    (struct.atomic.get $struct 0
+      (local.get 0)
+    )
+  )
+
+  ;; CHECK:      (func $get-seqcst (type $3) (param $0 (ref null $struct)) (result i32)
+  ;; CHECK-NEXT:  (struct.atomic.get $struct 0
+  ;; CHECK-NEXT:   (local.get $0)
+  ;; CHECK-NEXT:  )
+  ;; CHECK-NEXT: )
+  (func $get-seqcst (param (ref null $struct)) (result i32)
+    (struct.atomic.get seqcst $struct 0
+      (local.get 0)
+    )
+  )
+
+  ;; CHECK:      (func $get-acqrel (type $3) (param $0 (ref null $struct)) (result i32)
+  ;; CHECK-NEXT:  (struct.atomic.get acqrel $struct 0
+  ;; CHECK-NEXT:   (local.get $0)
+  ;; CHECK-NEXT:  )
+  ;; CHECK-NEXT: )
+  (func $get-acqrel (param (ref null $struct)) (result i32)
+    (struct.atomic.get acqrel $struct 0
+      (local.get 0)
+    )
+  )
+
+  ;; CHECK:      (func $get-s (type $2) (param $0 (ref null $packed)) (result i32)
+  ;; CHECK-NEXT:  (struct.atomic.get_s $packed 0
+  ;; CHECK-NEXT:   (local.get $0)
+  ;; CHECK-NEXT:  )
+  ;; CHECK-NEXT: )
+  (func $get-s (param (ref null $packed)) (result i32)
+    (struct.atomic.get_s $packed 0
+      (local.get 0)
+    )
+  )
+
+  ;; CHECK:      (func $get-s-seqcst (type $2) (param $0 (ref null $packed)) (result i32)
+  ;; CHECK-NEXT:  (struct.atomic.get_s $packed 0
+  ;; CHECK-NEXT:   (local.get $0)
+  ;; CHECK-NEXT:  )
+  ;; CHECK-NEXT: )
+  (func $get-s-seqcst (param (ref null $packed)) (result i32)
+    (struct.atomic.get_s seqcst $packed 0
+      (local.get 0)
+    )
+  )
+
+  ;; CHECK:      (func $get-s-acqrel (type $2) (param $0 (ref null $packed)) (result i32)
+  ;; CHECK-NEXT:  (struct.atomic.get_s acqrel $packed 0
+  ;; CHECK-NEXT:   (local.get $0)
+  ;; CHECK-NEXT:  )
+  ;; CHECK-NEXT: )
+  (func $get-s-acqrel (param (ref null $packed)) (result i32)
+    (struct.atomic.get_s acqrel $packed 0
+      (local.get 0)
+    )
+  )
+
+  ;; CHECK:      (func $get-u (type $2) (param $0 (ref null $packed)) (result i32)
+  ;; CHECK-NEXT:  (struct.atomic.get_u $packed 0
+  ;; CHECK-NEXT:   (local.get $0)
+  ;; CHECK-NEXT:  )
+  ;; CHECK-NEXT: )
+  (func $get-u (param (ref null $packed)) (result i32)
+    (struct.atomic.get_u $packed 0
+      (local.get 0)
+    )
+  )
+
+  ;; CHECK:      (func $get-u-seqcst (type $2) (param $0 (ref null $packed)) (result i32)
+  ;; CHECK-NEXT:  (struct.atomic.get_u $packed 0
+  ;; CHECK-NEXT:   (local.get $0)
+  ;; CHECK-NEXT:  )
+  ;; CHECK-NEXT: )
+  (func $get-u-seqcst (param (ref null $packed)) (result i32)
+    (struct.atomic.get_u seqcst $packed 0
+      (local.get 0)
+    )
+  )
+
+  ;; CHECK:      (func $get-u-acqrel (type $2) (param $0 (ref null $packed)) (result i32)
+  ;; CHECK-NEXT:  (struct.atomic.get_u acqrel $packed 0
+  ;; CHECK-NEXT:   (local.get $0)
+  ;; CHECK-NEXT:  )
+  ;; CHECK-NEXT: )
+  (func $get-u-acqrel (param (ref null $packed)) (result i32)
+    (struct.atomic.get_u acqrel $packed 0
+      (local.get 0)
+    )
+  )
+
+  ;; CHECK:      (func $set (type $4) (param $0 (ref null $struct))
+  ;; CHECK-NEXT:  (struct.atomic.set $struct 0
+  ;; CHECK-NEXT:   (local.get $0)
+  ;; CHECK-NEXT:   (i32.const 0)
+  ;; CHECK-NEXT:  )
+  ;; CHECK-NEXT: )
+  (func $set (param (ref null $struct))
+    (struct.atomic.set $struct 0
+      (local.get 0)
+      (i32.const 0)
+    )
+  )
+
+  ;; CHECK:      (func $set-seqcst (type $4) (param $0 (ref null $struct))
+  ;; CHECK-NEXT:  (struct.atomic.set $struct 0
+  ;; CHECK-NEXT:   (local.get $0)
+  ;; CHECK-NEXT:   (i32.const 0)
+  ;; CHECK-NEXT:  )
+  ;; CHECK-NEXT: )
+  (func $set-seqcst (param (ref null $struct))
+    (struct.atomic.set seqcst $struct 0
+      (local.get 0)
+      (i32.const 0)
+    )
+  )
+
+  ;; CHECK:      (func $set-acqrel (type $4) (param $0 (ref null $struct))
+  ;; CHECK-NEXT:  (struct.atomic.set acqrel $struct 0
+  ;; CHECK-NEXT:   (local.get $0)
+  ;; CHECK-NEXT:   (i32.const 0)
+  ;; CHECK-NEXT:  )
+  ;; CHECK-NEXT: )
+  (func $set-acqrel (param (ref null $struct))
+    (struct.atomic.set acqrel $struct 0
+      (local.get 0)
+      (i32.const 0)
+    )
+  )
+)