[llvm] d57a87e - [X86][ISel] Lowering llvm.thread.pointer

[Freddy Ye via llvm-commits]

Author: Freddy Ye
Date: 2021-10-12T11:01:18+08:00
New Revision: d57a87ea89c750b6d1cf359b619eea4eda806e1a

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

LOG: [X86][ISel] Lowering llvm.thread.pointer

Reviewed By: pengfei

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

Added: 
    llvm/test/CodeGen/X86/thread_pointer-error.ll
    llvm/test/CodeGen/X86/thread_pointer.ll

Modified: 
    llvm/lib/Target/X86/X86ISelLowering.cpp

Removed: 
    


################################################################################
diff  --git a/llvm/lib/Target/X86/X86ISelLowering.cpp b/llvm/lib/Target/X86/X86ISelLowering.cpp
index 2b34f9a07f6bf..d1579d4c3e504 100644
--- a/llvm/lib/Target/X86/X86ISelLowering.cpp
+++ b/llvm/lib/Target/X86/X86ISelLowering.cpp
@@ -26761,6 +26761,19 @@ SDValue X86TargetLowering::LowerINTRINSIC_WO_CHAIN(SDValue Op,
                                              getPointerTy(DAG.getDataLayout())),
                        Op.getOperand(1), ShAmt);
   }
+  case Intrinsic::thread_pointer: {
+    if (Subtarget.isTargetELF()) {
+      SDLoc dl(Op);
+      EVT PtrVT = getPointerTy(DAG.getDataLayout());
+      // Get the Thread Pointer, which is %gs:0 (32-bit) or %fs:0 (64-bit).
+      Value *Ptr = Constant::getNullValue(Type::getInt8PtrTy(
+          *DAG.getContext(), Subtarget.is64Bit() ? X86AS::FS : X86AS::GS));
+      return DAG.getLoad(PtrVT, dl, DAG.getEntryNode(),
+                         DAG.getIntPtrConstant(0, dl), MachinePointerInfo(Ptr));
+    }
+    report_fatal_error(
+        "Target OS doesn't support __builtin_thread_pointer() yet.");
+  }
   }
 }
 

diff  --git a/llvm/test/CodeGen/X86/thread_pointer-error.ll b/llvm/test/CodeGen/X86/thread_pointer-error.ll
new file mode 100644
index 0000000000000..77b3c16c515ab
--- /dev/null
+++ b/llvm/test/CodeGen/X86/thread_pointer-error.ll
@@ -0,0 +1,16 @@
+; FIXME: Targets below should support this builtin after they document the thread pointer for X86.
+; CHECK-ERROR: LLVM ERROR: Target OS doesn't support __builtin_thread_pointer() yet.
+
+; RUN: not --crash llc < %s -o /dev/null -mtriple=i686-pc-win32 2>&1 | FileCheck %s --check-prefix=CHECK-ERROR
+; RUN: not --crash llc < %s -o /dev/null -mtriple=x86_64-pc-win32 2>&1 | FileCheck %s --check-prefix=CHECK-ERROR
+; RUN: not --crash llc < %s -o /dev/null -mtriple=i686-pc-windows-gnu 2>&1 | FileCheck %s --check-prefix=CHECK-ERROR
+; RUN: not --crash llc < %s -o /dev/null -mtriple=x86_64-pc-windows-gnu 2>&1 | FileCheck %s --check-prefix=CHECK-ERROR
+; RUN: not --crash llc < %s -o /dev/null -mtriple=i686-apple-darwin 2>&1 | FileCheck %s --check-prefix=CHECK-ERROR
+; RUN: not --crash llc < %s -o /dev/null -mtriple=x86_64-pc-apple-darwin 2>&1 | FileCheck %s --check-prefix=CHECK-ERROR
+
+declare i8* @llvm.thread.pointer()
+
+define i8* @thread_pointer() nounwind {
+  %1 = tail call i8* @llvm.thread.pointer()
+  ret i8* %1
+}

diff  --git a/llvm/test/CodeGen/X86/thread_pointer.ll b/llvm/test/CodeGen/X86/thread_pointer.ll
new file mode 100644
index 0000000000000..76dc6d69ed157
--- /dev/null
+++ b/llvm/test/CodeGen/X86/thread_pointer.ll
@@ -0,0 +1,51 @@
+; NOTE: Assertions have been autogenerated by utils/update_llc_test_checks.py
+; RUN: llc < %s -mtriple=x86_64-linux-gnu | FileCheck %s --check-prefix=X64
+; RUN: llc < %s -mtriple=x86_64-linux-gnux32 | FileCheck %s --check-prefix=X32
+; RUN: llc < %s -mtriple=i386-linux-gnu | FileCheck %s --check-prefix=X86
+
+declare i8* @llvm.thread.pointer()
+
+define i8* @thread_pointer() nounwind {
+; X64-LABEL: thread_pointer:
+; X64:       # %bb.0:
+; X64-NEXT:    movq %fs:0, %rax
+; X64-NEXT:    retq
+;
+; X32-LABEL: thread_pointer:
+; X32:       # %bb.0:
+; X32-NEXT:    movl %fs:0, %eax
+; X32-NEXT:    retq
+;
+; X86-LABEL: thread_pointer:
+; X86:       # %bb.0:
+; X86-NEXT:    movl %gs:0, %eax
+; X86-NEXT:    retl
+  %1 = tail call i8* @llvm.thread.pointer()
+  ret i8* %1
+}
+
+define i32 @thread_pointer2(i32 %i) nounwind {
+; X64-LABEL: thread_pointer2:
+; X64:       # %bb.0: # %entry
+; X64-NEXT:    movslq %edi, %rax
+; X64-NEXT:    movl %fs:(,%rax,4), %eax
+; X64-NEXT:    retq
+;
+; X32-LABEL: thread_pointer2:
+; X32:       # %bb.0: # %entry
+; X32-NEXT:    movl %fs:0, %eax
+; X32-NEXT:    movl (%eax,%edi,4), %eax
+; X32-NEXT:    retq
+;
+; X86-LABEL: thread_pointer2:
+; X86:       # %bb.0: # %entry
+; X86-NEXT:    movl {{[0-9]+}}(%esp), %eax
+; X86-NEXT:    movl %gs:(,%eax,4), %eax
+; X86-NEXT:    retl
+entry:
+  %0 = tail call i8* @llvm.thread.pointer()
+  %1 = bitcast i8* %0 to i32*
+  %arrayidx = getelementptr inbounds i32, i32* %1, i32 %i
+  %2 = load i32, i32* %arrayidx, align 4
+  ret i32 %2
+}