Add SIMD load and store support to Memory64Lowering pass (#8303)

These are handled with a different visitor than the scalar loads and
stores.

Author: dschuff

src/passes/Memory64Lowering.cpp

@@ -87,6 +87,12 @@ struct Memory64Lowering : public WalkerPass<PostWalker<Memory64Lowering>> {
 
   void visitStore(Store* curr) { wrapAddress64(curr->ptr, curr->memory); }
 
+  void visitSIMDLoad(SIMDLoad* curr) { wrapAddress64(curr->ptr, curr->memory); }
+
+  void visitSIMDLoadStoreLane(SIMDLoadStoreLane* curr) {
+    wrapAddress64(curr->ptr, curr->memory);
+  }
+
   void visitMemorySize(MemorySize* curr) {
     auto& module = *getModule();
     auto* memory = module.getMemory(curr->memory);

test/lit/passes/memory64-lowering.wast

@@ -1,13 +1,15 @@
 ;; NOTE: Assertions have been generated by update_lit_checks.py --all-items and should not be edited.
 
-;; RUN: foreach %s %t wasm-opt --memory64-lowering --enable-memory64 --enable-threads --enable-bulk-memory --enable-reference-types -S -o - | filecheck %s
+;; RUN: foreach %s %t wasm-opt --memory64-lowering --enable-simd --enable-memory64 --enable-threads --enable-bulk-memory --enable-reference-types -S -o - | filecheck %s
 
 ;; Check the --table64-lowering alias
-;; RUN: foreach %s %t wasm-opt --table64-lowering --enable-memory64 --enable-threads --enable-bulk-memory --enable-reference-types -S -o - | filecheck %s
+;; RUN: foreach %s %t wasm-opt --table64-lowering --enable-simd --enable-memory64 --enable-threads --enable-bulk-memory --enable-reference-types -S -o - | filecheck %s
 
 (module
  ;; CHECK:      (type $0 (func))
 
+ ;; CHECK:      (type $1 (func (param i64 v128)))
+
  ;; CHECK:      (import "env" "__memory_base" (global $__memory_base i64))
  (import "env" "__memory_base" (global $__memory_base i64))
  ;; CHECK:      (import "env" "__memory_base32" (global $__memory_base32 i32))
@@ -292,7 +294,67 @@
   (memory.fill (i64.const 1) (i32.const 2) (i64.const 3))
   (memory.copy (i64.const 1) (i64.const 2) (i64.const 3))
  )
-)
+
+  ;; CHECK:      (func $simd (param $ptr i64) (param $val v128)
+  ;; CHECK-NEXT:  (v128.store
+  ;; CHECK-NEXT:   (i32.wrap_i64
+  ;; CHECK-NEXT:    (local.get $ptr)
+  ;; CHECK-NEXT:   )
+  ;; CHECK-NEXT:   (local.get $val)
+  ;; CHECK-NEXT:  )
+  ;; CHECK-NEXT:  (v128.store8_lane 0
+  ;; CHECK-NEXT:   (i32.wrap_i64
+  ;; CHECK-NEXT:    (local.get $ptr)
+  ;; CHECK-NEXT:   )
+  ;; CHECK-NEXT:   (local.get $val)
+  ;; CHECK-NEXT:  )
+  ;; CHECK-NEXT:  (drop
+  ;; CHECK-NEXT:   (v128.load
+  ;; CHECK-NEXT:    (i32.wrap_i64
+  ;; CHECK-NEXT:     (local.get $ptr)
+  ;; CHECK-NEXT:    )
+  ;; CHECK-NEXT:   )
+  ;; CHECK-NEXT:  )
+  ;; CHECK-NEXT:  (drop
+  ;; CHECK-NEXT:   (v128.load8_lane 0
+  ;; CHECK-NEXT:    (i32.wrap_i64
+  ;; CHECK-NEXT:     (local.get $ptr)
+  ;; CHECK-NEXT:    )
+  ;; CHECK-NEXT:    (local.get $val)
+  ;; CHECK-NEXT:   )
+  ;; CHECK-NEXT:  )
+  ;; CHECK-NEXT:  (drop
+  ;; CHECK-NEXT:   (v128.load8_splat
+  ;; CHECK-NEXT:    (i32.wrap_i64
+  ;; CHECK-NEXT:     (local.get $ptr)
+  ;; CHECK-NEXT:    )
+  ;; CHECK-NEXT:   )
+  ;; CHECK-NEXT:  )
+  ;; CHECK-NEXT:  (drop
+  ;; CHECK-NEXT:   (v128.load8x8_s
+  ;; CHECK-NEXT:    (i32.wrap_i64
+  ;; CHECK-NEXT:     (local.get $ptr)
+  ;; CHECK-NEXT:    )
+  ;; CHECK-NEXT:   )
+  ;; CHECK-NEXT:  )
+  ;; CHECK-NEXT:  (drop
+  ;; CHECK-NEXT:   (v128.load32_zero
+  ;; CHECK-NEXT:    (i32.wrap_i64
+  ;; CHECK-NEXT:     (local.get $ptr)
+  ;; CHECK-NEXT:    )
+  ;; CHECK-NEXT:   )
+  ;; CHECK-NEXT:  )
+  ;; CHECK-NEXT: )
+  (func $simd (param $ptr i64) (param $val v128)
+   (v128.store (local.get $ptr) (local.get $val))
+   (v128.store8_lane 0 (local.get $ptr) (local.get $val))
+   (drop (v128.load (local.get $ptr)))
+   (drop (v128.load8_lane 0 (local.get $ptr) (local.get $val)))
+   (drop (v128.load8_splat (local.get $ptr)))
+   (drop (v128.load8x8_s (local.get $ptr)))
+   (drop (v128.load32_zero (local.get $ptr)))
+  )
+ )
 
 (module
   ;; CHECK:      (memory $0 1 65536)