WebAssembly: generate load/store

Summary: This handles all load/store operations that WebAssembly defines, and handles those necessary for C++ such as i1. I left a FIXME for outstanding features which aren't required for now.

Reviewers: sunfish

Subscribers: jfb, llvm-commits, dschuff
llvm-svn: 246500

5 files changed, 296 insertions, 0 deletions

diff --git a/llvm/test/CodeGen/WebAssembly/load-ext.ll b/llvm/test/CodeGen/WebAssembly/load-ext.ll
new file mode 100644
index 00000000000..b673cfb1919
--- /dev/null
+++ b/llvm/test/CodeGen/WebAssembly/load-ext.ll
@@ -0,0 +1,86 @@
+; RUN: llc < %s -asm-verbose=false | FileCheck %s
+
+; Test that extending loads are assembled properly.
+
+target datalayout = "e-p:32:32-i64:64-v128:8:128-n32:64-S128"
+target triple = "wasm32-unknown-unknown"
+
+; CHECK-LABEL: (func $sext_i8_i32
+; CHECK: (setlocal @1 (load_sx_i8_i32 @0))
+define i32 @sext_i8_i32(i8 *%p) {
+  %v = load i8, i8* %p
+  %e = sext i8 %v to i32
+  ret i32 %e
+}
+
+; CHECK-LABEL: (func $zext_i8_i32
+; CHECK: (setlocal @1 (load_zx_i8_i32 @0))
+define i32 @zext_i8_i32(i8 *%p) {
+  %v = load i8, i8* %p
+  %e = zext i8 %v to i32
+  ret i32 %e
+}
+
+; CHECK-LABEL: (func $sext_i16_i32
+; CHECK: (setlocal @1 (load_sx_i16_i32 @0))
+define i32 @sext_i16_i32(i16 *%p) {
+  %v = load i16, i16* %p
+  %e = sext i16 %v to i32
+  ret i32 %e
+}
+
+; CHECK-LABEL: (func $zext_i16_i32
+; CHECK: (setlocal @1 (load_zx_i16_i32 @0))
+define i32 @zext_i16_i32(i16 *%p) {
+  %v = load i16, i16* %p
+  %e = zext i16 %v to i32
+  ret i32 %e
+}
+
+; CHECK-LABEL: (func $sext_i8_i64
+; CHECK: (setlocal @1 (load_sx_i8_i64 @0))
+define i64 @sext_i8_i64(i8 *%p) {
+  %v = load i8, i8* %p
+  %e = sext i8 %v to i64
+  ret i64 %e
+}
+
+; CHECK-LABEL: (func $zext_i8_i64
+; CHECK: (setlocal @1 (load_zx_i8_i64 @0))
+define i64 @zext_i8_i64(i8 *%p) {
+  %v = load i8, i8* %p
+  %e = zext i8 %v to i64
+  ret i64 %e
+}
+
+; CHECK-LABEL: (func $sext_i16_i64
+; CHECK: (setlocal @1 (load_sx_i16_i64 @0))
+define i64 @sext_i16_i64(i16 *%p) {
+  %v = load i16, i16* %p
+  %e = sext i16 %v to i64
+  ret i64 %e
+}
+
+; CHECK-LABEL: (func $zext_i16_i64
+; CHECK: (setlocal @1 (load_zx_i16_i64 @0))
+define i64 @zext_i16_i64(i16 *%p) {
+  %v = load i16, i16* %p
+  %e = zext i16 %v to i64
+  ret i64 %e
+}
+
+; CHECK-LABEL: (func $sext_i32_i64
+; CHECK: (setlocal @1 (load_sx_i32_i64 @0))
+define i64 @sext_i32_i64(i32 *%p) {
+  %v = load i32, i32* %p
+  %e = sext i32 %v to i64
+  ret i64 %e
+}
+
+; CHECK-LABEL: (func $zext_i32_i64
+; CHECK: (setlocal @1 (load_zx_i32_i64 @0))
+define i64 @zext_i32_i64(i32 *%p) {
+  %v = load i32, i32* %p
+  %e = zext i32 %v to i64
+  ret i64 %e
+}
diff --git a/llvm/test/CodeGen/WebAssembly/load-store-i1.ll b/llvm/test/CodeGen/WebAssembly/load-store-i1.ll
new file mode 100644
index 00000000000..99b689e0cb2
--- /dev/null
+++ b/llvm/test/CodeGen/WebAssembly/load-store-i1.ll
@@ -0,0 +1,68 @@
+; RUN: llc < %s -asm-verbose=false | FileCheck %s
+
+; Test that i1 extending loads and truncating stores are assembled properly.
+
+target datalayout = "e-p:32:32-i64:64-v128:8:128-n32:64-S128"
+target triple = "wasm32-unknown-unknown"
+
+; CHECK-LABEL: (func $load_unsigned_i1_i32
+; CHECK:      (setlocal @1 (load_zx_i8_i32 @0))
+; CHECK-NEXT: (return @1)
+define i32 @load_unsigned_i1_i32(i1* %p) {
+  %v = load i1, i1* %p
+  %e = zext i1 %v to i32
+  ret i32 %e
+}
+
+; CHECK-LABEL: (func $load_signed_i1_i32
+; CHECK:      (setlocal @1 (load_zx_i8_i32 @0))
+; CHECK-NEXT: (setlocal @2 (immediate 31))
+; CHECK-NEXT: (setlocal @3 (shl @1 @2))
+; CHECK-NEXT: (setlocal @4 (sar @3 @2))
+; CHECK-NEXT: (return @4)
+define i32 @load_signed_i1_i32(i1* %p) {
+  %v = load i1, i1* %p
+  %e = sext i1 %v to i32
+  ret i32 %e
+}
+
+; CHECK-LABEL: (func $load_unsigned_i1_i64
+; CHECK:      (setlocal @1 (load_zx_i8_i64 @0))
+; CHECK-NEXT: (return @1)
+define i64 @load_unsigned_i1_i64(i1* %p) {
+  %v = load i1, i1* %p
+  %e = zext i1 %v to i64
+  ret i64 %e
+}
+
+; CHECK-LABEL: (func $load_signed_i1_i64
+; CHECK:      (setlocal @1 (load_zx_i8_i64 @0))
+; CHECK-NEXT: (setlocal @2 (immediate 63))
+; CHECK-NEXT: (setlocal @3 (shl @1 @2))
+; CHECK-NEXT: (setlocal @4 (sar @3 @2))
+; CHECK-NEXT: (return @4)
+define i64 @load_signed_i1_i64(i1* %p) {
+  %v = load i1, i1* %p
+  %e = sext i1 %v to i64
+  ret i64 %e
+}
+
+; CHECK-LABEL: (func $store_i32_i1
+; CHECK:      (setlocal @2 (immediate 1))
+; CHECK-NEXT: (setlocal @3 (and @1 @2))
+; CHECK-NEXT: (store_i8 @0 @3)
+define void @store_i32_i1(i1* %p, i32 %v) {
+  %t = trunc i32 %v to i1
+  store i1 %t, i1* %p
+  ret void
+}
+
+; CHECK-LABEL: (func $store_i64_i1
+; CHECK:      (setlocal @2 (immediate 1))
+; CHECK-NEXT: (setlocal @3 (and @1 @2))
+; CHECK-NEXT: (store_i8 @0 @3)
+define void @store_i64_i1(i1* %p, i64 %v) {
+  %t = trunc i64 %v to i1
+  store i1 %t, i1* %p
+  ret void
+}
diff --git a/llvm/test/CodeGen/WebAssembly/load.ll b/llvm/test/CodeGen/WebAssembly/load.ll
new file mode 100644
index 00000000000..e153b8becc5
--- /dev/null
+++ b/llvm/test/CodeGen/WebAssembly/load.ll
@@ -0,0 +1,46 @@
+; RUN: llc < %s -asm-verbose=false | FileCheck %s
+
+; Test that basic loads are assembled properly.
+
+target datalayout = "e-p:32:32-i64:64-v128:8:128-n32:64-S128"
+target triple = "wasm32-unknown-unknown"
+
+; CHECK-LABEL: (func $ldi32
+; CHECK-NEXT: (param i32) (result i32)
+; CHECK-NEXT: (setlocal @0 (argument 0))
+; CHECK-NEXT: (setlocal @1 (load_i32 @0))
+; CHECK-NEXT: (return @1)
+define i32 @ldi32(i32 *%p) {
+  %v = load i32, i32* %p
+  ret i32 %v
+}
+
+; CHECK-LABEL: (func $ldi64
+; CHECK-NEXT: (param i32) (result i64)
+; CHECK-NEXT: (setlocal @0 (argument 0))
+; CHECK-NEXT: (setlocal @1 (load_i64 @0))
+; CHECK-NEXT: (return @1)
+define i64 @ldi64(i64 *%p) {
+  %v = load i64, i64* %p
+  ret i64 %v
+}
+
+; CHECK-LABEL: (func $ldf32
+; CHECK-NEXT: (param i32) (result f32)
+; CHECK-NEXT: (setlocal @0 (argument 0))
+; CHECK-NEXT: (setlocal @1 (load_f32 @0))
+; CHECK-NEXT: (return @1)
+define float @ldf32(float *%p) {
+  %v = load float, float* %p
+  ret float %v
+}
+
+; CHECK-LABEL: (func $ldf64
+; CHECK-NEXT: (param i32) (result f64)
+; CHECK-NEXT: (setlocal @0 (argument 0))
+; CHECK-NEXT: (setlocal @1 (load_f64 @0))
+; CHECK-NEXT: (return @1)
+define double @ldf64(double *%p) {
+  %v = load double, double* %p
+  ret double %v
+}
diff --git a/llvm/test/CodeGen/WebAssembly/store-trunc.ll b/llvm/test/CodeGen/WebAssembly/store-trunc.ll
new file mode 100644
index 00000000000..21988db66b7
--- /dev/null
+++ b/llvm/test/CodeGen/WebAssembly/store-trunc.ll
@@ -0,0 +1,46 @@
+; RUN: llc < %s -asm-verbose=false | FileCheck %s
+
+; Test that truncating stores are assembled properly.
+
+target datalayout = "e-p:32:32-i64:64-v128:8:128-n32:64-S128"
+target triple = "wasm32-unknown-unknown"
+
+; CHECK-LABEL: (func $trunc_i8_i32
+; CHECK: (store_i8 @1 @0)
+define void @trunc_i8_i32(i8 *%p, i32 %v) {
+  %t = trunc i32 %v to i8
+  store i8 %t, i8* %p
+  ret void
+}
+
+; CHECK-LABEL: (func $trunc_i16_i32
+; CHECK: (store_i16 @1 @0)
+define void @trunc_i16_i32(i16 *%p, i32 %v) {
+  %t = trunc i32 %v to i16
+  store i16 %t, i16* %p
+  ret void
+}
+
+; CHECK-LABEL: (func $trunc_i8_i64
+; CHECK: (store_i8 @1 @0)
+define void @trunc_i8_i64(i8 *%p, i64 %v) {
+  %t = trunc i64 %v to i8
+  store i8 %t, i8* %p
+  ret void
+}
+
+; CHECK-LABEL: (func $trunc_i16_i64
+; CHECK: (store_i16 @1 @0)
+define void @trunc_i16_i64(i16 *%p, i64 %v) {
+  %t = trunc i64 %v to i16
+  store i16 %t, i16* %p
+  ret void
+}
+
+; CHECK-LABEL: (func $trunc_i32_i64
+; CHECK: (store_i32 @1 @0)
+define void @trunc_i32_i64(i32 *%p, i64 %v) {
+  %t = trunc i64 %v to i32
+  store i32 %t, i32* %p
+  ret void
+}
diff --git a/llvm/test/CodeGen/WebAssembly/store.ll b/llvm/test/CodeGen/WebAssembly/store.ll
new file mode 100644
index 00000000000..f93f67b03fc
--- /dev/null
+++ b/llvm/test/CodeGen/WebAssembly/store.ll
@@ -0,0 +1,50 @@
+; RUN: llc < %s -asm-verbose=false | FileCheck %s
+
+; Test that basic stores are assembled properly.
+
+target datalayout = "e-p:32:32-i64:64-v128:8:128-n32:64-S128"
+target triple = "wasm32-unknown-unknown"
+
+; CHECK-LABEL: (func $sti32
+; CHECK-NEXT: (param i32) (param i32)
+; CHECK-NEXT: (setlocal @0 (argument 1))
+; CHECK-NEXT: (setlocal @1 (argument 0))
+; CHECK-NEXT: (store_i32 @1 @0)
+; CHECK-NEXT: (return)
+define void @sti32(i32 *%p, i32 %v) {
+  store i32 %v, i32* %p
+  ret void
+}
+
+; CHECK-LABEL: (func $sti64
+; CHECK-NEXT: (param i32) (param i64)
+; CHECK-NEXT: (setlocal @0 (argument 1))
+; CHECK-NEXT: (setlocal @1 (argument 0))
+; CHECK-NEXT: (store_i64 @1 @0)
+; CHECK-NEXT: (return)
+define void @sti64(i64 *%p, i64 %v) {
+  store i64 %v, i64* %p
+  ret void
+}
+
+; CHECK-LABEL: (func $stf32
+; CHECK-NEXT: (param i32) (param f32)
+; CHECK-NEXT: (setlocal @0 (argument 1))
+; CHECK-NEXT: (setlocal @1 (argument 0))
+; CHECK-NEXT: (store_f32 @1 @0)
+; CHECK-NEXT: (return)
+define void @stf32(float *%p, float %v) {
+  store float %v, float* %p
+  ret void
+}
+
+; CHECK-LABEL: (func $stf64
+; CHECK-NEXT: (param i32) (param f64)
+; CHECK-NEXT: (setlocal @0 (argument 1))
+; CHECK-NEXT: (setlocal @1 (argument 0))
+; CHECK-NEXT: (store_f64 @1 @0)
+; CHECK-NEXT: (return)
+define void @stf64(double *%p, double %v) {
+  store double %v, double* %p
+  ret void
+}