From c8301d17a861ef47c7b01a486391f288b035e692 Mon Sep 17 00:00:00 2001
From: Anton Korobeynikov <asl@math.spbu.ru>
Date: Thu, 16 Jul 2009 13:47:14 +0000
Subject: [PATCH] Add mem-imm stores

git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@75933 91177308-0d34-0410-b5e6-96231b3b80d8
---
 lib/Target/SystemZ/SystemZISelLowering.cpp |  1 -
 lib/Target/SystemZ/SystemZInstrInfo.td     | 14 +++++++++
 test/CodeGen/SystemZ/05-MemImmStores.ll    | 35 ++++++++++++++++++++++
 3 files changed, 49 insertions(+), 1 deletion(-)
 create mode 100644 test/CodeGen/SystemZ/05-MemImmStores.ll

diff --git a/lib/Target/SystemZ/SystemZISelLowering.cpp b/lib/Target/SystemZ/SystemZISelLowering.cpp
index 69beda338e9..bc396ae4ca0 100644
--- a/lib/Target/SystemZ/SystemZISelLowering.cpp
+++ b/lib/Target/SystemZ/SystemZISelLowering.cpp
@@ -55,7 +55,6 @@ SystemZTargetLowering::SystemZTargetLowering(SystemZTargetMachine &tm) :
   setSchedulingPreference(SchedulingForLatency);
 
   setOperationAction(ISD::RET,              MVT::Other, Custom);
-
 }
 
 SDValue SystemZTargetLowering::LowerOperation(SDValue Op, SelectionDAG &DAG) {
diff --git a/lib/Target/SystemZ/SystemZInstrInfo.td b/lib/Target/SystemZ/SystemZInstrInfo.td
index 5ba96f6dee9..f1baa457346 100644
--- a/lib/Target/SystemZ/SystemZInstrInfo.td
+++ b/lib/Target/SystemZ/SystemZInstrInfo.td
@@ -223,6 +223,20 @@ def MOV64mr : Pseudo<(outs), (ins rriaddr:$dst, GR64:$src),
                      "stg\t{$src, $dst}",
                      [(store GR64:$src, rriaddr:$dst)]>;
 
+// FIXME: displacements here are really 12 bit, not 20!
+def MOV8mi    : Pseudo<(outs), (ins riaddr:$dst, i32imm:$src),
+                       "mvi\t{$dst, $src}",
+                       [(truncstorei8 (i32 imm:$src), riaddr:$dst)]>;
+def MOV16mi   : Pseudo<(outs), (ins riaddr:$dst, i32imm:$src),
+                       "mvhhi\t{$dst, $src}",
+                       [(truncstorei16 (i32 imm:$src), riaddr:$dst)]>;
+def MOV32mi16 : Pseudo<(outs), (ins riaddr:$dst, i32imm:$src),
+                       "mvhi\t{$dst, $src}",
+                       [(store (i32 immSExt16:$src), riaddr:$dst)]>;
+def MOV64mi16 : Pseudo<(outs), (ins riaddr:$dst, i64imm:$src),
+                       "mvghi\t{$dst, $src}",
+                       [(store (i64 immSExt16:$src), riaddr:$dst)]>;
+
 // extloads
 def MOVSX64rm8  : Pseudo<(outs GR64:$dst), (ins rriaddr:$src),
                          "lgb\t{$dst, $src}",
diff --git a/test/CodeGen/SystemZ/05-MemImmStores.ll b/test/CodeGen/SystemZ/05-MemImmStores.ll
new file mode 100644
index 00000000000..eeea4cfec85
--- /dev/null
+++ b/test/CodeGen/SystemZ/05-MemImmStores.ll
@@ -0,0 +1,35 @@
+; RUN: llvm-as < %s | llc | grep mvghi | count 1
+; RUN: llvm-as < %s | llc | grep mvhi  | count 1
+; RUN: llvm-as < %s | llc | grep mvhhi | count 1
+; RUN: llvm-as < %s | llc | grep mvi   | count 1
+
+target datalayout = "E-p:64:64:64-i1:8:8-i16:16:16-i32:32:32-i64:64:64-f32:32:32-f64:64:64-f128:128:128"
+target triple = "s390x-unknown-linux-gnu"
+
+define void @foo1(i64* nocapture %a, i64 %idx) nounwind {
+entry:
+	%add.ptr = getelementptr i64* %a, i64 1		; <i64*> [#uses=1]
+	store i64 1, i64* %add.ptr
+	ret void
+}
+
+define void @foo2(i32* nocapture %a, i64 %idx) nounwind {
+entry:
+	%add.ptr = getelementptr i32* %a, i64 1		; <i32*> [#uses=1]
+	store i32 2, i32* %add.ptr
+	ret void
+}
+
+define void @foo3(i16* nocapture %a, i64 %idx) nounwind {
+entry:
+	%add.ptr = getelementptr i16* %a, i64 1		; <i16*> [#uses=1]
+	store i16 3, i16* %add.ptr
+	ret void
+}
+
+define void @foo4(i8* nocapture %a, i64 %idx) nounwind {
+entry:
+	%add.ptr = getelementptr i8* %a, i64 1		; <i8*> [#uses=1]
+	store i8 4, i8* %add.ptr
+	ret void
+}
-- 
2.34.1