[llvm] 487ace4 - [SystemZ][z/OS] Add llvm.read_register() intrinsic support for zOS

[Yusra Syeda via llvm-commits]

Author: Yusra Syeda
Date: 2022-06-10T12:30:07-04:00
New Revision: 487ace4c73625905b11f41baf69f286bce7882bc

URL: https://github.com/llvm/llvm-project/commit/487ace4c73625905b11f41baf69f286bce7882bc
DIFF: https://github.com/llvm/llvm-project/commit/487ace4c73625905b11f41baf69f286bce7882bc.diff

LOG: [SystemZ][z/OS] Add llvm.read_register() intrinsic support for zOS

Differential Revision: https://reviews.llvm.org/D127412

Added: 
    llvm/test/CodeGen/SystemZ/zos-stackpointer.ll

Modified: 
    llvm/lib/Target/SystemZ/SystemZISelLowering.cpp

Removed: 
    


################################################################################
diff  --git a/llvm/lib/Target/SystemZ/SystemZISelLowering.cpp b/llvm/lib/Target/SystemZ/SystemZISelLowering.cpp
index 9a1be95b67da2..7b16a0fc55911 100644
--- a/llvm/lib/Target/SystemZ/SystemZISelLowering.cpp
+++ b/llvm/lib/Target/SystemZ/SystemZISelLowering.cpp
@@ -1260,12 +1260,17 @@ SystemZTargetLowering::getRegForInlineAsmConstraint(
 
 // FIXME? Maybe this could be a TableGen attribute on some registers and
 // this table could be generated automatically from RegInfo.
-Register SystemZTargetLowering::getRegisterByName(const char *RegName, LLT VT,
-                                                  const MachineFunction &MF) const {
+Register
+SystemZTargetLowering::getRegisterByName(const char *RegName, LLT VT,
+                                         const MachineFunction &MF) const {
+  const SystemZSubtarget *Subtarget = &MF.getSubtarget<SystemZSubtarget>();
+
+  Register Reg =
+      StringSwitch<Register>(RegName)
+          .Case("r4", Subtarget->isTargetXPLINK64() ? SystemZ::R4D : 0)
+          .Case("r15", Subtarget->isTargetELF() ? SystemZ::R15D : 0)
+          .Default(0);
 
-  Register Reg = StringSwitch<Register>(RegName)
-                   .Case("r15", SystemZ::R15D)
-                   .Default(0);
   if (Reg)
     return Reg;
   report_fatal_error("Invalid register name global variable");

diff  --git a/llvm/test/CodeGen/SystemZ/zos-stackpointer.ll b/llvm/test/CodeGen/SystemZ/zos-stackpointer.ll
new file mode 100644
index 0000000000000..2fcc4c789518a
--- /dev/null
+++ b/llvm/test/CodeGen/SystemZ/zos-stackpointer.ll
@@ -0,0 +1,16 @@
+; RUN: llc < %s -mtriple=s390x-ibm-zos | FileCheck %s
+
+; CHECK-LABEL: get_stack:
+; CHECK: lgr   3, 4
+; CHECK: b 2(7)
+
+define i8* @get_stack() nounwind {
+entry:
+        %0 = call i64 @llvm.read_register.i64(metadata !0)
+        %1 = inttoptr i64 %0 to i8*
+  ret i8* %1
+}
+
+declare i64 @llvm.read_register.i64(metadata) nounwind
+
+!0 = !{!"r4"}