[llvm] WebAssembly: Remove threadlocal_address intrinsics when disabling TLS (PR #88209)

[Matthias Braun via llvm-commits]

https://github.com/MatzeB created https://github.com/llvm/llvm-project/pull/88209

None

>From 1894bcc63bd112122c2012de5c43870cddb9031b Mon Sep 17 00:00:00 2001
From: Matthias Braun <matze at braunis.de>
Date: Tue, 9 Apr 2024 15:46:54 -0700
Subject: [PATCH] WebAssembly: Remove threadlocal_address intrinsics when
 disabling TLS

---
 .../WebAssembly/WebAssemblyTargetMachine.cpp  | 13 +++++++++
 llvm/test/CodeGen/WebAssembly/strip-tls.ll    | 28 +++++++++++++++++++
 2 files changed, 41 insertions(+)
 create mode 100644 llvm/test/CodeGen/WebAssembly/strip-tls.ll

diff --git a/llvm/lib/Target/WebAssembly/WebAssemblyTargetMachine.cpp b/llvm/lib/Target/WebAssembly/WebAssemblyTargetMachine.cpp
index 70685b2e3bb2de..27b0bb2501d4ed 100644
--- a/llvm/lib/Target/WebAssembly/WebAssemblyTargetMachine.cpp
+++ b/llvm/lib/Target/WebAssembly/WebAssemblyTargetMachine.cpp
@@ -289,6 +289,19 @@ class CoalesceFeaturesAndStripAtomics final : public ModulePass {
 
   bool stripThreadLocals(Module &M) {
     bool Stripped = false;
+    for (auto &F : M) {
+      for (auto &B : F) {
+        for (auto &I : make_early_inc_range(B)) {
+          if (IntrinsicInst *II = dyn_cast<IntrinsicInst>(&I)) {
+            if (II->getIntrinsicID() == Intrinsic::threadlocal_address) {
+              II->replaceAllUsesWith(II->getArgOperand(0));
+              II->eraseFromParent();
+              Stripped = true;
+            }
+          }
+        }
+      }
+    }
     for (auto &GV : M.globals()) {
       if (GV.isThreadLocal()) {
         Stripped = true;
diff --git a/llvm/test/CodeGen/WebAssembly/strip-tls.ll b/llvm/test/CodeGen/WebAssembly/strip-tls.ll
new file mode 100644
index 00000000000000..281fe0aee7a497
--- /dev/null
+++ b/llvm/test/CodeGen/WebAssembly/strip-tls.ll
@@ -0,0 +1,28 @@
+; NOTE: Assertions have been autogenerated by utils/update_llc_test_checks.py UTC_ARGS: --version 4
+; RUN: llc -o - %s -mattr=-bulk-memory | FileCheck --check-prefix=NOTLS %s
+; RUN: llc -o - %s -mattr=+bulk-memory,+atomics | FileCheck --check-prefix=TLS %s
+
+target triple = "wasm32--"
+
+ at tls = internal thread_local global i32 0
+
+define i32 @func() {
+; NOTLS-LABEL: func:
+; NOTLS:         .functype func () -> (i32)
+; NOTLS-NEXT:  # %bb.0:
+; NOTLS-NEXT:    i32.const 0
+; NOTLS-NEXT:    i32.load tls
+; NOTLS-NEXT:    # fallthrough-return
+;
+; TLS-LABEL: func:
+; TLS:         .functype func () -> (i32)
+; TLS-NEXT:  # %bb.0:
+; TLS-NEXT:    global.get __tls_base
+; TLS-NEXT:    i32.const tls at TLSREL
+; TLS-NEXT:    i32.add
+; TLS-NEXT:    i32.load 0
+; TLS-NEXT:    # fallthrough-return
+  %p = call ptr @llvm.threadlocal.address.p0(ptr @tls)
+  %v = load i32, ptr %p
+  ret i32 %v
+}